o ưiS@sdZddlmZmZmZmZmZddlZddlm Z ddl m Z m Z mZddlmZGdddeZGd d d ed d ZGd dde ZdS)z] Calls Serper's /search endpoint to search Google. Serper API Reference: https://serper.dev )DictListOptional TypedDictUnionN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc@seZdZUdZeed<dS)_SerperSearchRequestRequiredz.Required fields for Serper Search API request.qN)__name__ __module__ __qualname____doc__str__annotations__rr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/serper/search/transformation.pyr s  r c@sJeZdZUdZeed<eed<eed<eed<eed<eed<eed<d S) SerperSearchRequestzL Serper Search API request format. Based on: https://serper.dev numpageglZhllocationZ autocorrectZtbsN)rrrrintrrboolrrrrrs  rF)totalc @seZdZdZedefddZ  ddedeedeedefd d Z ddeed e d ee ee efdefd dZ de ee efd e defddZdejdedefddZdS)SerperSearchConfigzhttps://google.serper.devreturncCsdS)NZSerperrrrrrui_friendly_name)sz#SerperSearchConfig.ui_friendly_nameNheadersapi_keyapi_basecKs,|ptd}|s td||d<d|d<|S)z: Validate environment and return headers. ZSERPER_API_KEYzESERPER_API_KEY is not set. Set `SERPER_API_KEY` environment variable.z X-API-KEYzapplication/jsonz Content-Type)r ValueError)selfr!r"r#kwargsrrrvalidate_environment-s z'SerperSearchConfig.validate_environmentoptional_paramsdatacKs4|ptdp|j}|d}|ds|d}|S)z7 Get complete URL for Search endpoint. SERPER_API_BASE/z/search)r r*rstripendswith)r%r#r(r)r&rrrget_complete_url>s    z#SerperSearchConfig.get_complete_urlqueryc Kst|tr d|}d|i}d|vr|d|d<d|vr$|d|d<d|vrN|d}t|trNt|dkrNd d d |D}d |dd |d|d<t|}|D]\}} ||vrh||vrh| ||<qV|S)aw Transform Search request to Serper API format. Args: query: Search query (string or list of strings). Serper only supports single string queries. optional_params: Optional parameters for the request - max_results: Maximum number of search results -> maps to `num` - search_domain_filter: List of domains -> appended as site: clauses to `q` - country: Country code filter (e.g., 'US', 'GB') -> maps to `gl` (lowercased) Returns: Dict with typed request data following SerperSearchRequest spec  r Z max_resultsrZcountryrZsearch_domain_filterrz OR css|]}d|VqdS)zsite:Nr).0drrr ssz>SerperSearchConfig.transform_search_request..(z) ()) isinstancelistjoinlowerlendictitemsZ(get_supported_perplexity_optional_params) r%r/r(r&Z request_datadomainsZdomain_clausesZ result_dataparamvaluerrrtransform_search_requestPs&   z+SerperSearchConfig.transform_search_request raw_response logging_objcKsd|}g}|dgD]}t|dd|dd|dd|ddd}||q t|d d S) a@ Transform Serper API response to LiteLLM unified SearchResponse format. Serper -> LiteLLM mappings: - organic[].title -> SearchResult.title - organic[].link -> SearchResult.url - organic[].snippet -> SearchResult.snippet - organic[].date -> SearchResult.date (optional, not always present) Args: raw_response: Raw httpx response from Serper API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format ZorganictitlelinksnippetdateN)rCurlrFrGZ last_updatedsearch)resultsobject)jsongetr appendr )r%rArBr&Z response_jsonrJresultZ search_resultrrrtransform_search_responses    z,SerperSearchConfig.transform_search_response)NN)N)rrrr* staticmethodrr rrr'r;rrr.r@httpxResponseLiteLLMLoggingObjr rPrrrrr&sN   0r)rtypingrrrrrrRZ*litellm.litellm_core_utils.litellm_loggingrrTZ+litellm.llms.base_llm.search.transformationrr r Zlitellm.secret_managers.mainr r rrrrrrs