AHjF*dZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z ddl Z ddlmZddlmZddlmZddlmZmZdd lmZdd lmZdd lmZmZmZeZ dd e e!d e e e"de e!de e"dee"eff dZ#e dde e"e e"fde"d e e!d e e e"de e!de e"de e"de e"de e e$e jJfde ee"efdefdZ&e dde e"e e"fde"d e e!d e e e"de e!de e"de e"de e"de e e$e jJfde ee"efde eeeeefffdZ'y)z# Main Search function for LiteLLM. N)partial)Any CoroutineDictListOptionalUnion)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 K/root/tools/cai/cai_env/lib/python3.12/site-packages/litellm/search/main.py_build_search_optional_paramsrsQ$')O)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) rr rrrrr!r"r#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)rr rrrrr!r"r#r$kwargs local_varsloopfuncctxfunc_with_context init_responseresponsee model_names rr&r&9s)vJ* %%' y   +#!5 3'    &&(#CGGT2"2249JKK   } -**H$H  GzR L+  '(0 $$ / (    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: )rrrr)r;zSearch optional_params: )r!r"headers)r"rr')rGr")r;r)rlitellm_paramsr*) rrr# logging_objr!r"r*r&rOprovider_configr()r.popgetrIrJlistr7typeallrget_provider_search_configrr debugrritemsvalidate_environmentget_complete_urlupdate_from_kwargsbase_llm_http_handlerr1r r8r9r:)rr rrrrr!r"r#r$r;r<rFrG _is_asyncsearch_provider_configrfiltered_kwargskeyvaluerO complete_urlrDrBrCs rr1r1sRFJ] 17 .>>+? ((0 ..+#2(!0 / )//+.+! /20    '(0 $$ / (    sD F+BF++ G4GG)NNNN)NNNNNNNN)(__doc__r/r2 functoolsrtypingrrrrrr httpxr9litellm._loggingr litellm.constantsr *litellm.litellm_core_utils.litellm_loggingr LiteLLMLoggingObj+litellm.llms.base_llm.search.transformationr r*litellm.llms.custom_httpx.llm_http_handlerrlitellm.types.utilsr litellm.utilsrrrr^intrJrfloatTimeoutr&r1rrrrts>> +-SXI/RR+, "&04)-! #"49-"#c]   #s(^ @"&04)-!!"59.2e d3i e e #e #49- e "# e c] e c]e sme eE5==01 2e DcN+e e e P"&04)-!!"59.2` d3i ` ` #` #49- ` "# ` c] ` c]` sm` eE5==01 2` DcN+`  >9S#~%=> >?` ` r