߲j]K dUdZddlmZddlZddlZddlZddlZddlZddl Zddl m Z m Z ddl mZmZddlmZddlmZmZej,eZd Zd ZdEd ZdEd ZdFd ZdGdZdHdZdIdJdZ dKdZ!dLdZ"dMdZ#dNdZ$dOdZ%dPdZ&dQdZ'dZ(e(e(ddZ)dRdZ*dRd Z+dIdSd!Z,dRd"Z-dTd$Z.dUdVd%Z/dRd&Z0dWd'Z1e*d(de1d)idgfe+d*e1d+e1d,e1d-d.ddgfe,d/igfe-d0e1d1d2d3d4d5dgfe0d6e1d7e1d8d9gd:d;d<e1d=d>d#dgfd?Z2d@e3dA<dXdBZ4dYdCZ5ddDl m6Z6y)Za(A2A client tools (``a2a`` toolset): a2a_discover/call/list/history/orchestrate talk to *other* agents. Peers come from config.yaml ``a2a_agents: {name: {url, auth: {type: bearer, token}, timeout, capabilities}}``. Stdlib urllib; wire format is A2A v1.0 ``SendMessage`` (v0.3 replies still parse).) annotationsN)ThreadPoolExecutor as_completed)AnyOptional) coerce_port)protocolsecurityxc$ddlm}|xsiS)zORead-only view of config.yaml; peers are only read, never mutated (cache-safe).rload_config_readonly)hermes_cli.configrrs @/home/ubuntu/.hermes/hermes-agent/plugins/platforms/a2a/tools.py _load_configrs6  ! 'R'c<tjdxsiS)N a2a_agents)rgetrr_configured_peersr s >  l + 1r1rc |jdd|jdixsit|jdtd|S)Nurlauthtimeout)rrr)rint_DEFAULT_TIMEOUT)entryextras r_peer_from_entryr#$sK99UB'621F1L"599Y0@AB MFK MMragentc|jdr |itgdStj|}|r2t ||jdgxsg|jddSdS)zUPeer name -> {url, auth, timeout, capabilities, tenant}, or treat ``agent`` as a URL.)zhttp://zhttps://)rrr capabilitiesr&tenantr)r&r'N) startswithr rrr#)r$r!s r _resolve_peerr))sx /0b5EWYZZ   # #E *Ex} E ."0M0SQS\a\e\efnpr\s tHDHHrch|r/|jddk(r|jdr dd|diSiS)Ntypebearertoken AuthorizationzBearer r)rs r _auth_headerr01s@;?DHHVDTX`D`eiememnuevOwtG}o6 7~|~~rrctjj||||}tjj||5}t j |j jdcdddS#1swYyxYw)N)dataheadersmethod)rutf-8)urllibrequestRequesturlopenjsonloadsreaddecode)rr3rr4r2reqresps r _http_jsonr@5sj .. 4 PC   W  57zz$))+,,W56777s 2BB ct|||dS)NGET)r@)rr3rs r_http_get_jsonrC;s c7GU 33rc dtjd|}t|||dtj|j dS)Nzapplication/json)z Content-Typez A2A-VersionPOSTr5)r PROTOCOL_VERSIONr@r:dumpsencode)rbodyr3rhdrss r_http_post_jsonrK?s?.x?X?X d\c dD c4&$**T2B2I2I'2R SSrc|jd} t|dz||S#tjj$r}|j dk7rYd}~nd}~wwxYwt|dz||S)zLGET the v1.0 agent-card.json; on 404 fall back to the v0.2 agent.json alias./z/.well-known/agent-card.jsoniNz/.well-known/agent.json)rstriprCr6error HTTPErrorcode)base_urlr3rbasees r _fetch_cardrUDsn ??3 Dd%CCWgVV << ! ! 66S=   $!::GW MMs#AAAct|trV|jdgxsgD]=}t|ts|jddk(s)|jds;|cSy)NsupportedInterfacesprotocolBindingJSONRPCr isinstancedictr)cardifaces r_select_jsonrpc_interfacer_Os`$XX3R8>B E%&5995F+G9+TY^YbYbchYi   rct|x}rt|dSt|tr)t|j dtr |dr|dS|j dS)zYCard's JSONRPC interface (v1.0 supportedInterfaces) > card's legacy top-level url > base.rrM)r_strr[r\rrN)rRr]r^s r_rpc_urlrbWs[)$//u/5<  $*TXXe_c"BtE{E{ ??3 rmessage context_idc *|jdd}t|jdixsi}t|jdt} t ||t |d}|xstj}tj|} dtjdd tjtj| | id } t|} | r| jd rt| d nt|jd xsd} | r| | d d <tj d|| d| tj"|d| | dtj$xj&dz c_t)t+||| ||} d| vr't-d|d| djd | dtj.| jdi}t1|}|d}}t3|t4r7|jd|}|jdxsijdd}tj"|d|| dtj$xj6dz c_|||fS#t $rd}Y(wxYw)zOne SendMessage to a peer -> (reply_text, context_id, state). Raises urllib errors / ValueError for the caller to format; handles redaction, audit, persistence, metrics.rrrrNz2.0 SendMessagerc)rd)jsonrpcidr4paramsr'rjoutboundriuserr rOzPeer 'z' returned an error: result contextIdstatusstater$)rr0rr rUmin Exceptionr new_context_idr redact_outbound new_task_id text_message ROLE_USERr_raauditpersist_messagemetricsoutbound_totalrKrb ValueErrorunwrap_send_message_response_reply_text_from_resultr[r\ inbound_total) agent_labelpeerrcrdrRr3rr]ctx safe_messagerpc_bodyr^r'r?payloadreply reply_ctxrps r _send_taskr`sXxxr"H488FB/526G$((9&678G8Wc'2.>?  1//1C++G4L (<(<(>-$h&;&;H YG ""a'" )U ""3 sJ JJcFtj|}t|ts t |S|j dgxsgD]}tj |}|s|cStj |j dixsij dxs|S)N artifactsrorc)r r}r[r\rar extract_text)rmartifacttxts rr~r~s  2 26 :F fd #6{JJ{B/52##H- J  &**Xr":"@b!E!Ei!P!ZTZ [[rzNError: peer '{agent}' rejected auth (HTTP {code}). Check the configured token.z>Error: peer '{agent}' rate limited us (HTTP 429). Retry later.)iiic t|jdxsdj}|sy t|it}|jd ixsi}|jd gxsg}|jd rd nd}djd|jdgxsgDxsd|jddd}d|jddd|jdddt||d|dt|jddt|jdd |d!t|d"g} | jd#|dd$Dd%j| S#t j j$r}d|jd|dcYd}~Sd}~wt$r}d|d |dcYd}~Sd}~wwxYw)&z.Fetch and summarize the Agent Card at ``url``.rrz6Error: 'url' is required (e.g. http://localhost:9999).u!Error: discovery failed — HTTP z from .NzError: could not reach u — r&skillsr yesno, c3K|];}t|tr)|jddd|jdd=yw)rX?z vprotocolVersionNrZ).0is r za2a_discover..sF 1d@S 55"C ( )AEE2CS,I+JKsAArWvrrz (pre-1.0 card)zAgent: namez Description: descriptionzURL: z Protocol: z Streaming: streamingz Push: pushNotificationsz Auth required: zSkills ():c 3K|]<}d|jd|jddd|jdd>yw) - rrir: rrNr/)rss rrza2a_discover..sBkZ[4faeeD#&678155PR;S:TUksAA )rarstriprUr r6rOrPrQrrjoinrbboollenextend) args_rr]rTcapsrrprotoliness r a2a_discoverrs dhhuo# $ * * ,C G83$45 88NB ' -2D XXh # )rFHHZ(5dD II((0"5;  ? TXX' - .o > txx,- .-XZ@[?\0]afgopsuygzf{_|% !4 567xTXXNaEb@c?dduvzu{ |F }B ' )E LLk_efigi_jkk 99U ! << ! !H2166(&QGG 8(U1#Q778s/E77G F,&G , G 8GG G c Rt|jdxs(|jdxs|jdxsdj}t|jdxs(|jdxs|jdxsdj}t|jdxs|jd xsdj}|r|sy t|}|r|jd sd |d S t ||||\}}}|jddjddj} d|d||rd| ndzdz} |xsd} |tj k(r | d|dz } | d| S#t j j$rE} tj| jdj|| jcYd} ~ Sd} ~ wt$r} t| cYd} ~ Sd} ~ wt$r} d|d| dcYd} ~ Sd} ~ wwxYw)zaSend a task to a peer (configured name or direct URL); ``context_id`` continues a prior exchange.r$ agent_namerrrctexttaskrdrnz/Error: both 'agent' and 'message' are required.rzError: unknown agent 'zP'. Configure it under 'a2a_agents' in config.yaml or pass a full http(s):// URL.u0Error: call to '{agent}' failed — HTTP {code}.)r$rQNzError: call to 'u ' failed — r TASK_STATE_r-[u · context u · ]z(no text reply)uS (The peer needs more input — answer by calling a2a_call again with context_id 'z'.)r)rarrr)rr6rOrP_HTTP_CALL_ERRORSrQformatr|rrreplacelowerr STATE_INPUT_REQUIRED) rrr$rcrdrrrrprT short_stateheaderrIs ra2a_callrs !UTXXl%;Utxx?OUSU V \ \ ^E$((9%S&)9STXXf=MSQSTZZ\GTXXl+Jtxx /DJKQQSJ @  D txx'w.~;",UD':"Ny%-- r2::3DJJLK|I; /5T+3GVX Y\_ _F  %%D --- ghqgrruvvXRv  << ! !B $$QVV-_`ggnsz{{A{AgB B 1v ;!% aS::;sBFH&+:G+%H&+ H&7 HH& H&H!H&!H&c pt}g}|r|jdt|d|jD]~\}}|j dg}|jd|d|j ddd|j d xsij d d d |rd dj |ndzn|jdt jx}rJ|jd|jdt|d|jd|ddDt jj}|jd|jd|dd|dd|dd|dd|dd |d!d"|d#d$|d%d&|d'd(d)j |S)*z@List configured A2A peers, persisted conversations, and metrics.zConfigured peers (rr&rrrrz (auth: rr+none)z caps: rrz@No peers configured. Add them under 'a2a_agents' in config.yaml.zPersisted conversations (u) — recall with a2a_history:c3&K|] }d| yw)rNr)rcs rrza2a_list..s5AtA3Z5sNz Metrics: rz in / r{z out, tasks_completedz completed, tasks_failedz failed, streams_startedz streams, push_sentz push sent, anti_loop_triggersz anti-loop, rate_limit_triggersz rate-limited, avg avg_latency_msmsr) rappendritemsrrr list_conversationsrrzsnapshot) rrpeersrrr!rconvosms ra2a_listrs  E E  )#e*R89 ;;= JKD%99^R0D LL4vR %(='>h RXHYH_]_GdGdekmsGtFuuvw;?gdiio%67RI J J  WX,,..v. R 0V =[\] 555!!#A LL LL9Q/0q9I7J6K6RSTeRfQggsn%&i2C0D/EZPQR]P^O__k*+,L;P9Q8RRefghxfyezz|~ 99U rc @t|jdxs|jdxsdj}|sytdt t |jdxsddd}t j|| }|sd |d Sd |d t|dg}|D]b}|jdxsdj}|jd|jdddt|dkDr|dddzn|ddj|S)zCRecall a persisted A2A conversation (survives compaction/restarts).rdrnrzGError: 'context_id' is required (see a2a_list for known conversations).r limit2)rz'No persisted conversation for context ''.z Conversation z (last z messages):rrrolerz] iNu …[truncated]r) rarrmaxrq _coerce_intr load_conversationrrr)rrrdrmessagesrrrs r a2a_historyrsTXXl+Jtxx /DJKQQSJ X 3{488G#4#:B?E FE))*EBH 8 BGGZLH k J KE pf #**, qvs+,BRUVZR[^bRbtET{=N/Nhl.mnop 99U r capabilityc tjDcgc]&\}}||jdgxsgvs|dk(r||f(c}}Scc}}w)zAConfigured peers that advertise the capability ('*' matches all).r&*)rrr)rrr!s r_match_peers_by_capabilityrsY->-@-F-F-H YkdEeii;ArBjTWFW5M YY Ys+A c t|t|||\}}}||xsdfS#t$r}|d|fcYd}~Sd}~wwxYw)z=Call a single peer synchronously -> (agent_name, reply_text).z (no reply)Error: N)rr#rr)r peer_entryrcrdr_ctx_staterTs r_call_peer_syncrsW+(5Ej5QSZ\fgtVE1\22 +gaSM**+s"% ?:??c t|jdxsdj}t|jdxs|jdxsdj}t|jdxsdjj}|dvr|nd}t|jdxsdj}|sy |sy t |x}sd |d Sg}t t t|t 5}|D cic]\} } |jt| | ||| !} } } t| D]\} | | } |j| j|dk(r0|ddjds| D]} | jn^ddd|j#d|D cgc]\} }|jdr| |f}} }|dvrV|s.dj%dg|D cgc] \} }d| d|c}} zS|dk(rt'|dn|d\} }d|d| d|Sdj%d |d!t|d"g|D cgc]\} }d#| d$|fD]}|c}}} zScc} } w#t $r!}|j| d|fYd}~nd}~wwxYw#1swYxYwcc}} wcc}} wcc}}} w)%uFan-out a task to peers matching a capability. Modes: ``all``, ``first`` (first successful), ``best`` (longest successful — coarse; use ``all`` to judge yourself).rrrcrmodeallrfirstbestrdzError: 'message' is required.z;Error: 'capability' is required (or use '*' for all peers).z1Error: no configured peers advertise capability 'r) max_workersrr zError:rNc |dS)Nrrrs rz!a2a_orchestrate..s qtr)key)rrrzAll peers failed:z rrct|dS)Nr )rrs rrz!a2a_orchestrate..s3qt9rrrz] zOrchestrated 'z' to z peer(s):z --- z ---)rarrrrrrqr_ORCHESTRATE_MAX_WORKERSsubmitrrrrmr(cancelrrsortrr)rrrrcrrdmatchesresultspoolrr!futuresfutfrTr successeslines ra2a_orchestratersTXXl+1r288:J$((9%?&)9?R@FFHG txx(5 ) / / 1 7 7 9D334DTXXl+1r288:J . L1*= =G =B:,bQQ%'G CL:R(S T 6X\krs\g\`bg4;;eWjQSWWss( 6C3+D+DX+N$# # 6 6 LL^L$29\;4AQAQRZA[$\I\   9912^e5f{tUZ4&5'6J5ffg gAEc))<=U^_`Ua e4&4&E7++ 99zl%G ~YOP07aauvdVSWEXZ_D`aDaab cc%t 6sm455 6 6 6]6gbsaK$JKAJ+KK)K K /KK K"J>8K>KKKc d|dS)Nstringr+rr)rs r_strr %s [ 99rzFetch and summarize another agent's A2A Agent Card from a URL (its name, description, capabilities, and skills). Use this to find out what a remote agent can do before calling it.zDllnM::w Z.6j%Q %-Jz " tUGQ\*.{Zd!e ,7K  M Mr) TypedDict)returnr\)r!r\r"rr'r\)r$rar'Optional[dict])rr\r'r\)N) rrar3r\rrr4rar2zOptional[bytes]r'r\)rrar3r\rrr'r\) rrarIr\r3r\rrr'r\)rRrar3r\rrr'r\)r]r(r'r()rRrar]r(r'ra) rrarr\rcrardrar'ztuple[str, str, str])rmrr'ra)rr\rrr'ra)rz dict | Nonerrr'ra)rrar'zlist[tuple[str, dict]])r) rrarr\rcrardrar'ztuple[str, str])rrar'r\)r'r)r'None)7__doc__ __future__rrr:loggingr urllib.errorr6urllib.requestconcurrent.futuresrrtypingrrgateway.platforms._sharedrrrr r getLogger__name__loggerr rrrr#r)r0r@rCrKrUr_rbrr~ _AUTH_ERRrrrrrrrrr r __annotations__rr%r&rrrr7sh#  ? @   8 $( 2M H7 4T N !#H \ ] #):z{460  Y +#cL: "uT"`abejdkm< bc!"]^ $%m np9%'gikmopR#'(S#T'0Apqs!N $ (=(,,q'r$()R$S*2