uPjdZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z GddeZGd d ed ZGd de Zy)z] Calls Serper's /search endpoint to search Google. Serper API Reference: https://serper.dev )Final TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strceZdZUdZeed<y)_SerperSearchRequestRequiredz.Required fields for Serper Search API request.qN)__name__ __module__ __qualname____doc__str__annotations__n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/serper/search/transformation.pyr r s 8 Frr cXeZdZUdZeed<eed<eed<eed<eed<eed<eed<y ) SerperSearchRequestzL Serper Search API request format. Based on: https://serper.dev numpageglhllocation autocorrecttbsN)r rrrintrrboolrrrrrs- H I G GM HrrF)totalc eZdZdZedefdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)SerperSearchConfigzhttps://google.serper.devreturncy)NSerperrrrrui_friendly_namez#SerperSearchConfig.ui_friendly_name,srNheadersapi_keyapi_basec t|j||dd|j}|s td||d<d|d<|S)z: Validate environment and return headers. )SERPER_API_KEYSERPER_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezESERPER_API_KEY is not set. Set `SERPER_API_KEY` environment variable.z X-API-KEYzapplication/jsonz Content-Type)resolve_server_api_keyr- ValueError)selfr(r)r*kwargss rvalidate_environmentz'SerperSearchConfig.validate_environment0sX--"$,*!11 . de e& "4roptional_paramsdatac |xstdxs |j}|jd}|jds|d}|S)z7 Get complete URL for Search endpoint. r-/z/search)r r-rstripendswith)r5r*r8r9r6s rget_complete_urlz#SerperSearchConfig.get_complete_urlGsLX~.?@XDDXDX??3'  +"7+Hrqueryc t|trdj|}d|i}d|vr|d|d<d|vr|dj|d<d|vrJ|d}t|tr5t |dkDr'd jd |D}d |dd |d |d<t |}|j D]"\}} ||jvs||vs| ||<$|S)ac 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 max_resultsrcountryrsearch_domain_filterrz OR c3&K|] }d| yw)zsite:Nr).0ds r z>SerperSearchConfig.transform_search_request..|s3QAeA3K3Qs(z) ()) isinstancelistjoinlowerlendictitems(get_supported_perplexity_optional_params) r5r?r8r6 request_datadomainsdomain_clauses result_dataparamvalues rtransform_search_requestz+SerperSearchConfig.transform_search_requestYs& eT "HHUOE 4  O +"1-"@L   '!0!;!A!A!CL  !_ 4,-CDG'4(S\A-=(. 3Q3Q(Q&' S(9':#n=MQ$O S!",/ ,113 +LE5DIIKKPU]hPh%* E" +r raw_response logging_objc .|j}g}|jdgD]b}t|jdd|jdd|jdd|jdd}|j|dt |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 organictitlelinksnippetdateN)r^urlrarb last_updatedsearch)resultsobject)jsongetrappendr)r5rZr[r6 response_jsonrfresult search_results rtransform_search_responsez,SerperSearchConfig.transform_search_responses, ,002 #'' 26 *F(jj"-JJvr* 9b1ZZ'! M NN= ) *  r)NN)N)r rrr- staticmethodrr'rPr7rLr>rYhttpxResponseLiteLLMLoggingObjrrnrrrr#r#)s1Oc ## t*   6*. *T$Z$&  $.T#Y..  .`& nn& '&  & rr#)rtypingrrrp*litellm.litellm_core_utils.litellm_loggingrrr+litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainr r rr#rrrrwsN $ S 8 9  6e  F )F r