From fcc499e4011b1059ef73dbcd58056af0cf5268e8 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 10 Jan 2025 15:51:53 -0800 Subject: [PATCH] refactor: normalize deepseek model names to deepseek-chat --- scripts/my_models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/my_models.py b/scripts/my_models.py index f3a7fbf13..750c3b2ea 100755 --- a/scripts/my_models.py +++ b/scripts/my_models.py @@ -19,9 +19,11 @@ def collect_model_stats(n_lines=1000): properties = event["properties"] main_model = properties.get("main_model") - # combine deepseek/deepseek-coder into ...-chat ai! total_tokens = properties.get("total_tokens", 0) if main_model: + # Normalize deepseek model names + if "deepseek" in main_model.lower(): + main_model = "deepseek-chat" model_stats[main_model] += total_tokens except json.JSONDecodeError: continue