uPj !dZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z GddeZGd d ed ZGd de Zy)zz Calls SearXNG's /search endpoint to search the web. SearXNG API Reference: https://docs.searxng.org/dev/search_api.html )Final TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strceZdZUdZeed<y)_SearXNGSearchRequestRequiredz/Required fields for SearXNG Search API request.qN)__name__ __module__ __qualname____doc__str__annotations__o/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/searxng/search/transformation.pyr r s 9 Frr cNeZdZUdZeed<eed<eed<eed<eed<eed<y) SearXNGSearchRequestzg SearXNG Search API request format. Based on: https://docs.searxng.org/dev/search_api.html categoriesengineslanguagepageno time_rangeformatN)r rrrrrintrrrrrs' O LM KO KrrF)totalc eZdZedefdZdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)SearXNGSearchConfigreturncy)NSearXNGrrrrui_friendly_namez$SearXNGSearchConfig.ui_friendly_name)srcy)zW SearXNG supports both GET and POST, but we'll use GET for simplicity. GETr)selfs rget_http_methodz#SearXNGSearchConfig.get_http_method-srNheadersapi_keyapi_basec P|j||ddd}|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. )SEARXNG_API_KEYSEARXNG_API_BASEN)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezBearer Authorizationzapplication/jsonz Content-Type)resolve_server_api_key)r(r*r+r,kwargss rvalidate_environmentz(SearXNGSearchConfig.validate_environment3sM--"$-+! .  )0 ':GO $"4roptional_paramsdatac ddlm}|xs td}|s td|j ds|j dr|d}n|d}|r(t |t rd|vr|d}||}|d |S|S) z 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) urlencoder/zSEARXNG_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,r9r:r7r<params query_strings rget_complete_urlz$SearXNGSearchConfig.get_complete_urlLs +A~.@A~    +  %&Zv.&Zw/ JtT*/@D/H !23F"+F"3LZq/ /rqueryc t|trdj|}|dd}d|vrT|dj}|dk(s|dk(rd|d<n1|d k(rd |d<n&|d k(rd |d<n|d k(rd |d<n|d k(rd |d<n||d<d|vr t |}|j D]"\}}||j vs||vs|||<$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)r rcountryusukenrdefresjpja max_resultsr?)rDlistjoinlowerrEitems(get_supported_perplexity_optional_params) r(rIr9r7 request_datarM result_dataparamvalues rtransform_search_requestz,SearXNGSearchConfig.transform_search_requestrs2 eT "HHUOE5  ',Y7==?G$'T/+/ Z(D+/ Z(D+/ Z(D+/ Z(D+/ Z(+2 Z( O + ",/ ,113 +LE5DIIKKPU]hPh%* E" + ";//r raw_response logging_objc X|j}g}|jdgD]w}|jdxs|jd}t|jdd|jdd|jdd|d }|j|yt |d S) uk 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 results publishedDatepubdatetitleurlcontentN)rgrisnippetdate last_updatedr>)rdobject)rLgetrappendr) r(rarbr7 response_jsonrdresultrl search_results rtransform_search_responsez-SearXNGSearchConfig.transform_search_responses. ,002 #'' 26 *F::o.G&**Y2GD(jj"-JJub) 9b1! M NN= ) *  r)NN)N)r rr staticmethodrr%r)rEr8rWrHr`httpxResponseLiteLLMLoggingObjrrtrrrr!r!(sc## t*   :*. $*$$T$Z$& $ $LD0T#YD0D0  D0L- nn- '-  - rr!)rtypingrrrv*litellm.litellm_core_utils.litellm_loggingrrx+litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainr r rr!rrrr}sN $ S 8 I 8 } *} r