ソースを参照

Merge pull request #4149 from arsaboo/openrouter

chore: Add HTTP-Referer and X-Title headers for OpenRouter
Timothy Jaeryang Baek 9 ヶ月 前
コミット
2e4de209fe
1 ファイル変更3 行追加0 行削除
  1. 3 0
      backend/apps/openai/main.py

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

@@ -462,6 +462,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