forked from Rockachopa/Timmy-time-dashboard
248af9ed03
* chore: stop tracking runtime-generated self-modify reports These 65 files in data/self_modify_reports/ are auto-generated at runtime and already listed in .gitignore. Tracking them caused conflicts when pulling from main. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix: resolve 8 dashboard bugs from Round 4 testing report - Fix Ollama timeout regression: request_timeout → timeout (agno API) - Add Bootstrap JS to base.html (fixes creative UI tab switching) - Send initial_state on Swarm Live WebSocket connect - Add /api/queue/status endpoint (stops 404 log spam from chat panel) - Populate agent tools from registry on /tools page - Add notification bell dropdown with /api/notifications endpoint - All 1157 tests pass Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> --------- Co-authored-by: Trip T <trip@local> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
"""Test that Ollama model is created with a generous timeout.
|
|
|
|
The default httpx timeout is too short for complex prompts (30-60s generation).
|
|
This caused socket read errors in production. The agno Ollama class uses
|
|
``timeout`` (not ``request_timeout``).
|
|
"""
|
|
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
|
|
def test_base_agent_sets_timeout():
|
|
"""BaseAgent creates Ollama with timeout=300."""
|
|
with patch("timmy.agents.base.Ollama") as mock_ollama, \
|
|
patch("timmy.agents.base.Agent"):
|
|
mock_ollama.return_value = MagicMock()
|
|
|
|
# Import after patching to get the patched version
|
|
from timmy.agents.base import BaseAgent
|
|
|
|
class ConcreteAgent(BaseAgent):
|
|
async def handle_message(self, message: str) -> str:
|
|
return ""
|
|
|
|
# Trigger Ollama construction
|
|
try:
|
|
ConcreteAgent(
|
|
agent_id="test",
|
|
name="Test",
|
|
role="tester",
|
|
system_prompt="You are a test agent.",
|
|
tools=[],
|
|
)
|
|
except Exception:
|
|
pass # MCP registry may not be available
|
|
|
|
# Verify Ollama was called with timeout
|
|
if mock_ollama.called:
|
|
_, kwargs = mock_ollama.call_args
|
|
assert kwargs.get("timeout") == 300, (
|
|
f"Expected timeout=300, got {kwargs.get('timeout')}"
|
|
)
|
|
|
|
|
|
def test_main_agent_sets_timeout():
|
|
"""create_timmy() creates Ollama with timeout=300."""
|
|
with patch("timmy.agent.Ollama") as mock_ollama, \
|
|
patch("timmy.agent.SqliteDb"), \
|
|
patch("timmy.agent.Agent"):
|
|
mock_ollama.return_value = MagicMock()
|
|
|
|
from timmy.agent import create_timmy
|
|
try:
|
|
create_timmy()
|
|
except Exception:
|
|
pass
|
|
|
|
if mock_ollama.called:
|
|
_, kwargs = mock_ollama.call_args
|
|
assert kwargs.get("timeout") == 300, (
|
|
f"Expected timeout=300, got {kwargs.get('timeout')}"
|
|
)
|