diff --git a/src/llm/api_models.json b/src/llm/api_models.json index 97bd420ad..80de87ff3 100644 --- a/src/llm/api_models.json +++ b/src/llm/api_models.json @@ -59,6 +59,16 @@ "model_name": "gemini-2.5-pro-preview-06-05", "provider": "Google" }, + { + "display_name": "Gemini 2.5 Pro (OpenRouter)", + "model_name": "google/gemini-2.5-pro", + "provider": "OpenRouter" + }, + { + "display_name": "Gemini 2.5 Flash (OpenRouter)", + "model_name": "google/gemini-2.5-flash", + "provider": "OpenRouter" + }, { "display_name": "GLM-4.5 Air", "model_name": "z-ai/glm-4.5-air", @@ -83,5 +93,10 @@ "display_name": "Azure Open AI Deployment", "model_name": "", "provider": "Azure OpenAI" + }, + { + "display_name": "Custom OpenRouter Model", + "model_name": "-", + "provider": "OpenRouter" } ] \ No newline at end of file diff --git a/src/tools/api.py b/src/tools/api.py index 60ccbdc9b..e476273f4 100644 --- a/src/tools/api.py +++ b/src/tools/api.py @@ -3,6 +3,10 @@ import pandas as pd import requests import time +import urllib3 + +# Disable SSL warnings +urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) from src.data.cache import get_cache from src.data.models import ( @@ -42,9 +46,9 @@ def _make_api_request(url: str, headers: dict, method: str = "GET", json_data: d """ for attempt in range(max_retries + 1): # +1 for initial attempt if method.upper() == "POST": - response = requests.post(url, headers=headers, json=json_data) + response = requests.post(url, headers=headers, json=json_data, verify=False) else: - response = requests.get(url, headers=headers) + response = requests.get(url, headers=headers, verify=False) if response.status_code == 429 and attempt < max_retries: # Linear backoff: 60s, 90s, 120s, 150s...