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: Update cost extraction to capture non-zero input/output costs

Browse source
This commit is contained in:
Stefan Hladnik (aider) 2025-03-18 03:17:57 +07:00 • committed by Stefan Hladnik
parent 44e5525e6f
commit 2651d99676
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/models.py
View file

@ -254,10 +254,10 @@ class ModelInfoManager:
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
input_cost_match = re.search(r"\$\s*([\d.]+)\s*/M input 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
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}
print(f"Model '{model}': Parsed parameters: {params}")
return params