scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

Merge 6bb26bd6a3 into 5337c68dd3

Browse source
This commit is contained in:
Edward Yi 2026-08-28 03:09:07 +00:00 • committed by GitHub
commit 47d4539d66
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/integrations/websearch_interception/handler.py
View file

@ -1109,6 +1109,7 @@ class WebSearchInterceptionLogger(CustomLogger):
k: v
for k, v in kwargs.items()
if not k.startswith("_websearch_interception")
and k not in optional_params_clean
and k
not in {
"_agentic_loop_api_surface",

8
tests/test_litellm/integrations/websearch_interception/test_websearch_responses.py
View file

@ -218,12 +218,16 @@ async def test_build_responses_plan_produces_responses_input():
optional_params={
"tools": [{"type": "function", "name": "litellm_web_search"}],
"tool_choice": {"type": "function", "name": "litellm_web_search"},
"prompt_cache_key": "cache-key",
"prompt_cache_retention": "24h",
},
logging_obj=MagicMock(),
stream=False,
kwargs={
"custom_llm_provider": "openai",
"_agentic_loop_api_surface": RESPONSES_AGENTIC_SURFACE,
"prompt_cache_key": "cache-key",
"prompt_cache_retention": "24h",
},
)
@ -249,6 +253,10 @@ async def test_build_responses_plan_produces_responses_input():
assert patch_obj.tools == [{"type": "function", "name": "litellm_web_search"}]
assert "tool_choice" not in patch_obj.optional_params
assert "_agentic_loop_api_surface" not in patch_obj.kwargs
assert patch_obj.optional_params["prompt_cache_key"] == "cache-key"
assert patch_obj.optional_params["prompt_cache_retention"] == "24h"
assert "prompt_cache_key" not in patch_obj.kwargs
assert "prompt_cache_retention" not in patch_obj.kwargs
assert patch_obj.model == "openai/gpt-4o"