iqdZddlmZmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZerddlmZneZGdd eZGd d eZGd d Zy)z+ Base Search transformation configuration. ) TYPE_CHECKINGAnyDictListLiteralOptionalUnionN) PrivateAttrBaseLLMException)LiteLLMPydanticObjectBase)Loggingc`eZdZUdZeed<eed<eed<dZeeed<dZeeed<dd iZ y) SearchResultzSingle search result.titleurlsnippetNdate last_updatedextraallow) __name__ __module__ __qualname____doc__str__annotations__rrr model_config|/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/search/transformation.pyrrs: J H LD(3-"&L(3-&W%Lr rcTeZdZUdZeeed<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. resultssearchobjectrr)default_factory_hidden_paramsN) rrrrrrrr&rrr dictr(rr r!r#r#s9 , FCW%L'tsr c hdS)z Get the set of Perplexity unified search parameters. These are the standard parameters that providers should transform from. Returns: Set of parameter names that are part of the unified spec >country max_resultsmax_tokens_per_pagesearch_domain_filterrrr r!(get_supported_perplexity_optional_paramsz9BaseSearchConfig.get_supported_perplexity_optional_paramsHs   r headersapi_keyapi_basec |S)zq Validate environment and return headers. Override in provider-specific implementations. r)r0r>r?r@kwargss r!validate_environmentz%BaseSearchConfig.validate_environmentXs r optional_paramsdatac 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)r0r@rDrErBs r!get_complete_urlz!BaseSearchConfig.get_complete_urles2""TUUr queryc 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 providerrG)r0rJrDrBs r!transform_search_requestz)BaseSearchConfig.transform_search_requests"" F  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 providerrG)r0rMrNrBs r!transform_search_responsez*BaseSearchConfig.transform_search_responses" G  r error_message status_codect|||S)z-Get appropriate error class for the provider.)rRmessager>r )r0rQrRr>s r!get_error_classz BaseSearchConfig.get_error_classs #!  r )r,N)NNr.)rrrrr1 staticmethodrr3rr7setr=rrrCr)r rrIrLhttpxResponseLiteLLMLoggingObjr#rPint ExceptionrUrr r!r+r+-sf  )c))!7  c    $"&"&  # 3-   "37 V3-VVuT4:-./ V V6 S$s)^$  tT$Z  *  nn  '              r r+)rtypingrrrrrrr rXpydanticr )litellm.llms.base_llm.chat.transformationr litellm.types.llms.baser *litellm.litellm_core_utils.litellm_loggingrrZrr#r+rr r!rbsTLKK F=W &, & =. =A A r