o;i) vdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddl mZmZmZmZejeZde fdZd edefd Zd edefd Zd edefd Zd dlmZmZmZmZd dlm Z m!Z!ddl"m#Z#e GddZ$e GddZ%e&ej'ej(ej)ej*hZ+ de,fdZ-ddde%de,defdZ.e GddZ/dddde$d e,d!e,de,fd"Z0 d)de$d e,d!e,defd#Z1Gd$d%Z2 d*de$d&ed'ee/de%fd(Z3dS)+a Session management for the gateway. Handles: - Session context tracking (where messages come from) - Session storage (conversations persisted to disk) - Reset policy evaluation (when to start fresh) - Dynamic system prompt injection (agent knows its context) N)Path)datetime timedelta) dataclass)DictListOptionalAnyreturnc(tjS)zReturn the current local time.)rnow4/home/ubuntu/.hermes/hermes-agent/gateway/session.py_nowrs <>>rvaluectj|dddS)z0Deterministic 12-char hex hash of an identifier.utf-8N )hashlibsha256encode hexdigestrs r_hash_idr"s3 >%,,w// 0 0 : : < ``.user_)rrs r_hash_sender_idr's $8E?? $ $$rc|d}|dkr)|d|}|dt||dzdSt|S)uHash the numeric portion of a chat ID, preserving platform prefix. ``telegram:12345`` → ``telegram:`` ``12345`` → ```` :rN)findr)rcolonprefixs r _hash_chat_idr%,s^ JJsOOE qyyvv888E%!)**$566888 E??rr!)Platform GatewayConfigSessionResetPolicy HomeChannel)canonical_whatsapp_identifiernormalize_whatsapp_identifier)atomic_replaceceZdZUdZeed<eed<dZeeed<dZ eed<dZ eeed<dZ eeed <dZ eeed <dZ eeed <dZeeed <dZeeed <dZeed<dZeeed<dZeeed<dZeeed<edefdZdeeeffdZedeeefddfdZdS) SessionSourcez Describes where a message originated from. This information is used to: 1. Route responses back to the right place 2. Inject context into the system prompt 3. Track origin for cron job delivery platformchat_idN chat_namedm chat_typeuser_id user_name thread_id chat_topic user_id_alt chat_id_altFis_botguild_idparent_chat_id message_idr c |jtjkrdSg}|jdkr'|d|jp|jpdn|jdkr%|d|jp|jnQ|jdkr%|d|jp|jn!||jp|j|j r|d |j d |S) z)Human-readable description of the source.z CLI terminalr2DM with usergroupgroup: channel channel: zthread: , ) r/r&LOCALr3appendr5r4r1r0r6join)selfpartss r descriptionzSessionSource.description_s =HN * *!> >T ! ! LLNDN$Ldl$LfNN O O O O ^w & & LLC4>#AT\CC D D D D ^y ( ( LLET^%Ct|EE F F F F LL74< 8 8 8 > 6 LL4DN44 5 5 5yyrc  |jj|j|j|j|j|j|j|jd}|j r |j |d<|j r |j |d<|j r |j |d<|j r |j |d<|j r |j |d<|S)N)r/r0r1r3r4r5r6r7r8r9r;r<r=)r/rr0r1r3r4r5r6r7r8r9r;r<r=)rIds rto_dictzSessionSource.to_dictts +||/    0#/Am    0#/Am  = * MAjM   6"&"5A  ? ."oAlOrdatac"|t|dt|d|d|dd|d|d|d|d |d |d |d |d |d S)Nr/r0r1r3r2r4r5r6r7r8r9r;r<r=) r/r0r1r3r4r5r6r7r8r9r;r<r=)r&strget)clsrOs r from_dictzSessionSource.from_dictssd:.//Y((hh{++hh{D11HHY''hh{++hh{++xx --////XXj))88$455xx --    r)__name__ __module__ __qualname____doc__r&__annotations__rQr1r r3r4r5r6r7r8r9r:boolr;r<r=propertyrKrr rN classmethodrTrrrr.r.Fs LLL#Ix}###Is!GXc]!!!#Ix}####Ix}### $J $$$!%K#%%%!%K#%%%FD"Hhsm"""$(NHSM((( $J $$$  S   X (c3h. T#s(^    [   rr.ceZdZUdZeed<eeed<eee fed<dZ e ed<dZ e ed<dZe ed <d Zeeed <d Zeeed <d ee effdZd S)SessionContexta Full context for a session, used for dynamic system prompt injection. The agent receives this information to understand: - Where messages are coming from - What platforms are available - Where it can deliver scheduled task outputs sourceconnected_platforms home_channelsFshared_multi_user_session session_key session_idN created_at updated_atr c D|jd|jDd|jD|j|j|j|jr|j nd|j r|j nddS)Ncg|] }|j Srr).0ps r z*SessionContext.to_dict..s#N#N#NAG#N#N#NrcHi|]\}}|j| Sr)rrN)rjrkhcs r z*SessionContext.to_dict..s5*/!Rr)r_r`rarbrdrerfrg) r_rNr`raitemsrbrdrerf isoformatrgrIs rrNzSessionContext.to_dictsk))++#N#NT5M#N#N#N373E3K3K3M3M*.)G+/9=R$/33555d9=R$/33555d   r)rUrVrWrXr.rYrr&rr)rbrZrdrQrerfr rrgr rNrrrr^r^s h'''+-....&+t+++KJ%)J")))%)J")))  c3h       rr^ctjdpdsdS ddlm}ddlm}|}||dd}d|vpd |vS#t$rYdSwxYw) uDTrue iff the agent will actually have Discord tools this session. Two conditions must hold: 1. The `discord` or `discord_admin` toolset is enabled for the Discord platform via `hermes tools` (opt-in, default OFF). 2. `DISCORD_BOT_TOKEN` is set — the tool's `check_fn` gates on it at registry time, so the toolset being enabled in config is not enough if the token isn't configured. Returns False (safe default — keeps the stale-API disclaimer) on any error so a bad config can't silently promise tools the agent lacks. DISCORD_BOT_TOKENrcFr) load_config)_get_platform_toolsdiscord)include_default_mcp_servers discord_admin) osenvironrRstriphermes_cli.configruhermes_cli.tools_configrv Exception)rurvcfgenableds r_discord_tools_loadedrs JNN. / / 52 < < > >u111111??????kmm%%c9RWXXXG#A''AA uus+A## A10A1F) redact_piicontextrc  |jjtv}|sG ddlm}||jjj}|r |jrd}n#t$rYnwxYw|o|}ddg}|jjj }|jjtj kr| d|dn|j}|ru|j p|jrt|jnd}|jpt#|j} |jd krd |} n,|jd krd | } n|jd krd| } n | } n|j} | d|d| d|jjr"| d|jj|jr*|jjrdnd} | d| dnp|jj r#| d|jj nA|jjr5|jj} |rt| } | d| |jjtjkr,| d| dn|jjtjkr t5r|j}ddg} |jr| d|jd|jrD|jr=| d|jd| d|jdn| d|jd|jr| d |jd|| n| d| d!n|jjtjkr+| d| d"nD|jjtj kr*| d| d#d$g}|j!D]/}|tj kr| |jd%0| d&d'"||j#r| d| d(|j#$D]K\}}|rt#|jn|j}| d)|jd*|j%d+|dL| d| d,dd-l&m'}|jjtj kr| d.nL|jjp&|rt#|jjn |jj}| d/|d| d0|d1|j#$D]+\}}| d2|jd3|j%d,| d| d4d5"|S)6ay Build the dynamic system prompt section that tells the agent about its context. This is injected into the system prompt so the agent knows: - Where messages are coming from - What platforms are connected - Where it can deliver scheduled task outputs When *redact_pii* is True **and** the source platform is in ``_PII_SAFE_PLATFORMS``, phone numbers are stripped and user/chat IDs are replaced with deterministic hashes before being sent to the LLM. Platforms like Discord are excluded because mentions need real IDs. Routing still uses the original values (they stay in SessionSource). r)platform_registryTz## Current Session Contextrcz **Source:** z! (the machine running this agent)r@r2r?rArBrCrDz ()z**Channel Topic:** zMulti-user threadzMulti-user sessionz**Session type:** uN — messages are prefixed with [sender name]. Multiple users may participate.z **User:** z **User ID:** un**Platform notes:** You are running inside Slack. You do NOT have access to Slack-specific APIs — you cannot search channel history, pin/unpin messages, manage channels, or list users. Do not promise to perform these actions. The gateway may inline the current message's Slack block/attachment payload when available, but you still cannot call Slack APIs yourself.z<**Discord IDs (for the `discord` / `discord_admin` tools):**z - Guild: ``z - Parent channel: `z - Thread: `z/` (use as `channel_id` for fetch_messages etc.)z - Channel: `z - Triggering message: `uC**Platform notes:** You are running inside Discord. You do NOT have access to Discord-specific APIs — you cannot search channel history, pin messages, manage roles, or list server members. Do not promise to perform these actions. If the user asks, explain that you can only read messages sent directly to you and respond.u**Platform notes:** You are responding via iMessage. Keep responses short and conversational — think texts, not essays. Structure longer replies as separate short thoughts, each separated by a blank line (double newline). Each block between blank lines will be delivered as its own iMessage bubble, so write accordingly: one idea per bubble, 1–3 sentences each. If the user needs a detailed answer, give the short version first and offer to elaborate.z**Platform notes:** You are running inside Yuanbao. You CAN send private (DM) messages via the send_message tool. Use target='yuanbao:direct:' for DM and target='yuanbao:group:' for group chat.zlocal (files on this machine)u: Connected ✓z**Connected Platforms:** rEz)**Home Channels (default destinations):**z - z: z (ID: z)**Delivery options for scheduled tasks:**)display_hermes_homeu.- `"origin"` → Local output (saved to files)u$- `"origin"` → Back to this chat (u*- `"local"` → Save to local files only (z/cron/output/)z- `"u"` → Home channel (zb*For explicit targeting, use `"platform:chat_id"` format if the user provides a specific chat ID.* )(r_r/_PII_SAFE_PLATFORMSgateway.platform_registryrrRrpii_safertitler&rFrGr5r4rr1r%r0r3rKr7rbr6SLACKDISCORDrr;r<r=extend BLUEBUBBLESYUANBAOr`rHrarpnamehermes_constantsr)rr _is_pii_saferentrylines platform_namesrc_uname_cnamedesc session_labeluidid_linesplatforms_listrkr/homehc_idr _origin_labels rbuild_session_context_promptrsj*>*.AAL   C C C C C C%))'.*A*GHHE $ $#     D , J$  E N+17799M~(.00 TMTTTUUUUn  #]03 G ,,, ]@mCK&@&@F}$$*&**'))))))+++6++?D +DFFGGG( ,/6~/Ga++Ma   B B B B      !, <'.":<<====  ,n$  '!#&&C *S**+++~(.00 R  9      H$4 4 4 ! " " .CZ[H| @ >s| > > >???} A!3 A M8J M M MNNN n n n noooo ? ? ? ?@@@~ O MCN M M MNNN LL " " " " LL    LLT      H$8 8 8 R  &   H$4 4 4 R  F   66N  (??     ! !QW"="="= > > > LLHTYY~-F-FHHIIIM R @AAA%399;; M MNHd3=OM$,///4NPQRRRRRRRR S!%011  - -(0(>u(E(E%%z* - - -(,%%% -s   ]++ L))  -d<.@AAA  -d<.@AAA  6  .111  X hh{D111 .!444 ((?A666 #hh':A>>>  $xx(N^&/X->>>7s>7J7JKK]~x)9:I ~)((( +)*** +L  8 ..^,,--- 88I  rc eZdZdZ d-dedefdZd.dZd.dZd.d Z d e de fd Z d e defd Zd e d e dee fdZdefdZ d/d e dede fdZ d-de deddfdZde defdZ d0de de defdZde defdZdedefdZd1dedefdZde dee fd Zde d!e dee fd"Zd-d#eedee fd$Zd%e defd&Zd/d%e d'e e e!fd(eddfd)Z"d%e d*ee e e!fddfd+Z#d%e dee e e!ffd,Z$dS)2 SessionStorez Manages session storage and retrieval. Uses SQLite (via SessionDB) for session metadata and message transcripts. Falls back to legacy JSONL files if SQLite is unavailable. N sessions_dirconfigc ||_||_i|_d|_t j|_||_d|_ ddl m }||_dS#t$r}td|Yd}~dSd}~wwxYw)NFr) SessionDBzL[gateway] Warning: SQLite session store unavailable, falling back to JSONL: ) rr_entries_loaded threadingLock_lock_has_active_processes_fn_db hermes_staterrprint)rIrrhas_active_processes_fnrrs r__init__zSessionStore.__init__s( 13  ^%% (?% f . . . . . . y{{DHHH f f f dabdd e e e e e e e e e fsA B%A==Br cn|j5|ddddS#1swxYwYdS)z4Load sessions index from disk if not already loaded.N)r_ensure_loaded_lockedrrs r_ensure_loadedzSessionStore._ensure_loadeds Z ) )  & & ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s *..c|jrdS|jdd|jdz }|r t |dd5}t j|}|D]?\}} t ||j |<)#ttf$rY ID mapping).rNTrrc>i|]\}}||Sr)rN)rjrrs rroz&SessionStore._save..s&MMMeU]]__MMMrz.tmpz .sessions_)dirsuffixr$wrr)indentz!Could not remove temp file %s: %s)tempfilerrrrpmkstemprQrzfdopenrdumpflushfsyncfilenor, BaseExceptionunlinkOSErrorrr)rIrrrOfdtmp_pathrrs r_savezSessionStore._saves t<<<)O; MMt}7J7J7L7LMMM''D%&&vl(  H 2sW555 % $!,,,, $$$ % % % % % % % % % % % % % % % 8] 3 3 3 3 3    O (#### O O O @(ANNNNNNNN O   s[<DAC1% D1C55D8C59D E#D.-E#. E8EE#EE#r_c vt|t|jddt|jddS)z%Generate a session key from a source.rTrFr)rgetattrr)rIr_s r_generate_session_keyz"SessionStore._generate_session_keysB $+DK9RTX$Y$Y%,T[:TV[%\%\    rrc|jr||jrdS|j|j|j}|jdkrdSt}|jdvr%|jt|j z}||krdS|jdvrN| |j ddd }|j |j kr|td z}|j|krdSdS) u(Check if a session has expired based on its reset policy. Works from the entry alone — no SessionSource needed. Used by the background expiry watcher to proactively flush memories. Sessions with active background processes are never considered expired. Fr/ session_typenoneidlebothminutesTdailyrrhourminutesecond microsecondr!days)rrdrget_reset_policyr/r3moderrgr idle_minutesreplaceat_hourr)rIrpolicyr idle_deadline today_resets r_is_session_expiredz SessionStore._is_session_expireds  ( ,,U->?? u--^.   ;& 5ff ;* * *!,yAT/U/U/UUM]""t ;+ + +++^&Kx&.((ya0000 +--turc|jr,||}||rdS|j|j|j}|jdkrdSt}|jdvr%|jt|j z}||krdS|jdvrN| |j ddd }|j |j kr|td z}|j|krd SdS) a Check if a session should be reset based on policy. Returns the reset reason ("idle" or "daily") if a reset is needed, or None if the session is still valid. Sessions with active background processes are never reset. Nrrrrrrrrr!r#r)rrrr%r/r3r&rrgrr'r(r)r)rIrr_rdr*r r+r,s r _should_resetzSessionStore._should_resets,  ( 44V< 1``. r!N)r session_countrrrlenrrrs rhas_any_sessionszSessionStore.has_any_sessions<s 8  x--//!33    Z * *  & & ( ( (t}%%) * * * * * * * * * * * * * * * * * *s& 33,A66A:=A:F force_newc~||}t}d}d}|j5|||jvr|s|j|}|jrd}nF|jr)||_||cdddS| ||}|s)||_||cdddSd} |} |j dk} |j }nd} d} d} | ddtjjdd} t!|| ||||j|j|j| | | }||j|<|| |jj|jd }dddn #1swxYwY|jrQ|rO |j|d n2#t0$r%} t2d | Yd} ~ nd} ~ wwxYw|jr?|r= |jjdi|n)#t0$r} t9d | Yd} ~ nd} ~ wwxYw|S)z Get an existing session or create a new one. Evaluates reset policy to determine if the existing session is stale. Creates a session record in SQLite when a new session starts. NrTrF %Y%m%d_%H%M%S_) rdrerfrgrrr/r3rrrrer_r4 session_resetSession DB operation failed: %sz4[gateway] Warning: Failed to create SQLite session: r)rrrrrrrrgrr/rrestrftimeuuiduuid4hexrr1r/r3rr4r end_sessionrrrcreate_sessionr)rIr_r4rdr db_end_session_iddb_create_kwargsr reset_reasonrrrrers rget_or_create_sessionz"SessionStore.get_or_create_sessionRs0088 ff! ZA A   & & ( ( (dm++I+ k2? E#.LL) E(+E$JJLLL /A A A A A A A A 2$(#5#5eV#D#DL# 9'*E$JJLLL ;A A A A A A A A @&*N(4%).);a)?&(-(8%%!&$(!%*" LL99RRDJLL   {A A A A A A A A A A A A A A A H 8 C) C C$$%6HHHH C C C >BBBBBBBB C 8 R( R R'';;*:;;;; R R RPQPPQQQQQQQQ R sOAF4FB1FFF)G G4G//G4H H:H55H:rdrc|j5|||jvr=|j|}t|_|||_|ddddS#1swxYwYdS)z9Update lightweight session metadata after an interaction.N)rrrrrgrr)rIrdrrs rupdate_sessionzSessionStore.update_sessions Z    & & ( ( (dm++ k2#'66 %1/AE,                   sAA00A47A4c|j5|||jvr4d|j|_| ddddS dddn #1swxYwYdS)zMark a session as suspended so it auto-resets on next access. Used by ``/stop`` to prevent stuck sessions from being resumed after a gateway restart (#7536). Returns True if the session existed and was marked. TNF)rrrrr)rIrds rsuspend_sessionzSessionStore.suspend_sessionsZ    & & ( ( (dm++7; k*4          +                usAA''A+.A+restart_timeoutreasoncJ|j5|||jvre|j|}|jr ddddSd|_||_t |_| ddddS dddn #1swxYwYdS)aMark a session as resumable after a restart interruption. Unlike ``suspend_session()``, this preserves the existing ``session_id`` and the transcript. The next call to ``get_or_create_session()`` for this key returns the same entry so the user auto-resumes on the same conversation lane. Returns True if the session existed and was marked. NFT) rrrrrrrrr)rIrdrKrs rmark_resume_pendingz SessionStore.mark_resume_pendings#Z    & & ( ( (dm++ k2?!         (,$&,#.2ff+         +               us3B6BBBc$|j5||j|}||js ddddSd|_d|_d|_| ddddS#1swxYwYdS)a/Clear the resume-pending flag after a successful resumed turn. Called from the gateway after ``run_conversation()`` returns a final response for a session that had ``resume_pending=True``, signalling that recovery succeeded. Returns True if a flag was cleared. NFT)rrrrRrrrr)rIrdrs rclear_resume_pendingz!SessionStore.clear_resume_pendingsZ    & & ( ( (M%%k22E}E$8}         $)E "&E *.E ' JJLLL                  s9B*BB  B  max_age_daysc||dkrdSddlm}t||z }g}|j5|t |jD]\}}|jr |j U | |j r0n8#t$r+}t d|j |Yd}~nd}~wwxYw|j|kr|||D]}|j|d|r|dddn #1swxYwY|r)tdt'||t'|S)u=Drop SessionEntry records older than max_age_days. Pruning is based on ``updated_at`` (last activity), not ``created_at``. A session that's been active within the window is kept regardless of how old it is. Entries marked ``suspended`` are kept — the user explicitly paused them for later resume. Entries held by an active process (via has_active_processes_fn) are also kept so long-running background work isn't orphaned. Pruning is functionally identical to a natural reset-policy expiry: the transcript in SQLite stays, but the session_key → session_id mapping is dropped and the user starts a fresh session on return. ``max_age_days <= 0`` disables pruning; returns 0 immediately. Returns the number of entries removed. Nrrr#z6has_active_processes_fn raised during prune for %s: %sz1SessionStore pruned %d entries older than %d days)rrrrrlistrrprrrdrrrrgrGpoprinfor2)rIrPrcutoff removed_keysrrexcs rprune_old_entrieszSessionStore.prune_old_entriess"  <1#4#41&&&&&&))6666"$ Z    & & ( ( ("4=#6#6#8#899 - - U? 0<889JKK%$%$ T!-s #f,, '',,,# - - !!#t,,,,  1               4   KKCL!!<   <   s=AD<B!D<! C+!C D<CAD<<EExmax_age_secondscddlm}t||z }d}|j5||jD]B}|jr |js1|j |kr&d|_d|_ t|_ |dz }C|r| dddn #1swxYwY|S)aMark recently-active sessions as resumable after an unexpected exit. Called on gateway startup after a crash or fast restart to preserve in-flight sessions instead of destroying their conversation history (#7536). Only marks sessions updated within *max_age_seconds* to avoid touching long-idle sessions. Sets ``resume_pending=True`` so the next incoming message on the same session_key auto-resumes from the existing transcript. Entries already flagged ``resume_pending=True`` are skipped. Entries explicitly ``suspended=True`` (from /stop or stuck-loop escalation) are also skipped. Terminal escalation for genuinely stuck sessions is still handled by the existing ``.restart_failure_counts`` counter (threshold 3), which runs after this method and sets ``suspended=True``. Returns the number of sessions marked resumable. rrR)secondsTrestart_interruptedr!N) rrrrrrvaluesrrrgrrr)rIr[rrVcountrs rsuspend_recently_activez$SessionStore.suspend_recently_active@s)$ '&&&&&))O<<<< Z    & & ( ( (--//  '5+;v+E+E+/E(*?E'26&&E/QJE                  sBB>>CCc d}d}d}|j5|||jvr ddddS|j|}|j}t }|ddt jjdd}t|||||j |j |j |j d }||j|<|||j r |j jnd|j r |j jndd}dddn #1swxYwY|jrQ|rO |j|d n2#t&$r%}t(d |Yd}~nd}~wwxYw|jrH|rF |jjd i|n2#t&$r%}t(d |Yd}~nd}~wwxYw|S) z1Force reset a session, creating a new session ID.Nr6r7r8T) rdrerfrgrrr/r3rrr9r:r;r)rrrrerr<r=r>r?rrrr/r3rrr4rr@rrrrA) rIrdrBrC new_entry old_entryr rers r reset_sessionzSessionStore.reset_sessionds  Z    & & ( ( ($-//           k2I ) 4 &&CLL99RRDJLLBBBBBBBB C 8 C( C C'';;*:;;;; C C C >BBBBBBBB CsGDCDDD+E E6E11E6F G GGtarget_session_idc d}d}|j5|||jvr ddddS|j|}|j|kr|cdddS|j}t }t |||||j|j|j|j }||j|<| dddn #1swxYwY|j rQ|rO |j |dn2#t$r%}td|Yd}~nd}~wwxYw|j rN |j |n2#t$r%}td|Yd}~nd}~wwxYw|S)aSwitch a session key to point at an existing session ID. Used by ``/resume`` to restore a previously-named session. Ends the current session in SQLite (like reset), but instead of generating a fresh session ID, re-uses ``target_session_id`` so the old transcript is loaded on the next message. If the target session was previously ended, re-open it so gateway resume semantics match the CLI. N)rdrerfrgrrr/r3session_switchz!Session DB end_session failed: %sz$Session DB reopen_session failed: %s)rrrrerrrrr/r3rrr@rrrreopen_session)rIrdrfrBrcrdr rs rswitch_sessionzSessionStore.switch_sessionsc! Z    & & ( ( ($-//           k2I#'888         !* 4 &&C$', '&3"+#-   I*3DM+ & JJLLL7               : 8 E) E E$$%68HIIII E E E @!DDDDDDDD E 8 H H''(9:::: H H H CQGGGGGGGG HsMCCACC C C99 D(D##D(3E E=E88E=active_minutescH|j5|t|j}dddn #1swxYwY|-t t |z fd|D}|dd|S)z3List all sessions, optionally filtered by activity.Nrc*g|]}|jk |Srrg)rjrrVs rrlz.SessionStore.list_sessions..s%DDDQQ\V-C-Cq-C-C-Crc|jSNrn)rs rz,SessionStore.list_sessions..s1<rT)rreverse)rrrSrr_rrsort)rIrkentriesrVs @r list_sessionszSessionStore.list_sessionss Z 3 3  & & ( ( (4=//1122G 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3  %VVi????FDDDD'DDDG // >>>s;AAArec|j|dz S)z3Get the path to a session's legacy transcript file.z.jsonl)r)rIres rget_transcript_pathz SessionStore.get_transcript_paths j#8#8#888rmessageskip_dbc2|jr|s |j||dd|d|d|d|d|ddkr|dnd |ddkr|d nd |ddkr|d nd |ddkr|d nd |ddkr|d nd  n2#t$r%}td|Yd }~nd }~wwxYw||}t|dd5}|tj |ddzd d d d S#1swxYwYd S)azAppend a message to a session's transcript (SQLite + legacy JSONL). Args: skip_db: When True, only write to JSONL and skip the SQLite write. Used when the agent already persisted messages to SQLite via its own _flush_messages_to_session_db(), preventing the duplicate-write bug (#860). rolercontent tool_name tool_calls tool_call_id assistant reasoningNreasoning_contentreasoning_detailscodex_reasoning_itemscodex_message_items) rer{r|r}r~rrrrrrr;arrF ensure_asciir) rappend_messagerRrrrrwrwriterdumps)rIrerxryrtranscript_pathrs rappend_to_transcriptz!SessionStore.append_to_transcripts 8 CG C C'') VY77#KK 22%kk+66&{{<88!(^!BBBBBBBB C22:>> /3 9 9 9 DQ GGDJwU;;;dB C C C D D D D D D D D D D D D D D D D D Ds*E+E99 F(F##F(-H  HHmessagesc|jrO |j||n2#t$r%}td|Yd}~nd}~wwxYw||}t |dd5}|D].}|tj |ddz/ ddddS#1swxYwYdS) zReplace the entire transcript for a session with new messages. Used by /retry, /undo, and /compress to persist modified conversation history. Rewrites both SQLite and legacy JSONL storage. z&Failed to rewrite transcript in DB: %sNrrrFrr) rreplace_messagesrrrrwrrrr)rIrerrrrmsgs rrewrite_transcriptzSessionStore.rewrite_transcriptsU 8 J J))*h???? J J J EqIIIIIIII J22:>> /3 9 9 9 DQ D D 3U;;;dBCCCC D D D D D D D D D D D D D D D D D D Ds&% AAA>2B>>CCc g}|jrN |j|}n2#t$r%}td|Yd}~nd}~wwxYw||}g}|rt|dd5}|D]z}|}|rb | tj |A#tj $r't d||ddYvwxYw{ dddn #1swxYwYt|t|kr;|r7td|t|t||S|S) z.Load all messages from a session's transcript.z#Could not load messages from DB: %sNrrrz*Skipping corrupt line in transcript %s: %srZufSession %s: JSONL has %d messages vs SQLite %d — using JSONL (legacy session not yet fully migrated))rget_messages_as_conversationrrrrwrrr|rGrloadsJSONDecodeErrorwarningr2)rIre db_messagesrrjsonl_messagesrlines rload_transcriptzSessionStore.load_transcripts 8 G G"hCCJOO  G G G BAFFFFFFFF G 22:>>  ! ! # # osW===   D::< $+F4Mt$T$T%,V5OQV%W%W# # #   G6+7*5*5*5 Nr)TFrp)4rXrloggingrzrrr=pathlibrrr dataclassesrtypingrrr r getLoggerrUrrrQrrr%rr&r'r(r)whatsapp_identityr*r+utilsr,r.r^ frozensetrSIGNALTELEGRAMrrrZrrrrrrrrrrrs   ((((((((!!!!!!,,,,,,,,,,,,  8 $ $hBCBCBBBB %3%3%%%%       !  T T T T T T T  T p            F i  O   ! / t8~~~ ~~ ~~~~B Q Q Q Q Q Q Q  Q n%)%* ''' '"'# '  ''''.%)%*AA A!A#A AAAAHn n n n n n n n h-1## # #L)# ######r