mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 12:24:59 +00:00
refactor: normalize deepseek model names to deepseek-chat
This commit is contained in:
parent
6e8efe22aa
commit
fcc499e401
1 changed files with 3 additions and 1 deletions
|
@ -19,9 +19,11 @@ def collect_model_stats(n_lines=1000):
|
||||||
properties = event["properties"]
|
properties = event["properties"]
|
||||||
main_model = properties.get("main_model")
|
main_model = properties.get("main_model")
|
||||||
|
|
||||||
# combine deepseek/deepseek-coder into ...-chat ai!
|
|
||||||
total_tokens = properties.get("total_tokens", 0)
|
total_tokens = properties.get("total_tokens", 0)
|
||||||
if main_model:
|
if main_model:
|
||||||
|
# Normalize deepseek model names
|
||||||
|
if "deepseek" in main_model.lower():
|
||||||
|
main_model = "deepseek-chat"
|
||||||
model_stats[main_model] += total_tokens
|
model_stats[main_model] += total_tokens
|
||||||
except json.JSONDecodeError:
|
except json.JSONDecodeError:
|
||||||
continue
|
continue
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue