o ưit%@sdZddlmZmZmZmZmZmZddlm Z ddl Z ddl m Z ddlmZmZmZddlmZGdd d eZGd d d ed d ZGdddeZdS)zp Calls DuckDuckGo's Instant Answer API to search the web. DuckDuckGo API Reference: https://duckduckgo.com/api )DictListLiteralOptional TypedDictUnion) urlencodeN)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc@seZdZUdZeed<dS) _DuckDuckGoSearchRequestRequiredz2Required fields for DuckDuckGo Search API request.qN)__name__ __module__ __qualname____doc__str__annotations__rrd/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/duckduckgo/search/transformation.pyrs  rc@s:eZdZUdZeed<eed<eed<eed<eed<dS)DuckDuckGoSearchRequestz` DuckDuckGo Instant Answer API request format. Based on: https://duckduckgo.com/api formatpretty no_redirectno_html skip_disambigN)rrrrrrintrrrrrs  rF)totalc @seZdZdZedefddZdedfddZ  dd e d e ed e ede fd d Z dd e ede de e e ee fdefddZde eeefde de fddZdejdedefddZdS)DuckDuckGoSearchConfigzhttps://api.duckduckgo.comreturncCdS)NZ DuckDuckGorrrrrui_friendly_name(sz'DuckDuckGoSearchConfig.ui_friendly_name)GETPOSTcCr")z Get HTTP method for search requests. DuckDuckGo Instant Answer API uses GET requests. Returns: HTTP method 'GET' r$r)selfrrrget_http_method,sz&DuckDuckGoSearchConfig.get_http_methodNheadersapi_keyapi_basecKs d|d<|S)z Validate environment and return headers. DuckDuckGo Instant Answer API does not require authentication. zapplication/jsonz Content-Typer)r&r(r)r*kwargsrrrvalidate_environment6s z+DuckDuckGoSearchConfig.validate_environmentoptional_paramsdatacKsN|ptdp|j}|r%t|tr%d|vr%|d}t|dd}|d|S|S)z Get complete URL for Search endpoint. DuckDuckGo uses query parameters, so we construct the URL with the query. DUCKDUCKGO_API_BASE_duckduckgo_paramsT)doseqz/?)r r/ isinstancedictr)r&r*r-r.r+paramsZ query_stringrrrget_complete_urlEs   z'DuckDuckGoSearchConfig.get_complete_urlquerycKsht|tr d|}|dd}t|}d|vr|d|d<gd}|D] }||vr/||||<q#d|iS)a Transform Search request to DuckDuckGo API format. Args: query: Search query (string or list of strings). DuckDuckGo only supports single string queries. optional_params: Optional parameters for the request - max_results: Maximum number of search results (DuckDuckGo API doesn't directly support this, used for filtering) - format: Output format ('json', 'xml') - pretty: Pretty print (0 or 1) - no_redirect: Skip HTTP redirects (0 or 1) - no_html: Remove HTML from text (0 or 1) - skip_disambig: Skip disambiguation results (0 or 1) Returns: Dict with typed request data following DuckDuckGoSearchRequest spec  json)rr max_results _max_results)rrrrr0)r2listjoinr3)r&r6r-r+Z request_dataZ result_dataZ ddg_paramsparamrrrtransform_search_request[s    z/DuckDuckGoSearchConfig.transform_search_request raw_response logging_objc Ksb|}|jr |jjjni}d}d|vr(zt|d}Wn ttfy'Ynwg}|drN|drNt|dd|dd|ddddd}| ||dg} | D]} |durdt ||krdnt | t r*d | vrd | vr| d d} | d d} d | vr| d d } | d }t | d kr| d n| }nt | dkr| dddn| }| }t|| |ddd}| |qVd| vr*| dg}|D]b}|durt ||krnUd |vr)d |vr)|d d} |d d} d | vr| d d } | d }t | d kr| d n| }nt | dkr| dddn| }| }t|| |ddd}| |qqVt|ddS)ui Transform DuckDuckGo API response to LiteLLM unified SearchResponse format. DuckDuckGo → LiteLLM mappings: - RelatedTopics[].Text → SearchResult.title + snippet - RelatedTopics[].FirstURL → SearchResult.url - RelatedTopics[].Text → SearchResult.snippet - No date/last_updated fields in DuckDuckGo response (set to None) Args: raw_response: Raw httpx response from DuckDuckGo API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format Nr:Z AbstractURLZ AbstractTextHeading)titleurlsnippetdateZ last_updatedZ RelatedTopicsZFirstURLTextz - r2z...ZTopicssearch)resultsobject)r8requestrDr4r ValueError TypeErrorgetr appendlenr2r3splitr )r&r?r@r+Z response_jsonZ query_paramsr9rKZabstract_resultZrelated_topicsZtopictextrDpartsrCrEZ search_resultZ nested_topicsZ nested_topicrrrtransform_search_responses                 " z0DuckDuckGoSearchConfig.transform_search_response)NN)N)rrrr/ staticmethodrr#rr'rrr,r3rrr5r>httpxResponseLiteLLMLoggingObjr rVrrrrr %sP    /r )rtypingrrrrrr urllib.parserrXZ*litellm.litellm_core_utils.litellm_loggingr rZZ+litellm.llms.base_llm.search.transformationr r r Zlitellm.secret_managers.mainr rrr rrrrs