Support pricing in alias0

This commit is contained in:
luijait 2025-05-21 22:32:11 +02:00
parent a421eb7fdf
commit b4571bb4b7
1 changed files with 2 additions and 0 deletions

View File

@ -283,6 +283,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
@ -386,6 +387,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