o ưit+@sdZddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z ddlZddlZedZedZgd Zdd lmZdd lmZmZmZdd lmZd d ddddZGddde ZGddded dZGdddeZ dS) z Brave Search /web/search endpoint. Documentation: https://api-dashboard.search.brave.com/app/documentation/web-search/get-started ) annotations)datetimetimezone)parser)DictListLiteralOptional TypedDictUnionNz#^\s*\d{4}[-/]\d{1,2}[-/]\d{1,2}\s*$z^\s*-?\d+(\.\d+)?\s*$)ZwebZ discussionsZfaqsZfaqnewsZvideos)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strF)dayfirst yearfirstsUnion[str, int, float, None]rboolrreturn Optional[str]cCs|sdSt|}t|r7zt|}|dks|dkr |d}tj|tjd WSt y6YdSwzt |rGt j|dddd}n t j|||dd}| WSt y`YdSw) zQ Convert a string/int/float to YYYY-MM-DD; return None if parsing fails. NgvH7BgvH7g@@)tzTF)rrZfuzzy)strstrip_UNIX_TIMESTAMPmatchfloatr fromtimestamprutcdate isoformat Exception_ISO_YMDrparse)rrrZts_floatdtr'_/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/brave/search/transformation.py to_yyyy_mm_dds(      r)c@seZdZUdZded<dS)_BraveSearchRequestRequiredz-Required fields for Brave Search API request.rqN__name__ __module__ __qualname____doc____annotations__r'r'r'r(r*?s  r*c@seZdZUdZded<ded<ded<ded<ded<ded <ded <ded <d ed <ded<ded<ded<ded<d ed<d ed<d ed<d ed<d ed<dS)BraveSearchRequestz Brave Search API request format. Based on: https://api-dashboard.search.brave.com/app/documentation/web-search/get-started intcountoffsetrZcountryZ search_langZui_langZ freshnessZ safesearchZ spellcheckrZtext_decorations result_filterZunitsZ goggles_idgogglesZextra_snippetssummaryZenable_rich_callbackinclude_fetch_metadata operatorsNr,r'r'r'r(r2Es(  r2)totalc@s|eZdZdZed*ddZd+ddZ d,d-ddZ d.d/ddZ d,d0ddZ ed1ddZ d2d%d&Z ed3d(d)Z d S)4BraveSearchConfigz.https://api.search.brave.com/res/v1/web/searchrrcCdS)Nz Brave Searchr'r'r'r'r(ui_friendly_namebsz"BraveSearchConfig.ui_friendly_nameLiteral['GET', 'POST']cCr=)z@ Brave Search API uses GET requests for search. GETr')selfr'r'r(get_http_methodfsz!BraveSearchConfig.get_http_methodNheadersrapi_keyrapi_basecKs<|ptd}|s td||d<d|d<d|d<d|d<|S) z: Validate environment and return headers. Z BRAVE_API_KEYzCBRAVE_API_KEY is not set. Set `BRAVE_API_KEY` environment variable.zX-Subscription-Tokenzapplication/jsonAcceptgzipzAccept-Encodingz Content-Type)r ValueError)rArCrDrEkwargsr'r'r(validate_environmentls z&BraveSearchConfig.validate_environmentoptional_paramsdictdata!Optional[Union[Dict, List[Dict]]]cKsZddlm}|ptdp|j}|r+t|tr+d|vr+|d}||dd}|d|S|S)z Get complete URL for Search endpoint with query parameters. The Brave Search API uses GET requests and therefore needs the request body (data) to construct query parameters in the URL. r) urlencodeBRAVE_API_BASE _brave_paramsT)doseq?) urllib.parserOrrP isinstancerL)rArErKrMrIrOparamsZ query_stringr'r'r(get_complete_urls  z"BraveSearchConfig.get_complete_urlqueryUnion[str, List[str]]search_engine_idc Kst|tr d|}d|i}d|vr|ddurd|d<nd|d<d|vr0t|dd}||d<d |vrM|d }t|trMt|d krM||d||d<t|} |D]\} } | |vrg| | vrg| | | <qUd | iS) u Transform Search request to Brave Search API format. Transforms Perplexity unified spec parameters: - query → q (same) - max_results → count - search_domain_filter → q (append domain filters) - country → country - max_tokens_per_page → (not applicable, ignored) All other Brave Search API-specific parameters are passed through as-is. Args: query: Search query (string or list of strings). Brave Search API supports single string queries. optional_params: Optional parameters for the request Returns: Dict with typed request data following Brave Search API spec  r+r9FT max_resultsr4Zsearch_domain_filterrrQ) rUlistjoinminlen_append_domain_filtersrLitemsZ(get_supported_perplexity_optional_params) rArXrKrDrZrIZ request_dataZ num_resultsdomainsZ result_dataparamvaluer'r'r(transform_search_requests2     z*BraveSearchConfig.transform_search_requestrd List[str]cCs*dd|D}d|}d|d|dS)zK Add site: filters to emulate domain restriction in Brave. cSsg|]}d|qS)zsite:r').0domainr'r'r( sz.z OR (z) AND ())r_)rXrdZdomain_clausesZ domain_queryr'r'r(rbs z(BraveSearchConfig._append_domain_filters raw_responsehttpx.Response logging_objOptional[LiteLLMLoggingObj]rc Ks|}g}|jr|jjjni}|t|}tdtt| ddd}|D]K} | | i dgD]>} t ||kr=n5| dd} | dd} | dd} t | d pY| d }t | d d}t | | | ||d }| |q3q't|d dS)z_ Transform Brave Search API response to LiteLLM unified SearchResponse format. r4r]resultstitleurl descriptionZpage_ageZageZfetched_content_timestamp)rtrvsnippetr! last_updatedsearch)rsobject)jsonrequestrvrV_sections_from_paramsrLmaxr`r3getrar)rappendr)rArnrprIZ response_jsonrs query_paramsZsections_to_processr\sectionresultrtrvrxr!ryZ search_resultr'r'r(transform_search_responses:       z+BraveSearchConfig.transform_search_responsercCsF|d}g}|rt|trdd|dD}dd|D}|p"tS)z Returns a list of sections the user has requested via the Brave Search API's `result_filter` parameter. If no `result_filter` parameter is provided, returns all sections. r6cSsg|]}|qSr')r)ripartr'r'r(rk0sz;BraveSearchConfig._sections_from_params..,cSs g|] }|tvr|qSr')lowerBRAVE_SECTIONS)rirr'r'r(rk2s )rrUrsplitr)rZ raw_filterZrequested_filterssectionsr'r'r(r~%s z'BraveSearchConfig._sections_from_params)rr)rr?)NN)rCrrDrrErrr)N)rErrKrLrMrNrr) rXrYrKrLrDrrZrrr)rXrrdrhrr)rnrorprqrr)rrLrrh) r-r.r/rP staticmethodr>rBrJrWrgrbrr~r'r'r'r(r<_s$      L  2r<)rrrrrrrr)!r0 __future__rrrZdateutilrtypingrrrr r r Zhttpxrecompiler$rrZ*litellm.litellm_core_utils.litellm_loggingr ZLiteLLMLoggingObjZ+litellm.llms.base_llm.search.transformationrrrZlitellm.secret_managers.mainrr)r*r2r<r'r'r'r(s&       $