diff --git a/ra_aid/llm.py b/ra_aid/llm.py index 2765010..c7267fb 100644 --- a/ra_aid/llm.py +++ b/ra_aid/llm.py @@ -122,6 +122,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, @@ -132,6 +137,7 @@ def create_openrouter_client( model=model_name, timeout=LLM_REQUEST_TIMEOUT, max_retries=LLM_MAX_RETRIES, + default_headers=default_headers, ) return ChatOpenAI( @@ -140,6 +146,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 {}), )