Rockachopa/Timmy-time-dashboard
1
0
Fork 2
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: update agno to v2 and fix airllm availability tests (#170)

Browse Source 
The agno dependency was pinned to <2.0 but the code uses agno.db.sqlite
(a 2.x API), breaking all tests in CI. Also fix airllm provider tests
to patch importlib.util.find_spec (what the production code uses) instead
of builtins.__import__.

Co-authored-by: Trip T <trip@local>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Alexander Whitestone
2026-03-11 12:40:45 -04:00
committed by GitHub
parent 9d78eb31d1
commit 68115fe477
2 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@@ -24,7 +24,7 @@ packages = [
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.11,<4" python = ">=3.11,<4"
agno = { version = ">=1.4.0,<2.0", extras = ["sqlite"] } agno = { version = ">=2.0.0,<3.0", extras = ["sqlite"] }
ollama = ">=0.3.0,<1.0" ollama = ">=0.3.0,<1.0"
openai = ">=1.0.0" openai = ">=1.0.0"
fastapi = ">=0.115.0,<1.0" fastapi = ">=0.115.0,<1.0"

11
tests/infrastructure/test_router_cascade.py
View File

@@ -500,8 +500,7 @@ class TestProviderAvailabilityCheck:
priority=1, priority=1,
) )
with patch("builtins.__import__") as mock_import: with patch("importlib.util.find_spec", return_value=MagicMock()):
mock_import.return_value = MagicMock()
assert router._check_provider_available(provider) is True assert router._check_provider_available(provider) is True
def test_check_airllm_not_installed(self): def test_check_airllm_not_installed(self):
@@ -515,11 +514,5 @@ class TestProviderAvailabilityCheck:
priority=1, priority=1,
) )
# Patch __import__ to simulate airllm not being available with patch("importlib.util.find_spec", return_value=None):
def raise_import_error(name, *args, **kwargs):
if name == "airllm":
raise ImportError("No module named 'airllm'")
return __builtins__.__import__(name, *args, **kwargs)
with patch("builtins.__import__", side_effect=raise_import_error):
assert router._check_provider_available(provider) is False assert router._check_provider_available(provider) is False