o ưiU@sdZddlmZmZmZmZmZddlZddlm Z ddl m Z m Z mZddlmZGdddeZGd d d ed d ZGd dde ZdS)z8 Calls Perplexity's /search endpoint to search the web. )DictListOptional TypedDictUnionN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc@s&eZdZUdZeeeefed<dS) _PerplexitySearchRequestRequiredz2Required fields for Perplexity Search API request.queryN)__name__ __module__ __qualname____doc__rstrr__annotations__rrd/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/perplexity/search/transformation.pyr s r c@s6eZdZUdZeed<eeed<eed<eed<dS)PerplexitySearchRequestzr Perplexity Search API request format. Based on: https://docs.perplexity.ai/api-reference/search-post max_resultssearch_domain_filtermax_tokens_per_pagecountryN)rrrrintrrrrrrrrs   rF)totalc @seZdZdZedefddZ  ddedeedeedefd d Z ddeed e d ee ee efdefd dZ de ee efd e defddZdejdedefddZdS)PerplexitySearchConfigzhttps://api.perplexity.aireturncCsdS)NZ Perplexityrrrrrui_friendly_name$sz'PerplexitySearchConfig.ui_friendly_nameNheadersapi_keyapi_basecKs2|ptd}|s tdd||d<d|d<|S)z: Validate environment and return headers. ZPERPLEXITYAI_API_KEYzQPERPLEXITYAI_API_KEY is not set. Set `PERPLEXITYAI_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)r ValueError)selfr r!r"kwargsrrrvalidate_environment(s z+PerplexitySearchConfig.validate_environmentoptional_paramsdatacKs*|ptdp|j}|ds|d}|S)z7 Get complete URL for Search endpoint. PERPLEXITY_API_BASEz/search)r r*endswith)r%r"r(r)r&rrrget_complete_url9s  z'PerplexitySearchConfig.get_complete_urlr c Ksxd|i}|d}|dur||d<|d}|dur||d<|d}|dur+||d<|d}|dur8||d<t|S)a4 Transform Search request to Perplexity API format. Note: LiteLLM's native spec is the perplexity search spec. There's no transformation needed for the request data. https://docs.perplexity.ai/api-reference/search-post Args: query: Search query (string or list of strings) optional_params: Optional parameters for the request - max_results: Maximum number of search results (1-20) - search_domain_filter: List of domains to filter (max 20) - max_tokens_per_page: Max tokens per page (default 1024) - country: Country code filter (e.g., 'US', 'GB', 'DE') Returns: Dict with typed request data following PerplexitySearchRequest spec r rNrrr)getdict) r%r r(r&Z request_datarrrrrrrtransform_search_requestLs    z/PerplexitySearchConfig.transform_search_request raw_response logging_objc Ksj|}g}|dgD]"}t|dd|dd|dd|d|dd}||q t|d d S) a5 Transform Perplexity API response to standard SearchResponse format. Args: raw_response: Raw httpx response from Perplexity API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format resultstitleurlsnippetdate last_updated)r3r5r6r7r8search)r2object)jsonr-r appendr )r%r0r1r&Z response_jsonr2resultZ search_resultrrrtransform_search_response}s    z0PerplexitySearchConfig.transform_search_response)NN)N)rrrr* staticmethodrrrrr'r.rrr,r/httpxResponseLiteLLMLoggingObjr r>rrrrr!sN   1r)rtypingrrrrrr@Z*litellm.litellm_core_utils.litellm_loggingrrBZ+litellm.llms.base_llm.search.transformationrr r Zlitellm.secret_managers.mainr r rrrrrrs