zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add app headers for openrouter

This commit is contained in:
AI Christianson 2025-02-19 19:21:16 -05:00
parent 6d095eab46
commit 9dee031a2a
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ra_aid/llm.py
View File

@ -119,6 +119,11 @@ def create_openrouter_client(
is_expert: bool = False,
) -> BaseChatModel:
"""Create OpenRouter client with appropriate configuration."""
default_headers = {
"HTTP-Referer": "https://ra-aid.ai",
"X-Title": "RA.Aid"
}
if model_name.startswith("deepseek/") and "deepseek-r1" in model_name.lower():
return ChatDeepseekReasoner(
api_key=api_key,
@ -129,6 +134,7 @@ def create_openrouter_client(
model=model_name,
timeout=LLM_REQUEST_TIMEOUT,
max_retries=LLM_MAX_RETRIES,
default_headers=default_headers,
)
return ChatOpenAI(
@ -137,6 +143,7 @@ def create_openrouter_client(
model=model_name,
timeout=LLM_REQUEST_TIMEOUT,
max_retries=LLM_MAX_RETRIES,
default_headers=default_headers,
**({"temperature": temperature} if temperature is not None else {}),
)