uPj_dZddlmZmZmZddlZddlmZddl m Z m Z m Z ddl mZGddeZGd d ed ZGd de Zy)z Calls Linkup's /search endpoint to search the web. Linkup API Reference: https://docs.linkup.so/pages/documentation/api-reference/endpoint/post-search )FinalLiteral TypedDictN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc<eZdZUdZeed<eded<eded<y)_LinkupSearchRequestRequiredz.Required fields for Linkup Search API request.q)deepstandarddepth) searchResults sourcedAnswer structured outputTypeN)__name__ __module__ __qualname____doc__str__annotations__rn/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/linkup/search/transformation.pyr r s"8 F % &&FGGrr cxeZdZUdZeed<eed<eed<eed<eed<eeed<eeed<eed <eed <y ) LinkupSearchRequestz Linkup Search API request format. Based on: https://docs.linkup.so/pages/documentation/api-reference/endpoint/post-search structuredOutputSchemaincludeSources includeImagesfromDatetoDateincludeDomainsexcludeDomainsincludeInlineCitations maxResultsN) rrrrrrboollistintrrrrrsD  M KII  OrrF)totalc eZdZdZedefdZ ddededzdedzdefdZ ddedzd ed ee ezdzdefd Z d ee ezd edefd Z de jdedefdZy)LinkupSearchConfigzhttps://api.linkup.so/v1returncy)NLinkuprrrrui_friendly_namez#LinkupSearchConfig.ui_friendly_name0srNheadersapi_keyapi_basec z|j||dd|j}|s tdd||d<d|d<|S) z: Validate environment and return headers. )LINKUP_API_KEYLINKUP_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezELINKUP_API_KEY is not set. Set `LINKUP_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)resolve_server_api_keyr8 ValueError)selfr3r4r5kwargss rvalidate_environmentz'LinkupSearchConfig.validate_environment4s_--"$,*!11 . de e%,WI#6 "4roptional_paramsdatac l|xstdxs |j}|jds|d}|S)z7 Get complete URL for Search endpoint. r8z/search)r r8endswith)rAr5rDrErBs rget_complete_urlz#LinkupSearchConfig.get_complete_urlKs@X~.?@XDDXDX  +"7+Hrqueryc Dt|trdj|}||jdd|jddd}d|vr|d|d<d |vr|d |d <t |}|j D]"\}}||j vs||vs|||<$|S) a Transform Search request to Linkup API format. Transforms Perplexity unified spec parameters: - query -> q - max_results -> maxResults - search_domain_filter -> includeDomains - country -> (not directly supported) - max_tokens_per_page -> (not applicable) All other Linkup-specific parameters are passed through as-is. Args: query: Search query (string or list of strings). Linkup only supports single string queries. optional_params: Optional parameters for the request Returns: Dict with typed request data following LinkupSearchRequest spec  rrrr)r rr max_resultsr(search_domain_filterr%) isinstancer*joingetdictitems(get_supported_perplexity_optional_params)rArIrDrB request_data result_dataparamvalues rtransform_search_requestz+LinkupSearchConfig.transform_search_request]s2 eT "HHUOE$((*=)--lOL4  O +)8)GL & !_ 4-<=S-TL) *",/ ,113 +LE5DIIKKPU]hPh%* E" +r raw_response logging_objc |j}g}|jdg}|D]}|jdd}|dk(rRt|jdd|jdd|jdddd } |j| l|d k(srt|jd|jdd|jdd|jdddd } |j| t |d S) aF Transform Linkup API response to LiteLLM unified SearchResponse format. Linkup -> LiteLLM mappings: - results[].name -> SearchResult.title - results[].url -> SearchResult.url - results[].content -> SearchResult.snippet - No date field in results (set to None) - No last_updated field in Linkup response (set to None) Args: raw_response: Raw httpx response from Linkup API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format resultstypetextnameurlcontentN)titlerasnippetdate last_updatedimagesearch)r\object)jsonrPr appendr) rArYrZrB response_jsonr\ raw_resultsresult result_type search_results rtransform_search_responsez,LinkupSearchConfig.transform_search_responses . ,002 +..y"= ! .F **VV4Kf$ , **VR0 5"-"JJy"5!% ! }-' , **VVZZr-BC 5"-"JJy"5!% ! }-- .0  r)NN)N)rrrr8 staticmethodrr2rQrCr*rHrXhttpxResponseLiteLLMLoggingObjrrqrrrr.r.-s0Oc ## t*   6*. *T$Z$&  $2T#Y22  2h: nn: ':  : rr.)rtypingrrrrs*litellm.litellm_core_utils.litellm_loggingrru+litellm.llms.base_llm.search.transformationrrr litellm.secret_managers.mainr r rr.rrrrzsS -, S 8H9H6e"^ )^ r