`gj*(dZddlmZddlZddlZddlmZmZmZddl m Z eje Z d dZddZddZd d ZeZeZdd ZGd d e Zy)uParallel.ai web search + content extraction — plugin form. Subclasses :class:`agent.web_search_provider.WebSearchProvider`. Uses two distinct Parallel SDK clients: - ``Parallel`` (sync) — for :meth:`search` - ``AsyncParallel`` (async) — for :meth:`extract` This is the first plugin to exercise the **async-extract** code path in the ABC: :meth:`extract` is declared ``async def``, and the dispatcher in :func:`tools.web_tools.web_extract_tool` detects coroutines via :func:`inspect.iscoroutinefunction` and awaits. Config keys this provider responds to:: web: search_backend: "parallel" # explicit per-capability extract_backend: "parallel" # explicit per-capability backend: "parallel" # shared fallback # Optional: search mode (default "agentic"; also "fast" or "one-shot") # via the PARALLEL_SEARCH_MODE env var. Env vars:: PARALLEL_API_KEY=... # https://parallel.ai (required) PARALLEL_SEARCH_MODE=agentic # optional: agentic|fast|one-shot ) annotationsN)AnyDictList)WebSearchProviderc ddlm}|ddy#t$rYyt$r}tt |d}~wwxYw)aMTrigger lazy install of the parallel SDK if it isn't present. Mirrors the lazy-deps pattern used by the legacy implementation. Swallows benign ImportError from the lazy_deps helper itself; if the SDK is genuinely missing the subsequent ``from parallel import ...`` raises ImportError that the caller can handle. r)ensurezsearch.parallelF)promptN)tools.lazy_depsr ImportError Exceptionstr) _lazy_ensureexcs P/root/.hermes/venv/lib/python3.12/site-packages/plugins/web/parallel/provider.py_ensure_parallel_sdk_installedr.s?$:&u5   $#c(##$s ??:?cddlm}t|dd}||Sddlm}|d}|s t dt ddlm}||}||_ |S) zLazy-load + cache the sync Parallel client. Cache lives on :mod:`tools.web_tools` (as ``_parallel_client``) so unit tests that reset that name between cases keep working. rN_parallel_clientget_provider_envPARALLEL_API_KEYVPARALLEL_API_KEY environment variable not set. Get your API key at https://parallel.ai)Parallelapi_key) tools.web_tools web_toolsgetattragent.web_search_providerr ValueErrorrparallelrr)_wtcachedrrrclients r_get_sync_clientr%@sf " S,d 3F  :12G  6  #$! g &F!C Mcddlm}t|dd}||Sddlm}|d}|s t dt ddlm}||}||_ |S) z}Lazy-load + cache the async Parallel client. Cache lives on :mod:`tools.web_tools` (as ``_async_parallel_client``). rN_async_parallel_clientrrr) AsyncParallelr) rrrrrr rr!r)r()r"r#rrr)r$s r_get_async_clientr*]sf " S2D 9F  :12G  6  #$& 7 +F!'C Mr&c,ddlm}d|_d|_y)zDrop both cached clients so tests can re-instantiate cleanly. Clears the canonical slots on :mod:`tools.web_tools` (where :func:`_get_sync_client` / :func:`_get_async_client` read/write them). rN)rrrr()r"s r_reset_clients_for_testsr,ys "C!%Cr&cvtjddjj}|dvrd}|S)zDReturn the validated PARALLEL_SEARCH_MODE value (default "agentic").PARALLEL_SEARCH_MODEagentic>one-shotfastr/)osgetenvlowerstrip)modes r_resolve_search_moder7s8 99+Y 7 = = ? E E GD 22 Kr&creZdZdZed dZed dZd dZd dZd dZ d ddZ ddZ dd Z y )ParallelWebSearchProviderz,Parallel.ai search + async extract provider.cy)Nr!selfs rnamezParallelWebSearchProvider.namer&cy)Nrr;r<s r display_namez&ParallelWebSearchProvider.display_namer?r&c0ddlm}t|dS)zBReturn True when ``PARALLEL_API_KEY`` is set to a non-empty value.rrr)rrbool)r=rs r is_availablez&ParallelWebSearchProvider.is_availables>$%7899r&cyNTr;r<s rsupports_searchz)ParallelWebSearchProvider.supports_searchr&cyrFr;r<s rsupports_extractz*ParallelWebSearchProvider.supports_extractrHr&c  ddlm}|rdddSt}tj d|||t j j|g||t|d}g}t|jxsgD]]\}}|jxsg} |j|jxsd |jxsd | rd j| nd |d zd _d d|idS#t $r} dt#| dcYd} ~ Sd} ~ wt$$r} dd| dcYd} ~ Sd} ~ wt&$r(} tj)d| dd| dcYd} ~ Sd} ~ wwxYw)zExecute a Parallel search (sync). Uses the ``beta.search`` endpoint with the configured mode (``PARALLEL_SEARCH_MODE`` env var, default "agentic"). Limit is capped at 20 server-side. ris_interruptedF Interrupted)successerrorz)Parallel search: '%s' (mode=%s, limit=%d))search_queries objectiver6 max_results )urltitle descriptionpositionTweb)rOdataNParallel SDK not installed: zParallel search error: %szParallel search failed: )tools.interruptrMr7loggerinfor%betasearchmin enumerateresultsexcerptsappendrXrYjoinr rr r warning) r=querylimitrMr6response web_resultsiresultrgrs rrcz ParallelWebSearchProvider.searchsg' Q 6#(=AA')D KK;UD% ()..55 %wrN 6HK&x'7'7'=2>  6!??0b""%zz/R!'!3=Esxx'92$%E   $e[-AB B 9$s3x8 8  7u=  Q NN6 <$1I#/OP P QsGC*CC** E3 DE EDE E+EEEc $K ddlm}|r|Dcgc]}|ddd c}Stjdt |t j j|dd {}g}|jxsgD]m}|jxsd}|sd j|jxsg}|jxsd} |jxsd} |j| | ||| | d d o|jxsgD]S} |j| jxsddd| j xs| j"xsd d| jxsdidU|Scc}w7#t$$r.} |Dcgc]}|ddt'| dncc}wc}cYd } ~ Sd } ~ wt($r(} |Dcgc] }|ddd| dncc}wc}cYd } ~ Sd } ~ wt*$r>} tj-d| |Dcgc] }|ddd| dncc}wc}cYd } ~ Sd } ~ wwxYww)uSExtract content from one or more URLs via the async SDK. Returns the legacy list-of-results shape that :func:`tools.web_tools.web_extract_tool` expects: one entry per successful URL plus one entry per failed URL with an ``error`` field. Errors are not raised — they're returned as per-URL items. rrLrNrU)rXrPrYzParallel extract: %d URL(s)T)urls full_contentNz ) sourceURLrY)rXrYcontent raw_contentmetadatazextraction failedrt)rXrYrurPrw)rXrYrurPr^zParallel extract error: %szParallel extract failed: )r_rMr`ralenr*rbextractrfrsrirgrXrYrherrorsru error_typer rr r rj) r=rrkwargsrMurmrfrprurXrYrPrs rryz!ParallelWebSearchProvider.extracts57  6MQHIA C KK5s4y A.055==!>H -/G"**0b  --3$kk&//*?R@Gjj&B *"!&#*'.25$F    ".B $yyB!##%!&!Y%2B2B!YFY%0%))/r$B  NM D ]W[\RSArCHM\\ \ B2B^_b^c@de   NN7 =B2B[\_[`@ab  sHE" EE"HAE",E -C-E"HE"" H +F/FFH H H  G$F65G=H >H H H)G;:HH HH  HcdddddddgdS) Nrpaidz2Objective-tuned search + parallel page extraction.rzParallel API keyzhttps://parallel.ai)keyr rX)r>badgetagenv_varsr;r<s rget_setup_schemaz*ParallelWebSearchProvider.get_setup_schemas*G.00  r&Nreturnr)rrC))rkrrlintrDict[str, Any])rrz List[str]r|rrzList[Dict[str, Any]])rr) __name__ __module__ __qualname____doc__propertyr>rArDrGrJrcryrr;r&rr9r9sh6 : .Q`AA),A AF  r&r9)rNone)rrr)r __future__rloggingr2typingrrrrr getLoggerrr`rr%r*r,_get_parallel_client_get_async_parallel_clientr7r9r;r&rrsf8# ""7   8 $$$:8 &(.V 1V r&