Browse Source

chore: add additional headers only for Openrouter

Alok Saboo 11 months ago
parent
commit
bd82334647
1 changed files with 3 additions and 3 deletions
  1. 3 3
      backend/apps/openai/main.py

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

@@ -116,9 +116,9 @@ async def speech(request: Request, user=Depends(get_verified_user)):
         headers = {}
         headers = {}
         headers["Authorization"] = f"Bearer {app.state.OPENAI_API_KEYS[idx]}"
         headers["Authorization"] = f"Bearer {app.state.OPENAI_API_KEYS[idx]}"
         headers["Content-Type"] = "application/json"
         headers["Content-Type"] = "application/json"
-        headers['HTTP-Referer'] = "https://openwebui.com/"
-        headers['X-Title'] = "Open WebUI"
-
+        if "openrouter.ai" in app.state.OPENAI_API_BASE_URLS[idx]:
+            headers['HTTP-Referer'] = "https://openwebui.com/"
+            headers['X-Title'] = "Open WebUI"
         r = None
         r = None
         try:
         try:
             r = requests.post(
             r = requests.post(