Add HTTP-Referer and X-Title headers for OpenRouter

28714768 · Alok Saboo · 82079e64 · 28714768
Commit 28714768 authored Jul 27, 2024 by Alok Saboo
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

backend/apps/openai/main.py backend/apps/openai/main.py +3 -0

No files found.
--- a/backend/apps/openai/main.py
+++ b/backend/apps/openai/main.py
@@ -459,6 +459,9 @@ async def generate_chat_completion(
    headers = {}
    headers["Authorization"] = f"Bearer {key}"
    headers["Content-Type"] = "application/json"
+    if "openrouter.ai" in app.state.config.OPENAI_API_BASE_URLS[idx]:
+        headers["HTTP-Referer"] = "https://openwebui.com/"
+        headers["X-Title"] = "Open WebUI"

    r = None
    session = None