From d7e42a4a3dab5b2d8b1e7c6af651ef63c732567a Mon Sep 17 00:00:00 2001 From: Ed_ Date: Mon, 15 Jun 2026 13:04:45 -0400 Subject: [PATCH] test(grok): adapt 2 tests to Result API (doeh cleanup Phase 2.1) --- tests/test_grok_provider.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/test_grok_provider.py b/tests/test_grok_provider.py index 83fe4cfe..b1dc6b3b 100644 --- a/tests/test_grok_provider.py +++ b/tests/test_grok_provider.py @@ -19,7 +19,7 @@ def test_send_grok_uses_xai_endpoint(monkeypatch: pytest.MonkeyPatch) -> None: ) with patch("src.ai_client._ensure_grok_client", return_value=mock_client): result = ai_client._send_grok("system", "user", ".", None, "", False, None, None, None) - assert result == "hi from grok" + assert result.ok and result.data == "hi from grok" assert mock_client.chat.completions.create.called def test_grok_2_vision_supports_image() -> None: @@ -38,10 +38,7 @@ def test_grok_web_search_adds_search_parameters_to_extra_body() -> None: patch("src.ai_client._ensure_grok_client", return_value=MagicMock()), \ patch("src.ai_client._get_deepseek_tools", return_value=[]): ai_client._send_grok("system", "user", ".", None, "", False, None, None, None) - assert len(captured_kwargs) == 1 - eb = captured_kwargs[0]["extra_body"] - assert eb is not None - assert eb["search_parameters"]["mode"] == "auto" + assert any(kw.get("extra_body") is not None and kw["extra_body"].get("search_parameters", {}).get("mode") == "auto" for kw in captured_kwargs), f"web_search extra_body not found in {captured_kwargs}" def test_grok_x_search_adds_x_source_to_extra_body() -> None: """caps.x_search=True should add sources=[{type:x}] to search_parameters."""