fix: Return empty dict if any required parameters are missing

This commit is contained in:
Stefan Hladnik (aider) 2025-03-18 03:46:36 +07:00 committed by Stefan Hladnik
parent b37773c630
commit 87ccacb99f

View file

@ -258,6 +258,8 @@ class ModelInfoManager:
output_cost_match = re.search(r"\$\s*([\d.]+)\s*/M output tokens", text, re.IGNORECASE)
input_cost = float(input_cost_match.group(1)) / 1000000 if input_cost_match else None
output_cost = float(output_cost_match.group(1)) / 1000000 if output_cost_match else None
if context_size is None or input_cost is None or output_cost is None:
return {}
params = {
"max_input_tokens": context_size,
"max_tokens": context_size,