stream responses and all tests passing

tests/test_loop.py

@@ -1,4 +1,4 @@
-from unittest.mock import AsyncMock, patch
+from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 
@@ -16,13 +16,13 @@ async def test_run_turn_basic(mock_anthropic_client):
         result = await run_turn("What is 2+2?")
 
     # verify client was called
-    mock_anthropic_client.messages.create.assert_called_once()
+    mock_anthropic_client.messages.stream.assert_called_once()
 
     # verify message returned
     assert result == "42"
 
     # verify call has correct parameters
-    call_args = mock_anthropic_client.messages.create.call_args
+    call_args = mock_anthropic_client.messages.stream.call_args
 
 
 @pytest.mark.unit
@@ -34,7 +34,7 @@ async def test_run_turn_with_history(mock_anthropic_client, sample_history):
     with patch("agent.loop.client", mock_anthropic_client):
         result = await run_turn("What is 2+2?", history=sample_history)
 
-    call_args = mock_anthropic_client.messages.create.call_args
+    call_args = mock_anthropic_client.messages.stream.call_args
     messages = call_args.kwargs["messages"]
 
     # verify all history was included plus new message
@@ -53,7 +53,7 @@ async def test_run_turn_uses_settings(mock_anthropic_client, settings):
             await run_turn("test message")
 
     # Verify settings were used
-    call_args = mock_anthropic_client.messages.create.call_args
+    call_args = mock_anthropic_client.messages.stream.call_args
     assert call_args.kwargs["model"] == settings.model
     assert call_args.kwargs["max_tokens"] == settings.max_tokens
 
@@ -63,16 +63,18 @@ async def test_run_turn_uses_settings(mock_anthropic_client, settings):
 async def test_run_session_calls_run_turn_with_user_input():
     """Test that user input is passed to run_turn"""
 
-    # Mock the input()
-    with patch("builtins.input", side_effect=["hello", KeyboardInterrupt]):
-        # Mock run_turn to avoid actually calling the API
-        with patch("agent.loop.run_turn", new_callable=AsyncMock) as mock_run_turn:
-            mock_run_turn.return_value = AsyncMock(content=[AsyncMock(text="response")])
+    mock_history = MagicMock()
+    mock_history.get_all.return_value = []
+    mock_history.session_id = "test-session"
 
-            try:
+    with patch("agent.loop.ConversationHistory", return_value=mock_history):
+        with patch("agent.loop.run_turn", new_callable=AsyncMock) as mock_run_turn:
+            mock_run_turn.return_value = "response text"
+
+            # Provide inputs: command, then quit
+            inputs = iter(["hello", "/quit"])
+            with patch("builtins.input", side_effect=lambda _: next(inputs)):
                 await run_session()
-            except KeyboardInterrupt:
-                pass
 
     mock_run_turn.assert_called()
     assert mock_run_turn.call_args.args[0] == "hello"