߲j<UdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z ddl m Z ddlmZej dZej$Zdad ed <ej$ZiZd ed <d ZiZd ed<d&dZd'dZd(dZd)dZddd*dZd+dZd,dZ d(dZ!dddd d-dZ"d.dZ#e$hdZ%d/dZ&d0d1dZ'e$hdZ(d2dZ)d3d Z*d4d!Z+d5d"Z,d#d$d%Z-y)6zXLive-subagent registry + model-facing control plane (list/steer/stop) for delegate_task.) annotationsN)AnyDictListOptional)request_hard_interrupt) tool_errorztools.delegate_toolFbool _spawn_pausedzDict[str, Dict[str, Any]]_active_subagents_recent_subagentsc|rt|tsyt5tj |xst j |}dddy||j d|j ddS#1swY0xYw)z``{subagent_id, goal, delegation_id}`` for a process task_id that belongs to a live or recently-finished child (children run their terminal sessions under ``task_id == subagent_id``), else None.Ngoal delegation_id) subagent_idrr) isinstancestr_active_subagents_lockr getr)task_idrecords A/home/ubuntu/.hermes/hermes-agent/tools/delegate_tool_registry.pyget_subagent_attributionrs{ *Wc2 R"&&w/Q3D3H3H3QR ~"FJJv,>QWQ[Q[\kQl mm RRs -A66A?c\t5t|atcdddS#1swYyxYw)zfGlobally block/unblock NEW delegate_task spawns (active children keep running). Returns the new state.N)_spawn_pause_lockr r )pauseds rset_spawn_pausedr's( V s"+cFt5tcdddS#1swYyxYw)N)rr ris_spawn_pausedr".s s c|jd}|sy|jddt5|t|<dddy#1swYyxYw)Nraccepting_steerT)r setdefaultrr )rsids r_register_subagentr'2sJ **] #C  '. (!'#(((s AA )agentct5tj|}|||jd|us dddytj|d|jd}|s dddydDcic]}||j|c}t|<t tt kDrCtjtttdt tt kDrCdddycc}w#1swYyxYw)z[Drop the live record (exact agent identity when given) and keep a bounded attribution stub.Nr(r)rrowner_agent_session_id) rr rpoprlen_RECENT_SUBAGENTS_CAPnextiter)rr(rr&ks r_unregister_subagentr1:s  G"&&{3 >%-6::g3F%3O  G G k40jj'  G G=p!pq!VZZ]"2!p##$'<<  ! !$t,='>"? F#$'<< G G"q G Gs).C?*C?1C?5C:A"C?:C??Dct5tj|}||jd|ur dddyd|d<t|dd}t |s dddy |}t|tr|jr|ndcdddS#t $r)}t jd||Yd}~dddyd}~wwxYw#1swYyxYw)anAtomically close steer acceptance and drain its final durable artifact. ``steer_subagent`` holds the same registry lock through ``agent.steer``, so either acceptance wins and this drain sees its exact text, or closure wins and the caller is rejected. Exact agent identity prevents a finishing child with a recycled public id from closing its replacement.Nr(Fr$_drain_pending_steerz#final steer drain for %s failed: %s) rr rgetattrcallable Exceptionloggerdebugrrstrip)rr(rdrainpendingexcs r_close_subagent_steeringr=Hs Q"&&{3 >VZZ0= Q Q%* !5t< Q Q gG%Wc2w}}wD Q Q  LL> S Q Q Q  Q Qs:,C C $B+#C  C !C8C C  C  Cc t5tj|}dddr|jdnd}|y tt |d|dS#1swY;xYw#t $r!}t jd||Yd}~yd}~wwxYw)zRequest that one running subagent stop at its next iteration boundary (cooperative: the flag propagates to in-flight tools and recurses into grandchildren via AIAgent.interrupt()). True iff a matching subagent was found.Nr(FzInterrupted via TUI ()z!interrupt_subagent(%s) failed: %s)rr rr rr6r7r8)rrr(r<s rinterrupt_subagentr@\s 4"&&{34#)FJJw tE }*54I+VW2XYZZ 44  8+sKs"AA#A # B ,BB cddlm}|jdy|jd}t|tr|jdn|jd}||uxst||xr|j |S)a1Authority follows the owning session's LIVE transport slot, read at check time. ``owner_transport`` on the record is only the capture-time marker that a gateway session commissioned the child (``None`` = no RPC authority ever). The slot is authoritative because every reattach path (prompt.submit, queued drain, resume, activate, viewer failover) already mutates it; a per-record copy needed a matching registry sync at each of those sites and two were missed (#106663). Records whose owner is not a session dict keep the exact-object rule.r)FanoutTransportowner_transportFowner_session_record transport)tui_gateway.transportrBrrdictcontains)rrErBownerbounds r_subagent_transport_matchesrKksr6 zz#$, JJ- .E&0&=EIIk "6::N_C`E I  c*UO"D"bXaIbcr!)owner_session_idrCrDc&|r|jsyt5tj|}|r|jdds dddy|A|jd|k7s#|!t ||r||jd|ur dddy|jd}| dddy t |j |cdddS#t$r)}tjd||Yd}~dddyd}~wwxYw#1swYyxYw)uQueue steering text into a running subagent without stopping it. AIAgent.steer() appends the text to the child's last tool result at its next iteration boundary — the current tool call is never cut. True iff the text was QUEUED while the child still accepted work; False for unknown/closed id, ownership mismatch, no live agent, or empty text. ``owner_session_id=None`` keeps the in-process helper contract; gateway callers must pass exact authority. Acceptance and completion are linearized by the registry lock: if acceptance wins but no delivery boundary remains, the text lands in the entry as ``missed_steer``. Fr$NrLrDr(zsteer_subagent(%s) failed: %s) r9rr rrKr steerr6r7r8)rtextrLrCrDrr(r<s rsteer_subagentrP|s tzz| "&&{3VZZ(95A  ' JJ) *.> >&.vG#+zz019MM 7# =  D)*!"  LL8+s K'" #s;+D:DD/C DC?2D?DDDcD|sy ddlm}||S#t$rYywxYw)uExact request transport + live session generation, if any — an in-process bridge, not a serializable capability. Non-gateway hosts get ``(None, None)``.)NNr) _current_session_steer_authority)tui_gateway.serverrRr6)rLrRs r _capture_gateway_steer_authorityrTs0 G/0@AA s  >r(r$rCrLrDc t5tjDcgc]/}|jDcic]\}}|tvs||c}}1c}}}cdddScc}}wcc}}}w#1swYyxYw)zCopy of the running subagent tree ({subagent_id, parent_id, depth, goal, model, started_at, tool_count, status, ...}); safe from any thread.N)rr valuesitems_PRIVATE_RECORD_KEYS)rr0vs rlist_active_subagentsr[sh sWhWoWoWqrrRS!'')M$!Qq8L/LAMrssMrsss3A/A( A" A"A( A/"A( (A//A8c||y|}t|D]/}t|dd}t|r|nd}|y||ury|}1y)uTrue when *child_agent* sits below *parent_agent* in the spawn tree (walks the ``_delegate_parent_ref`` weakref chain stamped at build time). Identity only — a parent may steer/stop its own children and grandchildren, never a sibling tree owned by another conversation.NF_delegate_parent_refT)ranger4r5) child_agent parent_agentmax_hopscur_refancestors r_is_descendant_ofrfsfl2 C 8_c148$SM35t   | # r!>liststoprNct|xsd}t|dd}|r||S |j|}|r t|S|S#t$r|cYSwxYw)zTip of a session id's compression lineage via the parent's live SessionDB (best-effort; input unchanged when unavailable) so a delegation dispatched before a compression rotation still matches the rotated parent. _session_dbN)rr4resolve_resume_session_idr6) session_idr`r&dbresolveds r_resolve_session_lineagerpsg jB C }d 3B "* //4 (s8}1c1  sAA AAct|jd|ryt|jdxsd}tt|ddxsd}|r|sy||k(ryt |||t ||hvS)uTrue when *parent_agent*'s conversation owns this live-child record. Tier 1: identity — the ``_delegate_parent_ref`` weakref chain reaches *parent_agent* (fast path while the parent AIAgent survives the run). Tier 2: durable lineage — the record's ``owner_agent_session_id`` matches the caller's ``session_id`` after resolving compression-rotation lineage on both sides. Tier 2 exists because the identity chain is BRITTLE across parent rebuilds: the CLI sets ``self.agent = None`` mid-session (route change, credential refresh, /model, MoA one-shots) and builds a NEW AIAgent while the child keeps a weakref to the old one. Delivery routes by durable session id; control must use the same spine or running children go invisible/unsteerable. r(Tr*rjrmF)rfrrr4rp)rr` owner_sid parent_sids r_owns_subagent_recordrtsG,l;FJJ78>B?IW\<<BCJ JJ #I| <Mefpr~MAA AAr!ct5ttj}dddg}D]}t ||s|j d}|j |j d|j d|j d|j d|j dt|ttfr!ttj|z dndt|j dd t|j d d dd d t||d}|sd|d<|S#1swYxYw)N started_atr parent_idrmodelstatusr$Fr(_live_transcript_path)rrwrrxryrunning_secondsr$live_transcriptrg)actioncount subagentsuNo live subagents right now. Children that already finished have delivered (or will deliver) their results as normal completion messages — there is nothing to steer or stop.note)rrgr rVrtrappendrintfloatroundtimer r4r,)r`recordsentriesrYstartedpayloads r _list_payloadrs% 3(//123G   $Q 5 %% %55/{+EE&MUU7^eeHoBLWWZ\aVbBcuTYY[7%:A>im#AEE*;U$CD&quuW~7NPTU   *0#g,U\]G  I  N133s D;;Ec|dk(r tjt|dS|xsdj}|st d|dSt 5t j|}ddd t||st d|d S|d k(r|xsdjs t d Stj|}|t d |d S|\}}} |dk(r t|nt||j} | rtj||||ddSt | j|S#1swYxYw)zSynchronous control plane for delegate_task: list/steer/stop. Runs in-turn (never backgrounded) over the same registry the TUI overlay drives, scoped so a conversation can only control its own spawn tree.rgF) ensure_asciirjzaction='zK' requires subagent_id (from the spawn dispatch response or action='list').NzNo live subagent 'z' in this conversation's spawn tree. It may have already finished (its result arrives as a normal completion message). Use action='list' to see live children.rNzOaction='steer' requires a non-empty 'message' describing the course correction.zUnknown action 'z#'. Use spawn, list, steer, or stop.rh)r~rryr)r&) jsondumpsrr9r rr rrt_CONTROL_OUTCOMESr@rPformat) r~rmessager`r&routcomeryrfailureoks r_handle_control_actionrsXzz- 5EJJ  " # # %C HVH,wxyy ,"&&s+, ~26<H &K K  '-R!6!6!8kll##F+G,VH4WXYY#FD'$*f$4 C .gmmo:^B zzVC6[_`otuu gnnn- ..#,,s D??E)interrupt_requesteduThe subagent stops at its next iteration boundary (in-flight tool calls are asked to cancel). Its partial result still re-enters the conversation as a completion message — do not wait or poll.uyCould not interrupt '{sid}' — it likely finished in the last moment. Its result arrives as a normal completion message.)queueduSteering text queued. The subagent sees it appended to its next tool result — the current tool call is never cut. If the child finishes before a delivery boundary remains, the text is reported back as missed_steer in its completion entry.zSubagent '{sid}' is no longer accepting steering (finishing or already finished). Its result arrives as a normal completion message; re-delegate a follow-up task if more work is needed.)rhrN)r Optional[str]returnzOptional[Dict[str, Any]])rr rr )rr )rDict[str, Any]rNone)rrr(rrr)rrr(rrr)rrrr ) rrrOrrLrrCrrDrrr )rLrrztuple[Any, Any])rzList[Dict[str, Any]]))r_rr`rrarrr )rmrr`rrr)rrr`rrr )r`rrr) r~rrrrrr`rrr).__doc__ __future__rloggingr threadingrtypingrrrragent.interrupt_compatrtools.registryr getLoggerr7Lockrr __annotations__rr r-rrrr"r'r1r=r@rKrPrT frozensetrXr[rf_CONTROL_ACTIONSrprtrrrr r!rrsJ^"  ,,9%   0 1"INN$ t')02,1 /1,1 n(<@ GQ( d$GKcg $!!!6C!]`!! !F !!|}s &67 A06/> r!