i UdZddlmZddlmZddlmZmZddlmZm Z m Z m Z ddl m Z mZd d gZd d d dddZded<ddZddZddZ ddZ ddZy)aIncremental session-summary derivation for :class:`SessionStore` adapters. :func:`fold_session_summary` lets a store maintain a per-session :class:`SessionSummaryEntry` sidecar incrementally inside ``append()`` so ``list_sessions_from_store()`` can fetch all metadata in a single ``list_session_summaries()`` call instead of N per-session ``load()`` calls. Every derived field is append-incremental (set-once or last-wins) so adapters never need to re-read previously appended entries. ) annotations)datetime)Anycast)SDKSessionInfo SessionKeySessionStoreEntrySessionSummaryEntry)_COMMAND_NAME_RE_SKIP_FIRST_PROMPT_PATTERNfold_session_summarysummary_entry_to_sdk_info custom_titleai_title last_prompt summary_hint git_branch) customTitleaiTitle lastPromptsummary gitBranchzdict[str, str]_LAST_WINS_FIELDSct|tsy |jdr|jddn|}t t j |jdzS#t$rYywxYw)z>Parse an ISO-8601 timestamp string to Unix epoch milliseconds.NZz+00:00i) isinstancestrendswithreplaceintr fromisoformat timestamp ValueError)tsnorms q/volume1/homes/robertsu/coba/app/.venv/lib/python3.12/site-packages/claude_agent_sdk/_internal/session_summary.py_iso_to_epoch_msr)'sg b# ,.KK,z%_fold_first_prompt..Ys0- LSqJq$  BAEE&M]$B BGs02  command_fallbackr u… first_prompt)r/rr0r2anyr6r!stripr searchgrouprmatchlenrstrip)datar3r+r,rawresult cmd_matchs r(_fold_first_promptrRGsI xx%& yyF" yyd"eii0B&Ct&Kii "G'4 ++i( gt $- LS- *  !%(T3'--/ $++F3 88./+4??1+='(  % + +F 3  v; DS\((*X5F%^&* "#)r7c||d|dt|dd}n |ddid}|d}|D]}td|}t|jd}d |vr|jd d u|d <d |vr|||d <d |vr(|jd }t |t r|r||d <t ||tjD],\} } |j| } t | t s(| || <.|jddk(s|jd} t | t r | r| |d< |jdd|S)uFold a batch of appended entries into the running summary for ``key``. Stores call this from inside ``append()`` to keep a :class:`SessionSummaryEntry` sidecar up to date without re-reading the transcript. ``prev`` is the previous summary for the same key (or ``None`` for the first append). Do not call this for keys with a ``subpath`` — subagent transcripts must not contribute to the main session's summary. Guard with ``if key.get("subpath") is None:`` before calling. All derived state lives in the opaque ``data`` dict; stores persist it verbatim and do not interpret it. ``mtime`` is NOT touched by the fold — it is the sidecar's storage write time and must be stamped by the adapter after persisting. It has to share a clock with the ``mtime`` returned by :meth:`SessionStore.list_sessions` for the same session (typically file mtime, S3 ``LastModified``, Postgres ``updated_at``, or whatever native timestamp the adapter surfaces); deriving it from entry ISO timestamps would make every batched-write sidecar appear strictly older than the session's current mtime, defeating the fast-path staleness check. For a new session (``prev is None``) the fold returns ``mtime=0`` as a placeholder; the adapter is expected to overwrite it. ``created_at`` latches the first parseable entry timestamp. The disk lite-parse scans the head buffer for the first timestamp occurrence, so the two paths agree for any timestamp appearing within the head window. N session_idmtimerN)rTrUrNrdict[str, Any]r$ is_sidechain isSidechainT created_atcwdr-tag) r0rr)r/rrrRritemspop) prevkeyentriesrrNrOr3msrZsrcdstvaltag_vals r(rrpseF |,']f&(  "%\!2QK 6?D%s+ eii 4 5  %#(99]#;t#CD t #!#D   ))E"C#s#!U 4')//1HC))C.C#s#S 2 99V  %ii&G'3'G%U %=@ Nr7c|d}|jdry|jdr|jdn|jdxsd}|jdxs|jdxsd}|xs(|jd xs|jd xs|}|syt|d ||d d|||jd xsd|jdxs|xsd|jdxsd|jd S)zConvert a :class:`SessionSummaryEntry` to :class:`SDKSessionInfo`. Returns ``None`` for sidechain sessions or sessions with no extractable summary, matching ``_parse_session_info_from_lite``'s filtering. rNrWNr9rFrDrrrrrTrUrrZr[rY) rTr last_modified file_sizerrFrrZr[rY)r/r)r3 project_pathrNrFrrs r(rrs  =D xx 88) *  XX( )   88N+Ktxx /CKtL  88M "  88N #     &Gn!!88L)1T HHUO 3| 3t HHUO #t88L)  r7N)r&rreturnz int | None)r3rVrjz list[str])rNrVr3rVrjNone)r^zSessionSummaryEntry | Noner_r r`zlist[SessionStoreEntry]rjr )r3r riz str | NonerjzSDKSessionInfo | None)__doc__ __future__rrtypingrrtypesrr r r sessionsr r__all__r__annotations__r)r6rRrrr7r(rts # C !#> ? " %> (&RM $M M%M M`( (.8((r7