nSjdZddlmZddlZddlZddlmZmZmZddl m Z ej e Z d dZdd ZGd d e ZdS)u:Exa web search + content extraction — plugin form. Subclasses :class:`agent.web_search_provider.WebSearchProvider`. Uses the official Exa SDK (``exa-py``) which is lazy-loaded via :func:`tools.lazy_deps.ensure` so that cold-start CLI users don't pay the SDK import cost when Exa isn't configured. Config keys this provider responds to:: web: search_backend: "exa" # explicit per-capability extract_backend: "exa" # explicit per-capability backend: "exa" # shared fallback for both Env var:: EXA_API_KEY=... # https://exa.ai (paid tier; free trial available) The previous in-tree implementation lived at ``tools.web_tools._exa_search`` / ``_exa_extract``; this file is the canonical replacement. Behavior is bit-for-bit identical aside from the ABC method-name change. ) annotationsN)AnyDictList)WebSearchProviderreturnrcbddlm}t|dd}||Sddlm}|d}|st d ddlm}|dd n9#t$rYn-t$r!}tt|d}~wwxYwdd l m }|| }d |j d<||_|S)zLazy-import and cache an Exa SDK client. Cache lives on :mod:`tools.web_tools` (as ``_exa_client``) so unit tests that reset that name between cases keep working. Raises ``ValueError`` when ``EXA_API_KEY`` is unset. rN _exa_clientget_provider_env EXA_API_KEYzLEXA_API_KEY environment variable not set. Get your API key at https://exa.ai)ensurez search.exaF)prompt)Exa)api_keyz hermes-agentzx-exa-integration)tools.web_tools web_toolsgetattragent.web_search_providerr ValueErrortools.lazy_depsr ImportError Exceptionstrexa_pyrheadersr )_wtcachedr r _lazy_ensureexcrclients :/home/ice/.hermes/hermes-agent/plugins/web/exa/provider.py_get_exa_clientr#)s+"!!!!! S- . .F  ::::::}--G   1   $:::::: \%00000      $$$#c((###$ S ! ! !F*8FN&'CO MsA B  B (BB Nonec ddlm}d|_dS)z?Drop the cached Exa client so tests can re-instantiate cleanly.rN)rrr )rs r"_reset_client_for_testsr&Ps!!!!!!COOOcteZdZdZeddZeddZddZddZdd Z dddZ ddZ ddZ dS)ExaWebSearchProvideruExa search + extract provider. Both methods are sync — Exa's SDK is sync-only. The web_extract_tool dispatcher wraps sync extracts via ``asyncio.to_thread`` when it needs to keep the event loop responsive. rrcdS)Nexaselfs r"namezExaWebSearchProvider.name_ur'cdS)Nrr,r-s r" display_namez!ExaWebSearchProvider.display_namecr0r'boolc>ddlm}t|dS)z=Return True when ``EXA_API_KEY`` is set to a non-empty value.rr r )rr r3)r.r s r" is_availablez!ExaWebSearchProvider.is_availablegs.>>>>>>$$]33444r'cdSNTr,r-s r"supports_searchz$ExaWebSearchProvider.supports_searchmtr'cdSr7r,r-s r"supports_extractz%ExaWebSearchProvider.supports_extractpr9r'querylimitintDict[str, Any]c ddlm}|rdddStd||t ||ddi }g}t |jpgD]P\}}|jpg}| |j pd |j pd |rd |nd |d zd Qdd|idS#t$r} dt| dcYd} ~ Sd} ~ wt$r} dd| dcYd} ~ Sd} ~ wt $r-} td| dd| dcYd} ~ Sd} ~ wwxYw)zExecute an Exa search. Returns ``{"success": True, "data": {"web": [{...}, ...]}}`` on success, ``{"success": False, "error": str}`` on failure (incl. missing API key and SDK install errors). ris_interruptedF Interrupted)successerrorzExa search: '%s' (limit=%d) highlightsT) num_resultscontents )urltitle descriptionpositionweb)rEdataNExa SDK not installed: zExa search error: %szExa search failed: )tools.interruptrCloggerinfor#search enumerateresultsrGappendrMrNjoinrrrrwarning) r.r=r>rCresponse web_resultsiresultrGr s r"rWzExaWebSearchProvider.searchss! L 6 6 6 6 6 6~ B#(=AAA KK5ue D D D&((//!&-0H K&x'7'=2>>   6#.4" ""%z/R!'!3?I'Qsxx ';';';r$%E  $e[-ABB B 9 9 9$s3xx88 8 8 8 8 8 8 P P P$/N/N/NOO O O O O O O L L L NN13 7 7 7$/JS/J/JKK K K K K K K LsGC B1C D>C*$D>* D>7D>D> D>"D93D>9D>urls List[str]kwargsrList[Dict[str, Any]]c f ddlm}|r d|DStdt |t |d}g}|jpgD];}|jpd}|j pd}|j pd} | || |||| dd <|S#t$r fd |DcYd Sd wt$r fd |DcYd Sd wt$r3 td  fd|DcYd Sd wwxYw)zExtract content from one or more URLs via Exa. Returns a list of result dicts shaped for the legacy LLM post-processing pipeline. On per-URL or whole-batch failure, results carry an ``error`` field rather than raising. rrBcg|]}|ddd S)rDrJ)rMrFrNr,).0us r" z0ExaWebSearchProvider.extract..s/HIA CCr'zExa extract: %d URL(s)T)textrJ) sourceURLrN)rMrNcontent raw_contentmetadatac8g|]}|ddtdS)rJrMrNrlrF)rrgrhr s r"riz0ExaWebSearchProvider.extract..s,\\\RSArCHHMM\\\r'Nc$g|] }|dddd S)rJrSrpr,rqs r"riz0ExaWebSearchProvider.extract..s?B2@_Z]@_@_``r'zExa extract error: %sc$g|] }|dddd S)rJzExa extract failed: rpr,rqs r"riz0ExaWebSearchProvider.extract..s?B2@\WZ@\@\]]r')rTrCrUrVlenr# get_contentsrYrjrMrNrZrrrr\) r.rarcrCr]rYr`rlrMrNr s @r"extractzExaWebSearchProvider.extracts &  6 6 6 6 6 6~ MQ KK0#d)) < < <&((55d5FFH,.G"*0b   ++j&B *"!&#*'.25$F$F N ] ] ]\\\\W[\\\ \ \ \ \ \ \             NN2C 8 8 8       sGB3BB33 D0= C D0 D0 C0*D00 D0=(D+%D0+D0cdddddddgdS) Nrpaidz5Semantic + neural web search with content extraction.r z Exa API keyzhttps://exa.ai)keyrrM)r/badgetagenv_varsr,r-s r"get_setup_schemaz%ExaWebSearchProvider.get_setup_schemas4J)++   r'N)rr)rr3)r<)r=rr>r?rr@)rarbrcrrrd)rr@) __name__ __module__ __qualname____doc__propertyr/r2r5r8r;rWrvr}r,r'r"r)r)WsXX5555 (L(L(L(L(LT----^       r'r))rr)rr$)r __future__rloggingostypingrrrrr getLoggerr~rUr#r&r)r,r'r"rs0#""""" """"""""""777777  8 $ $$$$$NA A A A A ,A A A A A r'