uPjUdZddlmZmZddlZddlmZddlm Z m Z m Z ddl m Z Gdded ZGd d ed ZGd ded ZGdded ZdddZeed<Gdde Zy)z Calls Parallel AI's /v1/search endpoint to search the web. Parallel AI API Reference: https://docs.parallel.ai/api-reference/search/search )Final TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc8eZdZUeeed<eeed<eed<y)_ParallelAISourcePolicyinclude_domainsexclude_domains after_dateN)__name__ __module__ __qualname__liststr__annotations__s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/parallel_ai/search/transformation.pyr r s#Y#YOrr F)totalceZdZUeed<y)_ParallelAIExcerptSettingsmax_chars_per_resultN)rrrintrrrrrrsrrc@eZdZUeed<eed<eed<eed<eed<y)_ParallelAIAdvancedSettings source_policyexcerpt_settings fetch_policylocation max_resultsN) rrrr rrdictrrrrrrrs**00Mrrc^eZdZUdZeeed<eed<eed<eed<eed<eed<eed<y ) ParallelAISearchRequestzv Parallel AI v1 Search API request format. Based on: https://docs.parallel.ai/api-reference/search/search search_queries objectivemodemax_chars_total session_id client_modeladvanced_settingsN) rrr__doc__rrrrrrrrr&r&&s4 IN IO22rr&basicadvanced)baseproLEGACY_PROCESSOR_TO_MODEc eZdZdZedefdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)ParallelAISearchConfigzhttps://api.parallel.aireturncy)Nz Parallel AIrrrrui_friendly_namez'ParallelAISearchConfig.ui_friendly_name;srNheadersapi_keyapi_basec t|j||dd|j}|s td||d<d|d<|S)N)PARALLEL_AI_API_KEYPARALLEL_API_KEYPARALLEL_AI_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezIPARALLEL_API_KEY is not set. Set `PARALLEL_API_KEY` environment variable.z x-api-keyzapplication/jsonz Content-Type)resolve_server_api_keyr? ValueError)selfr9r:r;kwargss rvalidate_environmentz+ParallelAISearchConfig.validate_environment?sX--"$D/!66 . hi i& "4roptional_paramsdatac |xstdxs |j}|jd}|jds|j dd}|S)Nr?/z /v1/searchz/v1)r r?rstripendswith removesuffix)rGr;rJrKrHs rget_complete_urlz'ParallelAISearchConfig.get_complete_urlSsXb~.DEbIbIb??3'  ."//67zBHrqueryc t|}i}t|tr||d<n |g|d<||d<|jdd}|jdd}||tj ||}|xsd|d<i}d|vr|jd|d<d|vr|jd|d <d |vrd |jd i|d <i} d |vr|jd | d <d|vr|jd| d<| r| |d<|j |jdi|r||d<|jddt|} | j || S)a Transform Search request to Parallel AI v1 API format. Args: query: Search query (string or list of strings) - If string: maps to `search_queries` (single item) and `objective` - If list: maps to `search_queries` (keyword queries) optional_params: Optional parameters for the request - mode: Search mode ('turbo', 'basic', 'advanced'); defaults to 'basic' - processor: Legacy v1beta param; 'base' maps to mode 'basic', 'pro' to 'advanced' - max_results: Maximum number of search results -> `advanced_settings.max_results` - search_domain_filter: Domains to include -> `advanced_settings.source_policy.include_domains` - exclude_domains: Domains to exclude -> `advanced_settings.source_policy.exclude_domains` - country: ISO 3166-1 alpha-2 code -> `advanced_settings.location` - max_chars_per_result: -> `advanced_settings.excerpt_settings.max_chars_per_result` - Any other params are passed through to the request body as-is Returns: Dict with request data following the v1 search request spec r'r(r)N processorr/r#countryr"rr search_domain_filterr r rr-max_tokens_per_page)r$ isinstancerpopr3getupdate) rGrRrJrHparams request_datar)rTr-r result_datas rtransform_search_requestz/ParallelAISearchConfig.transform_search_requestbs4_-79 eT "-2L) *.3WL) *(-L %zz&$'!::k48  o .  ,?!DE 0AL, -  ($/#' #5 6"r raw_response logging_objc n|j}g}|jdgD]}|jdxsg}|rdj|nd}t|jdxsd|jdxsd||jdd } |j | t |d S) aj Transform Parallel AI v1 API response to LiteLLM unified SearchResponse format. Parallel AI -> LiteLLM mappings: - results[].title -> SearchResult.title - results[].url -> SearchResult.url - results[].excerpts (array) -> SearchResult.snippet (joined string) - results[].publish_date -> SearchResult.date resultsexcerptsz ... titleurl publish_dateN)rfrgsnippetdate last_updatedsearch)rcobject)jsonrZjoinrappendr) rGr`rarH response_jsonrcresultrdri search_results rtransform_search_responsez0ParallelAISearchConfig.transform_search_responses ,002 #'' 26 *Fzz*-3H08gll8,bG(jj)/RJJu%+ZZ/! M NN= ) *  r)NN)N)rrrr? staticmethodrr8r$rIrrQr_httpxResponseLiteLLMLoggingObjrrtrrrr5r58s4c ## t*   0*. *  T$Z$&  MT#YMM  M^" nn" '"  " rr5)r.typingrrrv*litellm.litellm_core_utils.litellm_loggingrrx+litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainr r rrr&r3rr5rrrr}s} $ S 8iu %)5 3iu 3,3:"F%F[ -[ r