fix: mock get_model in tests to avoid API key requirement in CI

tests/unit/agents/test_hypothesis_agent.py

@@ -48,19 +48,21 @@ async def test_hypothesis_agent_generates_hypotheses(sample_evidence, mock_asses
     """HypothesisAgent should generate mechanistic hypotheses."""
     store = {"current": sample_evidence, "hypotheses": []}
 
-    with patch("src.agents.hypothesis_agent.Agent") as mock_agent_class:
-        mock_result = MagicMock()
-        mock_result.output = mock_assessment
-        # pydantic-ai Agent returns an object with .output for structured output
-        mock_agent_class.return_value.run = AsyncMock(return_value=mock_result)
+    with patch("src.agents.hypothesis_agent.get_model") as mock_get_model:
+        with patch("src.agents.hypothesis_agent.Agent") as mock_agent_class:
+            mock_get_model.return_value = MagicMock()  # Mock model
+            mock_result = MagicMock()
+            mock_result.output = mock_assessment
+            # pydantic-ai Agent returns an object with .output for structured output
+            mock_agent_class.return_value.run = AsyncMock(return_value=mock_result)
 
-        agent = HypothesisAgent(store)
-        response = await agent.run("metformin cancer")
+            agent = HypothesisAgent(store)
+            response = await agent.run("metformin cancer")
 
-        assert isinstance(response, AgentRunResponse)
-        assert "AMPK" in response.messages[0].text
-        assert len(store["hypotheses"]) == 1
-        assert store["hypotheses"][0].drug == "Metformin"
+            assert isinstance(response, AgentRunResponse)
+            assert "AMPK" in response.messages[0].text
+            assert len(store["hypotheses"]) == 1
+            assert store["hypotheses"][0].drug == "Metformin"
 
 
 @pytest.mark.asyncio
@@ -68,12 +70,12 @@ async def test_hypothesis_agent_no_evidence():
     """HypothesisAgent should handle empty evidence gracefully."""
     store = {"current": [], "hypotheses": []}
 
-    with patch("src.agents.hypothesis_agent.Agent"):
-        agent = HypothesisAgent(store)
-        response = await agent.run("test query")
+    # No need to mock Agent/get_model - empty evidence returns early
+    agent = HypothesisAgent(store)
+    response = await agent.run("test query")
 
-        assert "No evidence" in response.messages[0].text
-        assert len(store["hypotheses"]) == 0
+    assert "No evidence" in response.messages[0].text
+    assert len(store["hypotheses"]) == 0
 
 
 @pytest.mark.asyncio
@@ -82,20 +84,22 @@ async def test_hypothesis_agent_uses_embeddings(sample_evidence, mock_assessment
     store = {"current": sample_evidence, "hypotheses": []}
     mock_embeddings = MagicMock()
 
-    with patch("src.agents.hypothesis_agent.Agent") as mock_agent_class:
-        # Mock format_hypothesis_prompt to check if embeddings were passed
-        with patch("src.agents.hypothesis_agent.format_hypothesis_prompt") as mock_format:
-            mock_format.return_value = "Prompt"
-
-            mock_result = MagicMock()
-            mock_result.output = mock_assessment
-            mock_agent_class.return_value.run = AsyncMock(return_value=mock_result)
-
-            agent = HypothesisAgent(store, embedding_service=mock_embeddings)
-            await agent.run("query")
-
-            mock_format.assert_called_once()
-            _args, kwargs = mock_format.call_args
-            assert kwargs["embeddings"] == mock_embeddings
-            assert _args[0] == "query"  # First positional arg is query
-            assert _args[1] == sample_evidence  # Second positional arg is evidence
+    with patch("src.agents.hypothesis_agent.get_model") as mock_get_model:
+        with patch("src.agents.hypothesis_agent.Agent") as mock_agent_class:
+            # Mock format_hypothesis_prompt to check if embeddings were passed
+            with patch("src.agents.hypothesis_agent.format_hypothesis_prompt") as mock_format:
+                mock_get_model.return_value = MagicMock()  # Mock model
+                mock_format.return_value = "Prompt"
+
+                mock_result = MagicMock()
+                mock_result.output = mock_assessment
+                mock_agent_class.return_value.run = AsyncMock(return_value=mock_result)
+
+                agent = HypothesisAgent(store, embedding_service=mock_embeddings)
+                await agent.run("query")
+
+                mock_format.assert_called_once()
+                _args, kwargs = mock_format.call_args
+                assert kwargs["embeddings"] == mock_embeddings
+                assert _args[0] == "query"  # First positional arg is query
+                assert _args[1] == sample_evidence  # Second positional arg is evidence