test(grok): adapt 2 tests to Result API (doeh cleanup Phase 2.1)
This commit is contained in:
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user