o ưi!@sdZddlmZmZmZmZmZddlZddlm Z ddl m Z m Z mZddlmZGdddeZGd d d ed d ZGd dde ZdS)zz Calls SearXNG's /search endpoint to search the web. SearXNG API Reference: https://docs.searxng.org/dev/search_api.html )DictListOptional TypedDictUnionN)Logging)BaseSearchConfigSearchResponse SearchResultget_secret_strc@seZdZUdZeed<dS)_SearXNGSearchRequestRequiredz/Required fields for SearXNG Search API request.qN)__name__ __module__ __qualname____doc__str__annotations__rra/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/searxng/search/transformation.pyr s  r c@sBeZdZUdZeed<eed<eed<eed<eed<eed<dS) SearXNGSearchRequestzg SearXNG Search API request format. Based on: https://docs.searxng.org/dev/search_api.html categoriesZengineslanguageZpagenoZ time_rangeformatN)rrrrrrintrrrrrs  rF)totalc @seZdZedefddZddZ  ddedeed eedefd d Z dd eed e d ee ee efdefddZ de ee efd e defddZdejdedefddZdS)SearXNGSearchConfigreturncCdS)NZSearXNGrrrrrui_friendly_name'sz$SearXNGSearchConfig.ui_friendly_namecCr)zW SearXNG supports both GET and POST, but we'll use GET for simplicity. GETr)selfrrrget_http_method+sz#SearXNGSearchConfig.get_http_methodNheadersapi_keyapi_basecKs*|ptd}|rd||d<d|d<|S)z Validate environment and return headers. SearXNG is open-source and doesn't require an API key by default. Some instances may require authentication via headers. ZSEARXNG_API_KEYzBearer Authorizationzapplication/jsonz Content-Typer )r"r$r%r&kwargsrrrvalidate_environment1s z(SearXNGSearchConfig.validate_environmentoptional_paramsdatacKsddlm}|p td}|std|ds'|dr"|d}n|d}|rAt|trAd|vrA|d}||}|d |S|S) a Get complete URL for Search endpoint with query parameters. SearXNG 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) urlencodeZSEARXNG_API_BASEzSEARXNG_API_BASE is not set. Please set the `SEARXNG_API_BASE` environment variable or pass `api_base` parameter. Example: os.environ['SEARXNG_API_BASE'] = 'https://your-searxng-instance.com'z/search/search_searxng_params?) urllib.parser,r ValueErrorendswith isinstancedict)r"r&r*r+r(r,paramsZ query_stringrrrget_complete_urlDs     z$SearXNGSearchConfig.get_complete_urlqueryc Kst|tr d|}|dd}d|vrN|d}|dks!|dkr&d|d<n(|d kr/d |d<n|d kr8d |d<n|d krAd |d<n |d krJd |d<n||d<d|vrS t|}|D]\}}||vrm||vrm|||<q[d|iS)u Transform Search request to SearXNG API format. Transforms Perplexity unified spec parameters: - query → q - max_results → (handled via pageno, SearXNG returns ~20 results per page) - search_domain_filter → (not directly supported) - country → language (approximate mapping) - max_tokens_per_page → (not applicable, ignored) All other SearXNG-specific parameters are passed through as-is. Args: query: Search query (string or list of strings). SearXNG only supports single string queries. optional_params: Optional parameters for the request Returns: Dict with typed request data following SearXNGSearchRequest spec  json)rrcountryusukenrdefresZjpjaZ max_resultsr/)r4listjoinlowerr5itemsZ(get_supported_perplexity_optional_params) r"r8r*r(Z request_datar;Z result_dataparamvaluerrrtransform_search_requestks4        z,SearXNGSearchConfig.transform_search_request raw_response logging_objc Ksr|}g}|dgD]&}|dp|d}t|dd|dd|dd|dd }||q t|d d S) u Transform SearXNG API response to LiteLLM unified SearchResponse format. SearXNG → LiteLLM mappings: - results[].title → SearchResult.title - results[].url → SearchResult.url - results[].content → SearchResult.snippet - results[].publishedDate OR results[].pubdate → SearchResult.date - No last_updated field in SearXNG response (set to None) Args: raw_response: Raw httpx response from SearXNG API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format resultsZ publishedDateZpubdatetitleurlcontentN)rMrOsnippetdateZ last_updatedr.)rLobject)r:getr appendr ) r"rJrKr(Z response_jsonrLresultrRZ search_resultrrrtransform_search_responses     z-SearXNGSearchConfig.transform_search_response)NN)N)rrr staticmethodrr r#rrr)r5rrr7rIhttpxResponseLiteLLMLoggingObjr rWrrrrr%sN   ' Fr)rtypingrrrrrrYZ*litellm.litellm_core_utils.litellm_loggingrr[Z+litellm.llms.base_llm.search.transformationrr r Zlitellm.secret_managers.mainr r rrrrrrs