`gjdZddlmZddlZddlZddlmZmZmZddl m Z eje Z d dZd dZGdd e Zy) 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)WebSearchProviderc4ddlm}t|dd}||Sddlm}|d}|s t d ddlm}|dd dd l m }|| }d |jd<||_|S#t$rY2t$r}tt|d}~wwxYw)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 K/root/.hermes/venv/lib/python3.12/site-packages/plugins/web/exa/provider.py_get_exa_clientr")s" S- .F  :}-G  1  $:\%0   !F*8FNN&'CO M   $#c(##$sA++ B6B>BBcddlm}d|_y)z?Drop the cached Exa client so tests can re-instantiate cleanly.rN)rrr )rs r!_reset_client_for_testsr$Ps !COcfeZdZdZed dZed dZd dZd dZd dZ d ddZ ddZ dd Z y )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. cy)Nexaselfs r!namezExaWebSearchProvider.name_r%cy)Nrr*r+s r! display_namez!ExaWebSearchProvider.display_namecr.r%c0ddlm}t|dS)z=Return True when ``EXA_API_KEY`` is set to a non-empty value.rr r )rr bool)r,r s r! is_availablez!ExaWebSearchProvider.is_availablegs>$]344r%cyNTr*r+s r!supports_searchz$ExaWebSearchProvider.supports_searchmr%cyr5r*r+s r!supports_extractz%ExaWebSearchProvider.supports_extractpr7r%cn ddlm}|rdddStjd||t j ||ddi }g}t |jxsgD]]\}}|jxsg}|j|jxsd |jxsd |rd j|nd |d zd _dd|idS#t$r} dt| dcYd} ~ Sd} ~ wt$r} dd| dcYd} ~ Sd} ~ wt $r(} tj#d| 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)r>dataNExa SDK not installed: zExa search error: %szExa search failed: )tools.interruptr<loggerinfor"search enumerateresultsr@appendrFrGjoinrrrrwarning) r,querylimitr<response web_resultsiresultr@rs r!rPzExaWebSearchProvider.searchssJ! L 6#(=AA KK5ue D&(//!&-0H K&x'7'7'=2>  6#..4" ""%zz/R!'!3?Isxx ';r$%E   $e[-AB B 9$s3x8 8 P$1H/NO O L NN13 7$1DSE/JK K LsGC B6C D4 C'!D4' D43D:D4 D4 D/)D4/D4c  ddlm}|r|Dcgc]}|ddd c}Stjdt |t j |d}g}|jxsgD]L}|jxsd}|jxsd} |jxsd} |j| | ||| | d d N|Scc}w#t$r.} |Dcgc]}|ddt| d ncc}wc}cYd } ~ Sd } ~ wt$r(} |Dcgc] }|ddd | d ncc}wc}cYd } ~ Sd } ~ wt$r>} tj!d| |Dcgc] }|ddd| d ncc}wc}cYd } ~ 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. rr;r=rC)rFr?rGzExa extract: %d URL(s)T)text) sourceURLrG)rFrGcontent raw_contentmetadata)rFrGr_r?NrLzExa extract error: %szExa extract failed: )rMr<rNrOlenr" get_contentsrRr]rFrGrSrrrrrU) r,urlskwargsr<urXrRr[r_rFrGrs r!extractzExaWebSearchProvider.extracts&  6MQHIA C KK0#d) <&(55d5FH,.G"**0b  +++jj&B *"!&#*'.25$F  N+, ]W[\RSArCHM\\ \ B2BYZ]Y^@_`   NN2C 8B2BVWZV[@\]  s{C CCBCC E0C7C*)C71E07 E0D&DD& E0& E02E+ EE+%E0+E0cdddddddgdS) Nrpaidz5Semantic + neural web search with content extraction.r z Exa API keyzhttps://exa.ai)keyrrF)r-badgetagenv_varsr*r+s r!get_setup_schemaz%ExaWebSearchProvider.get_setup_schemas*J)++  r%N)returnr)ror2))rVrrWintroDict[str, Any])rdz List[str]rerrozList[Dict[str, Any]])rorr) __name__ __module__ __qualname____doc__propertyr-r0r3r6r9rPrgrnr*r%r!r'r'WsP5 (LT-^  r%r')ror)roNone)rv __future__rloggingostypingrrrrr getLoggerrsrNr"r$r'r*r%r!r~sJ0# ""7   8 $$NA ,A r%