o ưi @sdZddlmZmZmZmZmZmZddlZddl m Z ddl m Z mZmZddlmZGdddeZGd d d ed d ZGd dde ZdS)z Calls Linkup's /search endpoint to search the web. Linkup API Reference: https://docs.linkup.so/pages/documentation/api-reference/endpoint/post-search )DictListLiteralOptional TypedDictUnionN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc@s2eZdZUdZeed<eded<eded<dS)_LinkupSearchRequestRequiredz.Required fields for Linkup Search API request.q)deepstandarddepth) searchResultsZ sourcedAnswerZ structured outputTypeN)__name__ __module__ __qualname____doc__str__annotations__rrr`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/linkup/search/transformation.pyr s   r c@sbeZdZUdZeed<eed<eed<eed<eed<eeed<eeed<eed <eed <d S) LinkupSearchRequestz Linkup Search API request format. Based on: https://docs.linkup.so/pages/documentation/api-reference/endpoint/post-search ZstructuredOutputSchemaZincludeSourcesZ includeImagesZfromDateZtoDateincludeDomainsZexcludeDomainsZincludeInlineCitations maxResultsN) rrrrrrboolrintrrrrrs    rF)totalc @seZdZdZedefddZ  ddedeedeedefd d Z ddeed e d ee ee efdefd dZ de ee efd e defddZdejdedefddZdS)LinkupSearchConfigzhttps://api.linkup.so/v1returncCsdS)NZLinkuprrrrrui_friendly_name1sz#LinkupSearchConfig.ui_friendly_nameNheadersapi_keyapi_basecKs2|ptd}|s tdd||d<d|d<|S)z: Validate environment and return headers. ZLINKUP_API_KEYzELINKUP_API_KEY is not set. Set `LINKUP_API_KEY` environment variable.zBearer Authorizationzapplication/jsonz Content-Type)r ValueError)selfr%r&r'kwargsrrrvalidate_environment5s z'LinkupSearchConfig.validate_environmentoptional_paramsdatacKs*|ptdp|j}|ds|d}|S)z7 Get complete URL for Search endpoint. LINKUP_API_BASEz/search)r r/endswith)r*r'r-r.r+rrrget_complete_urlHs    z#LinkupSearchConfig.get_complete_urlquerycKst|tr d|}||dd|ddd}d|vr"|d|d<d |vr,|d |d <t|}|D]\}}||vrF||vrF|||<q4|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)rrrZ max_resultsrZsearch_domain_filterr) isinstancelistjoingetdictitemsZ(get_supported_perplexity_optional_params)r*r2r-r+Z request_dataZ result_dataparamvaluerrrtransform_search_request\s"       z+LinkupSearchConfig.transform_search_request raw_response logging_objc Ks|}g}|dg}|D]I}|dd}|dkr5t|dd|dd|ddddd } || q|d krWt|d|dd|dd|ddddd } || qt|d 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)titlerDsnippetdateZ last_updatedimagesearch)r?object)jsonr7r appendr ) r*r=r>r+Z response_jsonr?Z raw_resultsresultZ result_typeZ search_resultrrrtransform_search_responses8         z,LinkupSearchConfig.transform_search_response)NN)N)rrrr/ staticmethodrr$rrr,r8rrr1r<httpxResponseLiteLLMLoggingObjr rOrrrrr".sN   7r")rtypingrrrrrrrQZ*litellm.litellm_core_utils.litellm_loggingrrSZ+litellm.llms.base_llm.search.transformationr r r Zlitellm.secret_managers.mainr r rr"rrrrs