ed/manual_slop
Private
Public Access
0
0
Fork 0
Code Activity

refactor(schemas): remove NormalizedResponse backward-compat __init__; use canonical API

Browse Source
This commit is contained in:
ed
2026-06-24 17:12:49 -04:00
parent 03dd44c642
commit 20236546d7
9 changed files with 29 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_grok_provider.py
+5 -3
View File
@@ -30,10 +30,11 @@ def test_grok_2_vision_supports_image() -> None:
def test_grok_web_search_adds_search_parameters_to_extra_body() -> None:
"""caps.web_search=True should populate search_parameters.mode=auto in extra_body."""
from src import openai_compatible as oc
from src.openai_schemas import UsageStats
captured_kwargs: list[dict] = []
def _fake_send(client, request, *, capabilities):
captured_kwargs.append({"extra_body": request.extra_body, "model": request.model})
return MagicMock(text="ok", tool_calls=[], usage_input_tokens=0, usage_output_tokens=0, usage_cache_read_tokens=0, usage_cache_creation_tokens=0, raw_response=None)
return MagicMock(text="ok", tool_calls=[], usage=UsageStats(input_tokens=0, output_tokens=0, cache_read_tokens=0, cache_creation_tokens=0), raw_response=None)
with patch.object(oc, "send_openai_compatible", side_effect=_fake_send), \
patch("src.ai_client._ensure_grok_client", return_value=MagicMock()), \
patch("src.ai_client._get_deepseek_tools", return_value=[]):
@@ -43,12 +44,13 @@ def test_grok_web_search_adds_search_parameters_to_extra_body() -> None:
def test_grok_x_search_adds_x_source_to_extra_body() -> None:
"""caps.x_search=True should add sources=[{type:x}] to search_parameters."""
from src import openai_compatible as oc
from src.openai_schemas import UsageStats
captured_kwargs: list[dict] = []
def _fake_send(client, request, *, capabilities):
captured_kwargs.append({"extra_body": request.extra_body})
return MagicMock(text="ok", tool_calls=[], usage_input_tokens=0, usage_output_tokens=0, usage_cache_read_tokens=0, usage_cache_creation_tokens=0, raw_response=None)
return MagicMock(text="ok", tool_calls=[], usage=UsageStats(input_tokens=0, output_tokens=0, cache_read_tokens=0, cache_creation_tokens=0), raw_response=None)
with patch.object(oc, "send_openai_compatible", side_effect=_fake_send), \
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 captured_kwargs[0]["extra_body"]["search_parameters"]["sources"] == [{"type": "x"}]
assert captured_kwargs[0]["extra_body"]["search_parameters"]["sources"] == [{"type": "x"}]