tPjMXdZddlZddlmZmZddlmZddlmZddl m Z GddZ y) zo WebSearch Tool Transformation Transforms between Anthropic/OpenAI tool_use format and LiteLLM search format. N)AnyFinal)verbose_logger)LITELLM_WEB_SEARCH_TOOL_NAME)SearchResponseceZdZdZe ddedededeee e ffdZ ededeee e ffdZ ededeee e ffdZ ededeee e ffd Ze dd e e d e eded e e d zdee e e e zff dZe dd e e d e ed e e d zdee e ffdZed e e d e edee e e ffdZededed zde eeffdZededefdZy )WebSearchTransformationu  Transformation class for WebSearch tool interception. Handles transformation between: - Anthropic tool_use format → LiteLLM search requests - OpenAI tool_calls format → LiteLLM search requests - LiteLLM SearchResponse → Anthropic/OpenAI tool_result format responsestreamresponse_formatreturnc|rtjddgfS|dk(rtj|S|dk(rtj |Stj |S)a Transform model response to extract WebSearch tool calls. Detects if response contains WebSearch tool_use/tool_calls blocks and extracts the search queries for execution. Args: response: Model response (dict, AnthropicMessagesResponse, or ModelResponse) stream: Whether response is streaming response_format: Response format - "anthropic" or "openai" (default: "anthropic") Returns: (has_websearch, tool_calls): has_websearch: True if WebSearch tool_use found tool_calls: List of tool_use/tool_calls dicts with id, name, input/function Note: Streaming requests are handled by converting stream=True to stream=False in the WebSearchInterceptionLogger.async_log_pre_api_call hook before the API request is made. This means by the time this method is called, streaming requests have already been converted to non-streaming. zKWebSearchInterception: Unexpected streaming response, skipping interceptionFopenai responses)rwarningr _detect_from_openai_response_detect_from_responses_response#_detect_from_non_streaming_response)r r r s /home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/websearch_interception/transformation.pytransform_requestz)WebSearchTransformation.transform_requestsg8   " "#p q"9  h &*GGQ Q  +*JJ8T T*NNxX Xc t|tr|jdg}nt|ddxsg}t|tsdgfSg}|D]2}t|trF|jd}|jd}|jd}|jdd}n4t|dd}t|dd}t|dd}t|dd}|d k7s |t k7rt|t r |rtj|ni}nt|tr|}ni}t|t r|ntj|} |j||d || |d tjd ||5t|d kD|fS#tj$rtjd |i}YwxYw)aEParse a Responses API response for ``litellm_web_search`` function calls. After pre-request conversion the native web search tool is replaced by a ``litellm_web_search`` function tool, so the model emits ``function_call`` items in ``response.output`` instead of a native ``web_search_call``. outputNFtypenamecall_id arguments function_callzBWebSearchInterception: Failed to parse function_call arguments: %s)idrrrrinputz=WebSearchInterception: Found %s function_call with call_id=%sr) isinstancedictgetgetattrlistrstrjsonloadsJSONDecodeErrorrrdumpsappenddebuglen) r r tool_callsitem item_type item_namerr parsed_input arguments_strs rrz7WebSearchTransformation._detect_from_responses_responseCs h %\\(B/FXx6<"F&$'"9 (* ' vD$% HHV,  HHV, ((9- HH["5 #D&$7 #D&$7 !$ 48#D+r: O+yBc (d|Dcgc]H}|dd|dt|dtrtj|dn t |dddJc}d}t t |Dcgc]}d ||d||d }}||fScc}wcc}w) zNTransform to OpenAI format (assistant with tool_calls, separate tool messages)rSr rCrr!rD)r rrC)rTr/tool)rT tool_call_idr6)r"r#r(r+r'rZr.)r/rLr\r]r^ tool_messagess rrOz2WebSearchTransformation._transform_response_openails % T(& "6 AKBwKY]A^djjG&=dghjkrhsdt!  $ ,3z?+    *1 d 3)!,   !-///  s A B .BrXsearch_responsec g}|}t|ddxsg}|D]g}t|ddxsd}t|ddxsd}t|ddxs t|dd}|jd|||dt|d dxsdd id ||d S) u. Build an Anthropic-native ``web_search_tool_result`` content block. Native Anthropic clients (Claude Desktop, the Anthropic SDK, the Anthropic Console) expect search-tool results to be returned as structured ``web_search_tool_result`` blocks so that citations and source links can be rendered. The agentic loop currently feeds the model a flat text blob in the follow-up call (which is correct — the model needs readable evidence). This helper produces the *additional* block that should accompany the model's text reply when the original request used a native ``web_search_*`` tool. The spec'd shape carries page text only in ``encrypted_content``, an opaque server-issued blob that we cannot mint. Emitting the four spec fields alone would drop the snippet entirely, leaving the client (and the model, on any replayed follow-up turn) with URLs and titles but no evidence to answer from, forcing a fetch per result. So the snippet is carried in an additive ``snippet`` key alongside the spec fields. ``encrypted_content`` stays empty rather than holding plaintext, which would assert encryption semantics that do not hold. Spec reference: https://docs.anthropic.com/en/api/web-search-tool Args: tool_use_id: The ``tool_use_id`` the model emitted on the first turn. Must match exactly so the client can pair the result with its tool_use block. search_response: Structured ``SearchResponse`` from ``litellm.asearch()``. If None or empty, the block is still emitted with an empty result list (signals "search ran, no results" rather than "search did not run"). Nresultsurlrtitledate last_updatedweb_search_resultsnippet)rrgrhpage_ageencrypted_contentrlweb_search_tool_resultrW)r%r,)rXrditemsrfrrgrhrms r"build_web_search_tool_result_blockz:WebSearchTransformation.build_web_search_tool_result_blocksL.0  &$_iFL"G a+1r7B/52"1fd3Wwq.RV7W 3"!&$,-/#*1i#<#B   -&  rresultc t|dr^|jrRdj|jDcgc]+}d|jd|jd|j -c}}|St |}|Scc}w)z Format SearchResponse as text for tool_result content. Args: result: SearchResponse from litellm.asearch() Returns: Formatted text with Title, URL, Snippet for each result rfz zTitle: z URL: z Snippet: )r9rfjoinrhrgrlr')rsrqsearch_result_texts rformat_search_responsez.WebSearchTransformation.format_search_responsesv 69 %&..!'RXR`R`aQ7177)7155'QYYKHa"  "!"%V !! bs0A8) anthropic)rxN)N)__name__ __module__ __qualname____doc__ staticmethodrboolr'tupler&r#rrrrrQrPrOrrrrwrrr r s` +'Y'Y'Y'Y tT$Z 'Y'YR;/;/ tT$Z ;/;/z5/5/ tT$Z 5/5/n\/\/ tT$Z \/\/| +-1 JS dd*  tTDJ&& ' B.2//J//S //dd*// tTz  ////b0J0S 0 tT$Z 00B: : '$.:  c3h: : x"~"#""rr ) r|r(typingrrlitellm._loggingrlitellm.constantsr+litellm.llms.base_llm.search.transformationrr rrrrs'  +:FP"P"r