uPj+TUdZddlmZddlZddlmZmZddlmZmZm Z ddl Z ddl m Z ejdZded <ejd Zded <gd Zded <ddlmZddlmZmZmZddlmZddd ddZGdde ZGddedZGddeZy)z Brave Search /web/search endpoint. Documentation: https://api-dashboard.search.brave.com/app/documentation/web-search/get-started ) annotationsN)datetimetimezone)FinalLiteral TypedDict)parserz#^\s*\d{4}[-/]\d{1,2}[-/]\d{1,2}\s*$r_ISO_YMDz^\s*-?\d+(\.\d+)?\s*$_UNIX_TIMESTAMP)web discussionsfaqsfaqnewsvideosBRAVE_SECTIONS)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strF)dayfirst yearfirstc&|syt|j}tj|r\ t |}|dkDs|dkr|dz}t j |tjjjS tj|rtj|ddd}ntj|||d}|jjS#t$rYywxYw#t$rYywxYw) zQ Convert a string/int/float to YYYY-MM-DD; return None if parsing fails. NgvH7BgvH7g@@)tzTF)rrfuzzy)strstripr matchfloatr fromtimestamprutcdate isoformat Exceptionr r parse)srrts_floatdts m/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/brave/search/transformation.py to_yyyy_mm_ddr+s  A AQ QxH$(U"2F"))(x||DIIKUUW W  >>! a4%tLBa9xtTBwwy""$$   s%AC5A%D5 DD DDceZdZUdZded<y)_BraveSearchRequestRequiredz-Required fields for Brave Search API request.rqN__name__ __module__ __qualname____doc____annotations__r*r-r-@s 7 Fr6r-ceZdZUdZded<ded<ded<ded<ded<ded <ded <ded <d ed <ded<ded<ded<ded<d ed<d ed<d ed<d ed<d ed<y)BraveSearchRequestz Brave Search API request format. Based on: https://api-dashboard.search.brave.com/app/documentation/web-search/get-started intcountoffsetrcountry search_langui_lang freshness safesearch spellcheckbooltext_decorations result_filterunits goggles_idgogglesextra_snippetssummaryenable_rich_callbackinclude_fetch_metadata operatorsNr/r5r6r*r8r8Fsi J K L LNOO JO L M  Or6r8)totalceZdZdZed dZd dZ d ddZ d ddZ d ddZ eddZ dd Z edd Z y)BraveSearchConfigz.https://api.search.brave.com/res/v1/web/searchcy)Nz Brave Searchr5r5r6r*ui_friendly_namez"BraveSearchConfig.ui_friendly_namecsr6cy)z@ Brave Search API uses GET requests for search. GETr5)selfs r*get_http_methodz!BraveSearchConfig.get_http_methodgsr6Nc |j||dd|j}|s td||d<d|d<d|d <d|d <|S) z: Validate environment and return headers. ) BRAVE_API_KEYBRAVE_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezCBRAVE_API_KEY is not set. Set `BRAVE_API_KEY` environment variable.zX-Subscription-Tokenzapplication/jsonAcceptgzipzAccept-Encodingz Content-Type)resolve_server_api_keyrX ValueError)rTheadersapi_keyapi_basekwargss r*validate_environmentz&BraveSearchConfig.validate_environmentmsn--"$+)!00 . bc c*1&'.%+!""4r6c ddlm}|xstdxs |j}|r*t |t rd|vr|d}||d}|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) urlencoderX _brave_paramsT)doseq?) urllib.parserhrrX isinstancedict)rTrdoptional_paramsdatarerhparams query_strings r*get_complete_urlz"BraveSearchConfig.get_complete_urlsc +V~.>?V4CVCV JtT*$/F 1F"+F$"?LZq/ /r6c t|trdj|}d|i}d|vr |ddurd|d<nd|d<d|vrt|dd}||d<d |vr;|d }t|tr&t |d kDr|j |d||d<t |} |jD]"\} } | |jvs| | vs| | | <$d | 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.rKFT max_resultsr:search_domain_filterrri) rmlistjoinminlen_append_domain_filtersrnitems(get_supported_perplexity_optional_params) rTqueryrorcsearch_engine_idre request_data num_resultsdomains result_dataparamvalues r*transform_search_requestz*BraveSearchConfig.transform_search_requests 6 eT "HHUOE 3 $ 6?Kc;dhm;m5:L1 259L1 2 O +!$_]%CR!HK$/L ! !_ 4,-CDG'4(S\A-=$($?$? S@QSZ$[ S!",/ ,113 +LE5DIIKKPU]hPh%* E" + [  r6cd|Dcgc]}d| }}dj|}d|d|dScc}w)zK Add site: filters to emulate domain restriction in Brave. zsite:z OR (z) AND ())rz)rrdomaindomain_clauses domain_querys r*r}z(BraveSearchConfig._append_domain_filterssI AH Hf5!1 H H$kk.9 5'a00!Is -c |j}g}|jr |jjjni}|j t |}t dtt|jddd}|D]} |j| ijdgD]} t||k\r8| jdd} | jdd} | jdd} t| jd xs| jd }t| jd d}t| | | || }|j|t|d S)z_ Transform Brave Search API response to LiteLLM unified SearchResponse format. r:rwresultstitleurl descriptionpage_ageagefetched_content_timestamp)rrsnippetr# last_updatedsearch)robject)jsonrequestrrq_sections_from_paramsrnmaxr{r9getr|r+rappendr)rT raw_response logging_objre response_jsonr query_paramssections_to_processrvsectionresultrrrr#r search_results r*transform_search_responsez+BraveSearchConfig.transform_search_responsesc ,002 .0AMAUAUl2266==[] %)%?%?\@R%S CL,<,rs # ',, "**CD%D#$<==OOS 8  !!! !  !H ) 4E4N*(N*r6