UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-22 05:08:14 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #4149 from arsaboo/openrouter

Browse Source 
chore: Add HTTP-Referer and X-Title headers for OpenRouter
This commit is contained in:
Timothy Jaeryang Baek 2024-07-27 16:06:17 +02:00 committed by GitHub
commit 2e4de209fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/apps/openai/main.py
View File

@ -462,6 +462,9 @@ async def generate_chat_completion(
headers = {} headers = {}
headers["Authorization"] = f"Bearer {key}" headers["Authorization"] = f"Bearer {key}"
headers["Content-Type"] = "application/json" 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 r = None
session = None session = None