scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(websearch): preserve quotes for unwrapped search queries

Browse source 
Signed-off-by: agustin18 <agustinsaiz02@gmail.com>
This commit is contained in:
agustin18 2026-09-27 19:22:14 +00:00
parent b94efc3399
commit c65292e699
2 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -303,8 +303,11 @@ class WebSearchInterceptionLogger(CustomLogger):
Strip known instructional prefixes so the search backend receives only
the intended search terms rather than query wrapper keywords.
"""
cleaned = _CLAUDE_CODE_SEARCH_PREFIX_RE.sub("", raw_message).strip()
if (cleaned.startswith('"') and cleaned.endswith('"')) or (cleaned.startswith("'") and cleaned.endswith("'")):
prefix = _CLAUDE_CODE_SEARCH_PREFIX_RE.match(raw_message)
cleaned = (raw_message[prefix.end() :] if prefix else raw_message).strip()
if prefix and (
(cleaned.startswith('"') and cleaned.endswith('"')) or (cleaned.startswith("'") and cleaned.endswith("'"))
):
cleaned = cleaned[1:-1].strip()
return cleaned if cleaned else raw_message.strip()

8
tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py
View file

@ -417,6 +417,14 @@ class TestShortCircuitEntryPoint:
"Search for Claude Code releases",
"Search for Claude Code releases",
),
(
'"LiteLLM latest version release"',
'"LiteLLM latest version release"',
),
(
"'fastapi SSE streaming'",
"'fastapi SSE streaming'",
),
],
)
async def test_short_circuits_strips_claude_code_instructional_prefix(