߲j .UdZddlZddlZddlZddlmZddlmZeje Z edZ e re jejeZeeed<iZeeefed<eZeeed<ej,Zej0d d Zej0edzed <ddd d ededzdedzddfdZdefdZdedzdefdZdeddfdZdedzdefdZdedzdefdZ degdfdefdZ!dedzfdZ"ddZ#y)a-Per-thread interrupt signaling for all tools: thread-scoped so interrupting one agent session does not kill tools in other sessions (the gateway runs many agents in one process). The agent passes its execution thread id to set_interrupt(); tools call is_interrupted(), which checks the CURRENT thread.N)Callable)env_var_enabledHERMES_DEBUG_INTERRUPT_interrupted_threads_interrupt_reasons_yield_threadshermes_interrupt_acting_for_tid)defaultacting_for_tid)reasonactive thread_idr returnc||ntjj}t5|rtj ntj ||r |r |t|<ntj|d|stj |trttnd}dddtr6tjd||tjjyy#1swYFxYw)zSet or clear the interrupt for *thread_id* (default: current thread); ``reason`` is an optional user-safe cause. Clearing also drops a pending yield request.Nz[[interrupt-debug] set_interrupt(active=%s, target_tid=%s) called_from_tid=%s current_set=%s) threadingcurrent_threadident_lockradddiscardrpopr_DEBUG_INTERRUPTsetloggerinfo)r rr tid _snapshots 4/home/ubuntu/.hermes/hermes-agent/tools/interrupt.py set_interruptr%s!,))2J2J2L2R2RC L MV  ! !1E1M1MsS f&, s #  " "3 -  " "3 '1AC,-t L  0 C113999 FLLs A;C))C2cttjjxsttj SN)is_thread_interruptedrrrr getris_interruptedr&9s1 !9!9!;!A!A B qF[\j\n\n\pFqqr%cP|yt5|tvcdddS#1swYyxYw)zWhether *thread_id* has an interrupt bit set (``None`` never is). Used when a wait moves onto a deadline worker (``run_bounded_sync``) so ``/stop`` targeting the original tool-worker tid still kills the subprocess. See #94285. NF)rrrs rr"r"=s. 100111%cdt5tj|dddy#1swYyxYw)aWAsk the tool running on *thread_id* to yield: a foreground terminal command hands its live process to the background registry and returns at once, so a user's mid-turn message (``redirect()`` during tool execution) is delivered instead of parked behind it. The command itself is never killed; that is what ``set_interrupt`` is for.N)rrrr(s r request_yieldr+Js* &9%&&&s&/cP|yt5|tvcdddS#1swYyxYw)z?Whether a yield is pending for *thread_id* (``None`` never is).NF)rrr(s ris_thread_yield_requestedr-Ss+ +N*+++r)c|yt5|tvrtj| dddy dddy#1swYyxYw)zKAtomically take the pending yield for *thread_id*; True if one was pending.NFT)rrrr(s r consume_yieldr/[sK   &  " "9 - s <<Acallbackctjj}t5|tvr dddy| dddy#1swYyxYw)aRun a state transition atomically with current-thread interruption. Returns ``False`` without calling ``callback`` when the current thread is already interrupted. The callback runs under the interrupt lock and must not block or re-enter any interrupt API. NFT)rrrrr)r0rs rrun_if_not_interruptedr2fsP  " " $ * *C  & &   s A A  Act5tjtjj cdddS#1swYyxYw)z;User-safe interrupt cause for the current thread, if known.N)rrr#rrrr$r%rget_interrupt_reasonr4us; H!%%i&>&>&@&F&FGHHHs 1AA ctdy)aClear any interrupt bit on the CURRENT thread: gives a user-approved command a clean slate right before it spawns its child, so a stale bit that landed during the blocking approval-wait cannot SIGINT the just-approved run. A *genuine* interrupt arriving after this call re-sets the bit and is still observed by the executor's poll loop. Call directly on the executing thread.FN)rr$r%rclear_current_thread_interruptr6{s %r%r!)rN)$__doc__ contextvarsloggingrcollections.abcrutilsr getLogger__name__rrsetLevelINFOrrint__annotations__rdictstrrLockr ContextVarr boolrr&r"r+r-r/r2r4r6r$r%rrGs6 $!   8 $##;< OOGLL!"%c#h&%'DcN'5C  6L[5K5K%t6 &&sTz2 FVZF$F3:FtF_cF(rr 1S4Z 1D 1&S&T&+t++S4ZD Xb$h%7 D HcDjH r%