xjWUdZddlmZddlZddlZddlZejeZej Z ej e Z iZ ded<ddZdd Zd d Zd dZddd!dZd"dZd#dZd$dddd%dZddZdS)&u+ Hermes WebUI memory-provider session lifecycle. Batch-extraction memory providers (OpenViking, Holographic) only extract memories when AIAgent.commit_memory_session() invokes provider on_session_end(). WebUI sessions can be reopened and continued many times, so the lifecycle must guarantee: 1. Only completed, non-ephemeral turns are committable. 2. A commit finishing late must not erase work completed while it was in flight. 3. A failed commit preserves the uncommitted generation and owning agent handle. 4. Replacement/reopened agents cannot steal older dirty generations. 5. Overlapping commits are serialised via a per-session in-flight guard. CLI-parity semantics — post-turn marking, boundary extraction/commit: - Completed turn: Hermes core still mirrors the exchange through run_agent.py::_sync_external_memory_for_turn(), MemoryManager sync_all(), and provider sync_turn() WITHOUT triggering extraction. WebUI then calls mark_turn_completed() after the saved/completed-turn boundary so later drains know the synced session has uncommitted work and which agent owns it. - Session boundary: commit_session_memory() triggers AIAgent.commit_memory_session(), which calls provider on_session_end(), posting /api/v1/sessions//commit and triggering extraction. This is called only at boundaries — /api/session/new with prev_session_id, explicit agent eviction, LRU cache eviction, and shutdown drain — matching the CLI's AIAgent.commit_memory_session()/shutdown_memory_provider() boundary. The design uses a monotonic generation counter per session plus per-generation agent ownership segments. mark_turn_completed() records which agent owns the new generation. commit_session_memory() commits the earliest uncommitted segment and compare-and-clears only that captured segment after success. ) annotationsNzdict[str, dict] _sessionsreturndictcddddgdS)NrF) generationcommitted_generationagent in_flightsegmentsr 8/Users/bertmahoney/hermes-webui/api/session_lifecycle.py _new_entryr1s  !   rNonect5ttddddS#1swxYwYdSN) _conditionrclear notify_allr rr_reset_for_testsr;s                     s3AA A  session_idstrc|sdSt5t|t}||d<tddddS#1swxYwYdS)a Register the current agent handle for future completed generations. Existing dirty generations keep their original segment owner. This prevents a rebuilt/reopened agent from overwriting the handle needed to retry older failed memory-provider work. Nr )rr setdefaultrr)rr entrys rregister_agentrAs    $$Z>>g                  sAAA#&A#c|sdSt5t|}|d|d<tddddS#1swxYwYdS)zClear the current future-generation agent handle. Dirty segment owners are intentionally preserved so failed work remains retryable even if the cache drops the current agent reference. Nr )rrgetrrrs runregister_agentr!Ps     j))  !E'N                   s;AAA)r intc|sdSt5t|t}|||d<||n|d}|dxxdz cc<|d}|d}|r1|ds)|dd|ur ||dd<n||||d t|cdddS#1swxYwYdS) Nrr rr r end)startr&r )rrrrrappendr)rr rownerrr s rmark_turn_completedr*_s` q   $$Z>>  "E'N* '0B0B lq <( $  VE+. V8B<3C3CG3L3LPU3U3U",HRL   OOjeTT U U U                  sCC''C+.C+boolc|sdSt5t|}| ddddS|d|dkcdddS#1swxYwYdS)NFrr )_lockrrr s rhas_uncommitted_workr.rs u CC j)) =CCCCCCCC\"U+A%BB CCCCCCCCCCCCCCCCCCsAAAAr dict | NonecL|d}|dD]}|d|kr|cSdS)Nr r r&r )r committedsegments r_first_uncommitted_segmentr3|sC,-I$ 5>I % %NNN & 4rF)waittimeoutr4r5 float | Nonec6|sdS|tj|znd}t5t|}| ddddS|dr|s ddddS|tnD|tjz }|dkr ddddSt|t|}| ddddS|d|d|dkr ddddSt |}| ddddS|d}| ||n|d}|||d<| ddddS|d} d|d<dddn #1swxYwY |n#t$rxt d |t5t|} | d| d<t dddn #1swxYwYYdSwxYwt5t|} | d| d<| | dkr| | d<| d} | d } | r;| dd| kr)| d| r| dd| k)| r | dd | kr| d z| dd <t dddn #1swxYwYdS) NFr rrr r r&Tz-commit_memory_session() failed for session %sr r'r$) time monotonicrrrr4r3commit_memory_session Exceptionlogger exceptionrpop) rr r4r5deadliner remainingr2effective_agentcaptured_generationre_entryr1r s rcommit_session_memoryrDs u-4-@t~'))dH "" j)) =""""""""K   """"""""!!!!$t~'7'77 >> """""""" ***MM*--E}!""""""""K   %(>"? ? ?%""""""""&-U33 ?+"""""""","++g..  "','8eeeii>P>PO*#2  "9"""""""":&en!k="""""""""""""""@ --//// H*UUU  $ $ }}Z00H#(-%  ! ! # # #  $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ uu    ==,,  $)H[ !"X.D%EEE3F/0 !78I +H x{51Y>> Q x{51Y>> 5HQK0I=='01} G$                 4sF F)9F/7F3FF::F FF!F&F;;,H=';H." H=.H2 2H=5H2 6H=<H=B;LLLc t5dtD}dddn #1swxYwY|sdSd}|D]}t|drd}|s*tdt |dS)NTc<g|]\}}|d|dk|S)rr r ).0sidrs r z)drain_all_on_shutdown..s2www U5CVY^_uYvCvCvCvCvCvrF)r4zrYsq   D#"""""  8 $ $  Y  ' '               37&CCCC<ufj<<<<<<~      r