dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 10:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: Add method to fetch model info from openrouter pages

Browse source
This commit is contained in:
Stefan Hladnik (aider) 2025-03-18 03:02:19 +07:00 • committed by Stefan Hladnik
parent 94bcaa71e6
commit c1b2ff20de
1 changed files with 42 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

45
aider/models.py
View file

@ -211,7 +211,7 @@ class ModelInfoManager:
def get_model_info(self, model):
cached_info = self.get_model_from_cached_json_db(model)
litellm_info = None
if litellm._lazy_module or not cached_info:
try:
@ -219,13 +219,52 @@ class ModelInfoManager:
except Exception as ex:
if "model_prices_and_context_window.json" not in str(ex):
print(str(ex))
if litellm_info:
return litellm_info
if not cached_info and model.startswith("openrouter/"):
openrouter_info = self.fetch_openrouter_model_info(model)
if openrouter_info:
return openrouter_info
return cached_info
def fetch_openrouter_model_info(self, model):
"""
Fetch model info by scraping the openrouter model page.
Expected URL: https://openrouter.ai/<model_route>
Example: openrouter/qwen/qwen-2.5-72b-instruct:free
Returns a dict with keys: max_input_tokens, input_cost, output_cost.
"""
url_part = model[len("openrouter/"):]
url = "https://openrouter.ai/" + url_part
import requests
try:
response = requests.get(url, timeout=5)
if response.status_code != 200:
return {}
html = response.text
from bs4 import BeautifulSoup
soup = BeautifulSoup(html, "html.parser")
text = soup.get_text()
import re
context_match = re.search(r"([\d,]+)\s*context", text)
if context_match:
context_str = context_match.group(1).replace(",", "")
context_size = int(context_str)
else:
context_size = None
input_cost_match = re.search(r"\$\s*0\s*/M input tokens", text, re.IGNORECASE)
output_cost_match = re.search(r"\$\s*0\s*/M output tokens", text, re.IGNORECASE)
input_cost = 0 if input_cost_match else None
output_cost = 0 if output_cost_match else None
return {"max_input_tokens": context_size, "input_cost": input_cost, "output_cost": output_cost}
except Exception as e:
print("Error fetching openrouter info:", str(e))
return {}
model_info_manager = ModelInfoManager()