uPj!(dZddlmZmZmZddlZddlmZddl m Z m Z m Z ddl mZGddeZGd d ed ZGd de Zy)z Calls Google Programmable Search Engine (PSE) API to search the web. Google PSE API Reference: https://developers.google.com/custom-search/v1/reference/rest/v1/cse/list )FinalLiteral TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc0eZdZUdZeed<eed<eed<y)_GooglePSESearchRequestRequiredz2Required fields for Google PSE Search API request.qcxkeyN)__name__ __module__ __qualname____doc__str__annotations__r/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/google_pse/search/transformation.pyr r s< F G Hrr ceZdZUdZeed<eed<eed<eed<eed<eed<eed<eed <eed <eed <eed <eed <eed<eed<eed<eed<eed<eed<eed<eed<eed<eed<y)GooglePSESearchRequestz Google Programmable Search Engine API request format. Based on: https://developers.google.com/custom-search/v1/reference/rest/v1/cse/list numstartcr dateRestrict exactTerms excludeTermsfileTypefilterglhqimgSizeimgTypelinkSitelrorTerms relatedSiterightssafe searchType siteSearchsiteSearchFiltersortN)rrrrintrrrrrrrs{ H J GOM K G G L LM G L K IOO IrrF)totalc eZdZdZedefdZdedfdZ dde dedzd edzde fd Z dd edzd e d e e e zdzdefd Z ddee ezd e dedzd edzdedzde f dZ dejdedefdZy)GooglePSESearchConfigz*https://www.googleapis.com/customsearch/v1returncy)Nz Google PSErrrrui_friendly_namez&GooglePSESearchConfig.ui_friendly_name=sr)GETPOSTcy)zE Google PSE uses GET requests with query parameters. r8r)selfs rget_http_methodz%GooglePSESearchConfig.get_http_methodAsrNheadersapi_keyapi_basec |j||dd|j}|s td|jdxs t d}|s tdd|d <|S) z Validate environment and return headers. Google PSE uses API key as a query parameter, not in headers. This method is called but headers are not used for authentication. GOOGLE_PSE_API_KEYGOOGLE_PSE_API_BASEcaller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezMGOOGLE_PSE_API_KEY is not set. Set `GOOGLE_PSE_API_KEY` environment variable.search_engine_idGOOGLE_PSE_ENGINE_IDzvGOOGLE_PSE_ENGINE_ID is not set. Set `GOOGLE_PSE_ENGINE_ID` environment variable or pass `search_engine_id` parameter.zapplication/jsonz Content-Type)resolve_server_api_keyrC ValueErrorgetr )r;r=r>r?kwargsrJs rvalidate_environmentz*GooglePSESearchConfig.validate_environmentGs--"$0.!55 . lm m#)**-?"@"jNSiDjI #5roptional_paramsdatac ddlm}|xstdxs |j}|r(t |t rd|vr|d}||}|d|S|S)z Get complete URL for Search endpoint with query parameters. Google PSE uses GET requests, so we build the full URL with query params here. The transformed request body (data) contains the parameters needed for the URL. r) urlencoderC_google_pse_params?) urllib.parserTr rC isinstancedict)r;r?rQrRrOrTparams query_strings rget_complete_urlz&GooglePSESearchConfig.get_complete_urlhsc +`~.CD`H`H` JtT*/Ct/K !56F"+F"3LZq/ /rqueryrJc lt|trdj|}|j||dd|j}|xs t d}|s t d|s t d|||d}d |vrt|d d }||d <d |vrK|d } t| trt| d kDr| d |d<d|d<nt| tr | |d<d|d<d|vr|dj|d<t|} |jD]"\} } | |jvs| | vs| | | <$d| iS)u Transform Search request to Google PSE API format. Transforms Perplexity unified spec parameters: - query → q (same) - max_results → num - search_domain_filter → siteSearch - country → gl - max_tokens_per_page → (not applicable, ignored) All other Google PSE-specific parameters are passed through as-is. Args: query: Search query (string or list of strings). Google PSE supports single string queries. optional_params: Optional parameters for the request api_key: Google API key search_engine_id: Google Programmable Search Engine ID (cx parameter) Returns: Dict with typed request data following GooglePSESearchRequest spec  rArCrDrKzGOOGLE_PSE_API_KEY is requiredz GOOGLE_PSE_ENGINE_ID is required)r rr max_results rsearch_domain_filterrr.ir/countryr#rU)rXlistjoinrLrCr rMminlenrupperrYitems(get_supported_perplexity_optional_params) r;r]rQr>r?rJrO request_data num_resultsdomains result_dataparamvalues rtransform_search_requestz.GooglePSESearchConfig.transform_search_requests< eT "HHUOE --"$0.!55 . ,U~>T/U=> >?@ @"7  O +!$_]%CR!HK"-L  !_ 4,-CDG'4(S\A-=-4QZ \*36 /0GS)-4 \*36 /0  '!0!;!A!A!CL ",/ ,113 +LE5DIIKKPU]hPh%* E" + !+  r raw_response logging_objc |j}g}|jdgD]S}t|jdd|jdd|jdddd}|j|Ut |d S) u+ Transform Google PSE API response to LiteLLM unified SearchResponse format. Google PSE → LiteLLM mappings: - items[].title → SearchResult.title - items[].link → SearchResult.url - items[].snippet → SearchResult.snippet - No date/last_updated fields in Google PSE response (set to None) Args: raw_response: Raw httpx response from Google PSE API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format rjtitlelinksnippetN)rvurlrydate last_updatedsearch)resultsobject)jsonrNr appendr)r;rsrtrO response_jsonr~item search_results rtransform_search_responsez/GooglePSESearchConfig.transform_search_responses, ,002 !%%gr2 *D(hhw+HHVR(B/! M NN= ) *  r)NN)N)NNN)rrrrC staticmethodrr7rr<rYrPrer\rrhttpxResponseLiteLLMLoggingObjrrrrrr4r4:s;Fc!7## t*   J*. *T$Z$&  :##'+ Y T#YY Y t Y * Y * Y  Y v' nn' ''  ' rr4)rtypingrrrr*litellm.litellm_core_utils.litellm_loggingrr+litellm.llms.base_llm.search.transformationrrr litellm.secret_managers.mainr r rr4rrrrsQ -, S 8 i