uPjddZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z GddeZGd d ed ZGd de Zy)z Calls Tavily's /search endpoint to search the web. Tavily API Reference: https://docs.tavily.com/documentation/api-reference/endpoint/search )Final TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strceZdZUdZeed<y)_TavilySearchRequestRequiredz.Required fields for Tavily Search API request.queryN)__name__ __module__ __qualname____doc__str__annotations__n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/tavily/search/transformation.pyr r s 8 Jrr ceZdZUdZeed<eeed<eeed<eed<eed<eezed<eezed<eed <eed <eed <eed <eed <eed<eed<y)TavilySearchRequestz} Tavily Search API request format. Based on: https://docs.tavily.com/documentation/api-reference/endpoint/search max_resultsinclude_domainsexclude_domainstopic search_depthinclude_answerinclude_raw_contentinclude_imagesinclude_image_descriptionsinclude_favicon time_range start_dateend_datecountryN) r rrrintrlistrboolrrrrrsh #Y#Y J3J# $$OOM LrrF)totalc eZdZdZedefdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)TavilySearchConfigzhttps://api.tavily.comreturncy)NTavilyrrrrui_friendly_namez#TavilySearchConfig.ui_friendly_name3srNheadersapi_keyapi_basec z|j||dd|j}|s tdd||d<d|d<|S) z: Validate environment and return headers. )TAVILY_API_KEYTAVILY_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezETAVILY_API_KEY is not set. Set `TAVILY_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)resolve_server_api_keyr5 ValueError)selfr0r1r2kwargss rvalidate_environmentz'TavilySearchConfig.validate_environment7s_--"$,*!11 . de e%,WI#6 "4roptional_paramsdatac l|xstdxs |j}|jds|d}|S)z7 Get complete URL for Search endpoint. r5z/search)r r5endswith)r>r2rArBr?s rget_complete_urlz#TavilySearchConfig.get_complete_urlNs@X~.?@XDDXDX  +"7+Hrr c 4t|trdj|}d|i}d|vr|d|d<d|vr|d|d<d|vr|dj|d<t |}|j D]"\}}||j vs||vs|||<$|S)a Transform Search request to Tavily API format. Args: query: Search query (string or list of strings). Tavily only supports single string queries. optional_params: Optional parameters for the request - max_results: Maximum number of search results (0-20) - search_domain_filter: List of domains to include (max 300) -> maps to `include_domains` - exclude_domains: List of domains to exclude (max 150) - topic: Category of search ('general', 'news', 'finance') - search_depth: Depth of search ('basic', 'advanced') - include_answer: Include LLM-generated answer (bool or 'basic', 'advanced') - include_raw_content: Include raw HTML content (bool or 'markdown', 'text') - include_images: Perform image search (bool) - include_image_descriptions: Add descriptions for images (bool) - include_favicon: Include favicon URL (bool) - time_range: Time range filter ('day', 'week', 'month', 'year', 'd', 'w', 'm', 'y') - start_date: Start date filter (YYYY-MM-DD) - end_date: End date filter (YYYY-MM-DD) - country: Country code filter (e.g., 'US', 'GB', 'DE') Returns: Dict with typed request data following TavilySearchRequest spec  r rsearch_domain_filterrr%) isinstancer'joinlowerdictitems(get_supported_perplexity_optional_params)r>r rAr? request_data result_dataparamvalues rtransform_search_requestz+TavilySearchConfig.transform_search_request`s< eT "HHUOE U4 O +*9-*HL ' !_ 4.=>T.UL* +  '&5i&@&F&F&HL #",/ ,113 +LE5DIIKKPU]hPh%* E" +r raw_response logging_objc |j}g}|jdgD]S}t|jdd|jdd|jdddd}|j|Ut |d S) u  Transform Tavily API response to LiteLLM unified SearchResponse format. Tavily → LiteLLM mappings: - results[].title → SearchResult.title - results[].url → SearchResult.url - results[].content → SearchResult.snippet - No date/last_updated fields in Tavily response (set to None) Args: raw_response: Raw httpx response from Tavily API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format resultstitleurlcontentN)rXrZsnippetdate last_updatedsearch)rWobject)jsongetrappendr)r>rTrUr? response_jsonrWresult search_results rtransform_search_responsez,TavilySearchConfig.transform_search_responses, ,002 #'' 26 *F(jj"-JJub) 9b1! M NN= ) *  r)NN)N)r rrr5 staticmethodrr/rLr@r'rErShttpxResponseLiteLLMLoggingObjrrgrrrr+r+0s.Oc ## t*   6*. *T$Z$&  $9T#Y99  9v' nn' ''  ' rr+)rtypingrrri*litellm.litellm_core_utils.litellm_loggingrrk+litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainr r rr+rrrrpsN $ S 89 6e,R )R r