tPj)dZddlmZmZddlmZdeeeffdZdeeeffdZ deeeffdZ deeefde fd Z deeefde fd Z deeefde fd Zdeeefde fd Zy )z LiteLLM Web Search Tool Definition This module defines the standard web search tool used across LiteLLM. Native provider tools (like Anthropic's web_search_20250305) are converted to this format for consistent interception and execution. )AnyFinalLITELLM_WEB_SEARCH_TOOL_NAMEreturnc*tddddddidgddS) a= Get the standard LiteLLM web search tool definition. This is the canonical tool definition that all native web search tools (like Anthropic's web_search_20250305, Claude Code's web_search, etc.) are converted to for interception. Returns: Dict containing the Anthropic-style tool definition with: - name: Tool name - description: What the tool does - input_schema: JSON schema for tool parameters Example: >>> tool = get_litellm_web_search_tool() >>> tool['name'] 'litellm_web_search' Search the web for information. Use this when you need current information or answers to questions that require up-to-date data.objectquerystringThe search query to executetype descriptionr propertiesrequired)namer input_schemarv/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/websearch_interception/tools.pyget_litellm_web_search_toolrs;(- P$#@ !    rc0dtddddddidgdd d S) aE Get the standard LiteLLM web search tool definition in OpenAI format. Used by async_pre_call_deployment_hook which runs in the chat completions path where tools must be in OpenAI format (type: "function" with function.parameters). Returns: Dict containing the OpenAI-style tool definition. functionr r r r r rr)rr parameters)rrrrrr"get_litellm_web_search_tool_openair4sC0T! ('D %I    rc,dtddddddidgdd S) a Get the standard LiteLLM web search tool definition in Responses API format. Used by async_pre_call_deployment_hook on the Responses API path, where a function tool is a flat object (``type: "function"`` with a top-level ``name`` and ``parameters``) rather than the nested ``function`` wrapper used by Chat Completions. Returns: Dict containing the Responses-style function tool definition. rr r r r r rr)rrrrrrrr%get_litellm_web_search_tool_responsesrUs>, P$#@ !    rtoolc|jdd}t|tsy|dk(r|jdtk(S|dk(xs|j dS)a Check if a tool is a web search tool for the Responses API. Detects: - OpenAI native Responses web search tools, whose ``type`` is one of ``web_search``, ``web_search_2025_08_26``, ``web_search_preview``, ``web_search_preview_2025_03_11`` (matched by the ``web_search`` prefix) - The LiteLLM standard function tool in Responses shape: ``{"type": "function", "name": "litellm_web_search"}`` Args: tool: Tool dictionary to check Returns: True if tool is a Responses-API web search tool Example: >>> is_web_search_tool_responses({"type": "web_search"}) True >>> is_web_search_tool_responses({"type": "web_search_preview"}) True >>> is_web_search_tool_responses({"type": "function", "name": "litellm_web_search"}) True >>> is_web_search_tool_responses({"type": "function", "name": "get_weather"}) False rFrr web_search web_search_)get isinstancestrr startswithr tool_types ris_web_search_tool_responsesr+usY6xx+I i %Jxx#???  $ K (<(<](KKrc|jdd}|jdd}|dk(r2d|vr.|jdi}|jdd}|tk(ry|tk(ryy)a Check if a tool is a web search tool for Chat Completions API (strict check). This is a stricter version that ONLY checks for the exact LiteLLM web search tool name. Use this for Chat Completions API to avoid false positives with user-defined tools. Detects ONLY: - LiteLLM standard: name == "litellm_web_search" (Anthropic format) - OpenAI format: type == "function" with function.name == "litellm_web_search" Args: tool: Tool dictionary to check Returns: True if tool is exactly the LiteLLM web search tool Example: >>> is_web_search_tool_chat_completion({"name": "litellm_web_search"}) True >>> is_web_search_tool_chat_completion({"type": "function", "function": {"name": "litellm_web_search"}}) True >>> is_web_search_tool_chat_completion({"name": "web_search"}) False >>> is_web_search_tool_chat_completion({"name": "WebSearch"}) False rr"rrTF)r%rr tool_namer* function_def function_names r"is_web_search_tool_chat_completionr1st6xx+Ixx+IJ:#5"hhz26 +//; 8 800 rcx|jdd}t|tsy|jdxr|dk7S)a Check if a tool is an Anthropic-native ``web_search_*`` tool. Native clients (Anthropic SDK, Claude Desktop, Anthropic Console) send tools like ``{"type": "web_search_20250305", "name": "web_search"}`` and expect the response to contain ``web_search_tool_result`` content blocks so that citations can be rendered. This helper identifies that contract so the agentic loop can emit native-format blocks for those clients without affecting clients that send the LiteLLM standard tool. Returns False for the LiteLLM standard tool (``litellm_web_search``), the OpenAI-shaped variant, the bare ``WebSearch`` legacy name, and the bare ``web_search`` name (Claude Code style). rr"Fr$r)r%r&r'r(r)s r#is_anthropic_native_web_search_toolr3s<xx+I i %    . J9 3JJrc|jdd}|jdd}|dk(r2d|vr.|jdi}|jdd}|tk(ry|tk(ry|jdry|dk(r|ry|dk(rd |vryy ) u Check if a tool is a web search tool (native or LiteLLM standard). Detects: - LiteLLM standard: name == "litellm_web_search" - OpenAI format: type == "function" with function.name == "litellm_web_search" - Anthropic native: type starts with "web_search_" (e.g., "web_search_20250305") - Claude Code: name == "web_search" with a type field - Custom: name == "WebSearch" (legacy interception marker — only matched when input_schema is absent; see note below) Note on the legacy ``WebSearch`` name: Clients like Claude Desktop / Cowork ship a *client-side* tool called ``WebSearch`` (a fully-formed Anthropic client tool with its own ``input_schema``) that they handle themselves. Treating that as our interception marker hijacks it server-side and the client's own tool handler never fires — which means Cowork's separate native ``web_search_20250305`` sub-request (where citation data actually flows) never gets made. Real Anthropic client tools always carry an ``input_schema`` (the API rejects them otherwise), so a bare ``{name: "WebSearch"}`` with no schema is the only thing that could be a legacy interception marker. Gate the match on schema absence to keep both groups working. Args: tool: Tool dictionary to check Returns: True if tool is a web search tool Example: >>> is_web_search_tool({"name": "litellm_web_search"}) True >>> is_web_search_tool({"type": "function", "function": {"name": "litellm_web_search"}}) True >>> is_web_search_tool({"type": "web_search_20250305", "name": "web_search"}) True >>> is_web_search_tool({"name": "calculator"}) False >>> is_web_search_tool({"name": "WebSearch"}) # legacy interception marker True >>> is_web_search_tool({"name": "WebSearch", "input_schema": {"type": "object"}}) # Cowork client tool False rr"rrTr$r# WebSearchrF)r%rr(r-s ris_web_search_toolr6s\xx+Ixx+IJ:#5"hhz26 +//; 8 800M*L YKN$$> rN)__doc__typingrrlitellm.constantsrdictr'rrrboolr+r1r3r6rrrr<s:#T#s(^#LDcNBtCH~@"LtCH~"L$"LJ)T#s(^))XKd38nKK*IT#s(^IIr