uPj;"UdZddlmZmZmZmZddlmZddlZddl m Z ddl m Z ddl mZddlmZerdd lmZneZed Zeeeed <d ed efdZdededzdedzd efdZGddeZGddeZGddZy)z+ Base Search transformation configuration. ) TYPE_CHECKINGAnyFinalLiteral)urlsplitN) PrivateAttrBaseLLMExceptionget_secret_str)LiteLLMPydanticObjectBase)Logging) max_resultssearch_domain_filtercountrymax_tokens_per_page_PERPLEXITY_UNIFIED_PARAMSurlreturncHt|jjSN)rnetloclower)rs p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/search/transformation.py _search_hostrs C=   % % ''caller_api_basedefault_api_base base_env_varct|}|sy||r t|ndfDchc]}|r t|}}||vScc}w)NF)rr )rrr candidatebasetrusteds r_is_trusted_search_api_baser$#s` $O4I  ,8N< (d    TG  s<c`eZdZUdZeed<eed<eed<dZedzed<dZedzed<dd iZy) SearchResultzSingle search result.titlersnippetNdate last_updatedextraallow) __name__ __module__ __qualname____doc__str__annotations__r)r* model_configrrr&r&6s: J H LD#*#L#*#W%Lrr&cTeZdZUdZeeed<dZeed<ddiZ e e Z e ed<y ) SearchResponsez Standard Search response format. Standardized to Perplexity Search format - other providers should transform to this format. resultssearchobjectr+r,)default_factory_hidden_paramsN) r-r.r/r0listr&r2r9r1r3rdictr;r4rrr6r6Bs9 , FCW%L'tz:BaseSearchConfig.resolve_server_api_key..s QSSS Qsc32K|]}t|ywrr )rTvars rrVz:BaseSearchConfig.resolve_server_api_key..sIS^C0IsNr)nextrO)rBrPrrQrr server_keys rresolve_server_api_keyz'BaseSearchConfig.resolve_server_api_keysW ! ! QILI Q     ;; -|\!_ rheadersapi_keyapi_basec |S)zq Validate environment and return headers. Override in provider-specific implementations. r4)rBr\r]r^kwargss rvalidate_environmentz%BaseSearchConfig.validate_environments roptional_params request_datac |dfS)u OPTIONAL Sign the request. Providers like Bedrock AgentCore need to SigV4-sign the request before sending it to the API. For all other providers, this is a no-op and we just return the headers. Returns: Tuple of (headers, signed_json_body). When signed_json_body is not None, the handler MUST send it verbatim as the request body — re-serializing the payload would invalidate the signature. Nr4)rBr\rbrcr^r]s r sign_requestzBaseSearchConfig.sign_requests*}rdatac td)a Get complete URL for Search endpoint. Args: api_base: Base URL for the API optional_params: Optional parameters for the request data: Transformed request body from transform_search_request(). Some providers (e.g., Google PSE) use GET requests and need the request body to construct query parameters in the URL. Can be a dict or list of dicts depending on provider. **kwargs: Additional keyword arguments Returns: Complete URL for the search endpoint Note: Override in provider-specific implementations. z0get_complete_url must be implemented by providerNotImplementedError)rBr^rbrfr`s rget_complete_urlz!BaseSearchConfig.get_complete_urls2""TUUrqueryc td)a? Transform Search request to provider-specific format. Override in provider-specific implementations. Args: query: Search query (string or list of strings) optional_params: Optional parameters for the request Returns: Dict with request data z8transform_search_request must be implemented by providerrh)rBrkrbr`s rtransform_search_requestz)BaseSearchConfig.transform_search_requests"""\]]r raw_response logging_objc td)z Transform provider-specific Search response to standard format. Override in provider-specific implementations. z9transform_search_response must be implemented by providerrh)rBrnror`s rtransform_search_responsez*BaseSearchConfig.transform_search_responses""]^^r error_message status_codect|||S)z-Get appropriate error class for the provider.)rsmessager\r )rBrrrsr\s rget_error_classz BaseSearchConfig.get_error_class s #!  r)rN)NNr)r-r.r/r0rC staticmethodr1rErrI frozensetrKrOtupler[r=rar9r<bytesrerjrmhttpxResponseLiteLLMLoggingObjr6rqint Exceptionrvr4rrr?r?Qs  )c))!7*in** t * Dj      2d t  CHo  Dj * t@##  t *   &# c3hc6k*3;'$tCK/@*AA    t  tCH~ut|+ ,6*. V*VVT$Z$& V V6^T#Y^^ T  ^& _nn _' _  _          rr?)r0typingrrrr urllib.parserr{pydanticr)litellm.llms.base_llm.chat.transformationr litellm.secret_managers.mainr litellm.types.llms.baser *litellm.litellm_core_utils.litellm_loggingrr}rxrr1r2rboolr$r&r6r?r4rrrs65! F7=W5>5E)C.1(c(c(  Dj *   & &, & =. =E E r