diff --git a/config/models.yaml b/config/models.yaml index cec5cfa..1ab472a 100644 --- a/config/models.yaml +++ b/config/models.yaml @@ -5,24 +5,13 @@ models: default: deepseek_r1 providers: - # Provider 1: MegaLLM (OpenAI-Like) - openai_like: - api_key: ${MEGALLM_API_KEY} - base_url: ${API_URL} - models: - deepseek_v3: - id: "deepseek-ai/deepseek-v3.1" - temperature: 0.7 - max_tokens: 4096 - supports_streaming: true - - # Provider 2: OpenRouter + # Provider: OpenRouter openrouter: api_key: ${OPENROUTER_API_KEY} base_url: ${OPENROUTER_BASE_URL} models: deepseek_r1: - id: "openai/gpt-oss-120b:free" + id: "openai/gpt-oss-120b" temperature: 0.6 max_tokens: 4096 diff --git a/config/production.env b/config/production.env index 47f14fc..0e03a59 100644 --- a/config/production.env +++ b/config/production.env @@ -11,7 +11,7 @@ API_URL=https://gpt.nwhco.ir OPENROUTER_BASE_URL=https://openrouter.ai/api/v1 # API KEYS -OPENROUTER_API_KEY=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 +OPENROUTER_API_KEY=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 MEGALLM_API_KEY=sk-mega-d124420e7b96640569860ec6dee778a41c10a67356380eedfba5c7a2024db3c3 # ---------------- Vector DB Settings ---------------- diff --git a/src/models/factory.py b/src/models/factory.py index 824af29..ffd0d94 100644 --- a/src/models/factory.py +++ b/src/models/factory.py @@ -72,15 +72,17 @@ class ModelFactory: elif provider_name == 'openrouter': return OpenRouter( - id =model_config_data['id'], + id=model_config_data['id'], api_key=api_key, base_url=base_url, collect_metrics_on_completion=True, max_tokens=model_config_data.get('max_tokens', 4096), default_headers={ - "Authorization": f"Bearer {api_key}", - "Content-Type": "application/json" - }, + "Authorization": f"Bearer {api_key}", + "Content-Type": "application/json", + "HTTP-Referer": os.getenv("APP_URL", "https://imamreza.ir"), + "X-Title": os.getenv("APP_TITLE", "Imam Reza Shrine Agent"), + }, ) raise ValueError(f"Model '{model_name}' not found in configuration") \ No newline at end of file