瀏覽代碼

Merge pull request #2211 from arsaboo/header

chore: add HTTP-Referer and X-Title headers for Open WebUI
Timothy Jaeryang Baek 11 月之前
父節點
當前提交
fabcbc300f
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      backend/apps/openai/main.py

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

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