uPj'*dZddlZddlZddlmZddlmZddlmZm Z ddl Z ddl Z ddl m Z ddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZmZeZ dd edzde e!dzdedzde!dzde"e!eff dZ#e dde!e e!zde!d edzde e!dzdedzde!dzde!dzde!dzde$e jJzdzde"e!efdzdefdZ&e dde!e e!zde!d edzde e!dzdedzde!dzde!dzde!dzde$e jJzdzde"e!efdzdeeeeefzfdZ'y)z# Main Search function for LiteLLM. N) Coroutine)partial)AnyFinal)verbose_logger)request_timeout)Logging)BaseSearchConfigSearchResponse)BaseLLMHTTPHandler)SearchProviders)ProviderConfigManagerclientfilter_out_litellm_params max_resultssearch_domain_filtermax_tokens_per_pagecountryreturncBi}|||d<|||d<|||d<|||d<|S)as Helper function to build optional_params dict from Perplexity Search API parameters. Args: max_results: Maximum number of results (1-20) search_domain_filter: List of domains to filter (max 20) max_tokens_per_page: Max tokens per page country: Country code filter Returns: Dict with non-None optional parameters rrrr)rrrroptional_paramss X/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/search/main.py_build_search_optional_paramsrsQ$.0O)4 &'2F./&1D-.%, " querysearch_providerapi_keyapi_basetimeout extra_headersc Kt} tj} d| d<ttf|||||||||| d | } t j }t|j| }| jd|d{}tj|r |d{}n|}|td||S787#t$r$}|d}tj|||| | d}~wwxYww)a Async Search function. Args: query: Search query (string or list of strings) search_provider: Provider name (e.g., "perplexity") max_results: Optional maximum number of results (1-20), default 10 search_domain_filter: Optional list of domains to filter (max 20) max_tokens_per_page: Optional max tokens per page, default 1024 country: Optional country code filter (e.g., 'US', 'GB', 'DE') api_key: Optional API key api_base: Optional API base URL timeout: Optional timeout extra_headers: Optional extra headers **kwargs: Additional parameters Returns: SearchResponse with results list following Perplexity format Example: ```python import litellm # Basic search response = await litellm.asearch( query="latest AI developments 2024", search_provider="perplexity" ) # Search with options response = await litellm.asearch( query="AI developments", search_provider="perplexity", max_results=10, search_domain_filter=["arxiv.org", "nature.com"], max_tokens_per_page=1024, country="US" ) # Access results for result in response.results: print(f"{result.title}: {result.url}") print(f"Snippet: {result.snippet}") ``` Tasearch) rrrrrrrrr r!Nz5Got an unexpected None response from the Search API: /searchmodelcustom_llm_provideroriginal_exceptioncompletion_kwargs extra_kwargs)localsasyncioget_event_looprsearch contextvars copy_contextrunrun_in_executor iscoroutine ValueError Exceptionlitellmexception_type)rrrrrrrrr r!kwargs local_varsloopfuncctxfunc_with_context init_responseresponsee model_names rr#r#;s%vJ( ,,. y   +#!5 3'    !--/#*377D#9%)%9%9$@Q%RR   } -**H$H  TU]T^_` ` S+  ./w7 $$ / (    sM C.A3B>B:B> B<!B>9C.:B><B>> C+C&&C++C.c t} | jd} | jdd} | jdddu}t|tt fst dt|t|t rtd|Ds t d tjt| }|t d |tjd |t|||| }t| }|j!D]\}}||vs |||<tjd||j#||| xsi}|j%|||}|d}| j'| ||| |d|t(j+|||xst,| |||||| }|S#t.$r$}|d}t1j2|||| | d}~wwxYw)a Synchronous Search function. Args: query: Search query (string or list of strings) search_provider: Provider name (e.g., "perplexity") max_results: Optional maximum number of results (1-20), default 10 search_domain_filter: Optional list of domains to filter (max 20) max_tokens_per_page: Optional max tokens per page, default 1024 country: Optional country code filter (e.g., 'US', 'GB', 'DE') api_key: Optional API key api_base: Optional API base URL timeout: Optional timeout extra_headers: Optional extra headers **kwargs: Additional parameters Returns: SearchResponse with results list following Perplexity format Example: ```python import litellm # Basic search response = litellm.search( query="latest AI developments 2024", search_provider="perplexity" ) # Search with options response = litellm.search( query="AI developments", search_provider="perplexity", max_results=10, search_domain_filter=["arxiv.org", "nature.com"], max_tokens_per_page=1024, country="US" ) # Multi-query search response = litellm.search( query=["AI developments", "machine learning trends"], search_provider="perplexity" ) # Access results for result in response.results: print(f"{result.title}: {result.url}") print(f"Snippet: {result.snippet}") if result.date: print(f"Date: {result.date}") ``` litellm_logging_objlitellm_call_idNr#FTz/query must be a string or list of strings, got c3<K|]}t|tyw)N) isinstancestr).0qs r zsearch..s.Qaz!S/A.Qsz'All items in query list must be strings)providerz&Search is not supported for provider: zSearch call - provider: %s)rrrr)r8zSearch optional_params: %s)rrheaders)rrrr$)rDr)r8r&rlitellm_paramsr') rrr logging_objrrr'r#rLprovider_configr%)r+popgetrFrGlistr4typeallrget_provider_search_configr rdebugrritemsvalidate_environmentget_complete_urlupdate_from_kwargsbase_llm_http_handlerr.rr5r6r7)rrrrrrrrr r!r8r9rCrD _is_asyncsearch_provider_configrfiltered_kwargskeyvaluerL complete_urlrAr?r@s rr.r.sGFJZ 8> CX8Y-3ZZ8I4-P!::i74? %#t-NtTY{m\] ] eT "3.Q5.Q+QFG GBWAqAq$_5B  " )EoEVWX X9?K"?#!5 3 " ";&!I*//1 -JC/)',$ - 9?K0DD!'RE 5EE+F ((0 ..+#2(!0 / 066+.+! /27    '(0 $$ / (    sD F(BF(( G1GG)NNNN)NNNNNNNN)(__doc__r,r/collections.abcr functoolsrtypingrrhttpxr6litellm._loggingrlitellm.constantsr*litellm.litellm_core_utils.litellm_loggingr LiteLLMLoggingObj+litellm.llms.base_llm.search.transformationr r *litellm.llms.custom_httpx.llm_http_handlerr litellm.types.utilsr litellm.utilsrrrr[intrRrGdictrfloatTimeoutr#r.rrrrss% +-SXI/RR+, #-1&* ts)d*t4Z   #s(^ @#-1&*,0+/c c?c c tc s)d* c t c 4Z c 4Zc Djc U]] "T )c S>D(c c c L#-1&*,0+/] c?] ] t] s)d* ] t ] 4Z ] 4Z] Dj] U]] "T )] S>D(] iS. 899] ] r