diff --git a/src/cai/util.py b/src/cai/util.py index 13e5f482..161b8e5c 100644 --- a/src/cai/util.py +++ b/src/cai/util.py @@ -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