i6dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddl mZddlmZddlmZmZmZmZd d lmZej0eZd Zd Zd Zej<dej>Z ej<dZ!ej<dZ"ej<dZ#dFdZ$dGdZ%dHdZ&dIdZ'dJdKdZ(dLdZ)dMdZ*dNdZ+dOdZ,dPdZ-dPdZ.dQdZ/GddZ0dRd Z1dSd!Z2 dJ dTd"Z3 dJ dUd#Z4 dVd$Z5 dW dXd%Z6 dYd&Z7dZd'Z8 d[ d\d(Z9 dJ d]d)Z:e;hd*ZefZ?d^d+Z@d_d,ZAd`d-ZB dad.ZCdbd/ZDdcd0ZE dd ded1ZF dfd2ZGdgd3ZHdgd4ZIdhd5ZJdad6ZK dJ did7ZL dd djd8ZM dfd9ZNdJdkd:ZOdld;ZPdmd<ZQdnd=ZRdod>ZS dpd?ZT dqd@ZU dd drdAZV dJ dsdBZW dd dtdCZX dJ dudDZY dd dvdEZZy)wzSession listing implementation. Scans ~/.claude/projects// for .jsonl session files and extracts metadata from stat + head/tail reads without full JSONL parsing. ) annotationsN)datetime)Path)Any)SDKSessionInfo SessionKeySessionMessage SessionStore)_store_implementsiz>^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$z^(?:||||\[Request interrupted by user[^\]]*\]|\s*[\s\S]*\s*$|\s*[\s\S]*\s*$)z"(.*?)z [^a-zA-Z0-9]c2tj|r|Sy)z4Returns the string if it is a valid UUID, else None.N)_UUID_REmatch) maybe_uuids j/volume1/homes/robertsu/coba/app/.venv/lib/python3.12/site-packages/claude_agent_sdk/_internal/sessions.py_validate_uuidrDs~~j! cd}|D](}t|}|dz|z |z}|dz}|dk\s$|dz}*t|}|dk(ryd}g}|}|dkDr"|j||dz|dz}|dkDr"d jt |S) zC32-bit integer hash to base36, matching the CLI's directory naming.rlll0$0123456789abcdefghijklmnopqrstuvwxyz$)ordabsappendjoinreversed)shchchardigitsoutns r _simple_hashr)Ps A2w !VqL4  N ?  A  AAAv 3F C A a% 6!b&>" b a% 778C= !!rctjd|}t|tkr|St |}|dtd|S)zMakes a string safe for use as a directory name. Replaces all non-alphanumeric characters with hyphens. For paths exceeding MAX_SANITIZED_LENGTH, truncates and appends a hash suffix. -N) _SANITIZE_REsublenMAX_SANITIZED_LENGTHr))name sanitizedr#s r_sanitize_pathr2gsN   d+I 9~--TA--. /q 44rc tjjd}|rtt j d|Stt j dt tjdz S)zAReturns the Claude config directory (respects CLAUDE_CONFIG_DIR).CLAUDE_CONFIG_DIRNFCz.claude)osenvirongetr unicodedata normalizestrhome) config_dirs r_get_claude_config_home_dirr>ysX 34JK))%<==  %%eSy1H-IJ KKrc|r5|jd}|r"ttjd|dz St dz S)zReturns the projects directory. ``env_override`` is consulted before ``os.environ`` so callers that pass ``CLAUDE_CONFIG_DIR`` to the subprocess via ``options.env`` resolve the same directory the subprocess will write to. r4r5projects)r8rr9r:r>) env_overrideoverrides r_get_projects_dirrCsF##$78  --eX>?*L L & (: 55rc.tt|z SN)rCr2) project_paths r_get_project_dirrGs   != ==rc tjj|}tjd|S#t $rtjd|cYSwxYw)zEResolves a directory path to its canonical form using realpath + NFC.r5)r6pathrealpathr9r:OSError)dresolveds r_canonicalize_pathrNsP/77##A&$$UH55 /$$UA../s47AAcRt|}|jr|St|}t|tkry|dt}t } |j D]5}|js|jj|dzs3|cS y#t$rYywxYw)urFinds the project directory for a given path. Tolerates hash mismatches for long paths (>200 chars). The CLI uses Bun.hash while the SDK under Node.js uses simpleHash — for paths that exceed MAX_SANITIZED_LENGTH, these produce different directory suffixes. This function falls back to prefix-based scanning when the exact match doesn't exist. Nr+) rGis_dirr2r.r/rCiterdirr0 startswithrK)rFexactr1prefix projects_direntrys r_find_project_dirrWs \ *E ||~ |,I 9~-- ,, -F$&L !))+E||~%**"7"7 "E ,     s$#B4BBB B&%B&cd|vr|S tjd|d}t|tr|S|S#tjt f$r|cYSwxYw)z3Unescape a JSON string value extracted as raw text.\")jsonloads isinstancer;JSONDecodeError ValueError)rawresults r_unescape_json_stringrbs\ 3 auAJ' fc "M  * - s*55AAc d|dd|dg}|D]r}|j|}|dkr|t|z}|}|t|ks9||dk(r|dz }||dk(rt|||cS|dz }|t|kr:ty) zExtracts a simple JSON string field value without full parsing. Looks for "key":"value" or "key": "value" patterns. Returns the first match, or None if not found. rZ":"": "rrYrr Nfindr.rb)textkeypatternspatternidx value_startis r_extract_json_string_fieldros C5 #dm,Hii  7 CL( #d)mAw$QAw#~,T+a-@AA FA #d)m rcd|dd|dg}d}|D]y}d} |j||}|dkr|t|z}|}|t|kr9||dk(r|dz }||dk(rt|||}n|dz }|t|kr9|dz}u|S) z>Like _extract_json_string_field but finds the LAST occurrence.rZrdreNrrYrr rf) rhrirj last_valuerk search_fromrlrmrns r_extract_last_json_string_fieldrssC5 #dm,H!J ))G[1CQwG ,KAc$i-7d?FA7c>!6tK7J!KJQc$i-a%K$ rc@d}d}t|}||kr|jd|}|dk\r |||}|dz}n||d}|}d|vrd|vr8d|vr=d |vsd |vrFd |vsd |vrO tj|}t |tr|jd dk7r|jd}t |ts|jd}g} t |tr| j|nqt |tra|D]\} t | ts| jd dk(s)t | jdtsI| j| d^| D]} | jddj} | s&tj| } | r|s| j!d}Qt"j%| rgt| dkDr| ddj'dz} | cS||kr|r|Sy#tjt f$rY wxYw)zExtracts the first meaningful user prompt from a JSONL head chunk. Skips tool_result messages, isMeta, isCompactSummary, command-name messages, and auto-generated patterns. Truncates to 200 chars. rr r Nz "type":"user"z"type": "user"z "tool_result"z "isMeta":truez"isMeta": truez"isCompactSummary":truez"isCompactSummary": truetypeusermessagecontentrh ru…)r.rgr[r\r^r_r]dictr8r;rlistreplacestrip_COMMAND_NAME_REsearchgroup_SKIP_FIRST_PROMPT_PATTERNrrstrip)headstartcommand_fallbackhead_len newline_idxlinerVrxrytextsblockr`ra cmd_matchs r_extract_first_prompt_from_headrs3 E4yH ( iie, ! k*D!OE  $ ,0Jd0R  JJt$E%&%))F*;v*E ))I&'4( ++i( gs # LL !  & ud+ &)V3"599V#4c:LLv/ !C[[s+113F)//7I''0q'9$)//76{S ,,.9M#W ( | Y$$j1   s HHHceZdZdZdZddZy)_LiteSessionFilez>Result of reading a session file's head, tail, mtime and size.mtimesizertailc<||_||_||_||_yrEr)selfrrrrs r__init__z_LiteSessionFile.__init__Ys    rN) rintrrrr;rr;returnNone)__name__ __module__ __qualname____doc__ __slots__rrrrrTsH1Irrc, |jd5}tj|j}|j}t |j dz}|jt}|s dddy|jdd}td|tz }|dk(r|}n9|j||jt} | jdd}t||||cdddS#1swYyxYw#t$rYywxYw) zoOpens a session file, stats it, and reads head + tail. Returns None on any error or if file is empty. rbNutf-8r}errorsrr)openr6fstatfilenost_sizerst_mtimereadLITE_READ_BUF_SIZEdecodemaxseekrrK) file_pathfstatrr head_bytesr tail_offsetr tail_bytess r_read_session_liter`s  ^^D !Q88AHHJ'D<z0_parse_session_info_from_lite..sS0BMM/4R0s  tag timestampZ+00:00r) session_idr last_modified file_size custom_title first_prompt git_branchrr created_at)rrrrrgrsrronextr!rendswithr}rr fromisoformatrr_r)rliterFrrrr first_newline first_linerrrr session_cwdtag_linerrfirst_timestamptss r_parse_session_info_from_liters%#iiDJJ ID$tIIdOM)6!);n}%Jz)-Bj-P (m<  *4 ?  *4 ;  +4 ;   348@DL   *4 >  *4 ;      (k:  %dK 8   -T59Q\QTKShtzz$/0S H GO (5 9 ATTX"J0{CO #++C0 ''X6$  X33B7AACdJKJ !!       s AF33 F?>F?c* t|j}g}|D]`}|j}|j ds!t |dd}|s2t |}|@t|||}|P|j|b|S#t$rgcYSwxYw)zReads session files from a single project directory. Each file gets a stat + head/tail read. Filters out sidechain sessions and metadata-only sessions (no title/summary/prompt). .jsonlNi) r|rQrKr0rrrrr) project_dirrFentriesresultsrVr0rrinfos r_read_sessions_from_dirrs{**,-%'Gzz}}X& #D"I.  !%( < ,Z|L   NN4  N+  sB BBci}|D]H}|j|j}||j|jkDs:|||j<Jt|j S)z=Deduplicates by session_id, keeping the newest last_modified.)r8rrr|values)sessionsby_idr"existings r_deduplicate_by_session_idr(s](*E 99Q\\*  q1G1GG"#E!,,    rcZ|jdd|dkDr||d}| |dkDr|d|}|S)zFSorts sessions by last_modified descending and applies offset + limit.c|jSrE)r)r"s rz*_apply_sort_limit_offset..:srTrireverserN)sort)rlimitoffsets r_apply_sort_limit_offsetr4sE  MM/M> zFG$ UQYFU# Orct|}|r t|}ng}t|dkr(t |}|gSt ||}t |||St}tjdk(} g} |D]4} t| } | r| jn| } | j| | f6| jdd |jDcgc]}|js|}}g}t#}t |}|L|j$}|j'| r|jn|t ||}|j)||D]}| r|j$jn |j$}||vr0| D]f\}} || k(xs(t| t*k\xr|j-| dz}|s8|j'|t ||}|j)|t/|}t |||S#t$rg}YwxYwcc}w#t $r8t |}|t g||cYSt ||}t |||cYSwxYw)zDLists sessions for a specific project directory (and its worktrees).r win32ct|dS)Nr )r.)xs rrz,_list_sessions_for_project..gs s1Q4yrTrr+)rNr Exceptionr.rWrrrCsysplatformr2lowerrrrQrPrKsetr0addextendr/rRr) directoryrrinclude_worktrees canonical_dirworktree_pathsrrrUcase_insensitiveindexedwtr1rTe all_dirents all_sessions seen_dirscanonical_project_dirdir_baserVdir_namewt_pathis_matchdedupeds r_list_sessions_for_projectrBs~'y1M 0?N >a' 6  I*; F'%@@%&L||w.G"2& &6"IF|$ LL($L7A"."6"6"8G"8QAHHJq"8 G*,L%I.m<((-- *:hnn&I*+@-PH%)95::##%uzz y &OGV 6)F 336'' 5  h'25'B##H- ' &)6G #GUF ;;K N 8H A' 6  +Bv> >*; F'%@@ AsA H H!H(H,H! HHH!!#I"I"!I"ct} |jDcgc]}|js|}}g}|D]}|j t |t |}t|||Scc}w#t$rgcYSwxYw)z.Lists sessions across all project directories.)rCrQrPrKrrrr)rrrUr project_dirsrrrs r_list_all_sessionsrs$&L#/#7#7#9H#9aQXXZ#9 H*,L# 3K@A$)6G #GUF ;;I  s&A:A5A5A:5A:: BBc:|rt||||St||S)aLists sessions with metadata extracted from stat + head/tail reads. When ``directory`` is provided, returns sessions for that project directory and its git worktrees. When omitted, returns sessions across all projects. Use ``limit`` and ``offset`` for pagination. Args: directory: Directory to list sessions for. When provided, returns sessions for this project directory (and optionally its git worktrees). When omitted, returns sessions across all projects. limit: Maximum number of sessions to return. offset: Number of sessions to skip from the start of the sorted result set. Use with ``limit`` for pagination. Defaults to 0. include_worktrees: When ``directory`` is provided and the directory is inside a git repository, include sessions from all git worktree paths. Defaults to ``True``. Returns: List of ``SDKSessionInfo`` sorted by ``last_modified`` descending. See Also: :func:`list_sessions_from_store` for the :class:`SessionStore`-backed async variant. Example: List sessions for a specific project:: sessions = list_sessions(directory="/path/to/project") Paginate:: page1 = list_sessions(limit=50) page2 = list_sessions(limit=50, offset=50) List sessions without scanning git worktrees:: sessions = list_sessions( directory="/path/to/project", include_worktrees=False, ) )rr)rrrrs r list_sessionsrs'b))UFDUVV eV ,,rc,t|}|sy|d}|r|t|}t|}|t||z }| t |||S t |}|D]5}||k(r t|} | t| |z }|(t |||cSyt} | jD cgc]} | js| } } | D] } t| |z }|t ||cSy#t $rg}YwxYwcc} w#t$rYywxYw)uReads metadata for a single session by ID. Wraps ``_read_session_lite`` for one file — no O(n) directory scan. Directory resolution matches ``get_session_messages``: ``directory`` is the project path; when omitted, all project directories are searched for the session file. Args: session_id: UUID of the session to look up. directory: Project directory path (same semantics as ``list_sessions(directory=...)``). When omitted, all project directories are searched for the session file. Returns: ``SDKSessionInfo`` for the session, or ``None`` if the session file is not found, is a sidechain session, or has no extractable summary. See Also: :func:`get_session_info_from_store` for the :class:`SessionStore`-backed async variant. Example: Look up a session in a specific project:: info = get_session_info( "550e8400-e29b-41d4-a716-446655440000", directory="/path/to/project", ) if info: print(info.summary) Search all projects for a session:: info = get_session_info("550e8400-e29b-41d4-a716-446655440000") Nr) rrNrWrrrrrCrQrPrK)rruuid file_name canonicalrrr r wt_project_dirrUrdirentsrVs rget_session_infor$sUN * %D &I&y1 ' 2  "%kI&=>D4T4KK 0;N!BY.r2N)).9*DE#8tRHH!%&L*224C4 14C!%)"34  0t< < / N D s< C1D/DD D1 C?>C?D DD>rwsystemprogress assistant attachmentcL ||z jdS#t$rYywxYw)z leaf). Note: logicalParentUuid (set on compact_boundary entries) is intentionally NOT followed. This matches VS Code IDE behavior — post-compaction, the isCompactSummary message replaces earlier messages, so following logical parents would duplicate content. r parentUuidNrvrwr' isSidechainteamNameisMetac|d}j|dd}|ddD]!}j|dd}||kDs|}|}#|S)Nrrr1r r8) candidatesbestbest_idxcurcur_idx entry_indexs r _pick_bestz-_build_conversation_chain.._pick_bests]!}??4<4ab>C!ooc&k26G!" "  r)r?list[_TranscriptEntry]r_TranscriptEntry) enumeraterr8rrr)rby_uuidrVrn parent_uuidsparentr terminalsleavesterminalwalk_cur walk_seenuidleaf main_leavesrEchain chain_seen chain_currDs @r_build_conversation_chainrWs(  ,.G!&f #%Kg&5%& E&M"'!UL<(    V $ $EGqqy 'DGIE&(F,4!e "6"Ci MM# ||F#'<< h'\\,/F.4w{{6*$H"  Dxx &$" '2:k "z&7ID%'E5J)-I   *  s Y|,+1GKK't    MMO LsF.s. G;G )rVr5s r_is_visible_messagerYsT6"JV k 9 yy yy yy$ $$rc|jd}|dk(rdnd}t||jdd|jdd|jdd S) z7Converts a transcript entry dict into a SessionMessage.rvrwr'rr sessionIdrxN)rvrrrxparent_tool_use_id)r8r )rVr5msg_types r_to_session_messager^s[6"J)F2F H   YYvr "99["- )$  rclt|sgSt||}|sgSt|}t|||S)a5Reads a session's conversation messages from its JSONL transcript file. Parses the full JSONL, builds the conversation chain via ``parentUuid`` links, and returns user/assistant messages in chronological order. Args: session_id: UUID of the session to read. directory: Project directory to find the session in. If omitted, searches all project directories under ``~/.claude/projects/``. limit: Maximum number of messages to return. offset: Number of messages to skip from the start. Returns: List of ``SessionMessage`` objects in chronological order. Returns an empty list if the session is not found, the session_id is not a valid UUID, or the transcript contains no visible messages. See Also: :func:`get_session_messages_from_store` for the :class:`SessionStore`-backed async variant. Example: Read all messages from a session:: messages = get_session_messages( "550e8400-e29b-41d4-a716-446655440000", directory="/path/to/project", ) for msg in messages: print(msg.type, msg.message) Read with pagination:: page = get_session_messages( session_id, limit=10, offset=20 ) )rr/r6_entries_to_session_messages)rrrrryrs rget_session_messagesras?V * %  Y7G  '0G ' ??rct|}|Dcgc]}t|s|}}|Dcgc] }t|}}| |dkDr||||zS|dkDr||dS|Scc}wcc}w)zBuilds the conversation chain from parsed entries and applies paging. Shared by the filesystem and SessionStore-backed paths. Nr)rWrYr^)rrrrTrvisiblemessagess rr`r`Ss &g .E:%Q#6q#9q%G:0781#A&H8 UQY%00 z  O;8sAAAc |d d fd }|rZt|}t|}| ||}||S t|}|D]$}||k(r t|}|||}|"|cSyt } t | j } | D]!} | js|| }||cSy#t$rg}YwxYw#t$rYywxYw)aIResolves the on-disk path of a session JSONL file. Directory resolution mirrors ``_read_session_file``: when ``directory`` is provided, looks in that project directory and its git worktrees; otherwise searches all project directories. Returns the path of the first non-empty match, or ``None`` if not found. rcp|z } |jjdkDr|S y#t$rYywxYw)Nr)rrrK)r candidater s r_stat_candidatez3_resolve_session_file_path.._stat_candidatewsN)+  ~~''!+  ,   s ) 55N)rrr Path | None) rNrWrrrCr|rQrKrP) rrrhr rfoundr r r"rUr#rVr s @r_resolve_session_file_pathrkms,f%I*95 ' 6  "#K0E   0?N!B]".r2N)'7$ L!$&L|++-.||~ &  L  7 N " s# B44C4 CC CCcLt||}|y|jd}|dz S)zResolves the subagents directory for a given session. The session file lives at ``/.jsonl`` and the subagents directory at ``//subagents/``. Returns ``None`` if the session cannot be found. Nr subagents)rk with_suffix)rrrM session_dirs r_resolve_subagents_dirrps4**i@H&&r*K  $$rc,gdfd |S)a Recursively collects ``agent-*.jsonl`` files from a directory tree. Subagent transcripts may live directly in ``subagents/`` or in nested subdirectories such as ``subagents/workflows//``. Returns a list of ``(agent_id, file_path)`` tuples. cr t|jd}|D]}|j}|j rN|j dr=|j dr,|tdtd }j||fm|js~|y#t$rYywxYw)Nc|jSrE)r0)ps rrz5_collect_agent_files.._walk..s!&&r)riagent-r) sortedrQrKr0is_filerRrr.rrP) current_dirr#rVr0agent_id_walkrs rrzz#_collect_agent_files.._walks [0028HIGE::D OOH-MM(+H X?%01e    sB** B65B6)rxrrrr)base_dirrzrs @@r_collect_agent_filesr|s')G" (O Nrct|sgSi}|D] }|||d< d}t|D]}|jddvs|}n|gSg}t}|}|U|d}||vrnK|j||j ||jd}|r|j|nd}|U|j |S)u Builds the conversation chain for a subagent transcript. Subagent transcripts are simpler than main sessions — no compaction, no sidechains, no preserved segments. Find the last user/assistant entry and walk ``parentUuid`` links back to the root. rNrvr9r8)r!r8rrrr) rrIrVrRrTseencurrentrQrKs r_build_subagent_chainrs  +-G!&f  %)D'" 99V  5 5D # | $&EUD'+G  fo $;    W\*)/'++f%T   MMO Lrct|sgSt||}|gSt|Dcgc]\}}| c}}Scc}}w)aLists subagent IDs for a given session by scanning the subagents directory. Subagent transcripts are stored at ``~/.claude/projects///subagents/agent-.jsonl`` (and may be nested in subdirectories such as ``workflows//``). Args: session_id: UUID of the parent session. directory: Project directory to find the session in. If omitted, searches all project directories under ``~/.claude/projects/``. Returns: List of subagent ID strings. Returns an empty list if the session is not found, the session_id is not a valid UUID, or the session has no subagents. See Also: :func:`list_subagents_from_store` for the :class:`SessionStore`-backed async variant. Example: List subagent IDs for a session:: agent_ids = list_subagents( "550e8400-e29b-41d4-a716-446655440000", directory="/path/to/project", ) )rrpr|)rr subagents_dirry_s rlist_subagentsrsL@ * % *:yAM (<](K L(K1H(K LL Ls <ct|sgS|sgSt||}|gSd}t|D]\}}||k(s |}n|gS |jd} | sgSt | } t | ||S#t$rgcYSwxYw)aReads a subagent's conversation messages from its JSONL transcript file. Parses the subagent transcript, builds the conversation chain via ``parentUuid`` links, and returns user/assistant messages in chronological order. Args: session_id: UUID of the parent session. agent_id: ID of the subagent (as returned by ``list_subagents``). directory: Project directory to find the session in. If omitted, searches all project directories under ``~/.claude/projects/``. limit: Maximum number of messages to return. offset: Number of messages to skip from the start. Returns: List of ``SessionMessage`` objects in chronological order. Returns an empty list if the session or subagent is not found, the session_id is not a valid UUID, or the transcript contains no user/assistant messages. See Also: :func:`get_subagent_messages_from_store` for the :class:`SessionStore`-backed async variant. Example: Read all messages from a subagent:: messages = get_subagent_messages( "550e8400-e29b-41d4-a716-446655440000", "abc123", directory="/path/to/project", ) Nrr*)rrpr|r,rKr6_entries_to_subagent_messages) rryrrrrrfound_idrryrs rget_subagent_messagesr*sP * %  *:yAM E3MB) x E  C } //7/3  '0G (% @@  sA33 BBct|}|Dcgc]!}|jddvst|#}}| |dkDr||||zS|dkDr||dS|Scc}w)zBuilds the subagent chain from parsed entries and applies paging. Shared by the filesystem and SessionStore-backed paths. rvr9Nr)rr8r^)rrrrTrrds rrrps} "' *E(-(-1vBW1WA  UQY%00 z  Os A AcHt| t|nd}t|S)a{Derive the :class:`SessionStore` ``project_key`` for a directory. Defaults to the current working directory. Uses the same realpath + NFC normalization + djb2-hashed sanitization the CLI uses for project directory names, so keys match between local-disk transcripts and store-mirrored transcripts even on filesystems that decompose Unicode (macOS HFS+). .)rNr;r2)rabs_paths rproject_key_for_directoryrs$"I4I#i.sSH ( ##rcFdddjfd|DdzS)aqSerialize store entries to a JSONL string (one ``json.dumps`` per line). The ``SessionStore.load`` contract permits adapters to reorder object keys (e.g. Postgres JSONB), but ``_parse_session_info_from_lite`` scans for ``{"type":"tag"`` as a line prefix. Hoist ``type`` to the front so the store path matches the byte shape the disk path produces. c@t|tr d|vr d|di|S|S)Nrv)r]r{)rs r _type_firstz&_entries_to_jsonl.._type_firsts+ a 6Q;AfI++ +rruc3XK|]!}tj|d#yw)),:) separatorsN)r[dumps)rrrs rrz$_entries_to_jsonl..s$UW$**[^ CCWs'*)rrrr)r )rrs @r_entries_to_jsonlrs) UWUU  rc|jd}t|}|dtjdd}|tkDr'|t d|tz djddn|}t ||||S)aBuild the head/tail/size lite shape from an in-memory JSONL string. Matches ``_read_session_lite``'s byte semantics so the store path exposes the same slice to ``_parse_session_info_from_lite`` as the disk path would for the same transcript. rNr}rrr)encoder.rrrr)jsonlrbufrrrs r_jsonl_to_liters ,,w C s8D "" # * *79 * ED $ $ C4,, - /077 7R  %dD IIrc(|j}||jddzd} tj|}t |trv|jd}t |trU |jdr|jddn|}ttj|jdzStt!j dzS#tjt f$rd}YwxYw#t $rYLwxYw)zBest-effort mtime: parse the last entry's ``timestamp`` field. Falls back to the current wall-clock time when absent or unparseable. rur Nrrrr)rrfindr[r\r^r_r]r{r8r;rr}rrrrtime)rtrimmed last_lineobjrnorms r_mtime_from_jsonl_tailrs llnG d+a/12Ijj##t WW[ ! b#  46KK4Drzz#x0"811$7AACdJKK tyy{T! ""  * -  s$C$0AD$DD DDcg}|D]\}t|ts|jdtvs,t|jdtsL|j |^|S)aFilter store-loaded entries to transcript message types with a ``uuid``. Mirrors ``_parse_transcript_entries`` for the already-parsed object path so chain-building never sees metadata-only entries (custom-title, tag, agent_metadata, etc.). rvr)r]r{r8r2r;r)rrars r_filter_transcript_entriesrsS&(F  q$ f !88155=#. MM!   Mrc|Kt|}||d}|j|d{}|syt|S7w)z{Load entries from a SessionStore and serialize to a JSONL string. Returns ``None`` if the session has no entries.  project_keyrN)rloadr)storerrrrirs r_load_store_entries_as_jsonlrsB,I6K&1LCJJsO#G  W %%$s %<:<c Ktjt d  fd tj fd|Dddid{}g}t ||dD]r\}}|d}|d} t |t r|jt|d | >|At|t|| |} | [| | _ |j| t|S7w) aDerive ``SDKSessionInfo`` for each ``listing`` entry via per-session ``store.load()`` + lite-parse. Loads run concurrently with a fixed bound so large listings don't exhaust adapter connection pools or hit backend rate limits; adapter errors degrade that row to an empty summary instead of failing the whole list. Sidechain and no-summary sessions are dropped. cK4d{t|d{cdddd{S7*77 #1d{7swYyxYwwrE)r)sidrsem session_stores r _bounded_loadz-_derive_infos_via_load.._bounded_loads535mS)TT33T333sHA5A;7; A9A;AA A A  Ac34K|]}|dyw)rNr)rrrs rrz)_derive_infos_via_load.. s :'Q-, ('sreturn_exceptionsTN)strictrrr)rrr)rr;r str | None) asyncio Semaphore_STORE_LIST_LOAD_CONCURRENCYgatherzipr] BaseExceptionrrrrr) rlistingrrFsettledrrVoutcomerrparsedrrs ` ` @@r_derive_infos_via_loadrs   8 9CUNN :' :G%'Ggwt<wL!g g} - NN#rO   ? . /  > $v%=& N1sAC C B CcKt| t|nd}t|}t|d}t|drddlm} |j |d{}|r9t|j|d{} | D cic] } | d| d } } ng} i} tjd g} t|D]k} | d}|r| j|}|| d|kr'|| |}|j|D| j| d|d j|m|r| jfd | D| j!d d |dkDr| |dn| }| |dkDr|d|}|Dcgc] }|d | }}|rNt#||||d{}|Dcic]}|j$|}}|D]}|j|d|d<|Dcgc] }|d |dc}S|s t)dt|j|d{} t#|| ||d{}t+|||S77cc} wcc}w7cc}wcc}w#t&$rYywxYw7V7>w)uList sessions from a :class:`SessionStore`. Async, store-backed counterpart to :func:`list_sessions`. Loads each session's entries to derive a real summary via the same lite-parse used by the filesystem path, so disk and store paths produce identical results for the same transcript content. Args: session_store: The store to read from. Must implement :meth:`SessionStore.list_session_summaries` or :meth:`SessionStore.list_sessions` (or both). directory: Project directory used to compute the ``project_key``. Defaults to the current working directory. limit: Maximum number of sessions to return. offset: Number of sessions to skip from the start of the sorted result set. Returns: List of ``SDKSessionInfo`` sorted by ``last_modified`` descending. Raises: ValueError: If ``session_store`` implements neither :meth:`SessionStore.list_session_summaries` nor :meth:`SessionStore.list_sessions`. Note: ``include_worktrees`` is a filesystem concept and is not honored on the store path — the store operates on a single ``project_key``. .. note:: If the store implements ``list_session_summaries``, this is one batch summary call plus one cheap ``list_sessions()`` enumeration to gap-fill sessions missing a sidecar or whose sidecar is stale (``summary.mtime < list_sessions.mtime``) — zero per-session ``load()`` calls when sidecars are complete and fresh. Otherwise falls back to one ``store.load()`` per session (bounded at 16 concurrent), which on remote backends with many or large sessions can be expensive (e.g., S3 egress, Postgres large-row reads). Gap-fill requires ``list_sessions``: if the store implements ``list_session_summaries`` but not ``list_sessions``, sessions without a sidecar cannot be discovered and will be absent from the result. Nrrlist_session_summariesr )summary_entry_to_sdk_inforrzjlist_session_summaries without list_sessions: gap-fill skipped; sessions lacking a sidecar will be omitted)rrc3DK|]}|dvr|d|dddyw)rrN)rrrr)rrfresh_summary_idss rrz+list_sessions_from_store..s6$.?? j,QUV$s c |dS)Nrr)sls rrz*list_sessions_from_store..sbkrTrrrzsession_store implements neither list_session_summaries() nor list_sessions() -- cannot list sessions. Provide a store with at least one of those methods.)rNr;r2r session_summaryrrr|rloggerdebugrr8rrrrrrNotImplementedErrorr_r)rrrrrFrhas_list_sessionsr summariesrr known_mtimesslotsr"rknownrpagerto_fillfilledrby_sidrrs @rlist_sessions_from_storer#sd& 8Mc)nSVWL .K)-I(@A>N I+BB;OOI!]%@%@%MMNELMW,7 ;W M!  J +-E*-%  o$(,,S1E}!zE) 0LA<%))#. qz4@A!%%c*!"! $ JJ14J @&,aZ5>UD UQYFU|$(?DbBvJ,>rDG?5!7I| 4::6a!,,/6:!B!'B|, ?D% 8Mc)nSVWL (T< HH UsA#A!AA#cKt|sgSt|}||d}|j|d{}|sgStt |||S7w)uRead a session's conversation messages from a :class:`SessionStore`. Async, store-backed counterpart to :func:`get_session_messages`. Feeds ``session_store.load()`` results directly into the chain builder — no JSONL round-trip. Args: session_store: The store to read from. session_id: UUID of the session to read. directory: Project directory used to compute the ``project_key``. Defaults to the current working directory. limit: Maximum number of messages to return. offset: Number of messages to skip from the start. Returns: List of ``SessionMessage`` objects in chronological order. Empty list if the session is not found or ``session_id`` is invalid. rN)rrrr`r)rrrrrrrirs rget_session_messages_from_storersb2 * % +I6K&1LC!&&s++G  '"7+UF ,s2AAAcKt|sgSt|ds tdt|}|j ||dd{}t }g}|D]p}|j ds|jddd}|j d s<|td d} | |vsO|j| |j| r|S7w) aList subagent IDs for a session from a :class:`SessionStore`. Async, store-backed counterpart to :func:`list_subagents`. Args: session_store: The store to read from. Must implement :meth:`SessionStore.list_subkeys`. session_id: UUID of the parent session. directory: Project directory used to compute the ``project_key``. Defaults to the current working directory. Returns: List of subagent ID strings. Empty list if ``session_id`` is invalid or the session has no subagents. Raises: ValueError: If ``session_store`` does not implement :meth:`SessionStore.list_subkeys`. list_subkeyszwsession_store does not implement list_subkeys() -- cannot list subagents. Provide a store with a list_subkeys() method.rN subagents//r r1ru) rr r_rrrrRrsplitr.rr) rrrrsubkeysr~idssubpathlastrys rlist_subagents_from_storers0 * % ]N ; G  ,I6K!..#:>GUDC!!,/ ~~c1%b) ??8 $CMO,Ht#" 8$ JsAC C A CC+'CcKt|sgS|sgSt|}d|}t|dr<|j||dd{}d|t fd|Dd} | gS| }|||d} |j | d{} | sgS| D cgc](} t | tr| jdd k(s| *} } | sgStt| ||S77Wcc} ww) aRead a subagent's conversation messages from a :class:`SessionStore`. Async, store-backed counterpart to :func:`get_subagent_messages`. Subagents may live at ``subagents/agent-`` or nested under ``subagents/workflows//agent-``. Scans subkeys when the store implements :meth:`SessionStore.list_subkeys`; otherwise tries the direct path. Args: session_store: The store to read from. session_id: UUID of the parent session. agent_id: ID of the subagent. directory: Project directory used to compute the ``project_key``. Defaults to the current working directory. limit: Maximum number of messages to return. offset: Number of messages to skip from the start. Returns: List of ``SessionMessage`` objects in chronological order. Empty list if the session/subagent is not found. zsubagents/agent-rrNruc3tK|]/}|jdr|jdddk(r|1yw)rrr r1N)rRr)rsktargets rrz3get_subagent_messages_from_store..^s< !B==.299S!3DR3HF3R!s58)rrrrvagent_metadata) rrr rrrr]r{r8rr)rrryrrrrrrrrirr transcriptrs @r get_subagent_messages_from_storer4s5: * %  +I6K  +G7%22'z B  ($ !     =I# C "&&s++G  A1d#f 9I(I   (":.v G *, s6AC&C >C&C C&-C!C&C&!C&)rr;rr)r"r;rr;)r0r;rr;)rrrE)rAzdict[str, str] | Nonerr)rFr;rr)rLr;rr;)rFr;rri)r`r;rr;)rhr;rir;rr)rr;rr;)rrrz_LiteSessionFile | None)rr;r list[str])rr;rrrFrrSDKSessionInfo | None)rrrFrrlist[SDKSessionInfo])rrrr)r)rrr int | Nonerrrr) rr;rrrrrboolrr)rrrrrr)NNrT) rrrrrrrrrr)rr;rrrr)rrr r;rr)rr;rrrr)ryr;rrF)rrFrrF)rVrGrr)rVrGrr )NNr) rr;rrrrrrrlist[SessionMessage])rrFrrrrrr)rr;rrrri)r{rrzlist[tuple[str, Path]])rr;rrrr) rr;ryr;rrrrrrrr)rzstr | Path | Nonerr;)r list[Any]rr;)rr;rrrr)rr;rr)rrrrF)rr rr;rrrr) rr rrrrrFr;rr) rr rrrrrrrr)rr rr;rrrr) rr rr;rrrrrrrr)rr rr;rrrr)rr rr;ryr;rrrrrrrr)[r __future__rrr[loggingr6rerrrr9rpathlibrtypingrtypesrr r r session_store_validationr getLoggerrrrrr/compile IGNORECASErrrr,rr)r2r>rCrGrNrWrbrorsrrrrrrrrrrrr$ frozensetr2r{r;rGr-r/r6rWrYr^rar`rkrpr|rrrrrrrrrrrrrrrrrrrrs #   LL7   8 $ "  2::EMM (RZZ62::CDrzz/* ". 5$L 6>/F 02jDY #YYx % "! 3@3@3@ 3@  3@  3@l #   49x % >%T!'M'M'M'MZ! CACACACA  CA  CA  CAL #   4 $(J$#,$ &  &%( &5? & &,, ,, ,  ,b! Z<Z<Z< Z<  Z<  Z<@!IIII IB! """"  "  "  "P!.... .j! JJJJ J  J  JJr