dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-06 04:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

fix: Update return keys in fetch_openrouter_model_info() to match JSON metadata

Browse source
This commit is contained in:
Stefan Hladnik (aider) 2025-03-18 03:24:04 +07:00 • committed by Stefan Hladnik
parent 2651d99676
commit 29587cd07c
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aider/models.py
View file

@ -236,7 +236,7 @@ class ModelInfoManager:
Fetch model info by scraping the openrouter model page. Fetch model info by scraping the openrouter model page.
Expected URL: https://openrouter.ai/<model_route> Expected URL: https://openrouter.ai/<model_route>
Example: openrouter/qwen/qwen-2.5-72b-instruct:free Example: openrouter/qwen/qwen-2.5-72b-instruct:free
Returns a dict with keys: max_input_tokens, input_cost, output_cost. Returns a dict with keys: max_tokens, max_input_tokens, max_output_tokens, input_cost_per_token, output_cost_per_token.
""" """
url_part = model[len("openrouter/"):] url_part = model[len("openrouter/"):]
url = "https://openrouter.ai/" + url_part url = "https://openrouter.ai/" + url_part
@ -258,7 +258,13 @@ class ModelInfoManager:
output_cost_match = re.search(r"\$\s*([\d.]+)\s*/M output tokens", text, re.IGNORECASE) output_cost_match = re.search(r"\$\s*([\d.]+)\s*/M output tokens", text, re.IGNORECASE)
input_cost = float(input_cost_match.group(1)) if input_cost_match else None input_cost = float(input_cost_match.group(1)) if input_cost_match else None
output_cost = float(output_cost_match.group(1)) if output_cost_match else None output_cost = float(output_cost_match.group(1)) if output_cost_match else None
params = {"max_input_tokens": context_size, "input_cost": input_cost, "output_cost": output_cost} params = {
"max_input_tokens": context_size,
"max_tokens": context_size,
"max_output_tokens": context_size,
"input_cost_per_token": input_cost,
"output_cost_per_token": output_cost,
}
print(f"Model '{model}': Parsed parameters: {params}") print(f"Model '{model}': Parsed parameters: {params}")
return params return params
except Exception as e: except Exception as e: