diff --git a/ra_aid/llm.py b/ra_aid/llm.py index a1c3603..8d7312d 100644 --- a/ra_aid/llm.py +++ b/ra_aid/llm.py @@ -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 {}), )