diff --git a/pyproject.toml b/pyproject.toml index 385cbd03..eec9fb02 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ dependencies = [ "numpy>=1.21, <3", "pandas>=1.3, <3", # core cai - "openai>=1.68.2", + "openai==1.75.0", "pydantic>=2.10, <3", "griffe>=1.5.6, <2", "typing-extensions>=4.12.2, <5", diff --git a/src/cai/util.py b/src/cai/util.py index 45206e7c..ee0ec2d9 100644 --- a/src/cai/util.py +++ b/src/cai/util.py @@ -288,7 +288,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 + "alias" not in model_str and "ollama" in model_str or "qwen" in model_str or "llama" in model_str or