߲jldZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z m Z ddlmZmZmZmZddlmZmZddlmZmZmZmZdd lmZmZmZmZm Z dd l!m"Z"dd l#m$Z$ejJd Z&e Gd dZ'd)dZ(d*dZ)d+dZ*d,dZ+d-dZ,ddd.dZ-dddZ.d/dZ/d0d1dZ0d2dZ1d3dZ2ddd d!d"Z3d4d#Z4d5d$Z5d6d%Z6d7d&Z7d8d'Z8d9d(Z9y):uLBatch execution + background dispatch for delegate_task: ``delegate_task`` builds a ``_Batch`` (children + origin identity) and hands it to ``_run_batch``, which runs it synchronously or as detached async units — one per task ``group`` (ungrouped tasks are a unit each), so a finished review does not wait for the slowest sibling.) annotationsN)FIRST_COMPLETEDwait) dataclassreplace)AnyDictListOptional)_new_delegation_idrecord_unit_child) _attach_child _detach_child_fabricated_entry_signal_child_stop)SUBAGENT_FAILURE_STATUSES_print_completion_line_quietdescribe_subagent_failureformat_batch_tag) _capture_gateway_steer_authority)_finalize_child_resultsztools.delegate_toolceZdZUdZded<ded<ded<ded <d ed <d ed <ded<d ed<ded<ded<d ed<d ed<ded<ded<ded<ded<dZd ed<dZd ed<d!dZd"d Zy)#_BatchzOne delegate_task call's built children plus everything needed to run them and assemble the combined result (shared by the sync path and the background runner).zList[Dict[str, Any]] task_listz List[tuple]childrenr parent_agentDict[str, Any]creds Optional[str]contextstrtop_roleint max_children live_deleg_idlist live_writers live_pathsorigin_wake_sidorigin_ui_session_idorigin_owner_transportorigin_owner_session_recordboolorigin_session_history_deliveryfloat overall_startNgroupunit_idcR|jxsd|j|jdS)zKSteer/stop authority of the originating session, passed to every child run.N)owner_session_idowner_transportowner_session_record)r+r,r-)selfs A/home/ubuntu/.hermes/hermes-agent/tools/delegate_tool_dispatch.py owner_kwargsz_Batch.owner_kwargs4s.!% 9 9 ATVZVqVq$($D$D  c^ddlm}|d||d||jd|jS)Nr)_run_single_childgoal) task_indexr>childr)tools.delegate_toolr=rr:)r8itaskr@r=s r9 run_childz_Batch.run_child;sK9 FADL\`\m\mFqurCrCrEF Fr;)returnr)rCr$rDrr@rrFr) __name__ __module__ __qualname____doc____annotations__r2r3r:rErAr;r9rrs]$#  M  !$$%))E=!G]! Fr;rcp|dkDr1|r.dt||d|d}t|t|dd||yyy)zPAnnounce the batch tag once so interleaved ``[tag n/N]`` lines are attributable.u 🔀 [z ] delegating z tasks_delegate_spinnerN) console_line)rrgetattr)rn_tasksr&_hdrs r9_announce_batchrS@sM{}*=,GH V]U^^de|W\CVX\-]_crvw%{r;cddlm}|}d}d}td5ddlm}m}|dd}|}ddd||gt ||S#1swYxYw)u``(wake_sid, ui_session_id, owner_transport, owner_session_record, session_history_delivery)`` of the ORIGINATING session, captured BEFORE building any child: AIAgent construction clobbers the HERMES_SESSION_ID ContextVar/os.environ with the subagent's id. The wake- capability flag rides the same request-scoped binding and is captured here for the same reason — and fails closed: a binding that never declared it (or a read error) leaves the session treated as non-wake-capable (#98619).r)_current_origin_session_idFN)get_session_env"session_history_delivery_supportedHERMES_UI_SESSION_ID)tools.async_delegationrUrgateway.session_contextrWrXr)rU_origin_wake_sid_origin_ui_session_id _origin_session_history_deliveryrWrXs r9_capture_originr_FsB13',$ P_ /0F K+M+O(P 3 Q6VWl6m QpP QQ PPs AAc h|d}|t|kr||nd|}|jdd} |r |d|dzd|n |dzd|} |jdd uxr| d k(} | rd n| d k(rd nd } | d| d|d|jddd} | tvr,t|jd|jddnd}| rd}|r| d|z } t ||| |rF|dkDr@t d5|j d|rd|zdznd|d|dk7rdndd d!d!d!y!y!y!#1swYy!xYw)"uPrint one completion line for a finished child and refresh the spinner text. Failed/errored/timed-out children say WHY on the same line — a bare ✗ reads as "silently dropped".r?zTask status?u · rM/ schema_validF completedu⚠u✓u✗z [z] z (duration_secondsrzs)failure_reasonerrorx) max_charsrVuFoutput_schema not satisfied — raw text returned (schema_valid=false)u — zSpinner update_text failed: %su🔀 [z tasksz remainingN)lengetrrrr update_text)r spinner_refentrytag task_labelsrQ remainingidxlabelra_slotschema_invalidiconcompletion_line _err_lines r9_report_child_doner|Ws  C #c+&6 6K eC5ME YYx %F-0se4Away )Qq 6JEYY~.%7QFkPRS4T3UUWXO . . "%)),<"=uyy?Q]`a46\ U9+..<oFy1} 4 5 H  # #eC#I,<+LYKW\dmqrdr]`xz\{|F%G H H H%{ H Hs 1,D((D1cr|s5|jr(t|trt|j|yyyy)uDetached (background) unit: durably record a child on the unit's own row the moment it finishes, so an owner death before the unit's join — or anywhere before its durable completion write, the whole remaining window for a one-child unit — loses only children still running, never finished work (#116000). Best-effort by construction: ``record_unit_child`` never raises into the join.N)r3 isinstancedictr )batchrqhonor_parent_interrupts r9_record_finished_childrms/ "emm 5$8O%--/9Pm !r;c ddlm}|jt|j}}|jDcgc] }|ddd }}t |dd}t |j|} |jD cic] \} } } | |  c} } } t|j} fd||j}d } |jD  cic]B\} }} |jtjj|j| || | Dc} }} t}|r_|r.t |d d d ur|j!fd |Dd }n/t#|d t$\}}|D]}||}|j'|t)|||t+|||| ||| t|z |rN|j,s[|j/dt0vsst|| kst3dd 5ddlm}|j/dd}t9|t:r0d|cxkrt|j<krnn|j<|nd}||j,i||rd|ini|ddd|r_|j?| ||jAdycc}wcc} } } wcc} }} w#1swY]xYw#|j?| |wxYw)aRun the batch's children in parallel, appending entries to ``results`` (sorted by task_index on return, one completion line printed per child). Polls futures with a short ``wait()`` timeout instead of ``as_completed()`` so a wedged child cannot block the parent forever after an interrupt; on parent interrupt the still-pending children are reported ``interrupted`` and abandoned (they already got the interrupt signal).r)DaemonThreadPoolExecutorr>N(rNc |jst|ddj|S |jS#t$r0}t|dt |j|cYd}~Sd}~wwxYw)N interruptedu9Parent agent interrupted — child did not finish in timerh)donerrnresult Exceptionr")futureruexc_child_by_indexs r9 _entry_ofz)_run_children_parallel.._entry_ofsu{{}$]$_apatatuxay  W==? " W$S'3s8_=P=PQT=UV V WsA A9 %A4.A94A9) max_workersF_interrupt_requestedTc36K|]}||ywNrA).0frfuturess r9 z)_run_children_parallel..sIAyGAJ7Isg?)timeout return_whenraztask failure notice failedexc_info)push_task_failure_noticer?live_transcript)rQ)rcancel_futuresc |dS)Nr?rA)rs r9z(_run_children_parallel..s qr;)key)!tools.daemon_poolrrrmrrPrr&rr%submit contextvars copy_contextrunrEsetextend_cf_waitrappendrr|r3rnrrrZrr~r$r)shutdownsort)rresultsrrrrQtrsrp_tagrC_r@n_hereexecutorrpendingrrrqr_i_liverrrs @@@r9_run_children_parallelrus;!..EOO0D'L+0??;a1V9Sb>;K;,(;TBK E// >D5:^^DDMQ5q%xDO  FW(E4F4FGH KLw|xFxFGGhshiklns8??;#;#;#=#A#A5??TUWXZ_`bccGg,%',@VX]*^bf*fIII" $WcWMD' x!&'&/:u%&ue5KL"<eT;X_agjmnujvavw.5==UYYxEX\uEuL61 >auq(9:; ugbk3IJugF\]GU__enneFXFXY4>>+e.A.AA1ENByyr* dC Q$%HU5G5G1H%HUM_M_`dMeMq9DI 9""4(11%8 9c%**+++0+;+;D+A'( BU00':+2n]HN'.\e;TUZ;[\4T\T\M\$1!7<~~c)1aSQVQaQaMbIb%""1%cJc'1#$ {{!KK O# 9 9]ds)H) H5!H;1H;)H2 u8background=true is not available in this session — it cannot receive a detached subagent result after the turn ends (a finite chat using -Q, --oneshot, or non-TTY stdio, `hermes -z`, a cron job, a Kanban worker, or a stateless HTTP endpoint). The subagent(s) ran SYNCHRONOUSLY and the result is included above.aThe background delegation pool was at capacity (delegation.max_concurrent_children), so the subagent(s) ran SYNCHRONOUSLY and the result is included above. Raise delegation.max_concurrent_children in config.yaml to allow more concurrent background delegations.)no_async at_capacityc~t|}t|tr t||d<t j |dS)zGInline fallback: run the batch now and explain why it was not detached.noteF ensure_ascii)rr~r_SYNC_FALLBACK_NOTESjsondumps)rreasonrs r9_run_sync_with_noters5 #E *F&$-f5v ::f5 11r;cddlm}|ddk(ry ddlm}|ry |r|rtj d||S|rtj d |y#t$rYywxYw) zDetached result target: empty for push, a resumable API id, or None for inline. API completion only persists a row; this does not authorize a model wake. The continuation must read that row, not an authoritative caller-owned snapshot.rrWHERMES_SINGLE_QUERY_SESSION1N)async_delivery_supportedrVudelegate_task: session %s resumes server history — detached result will be persisted for the next client turn (no model wake).udelegate_task: session %s has no declared server-history consumer — running the batch synchronously so the result returns in this turn.)r[rWrrloggerinfo)r*r/rWrs r9_resolve_async_wake_sidrs 8 45<  E # % &:  89H   @AP   s A A$#A$cddlm}|d}tt|ddxsd}t d5ddlm}|dd}|d dxs|}|d k(r|r|}ddd|xs||fS#1swYxYw) u[``(session_key, origin_ui_session_id)`` the async registry routes completions by. Desktop/TUI: the routable key is the durable AIAgent.session_id — compression can rotate it mid-turn before the TUI-side dict is re-anchored, and a stale approval-context key would orphan the completion. Gateway chats keep the platform conversation key (agent:main:...). The CLI has no bound approval contextvar and no HERMES_SESSION_KEY, so the key resolves empty; its drain is a positive-ownership filter on the durable session_id (empty would fail closed), so stamp the parent's durable id. r)get_current_session_keyrV)default session_idNrHERMES_SESSION_SOURCErYtui)tools.approval_contextrr"rPrr[rW)rr+r session_keyagent_session_idrWsources r9_resolve_async_session_keyrs?)"5K7<rBHbI +; !8"= //ErJbNb U?/*K+  **,@ @@++s &A++A4c:g}d}|D]f} |j}|jd}|j|jdd||jdf|xs t|}ht ||fS#t$r|jdYwxYw)aProgress token for the async registry's stale monitor: every child's (api_call_count, current_tool, last_activity_ts). last_activity_ts ticks on streamed chunks, tool transitions and API-call start/completion, so a child streaming a long response counts as alive; a fully frozen token past the threshold means the batch is wedged. ``in_tool`` is True while ANY child is inside a tool so slow tools get the higher ceiling (mirrors the sync heartbeat).F current_toolapi_call_countrlast_activity_tsN)get_activity_summaryrnrr.rtuple) child_agentspartsin_toolcsummarytools r9_batch_progress_tokenr*s EG  ,,.G;;~.D LL'++&6:D'++N`Bab c+dG  <    LL  sA#A==BBa*Subagent is running in the background; its full result re-enters the conversation as a new message when it finishes. Results are delivered only after you END YOUR TURN: do anything that does not depend on it, then stop with a one-line status. Do not poll its transcript or artifacts to wait for it.ai{n} subagents are running in parallel in the background as {k} completion unit(s); each unit's results re-enter the conversation as their own new message when THAT unit finishes. Results are delivered only after you END YOUR TURN: do anything that does not depend on them, then stop with a one-line status. Do not poll transcripts or artifacts to wait for them.zWhile a child runs you can orchestrate it live with this same tool: delegate_task(action='list') to see live children, action='steer' with subagent_id + message to redirect one, action='stop' with subagent_id to end one early.zEach subagent streams a human-readable transcript of its operations to the file listed above (append-only, one per task). Read or `tail -f` these paths at any time to watch a child work while it runs.)onemany control_hintlive_transcripts_hintc|jDcgc]}|d }}t|}dd||jxs|dd||dk(r tdn"tdj |t|d }t|dkDrH|D cgc]3\}}||j |j D cgc]\}} } | c} }d 5c} }}}|d <|j D cgc]\} } } t| d d } } } td| Dr| |d<td|d<|jr$t|j|d<td|d<|Scc}wcc} }wcc} }}}wcc} } w)zNModel-facing handle for an accepted background call: one entry per async unit.r> dispatched backgroundrrMrr)nk)ramodecount delegation_idgoalsr)rr2 task_indexesunits _subagent_idNc3DK|]}t|txr|ywr)r~r")rrls r9rz&_dispatched_payload..gs 2:a  #! # 2s  subagent_idsrrr) rrmr&_BACKGROUND_NOTESformatr2rrPanyr)r') rrrrrpayloadunituidrCrrsidss r9_dispatched_payloadrXsr % 01QvY 0E 0 E A q,,;a e,-F!%(8I&8Q8X8X[\`cdi`j8X8kG  5zA~#  c"DJJ\`\i\iHjyPQSTVWHj k ?Dnn M!QGA~t , MD M 2T 22"&"3N"C &*5+;+;&<"#+<=T+U'( N' 1Ik  Ns# E  #E , E9E EE c hddlm}|s|gSi}|jD]P\}}}|jd}|dvr dt |fnd|f}|j |gj |||fR|jDcgc] \}}t|||ddk(r|dnd "c}}Scc}}w) a-Partition the call's children into async units: one per distinct task ``group`` (first-appearance order) and one per ungrouped task. Each unit is a ``_Batch`` sharing the call's task_list/transcripts but owning a subset of ``children``, so a unit joins only on itself and its completion re-enters the conversation on its own. Off by default (``delegation.independent_completions``): the whole call is ONE unit and returns as one message. A per-task flurry of completions (one new turn each) fragmented orchestrators that had no plan for it.r)_get_independent_completionsr2)NrVgrCrMN)rr2) tools.delegate_tool_configrrrnr" setdefaultritemsr) rrmembersrCrrrrchs r9 _units_ofrosH ' )w&(G>>61a EE'N 2sCFma3#**Aq!956cjbobobq rW^WZ\^GEBQ3s1vD R rr rs%B.c ddlm}jDcgc]\}}}| c}} fd}|d jDcgc]}|d c}jdj j dfd|||tjtjkr jD cgc]\} }}|  c}} ndjD cic]r\} }}| tjkrTj| Ej| j,t| tj| jtc}} fdd |Scc}}wcc}wcc}} wcc}} w) zSHand ONE unit to the async registry; the runner joins on that unit's children only.r)dispatch_async_delegation_batchc,D]}t|dy)NzAsync delegation cancelled)r)rrs r9 _interruptz"_dispatch_unit.._interrupts @A q"> ? @r;r>NmodelctdS)NFr)r)r sr9rz _dispatch_unit..s-d5Qr;ctSr)r)rsr9rz _dispatch_unit..s 1,?r;) rr!toolsetsrolerrunner interrupt_fnrslot_keyrtask_transcripts progress_fnrA) rZrrrr!r#rrmr(pathr") r r3r%routingrrrrrrCrs ` @r9_dispatch_unitr*sPF'+}}5)1aA5L@ + "&..1Qqy14<< ]]$**W"5Qw;>t}};MPSTXTbTbPc;c7IQ1a7imPTP]P]EWQ1T%6%6!77Dc>cdH B C"5*55%%L(B,@@ w == /KAq% , . /!$7C << !\ 1<8O#? @ $T* KKl Wj1 'um< < XZ[^_Z_adejakmumymy{BnC D4T:9EF1G2"%4G$2 ! ::gE 22r;c\|r t|Stjt|dS)zPTool result JSON: a dispatch handle (background) or the joined combined results.Fr)r9rrr)rrs r9 _run_batchr;s'#E** ::,U3% HHr;)rQr$r&r rFNone)rFztuple[str, str, Any, Any, bool])rFr<)rrrqrrr.rFr<)rrrr'rr.rFr<)rrrr.rFr)rrrr"rFr")F)r*r"r/r.rFr )rrr+r"rFztuple[str, str])rz List[Any]rFr)rrrzList[tuple[_Batch, str]]rFr)rrrFz List[_Batch]) r rr3r r%r r)rrFr)r rrFr<)rrrFr")rrrr.rFr"):rJ __future__rrrloggingrconcurrent.futuresrrr dataclassesrrtypingrr r r rZr r tools.delegate_tool_child_runrrrrtools.delegate_tool_progressrrrrrtools.delegate_tool_registryrtools.delegate_tool_resultsr getLoggerrrrSr_r|rrrrrrrrr rrr*r,r9r;rAr;r9rGs1 #  @*,,HmmJ?   0 1 !F!F !FHx Q"H,0=0~MQ%R u  m 2"HA,!2 _  :    h%..s"00 23hIr;