o ưi*@s,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 ddZ#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 j%fde ee"efdefddZ&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 j%fde ee"efde eeeeefffddZ'dS)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_pagecountryreturncCsHi}|dur ||d<|dur||d<|dur||d<|dur"||d<|S)a 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 Nrrrr)rrrroptional_paramsrrJ/home/app/Keep/.python/lib/python3.10/site-packages/litellm/search/main.py_build_search_optional_paramssrquerysearch_providerapi_keyapi_basetimeout extra_headersc  st} zIt} d| d<ttf|||||||||| d | } t}t|j| }| d|IdH}t |r>|IdH}n|}|durKt d||WSt yh}z|d}t j |||| | dd}~ww)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_providerZoriginal_exceptionZcompletion_kwargs extra_kwargs)localsasyncioget_event_looprsearch contextvars copy_contextrunrun_in_executor iscoroutine ValueError Exceptionlitellmexception_type)rrrrrrrrr r!kwargs local_varsloopfuncctxZfunc_with_contextZ init_responseresponsee model_namerrrr"9sV;     r"c  Kst} z| d} | dd} | dddu}t|ttfs'tdt|t|tr9tdd |Ds9td t j t |d }|durLtd |t d |t||||d}t| d}|D] \}}||vrq|||<qet d||j||| pid}|j||d}|d}| j||| |d|dtj|||pt| ||||||d }|WSty}z|d}tj|||| | dd}~ww)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 css|]}t|tVqdS)N) isinstancestr).0qrrr szsearch..z'All items in query list must be strings)providerz&Search is not supported for provider: zSearch call - provider: )rrrr)r5zSearch optional_params: )rrheaders)rrr#)r>r)r%rZlitellm_paramsr&) rrr Z logging_objrrr&r"rEZprovider_configr$)r(popgetr?r@listr1typeallrZget_provider_search_configrr debugrritemsZvalidate_environmentZget_complete_urlZupdate_environment_variablesbase_llm_http_handlerr+r r2r3r4)rrrrrrrrr r!r5r6r=r>Z _is_asyncZsearch_provider_configrZfiltered_kwargskeyvaluerEZ complete_urlr<r:r;rrrr+sC       r+)NNNN)NNNNNNNN)(__doc__r)r, functoolsrtypingrrrrrrZhttpxr3Zlitellm._loggingr Zlitellm.constantsr Z*litellm.litellm_core_utils.litellm_loggingr ZLiteLLMLoggingObjZ+litellm.llms.base_llm.search.transformationr r Z*litellm.llms.custom_httpx.llm_http_handlerrZlitellm.types.utilsrZ litellm.utilsrrrrMintr@rfloatTimeoutr"r+rrrrs               h