o ưi6@sbdZddlZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZGdddZdS) zo WebSearch Tool Transformation Transforms between Anthropic/OpenAI tool_use format and LiteLLM search format. N)AnyDictListOptionalTupleUnion)verbose_logger)LITELLM_WEB_SEARCH_TOOL_NAME)SearchResponsec@sLeZdZdZe ddedededeee e ffddZ ededeee e ffd d Z ededeee e ffd d Z e  dde e de ededee e dee ee e e fff ddZe dde e de edee e dee e ffddZede e de edee e e ffddZededefddZd S)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 anthropicresponsestreamresponse_formatreturncCs2|r tddgfS|dkrt|St|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)rwarningr _detect_from_openai_response#_detect_from_non_streaming_response)r rrrq/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/websearch_interception/transformation.pytransform_requests  z)WebSearchTransformation.transform_requestc Cs t|tr |dg}nt|dstddgfS|jpg}|s*tddgfSg}|D]Y}t|trK|d}|d}|d}|di}nt|dd }t|dd }t|dd }t|di}|d kr|td d fvr|d ||d }| |td|d|dq.t |dk|fS)z3Parse non-streaming response for WebSearch tool_usecontentz8WebSearchInterception: Response has no content attributeFz1WebSearchInterception: Response has empty contenttypenameidinputNtool_use WebSearch web_search)rrrrWebSearchInterception: Found z tool_use with id=r) isinstancedictgethasattrrdebugrgetattrr appendlen) r r tool_callsblockZ block_typeZ block_nameZblock_idZ block_input tool_callrrrrBsL             z;WebSearchTransformation._detect_from_non_streaming_responsec CsDt|tr |dg}nt|dstddgfS|jpg}|s*tddgfS|d}t|tr:|di}nt|dd}|sKtddgfSt|trW|d g}nt|d dp^g}|sjtd dgfSg}|D]}t|tr|d }|d }|d i} t| tr| dnt| dd} t| tr| dnt| dd} n&t|d d}t|d d}t|d d} | rt| ddnd} | rt| ddnd} |d kr| tddfvrt| t rzt | } Wnt j yt d| i} Ynw| pi} |d | | | d| d} || td| d|qnt|dk|fS)z4Parse OpenAI-style response for WebSearch tool_callschoicesz8WebSearchInterception: Response has no choices attributeFz1WebSearchInterception: Response has empty choicesrmessageNz2WebSearchInterception: First choice has no messager)z0WebSearchInterception: Message has no tool_callsrrfunctionr argumentsrrz;WebSearchInterception: Failed to parse function arguments: rr/)rrrr.rr z tool_call with id=)r!r"r#r$rr%r,r&r strjsonloadsJSONDecodeErrorrr'r()r r,Z first_choicer-Zopenai_tool_callsr)r+Ztool_idZ tool_typer.Z function_nameZfunction_argumentsr/Ztool_call_dictrrrr{s           "      z4WebSearchTransformation._detect_from_openai_responseNr)search_resultsthinking_blockscCs$|dkr t||Stj|||dS)a Transform LiteLLM search results to Anthropic/OpenAI tool_result format. Builds the assistant and user/tool messages needed for the agentic loop follow-up request. Args: tool_calls: List of tool_use/tool_calls dicts from transform_request search_results: List of search result strings (one per tool_call) response_format: Response format - "anthropic" or "openai" (default: "anthropic") thinking_blocks: Optional list of thinking/redacted_thinking blocks from the model's response. When present, prepended to the assistant message content (required by Anthropic API when thinking is enabled). Returns: (assistant_message, user_or_tool_messages): For Anthropic: assistant_message with tool_use blocks, user_message with tool_result blocks For OpenAI: assistant_message with tool_calls, tool_messages list with tool results r)r6)r _transform_response_openai_transform_response_anthropic)r)r5rr6rrrtransform_responsesz*WebSearchTransformation.transform_responsecsZg}|r |||ddDd|d}dfddttDd}||fS)zKTransform to Anthropic format (single user message with tool_result blocks)cSs&g|]}d|d|d|ddqS)rrrr)rrrrr.0Ztcrrr szIWebSearchTransformation._transform_response_anthropic.. assistant)roleruserc$g|]}d|d|dqS)Z tool_resultr)rZ tool_use_idrrr;ir5r)rrr<( )extendranger()r)r5r6Zassistant_contentassistant_messageZ user_messagerrCrr8s"     z5WebSearchTransformation._transform_response_anthropiccs8dddDd}fddttD}||fS)zNTransform to OpenAI format (assistant with tool_calls, separate tool messages)r=c SsJg|]!}|dd|dt|dtrt|dnt|dddqS)rr.rrr0)rrr.)r!r"r2dumpsr1r:rrrr<=s &zFWebSearchTransformation._transform_response_openai..)r>r)cr@)Ztoolr)r>Z tool_call_idrrrArCrrr<KrD)rFr()r)r5rGZ tool_messagesrrCrr74s    z2WebSearchTransformation._transform_response_openairesultcCs6t|dr|jrddd|jD}|St|}|S)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 resultsz cSs(g|]}d|jd|jd|jqS)zTitle: z URL: z Snippet: )titleurlsnippet)r;rrrrr<eszBWebSearchTransformation.format_search_response..)r$rJjoinr1)rIZsearch_result_textrrrformat_search_responseVs  z.WebSearchTransformation.format_search_response)r )r N)N)__name__ __module__ __qualname____doc__ staticmethodrboolr1rrrrrrrrr9r8r7r rPrrrrr sx  ) 8 b  #   1 !r )rTr2typingrrrrrrZlitellm._loggingrZlitellm.constantsr Z+litellm.llms.base_llm.search.transformationr r rrrrs