Explorar o código

Merge pull request #4149 from arsaboo/openrouter

chore: Add HTTP-Referer and X-Title headers for OpenRouter
Timothy Jaeryang Baek hai 9 meses
pai
achega
2e4de209fe
Modificáronse 1 ficheiros con 3 adicións e 0 borrados
  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