o ưi7@sdZddlmZmZmZmZmZddlZddlm Z ddl m Z m Z mZddlmZGdddeZGd d d ed d ZGd dde ZdS)z Calls Tavily's /search endpoint to search the web. Tavily API Reference: https://docs.tavily.com/documentation/api-reference/endpoint/search )DictListOptional TypedDictUnionN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc@seZdZUdZeed<dS)_TavilySearchRequestRequiredz.Required fields for Tavily Search API request.queryN)__name__ __module__ __qualname____doc__str__annotations__rr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/tavily/search/transformation.pyr s  r c@seZdZUdZeed<eeed<eeed<eed<eed<ee efed<ee efed<e ed <e ed <e ed <eed <eed <eed<eed<dS)TavilySearchRequestz} Tavily Search API request format. Based on: https://docs.tavily.com/documentation/api-reference/endpoint/search max_resultsinclude_domainsZexclude_domainsZtopicZ search_depthZinclude_answerZinclude_raw_contentZinclude_imagesZinclude_image_descriptionsZinclude_faviconZ time_rangeZ start_dateend_datecountryN) rrrrintrrrrboolrrrrrs    rF)totalc @seZdZdZedefddZ  ddedeedeedefd d Z ddeed e d ee ee efdefd dZ de ee efd e defddZdejdedefddZdS)TavilySearchConfigzhttps://api.tavily.comreturncCsdS)NZTavilyrrrrrui_friendly_name0sz#TavilySearchConfig.ui_friendly_nameNheadersapi_keyapi_basecKs2|ptd}|s tdd||d<d|d<|S)z: Validate environment and return headers. ZTAVILY_API_KEYzETAVILY_API_KEY is not set. Set `TAVILY_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)r ValueError)selfr!r"r#kwargsrrrvalidate_environment4s z'TavilySearchConfig.validate_environmentoptional_paramsdatacKs*|ptdp|j}|ds|d}|S)z7 Get complete URL for Search endpoint. TAVILY_API_BASEz/search)r r+endswith)r&r#r)r*r'rrrget_complete_urlEs  z#TavilySearchConfig.get_complete_urlr cKst|tr d|}d|i}d|vr|d|d<d|vr"|d|d<d|vr.|d|d<t|}|D]\}}||vrH||vrH|||<q6|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 rZsearch_domain_filterrr) isinstancelistjoinlowerdictitemsZ(get_supported_perplexity_optional_params)r&r r)r'Z request_dataZ result_dataparamvaluerrrtransform_search_requestXs    z+TavilySearchConfig.transform_search_request raw_response logging_objcKs^|}g}|dgD]}t|dd|dd|ddddd}||q t|dd 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)r;r=snippetdateZ last_updatedsearch)r:object)jsongetr appendr )r&r8r9r'Z response_jsonr:resultZ search_resultrrrtransform_search_responses    z,TavilySearchConfig.transform_search_response)NN)N)rrrr+ staticmethodrr rrr(r3rrr-r7httpxResponseLiteLLMLoggingObjr rGrrrrr-sN   ;r)rtypingrrrrrrIZ*litellm.litellm_core_utils.litellm_loggingrrKZ+litellm.llms.base_llm.search.transformationrr r Zlitellm.secret_managers.mainr r rrrrrrs