)Xjqd|dZddlmZddlZddlZddlZddlmZddlm Z dddd Z e ejjd d Ze ejjd d ZeejjddZdZddZddZddZGddZGddZeZddZdZy)u Pre-warmed, supervised background workers — the agent pool. WHY THIS EXISTS (2026-07-14, Ahmed's call after three failed fixes): the old delegation path ran workers as background Task-tool subagents INSIDE the brain's CLI session. That broke in three ways at once: 1. SLOW START — a "spawn" was a full coordinator turn streaming the whole worker prompt into a Task call before the subagent even booted (10-20s). 2. BLIND — the SDK stream is only read while a brain turn is open, so a worker finishing/dying/stalling while the room was quiet sat unread: HUD chip stuck "running" forever, no report, Ahmed waiting on nothing. 3. HONOUR-SYSTEM REPORTS — a worker only reported by remembering to append to control/report.txt; a crashed or stuck worker wrote nothing, and nothing supervised it. The pool fixes all three by making workers SEPARATE ClaudeSDKClient sessions owned by the engine, exactly like Claude Code's own harness: - PRE-WARMED: WARM_SPARES (default 1) idle sessions sit fully connected (CLI subprocess + MCP servers booted). dispatch() hands the prompt to a warm spare INSTANTLY, and the moment a spare is taken a replacement starts warming — there is always one ready (Ahmed's N+1 rule). - SUPERVISED: every running worker has a dedicated reader task consuming its stream the whole time. Every tool call emits task_progress (live HUD chip note), a no-activity watchdog reports a stall LOUDLY instead of letting Ahmed wait forever, a crash/death reports LOUDLY, and the final result is delivered through report_sink (the engine inbox rail) the moment it lands — no report.txt honour system. - KILLABLE: kill() disconnects the CLI subprocess immediately and clears the HUD chip in the same breath — no zombie UI. The warm spare means capacity is instantly back. Tiers map to models exactly like the old _AGENTS: worker=sonnet (the only pre-warmed tier — it's the default), heavy=opus, genius=fable (cold-started on demand; a few seconds slower, which is fine for rare heavy jobs). Env knobs: WARM_SPARES (1), AGENT_MAX (4 running), WORKER_STALL_S (300, 0=off), AGENT_POOL=0 disables the whole pool (llm.py then restores the old Task-tool path). Wiring: main.py sets pool.report_sink = app.inbox.put_nowait and calls pool.start() at boot / pool.stop() at shutdown; llm.py registers build_server() as the "agents" MCP server on the BRAIN session only (workers never get it — no recursive fan-out). ) annotationsN task_ledger)emitsonnetopusfableworkerheavygenius WARM_SPARES1 AGENT_MAX4WORKER_STALL_S300g4@c@ ddlm}||y#t$rYywxYw)uGSame trail as llm.py's _log_agent — one lifecycle log for everything.r) _log_agentN) voice.llmr Exception)liners >/Users/ahmed/devFolder/Ultron/claude-voice/voice/agent_pool.py_logr@s$ (4    s  c|r|jdry|jds#|jds|jdsy ddlm}ddlm}|j syd |j |z}|j |d |j|jd d y#t$r$}tdt|ddYd}~yd}~wwxYw)uDrop a milestone into jarvis-memory when a task ends, so "what happened to X?" works days later from any device — not just this process's lifetime. Fire-and-forget (never blocks the pool); skips trivial no-op tasks and only checkpoints once per task. checkpointedNresulterrorstepsrr)memory_controluBackground task — task)kindidzPOOL checkpoint skipped: x) getvoicerr enabled summarize fire_savemark_checkpointedrrstr)rec_tlr textes r _checkpointr1Is #''.) GGH !1SWWW5E 9,(%%' % c(::  F 3 cggdB/0 9 (Q 7889s B0'AB00 C9CCct|ddxsd}t|ddxsi}d} |dk(rt|jdddd}n|d vrtxx 2./4D . .77##Cb(A$BCCRHD [ txx,-cr2D Z txxr*+CR0D[[]a%!'')779Sb>D# ::d B D $dV2dV .$. sCD<6D<D<< E  E c$eZdZddZddZddZy)_Workerc||_||_||_||_d|_t j |_t j |_d|_ d|_ d|_ y)NstartingFr$) r#tiertitleclientreadertime monotonicstarted last_activity last_note stall_warned ledger_id)selfwidrVrWrXs r__init__z_Worker.__init__zsX   +/ ~~' !^^-#!cttj|jz }|dk\r|dzd|dzddS|dS)N<m02ds)intrZr[r\)raris ragez _Worker.agesJ  4<</ 0-."W!r'!AF3s1d|$A MMQ M  $ $ & 2 24>>3C D%rdcK |jdd{}|jj|tdt |jd|xjd zc_y7W#t $r?}tdt |ddtdt |dddYd}~[d}~wwxYw#|xjd zc_wxYww) Nr zPOOL warm spare ready (spares=)zPOOL warm FAILED: z$ [agent pool: warm-up failed: r%]r) _connectr{appendrrrr,printr|)rarXr0s rrzAgentPool._warm_ones ==22F LL   ' 1#dll2C1DAF G MMQ M3 J %c!fTcl^4 5 8QaH I I J MMQ MsKCA2A0AA2C0A22 B:;5B50B=5B::B==CCcxKddlm}ddlm}|||}|j d{|S7w)Nr)ClaudeSDKClient)worker_options)options)claude_agent_sdkrrrconnect)rarVrrrXs rrzAgentPool._connects54, )=>nn  s /:8:c K|tvrd}t|jtk\r=dj d|jj D}dtd|dS|xj dz c_d|j }d }|dk(r*|jr|jjd }d }n |j|d{}t|||dd|} | |j|<tj|| j ||| _ |j%|d{t+j,j/|j1| | _t5d| j ||td |d||rdndd| j |j7d|d|d|rd d"Sd!d"S7#t$r<}td |d t|ddd|dt|dddcYd}~Sd}~wwxYw7#t$r}|jj|d |j'd{7n#t$rYnwxYwtj(| j"dt|ddtd |dt|dddt|dddcYd}~Sd}~wwxYww)#Nr , c3<K|]}d|jdyw)'N)rW.0rs r z%AgentPool.dispatch..sM6L!''!n6Lsz REFUSED: z agents already running (z)). Finish or kill one first (agent_kill).rrFrTPOOL z connect FAILED: rzFAILED to start a z agent: z0. Tell Ahmed plainly; do not pretend it started.Pznever started: z query FAILED: zFAILED to hand the task over: . task_started)descagentr#z DISPATCHED [z warmz coldz] title=zAgent z started (upre-warmed — already workingz fresh sessionz[). It reports back automatically when done; agent_status checks on it; agent_kill stops it.) _TIER_MODELSrr}rjoinrIr~r{poprrrr,rSrcreaterWr`r?rfailrrr_readrYrr) rar!rWrVnamesrbwarmrXr0rs rdispatchzAgentPool.dispatchs | #D t}}  *IIMdmm6J6J6LMME {*CE7K=> ? Q $))o 8  \\%%a(FD K#}}T22 CuSbz6 2 c"((aggtTB  D,,t$ $ $++-99$**Q-H ^!''#> uSEtfW',JKgg[" # ZvR7;3QR./ 0APQR./ 073 KuSE!23q6$3<.AB,TF(3q6$3<.IIJK K % D MM  c4 ( '')))     Q[[OCF4CL>*J K 5_SVDS\N; <3CF4CL>C C DsB9K<GG GAKH1H2H6BK G H1H H KHKH K"K ?IIIK  I%"K $I%%A K KK KKcXKg}|xsdjj}|dk(r$t|jj }n|dk(r4|jrt|jj dg}nb||jvr|j|g}nC|jj Dcgc] }||j jvr|"}}|s&d|jrd|j zSdzSg}|D]b}|j|dd d {|j|jd |j d td |jdd|jddj|zdzScc}w7vw)zAtarget: a worker id ('w3'), 'latest', 'all', or a title fragment.latestallrCzNo matching agent running.z Running: r$FkilledrstatusN 'rrz KILLED by requestzKilled: ; z. A warm spare is ready.) rKlowerrr}rIrW _status_shortrrr#rrr)ratargetvictimstrrs rkillzAgentPool.kills!#  x & & ( . . 0 :4==//12G (]}} 4 4 67;< $-- }}Q'(G"&--"6"6"80"8Qqww}}."8G00--$D$6$6$8#9:1 2-/1 2A//!eH/E E E LLADD6AGG9A. / 501 2 DIIe,,/III0 Fs CF*%F#)AF*1F(2A7F*c T|js(t|j}d|d|dk7rddSddSg}|jjD]}t |j }d|d|j }tr|tkDrd |d zd |j d }|j|jd |jd|jd|jd| |jdt|jddj|S)NzNo agents running. z warm sparerrir$z ready.zlast activity u s ago — u#POSSIBLY STUCK — no activity for rfz m (last: rz [z] 'u' — running r(z warm spare ready) )r}rr{rIrjror^STALL_Srr#rVrWrkr)rasparelinesrrostates rrzAgentPool.statuss(}} %E)% #qjs1: ;.01: ;%%'Aqvvx=D$TF*Q[[MBE4'>>trzlK##$;;-q2 LLADD6AFF83qwwi~aeegYb!7$ % (  qT\\*++=>?yyrdcbdjd|jjDS)Nrc3TK|] }|jd|jd"yw)rrN)r#rWrs rrz*AgentPool._status_short..(s(O8N1ADD6AGG9A.8Ns&()rr}rIrns rrzAgentPool._status_short's$yyO 8L8L8NOOOrdc D Kddlm}m}m}m}m}m}d}d} d} g} d} |jjj} | #tj| j} tj| htd{\}}|str|j s|j#tkDrd|_t%|j#dz}t'd|j(d |d t+d |j(|j, |j/d |j,d|d|j0d | j3} d} t7j8|_d|_t=||r|j>D]}t=||rtA||_t+d|j(|j0tCjD|jF|j0t'd|j(d|j0t=||s|jHjKs|jHjK} nt=||rtM|ddxsg}t=|tNr|D]}t=||stM|dds|j>}t=|tPr|n tQ|}| jS|jKddt'd|j(d|jKddn=t=||r1tM|ddrtM|ddxsd} ntM|ddxs| }n2| !| jWs| jY | s!|s| sd| rd| dnd|j0dz} | rt'd|j(d |j]d!| ddtCj^|jF| }ta||jc|dd"#d{| rd$d%je| dd&d'nd}|j/d(|j,d)|j]d!| dd*d'|d+ nt'd|j(d,|j]d!|xs| ddtCjf|jF|xs| }ta||jc|dd-#d{|j/d.|j,d/|j]d0|xs| dd1z|jiy7#t4$rYd} wxYw#d} wxYw#tjT$rI| | jWs| jYY| "| jWs| jYyyytZ$r}tQ|dd} Yd}~pd}~wwxYw#| "| jWs| jYwwwxYw77w)2zConsume the worker's whole stream: progress out, stall watchdog on, result/death delivered loudly. This is what makes agents visible.r)AssistantMessage ResultMessage TextBlockToolResultBlock UserMessage ToolUseBlockr$NT)timeoutrfrz STALL warning (zm idle) task_stalled)r#rz Heads-up: the agent working on 'u%' has gone quiet — no activity for z minutes (last: zB). It may be stuck; Ahmed can say the word to kill it or leave it.F task_progress)r#noteu … contentis_errorr%z TOOL-ERROR rzthe session ended with an errorrz1it ended without doing the job or saying anythingu — every tool call failed: z (last thing it did: rz FAILED after rDfailedrz Tool errors seen: rrzThe agent working on 'z' FAILED after i,uc Tell Ahmed plainly what broke, then relaunch a fresh agent for it — do NOT do the task yourself.z DONE after completedz Finished 'z' (z): i)5rrrrrrrrXreceive_response __aiter__r ensure_future __anext__wait_POLL_Srr_rorjrr#rrW_reportr^rStopAsyncIterationrZr[r]rJrrQrstepr`r/rKrErr,rCancelledErrordonecancelrrkrr1rrfinishr)rarrrrrrrreportr last_text tool_errorspendingstreamrrminsmsgrMrctxtr0r-errss rrzAgentPool._read.s 4 4 !# )-M !XX..0::2uQTTF*:4&HI^177C >qwwiH>>BVD&&'kk]3GHI #!..*C#G"&.."2!&c#34!$%e\:*4U*;AK QTT L',,Q[[!++F 5eAKK=!AB'y9ejj>N>N>P(- (8(8(:I"- [1 &c9d;ArG!'40%,E *5/ B(/z5(I$)MM+5a+=a3q6 + 2 2399;t3D E $uQTTF,syy{4C?P>Q%R S &- ]3sJ6")#x">#G%F")h!=!JET"7<<> fYI'9Q8HI0 Q?AF  5nQUUWIRt ~F G""1;;7C  //!eH/E E E"*$))KO*D)EQG(*  LL1!''/EEG9Bvds|nAdV<56 7 5l1557)2v7JDS6Q5TU V$$Q[[&2EIFC  //!eK/H H H LL:aggYc!%%'#>"/i$78 9 uI"*"G#GB%% "7<<> "7<<> ,:" "VDS\F ""7<<> ,:" F IsX A-U2 U CU2U"CU2:U2AU22 U2B9U29B9X 2X3CX XAX U2 U("U+#U2'U((U++U//U225W.'W1(%X  W.W)#W1)W..W11&XX X c||j |j|ytd|y#t$rYwxYw)zADeliver to the engine inbox rail (announced when Ahmed is quiet).NzTASK )rrr)rar/s rrzAgentPool._reportsI    '   & v  s / ;;c\K|jj|jdtj|j |t d|j||jY|jjs?d} tj}|j|ur|jj |jjd{|r |jd|j dyy#t$rYywxYw76#t$rY?wxYww)uTear a worker down NOW: reader cancelled, CLI subprocess gone, HUD chip cleared in the same breath — never a zombie chip again.N task_done)r#rzStopped the agent working on 'z'.)r}rr#rmarkr`rrYrr current_taskrrrXrrrW)rarrrcurrents rrzAgentPool._terminates !$$% f- [QTT&1 88  G !..0xxw&! ((%%' ' '  LL9!''"E F    (   s`BD, D (D,D$D%D)#D, DD,DD,D D)&D,(D))D,Nrprq)rVr,)r )r!r,rWr,rVr,rpr,)r)rr,rpr,rr)rrSrprq)r/r,rprq)stopped)rrSrboolrr,rprq)rtrurv__doc__rcrrrrrrrrrrrrrwrdrryrysdL  E -0^J4 "Pun (1G!$G59GrdrycHtjjdddk7S)N AGENT_POOLr0)rFenvironr&rwrdrr(r(s ::>>, , 33rdc (ddlm}m}d$d|dddddd dd d dgd d d dddgdd%fd }|dddidd%fd }|ddddddd igdd%fd }|ddddddd igdd%fd }|d!d"||||g#S)&uIn-process MCP server exposing the pool to the BRAIN session only. Explicit JSON schemas (dict-shorthand marks every param required — the Asana-era bug); `task` and `title` required, the rest optional.r)r4create_sdk_mcp_servercdd|dgiS)Nrr/)typer/rw)rs r_textzbuild_server.._textsVS9:;;rdruPHand a task to a pre-warmed background agent — THE way to delegate (the old Task tool is gone). Starts instantly and reports back automatically when done; you stay free to talk. `task` = the FULL self-contained job description (the agent can't hear the conversation — include every detail, path and constraint). `title` = a few words for the HUD chip ('scrape coffee shops'). `tier`: worker (default, Sonnet, pre-warmed = instant) | heavy (Opus, for genuinely complex/long jobs) | genius (Fable, hardest problems or when Ahmed asks for fable) — heavy/genius start fresh (a few seconds).objectstringzFull self-contained task prompt)r  descriptionz2-5 word label for the HUDr z!worker (default) | heavy | genius)r enumr)r!rWrVr!rW)r  propertiesrequiredc VKt|jddj}t|jddjxs|dd}|sdStj ||t|jddd{}|S7 w)Nr!r$rWrfzdispatch failed: empty task.rVr )r,r&rKpoolr)rNr!rWoutr s rrzbuild_server..dispatchs.488FB'(..0DHHWb)*002?d3Bi78 8MM$"%dhhvx&@"ACCSzCsBB)B' B) agent_statuszLive status of every background agent: what it's doing right now, how long it's been running, and whether it looks stuck. Use when Ahmed asks how a task is going.)r rc@KtjSwrm)rrrNr s rrz"build_server..agent_statuss T[[]##s agent_killzStop a background agent immediately (its HUD chip clears at once). `target`: an agent id from agent_status ('w3'), 'latest' (default), 'all', or a fragment of the task title.rz$'w3' | 'latest' | 'all' | title textc Ktjt|jddd{S7w)Nrr)rrr,r&rs rrz build_server..agent_kills3499S(H)E%FGGHHGs4AA A task_recalluJLook up what happened to a PAST or running background task from the durable ledger — the answer to 'did you do X?' / 'what happened to X?' / 'is that done yet?'. Survives restarts and crashes, unlike agent_status (which only sees agents alive right now). ALWAYS call this before telling Ahmed you don't know the state of something he asked you to do. `query` = a few words from the task (e.g. 'email Elie', 'coffee shops Khobar'); empty = the most recent tasks. Returns each match's status (done/running/failed/killed/interrupted), when, how long it took, and its result or the error.r?z(words from the task; empty = most recentc Ktjt|jddj Sw)Nr?r$)r recall_textr,r&rKrs rrz!build_server..task_recall s8 [,,S'21F-G-M-M-OPQQsAAagentsz1.0.0)r3versiontools)rr,rpdict)rNr rpr )rr4r)r4rrrrrr s @r build_serverr!sB=< *  ('HJ!)(DF ( ='JL  ( *+,-+, . ,2 . 0 $ 0 $ , 4"X%K%MN   II - A!H4:$;<  RR ! wz;? rd)rr,rprqrrr)rpr)r __future__rrrFrZr'r voice.eventsrrrjrr&rrrsrrrr1rQrSryrr(r!rwrdrr$s*X# "VwG "**..45  {C0 1  /7 8  9,/455*wGwGv {4Qrd