RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support pricing in alias0

This commit is contained in:
luijait 2025-05-21 22:32:11 +02:00
parent cde27c8ea1
commit caf0e054d7
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cai/util.py
View File

@ -279,6 +279,7 @@ class CostTracker:
# Check if this is a local/free model
model_str = model_name.lower()
is_local_model = (
"alias" not in model_str and
"ollama" in model_str or
"qwen" in model_str or
"llama" in model_str or
@ -382,6 +383,7 @@ class CostTracker:
# Check if this is a local model (always free) first
model_str = model_name.lower()
is_local_model = (
"alias" not in model_str and
"ollama" in model_str or
"qwen" in model_str or
"llama" in model_str or