diff --git a/auto_update_pricing.py b/auto_update_pricing.py index bda59d398ef..f59d4b4729b 100644 --- a/auto_update_pricing.py +++ b/auto_update_pricing.py @@ -33,14 +33,17 @@ def update_local_data(local_data, remote_data): model_suffix = model_name[len("openrouter/"):] for model in remote_data["data"]: if model["id"] == model_suffix: - model_info["max_tokens"] = model["context_length"] - model_info["input_cost_per_token"] = model["pricing"]["prompt"] - model_info["output_cost_per_token"] = model["pricing"]["completion"] + # Update only the values that need to be updated + model_info.update({ + "max_tokens": model["context_length"], + "input_cost_per_token": model["pricing"]["prompt"], + "output_cost_per_token": model["pricing"]["completion"] + }) break # Add models not in local data yet for model in remote_data["data"]: model_id = model["id"] - if model_id not in local_data: + if f"openrouter/{model_id}" not in local_data: local_data[f"openrouter/{model_id}"] = { "max_tokens": model["context_length"], "input_cost_per_token": model["pricing"]["prompt"],