From bd823346478772087772d574229acf93a8bb77cd Mon Sep 17 00:00:00 2001 From: Alok Saboo Date: Mon, 13 May 2024 08:28:24 -0400 Subject: [PATCH] chore: add additional headers only for Openrouter --- backend/apps/openai/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/apps/openai/main.py b/backend/apps/openai/main.py index e346fc860..b38c2bc2d 100644 --- a/backend/apps/openai/main.py +++ b/backend/apps/openai/main.py @@ -116,9 +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" - 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 try: r = requests.post(