diff --git a/tests/test_ai_client_tool_loop_builder.py b/tests/test_ai_client_tool_loop_builder.py index be30b880..e7fae125 100644 --- a/tests/test_ai_client_tool_loop_builder.py +++ b/tests/test_ai_client_tool_loop_builder.py @@ -9,6 +9,7 @@ from typing import Any from unittest.mock import MagicMock, patch from src.openai_compatible import NormalizedResponse, OpenAICompatibleRequest from src.ai_client import run_with_tool_loop +from src.result_types import Result from src.vendor_capabilities import VendorCapabilities def _make_normalized_response(text: str = "ok", tool_calls: list[dict[str, Any]] | None = None) -> NormalizedResponse: @@ -30,7 +31,7 @@ def test_run_with_tool_loop_calls_request_builder_each_round() -> None: def builder(round_idx: int) -> OpenAICompatibleRequest: builder_calls.append(round_idx) return OpenAICompatibleRequest(messages=[{"role": "user", "content": f"round={round_idx}"}], model="m") - with patch("src.openai_compatible.send_openai_compatible", side_effect=[tool_response, final]), \ + with patch("src.openai_compatible.send_openai_compatible", side_effect=[Result(data=tool_response), Result(data=final)]), \ patch("src.ai_client._execute_tool_calls_concurrently", return_value=[("noop", "c1", "r", "")]): result = run_with_tool_loop( client, builder, capabilities=caps,