klaude-code 2.8.0__py3-none-any.whl → 2.9.0__py3-none-any.whl

klaude_code/session/session.py

@@ -92,10 +92,6 @@ class Session(BaseModel):
             ]
         return self._user_messages_cache
 
-    @staticmethod
-    def _project_key() -> str:
-        return project_key_from_cwd()
-
     @classmethod
     def paths(cls) -> ProjectPaths:
         return get_default_store().paths
@@ -410,7 +406,9 @@ class Session(BaseModel):
                     )
                     yield from self._iter_sub_agent_history(tr, seen_sub_agent_sessions)
                 case message.UserMessage() as um:
-                    images = [part for part in um.parts if isinstance(part, message.ImageURLPart)]
+                    images = [
+                        part for part in um.parts if isinstance(part, (message.ImageURLPart, message.ImageFilePart))
+                    ]
                     yield events.UserMessageEvent(
                         content=message.join_text_parts(um.parts),
                         session_id=self.id,
@@ -608,27 +606,66 @@ class Session(BaseModel):
         return resolved[0]
 
     @classmethod
-    def clean_small_sessions(cls, min_messages: int = 5) -> int:
-        sessions = cls.list_sessions()
-        deleted_count = 0
+    def find_sessions_by_prefix(cls, prefix: str) -> list[str]:
+        """Find main session IDs matching a prefix.
+
+        Args:
+            prefix: Session ID prefix to match.
+
+        Returns:
+            List of matching session IDs, sorted alphabetically.
+        """
+        prefix = (prefix or "").strip().lower()
+        if not prefix:
+            return []
+
         store = get_default_store()
-        for session_meta in sessions:
-            if session_meta.messages_count < 0:
+        matches: set[str] = set()
+
+        for meta_path in store.iter_meta_files():
+            data = _read_json_dict(meta_path)
+            if data is None:
                 continue
-            if session_meta.messages_count < min_messages:
-                store.delete_session(session_meta.id)
-                deleted_count += 1
-        return deleted_count
+            # Exclude sub-agent sessions.
+            if data.get("sub_agent_state") is not None:
+                continue
+            sid = str(data.get("id", meta_path.parent.name)).strip()
+            if sid.lower().startswith(prefix):
+                matches.add(sid)
+
+        return sorted(matches)
 
     @classmethod
-    def clean_all_sessions(cls) -> int:
-        sessions = cls.list_sessions()
-        deleted_count = 0
+    def shortest_unique_prefix(cls, session_id: str, min_length: int = 4) -> str:
+        """Find the shortest unique prefix for a session ID.
+
+        Args:
+            session_id: The session ID to find prefix for.
+            min_length: Minimum prefix length (default 4).
+
+        Returns:
+            The shortest prefix that uniquely identifies this session.
+        """
         store = get_default_store()
-        for session_meta in sessions:
-            store.delete_session(session_meta.id)
-            deleted_count += 1
-        return deleted_count
+        other_ids: list[str] = []
+
+        for meta_path in store.iter_meta_files():
+            data = _read_json_dict(meta_path)
+            if data is None:
+                continue
+            if data.get("sub_agent_state") is not None:
+                continue
+            sid = str(data.get("id", meta_path.parent.name)).strip()
+            if sid != session_id:
+                other_ids.append(sid.lower())
+
+        session_lower = session_id.lower()
+        for length in range(min_length, len(session_id) + 1):
+            prefix = session_lower[:length]
+            if not any(other.startswith(prefix) for other in other_ids):
+                return session_id[:length]
+
+        return session_id
 
     @classmethod
     def exports_dir(cls) -> Path:

klaude_code/session/store.py

@@ -2,7 +2,6 @@ from __future__ import annotations
 
 import asyncio
 import json
-import shutil
 from collections.abc import Iterable, Sequence
 from dataclasses import dataclass
 from pathlib import Path
@@ -152,9 +151,6 @@ class JsonlSessionStore:
             return []
         return sessions_dir.glob("*/meta.json")
 
-    def delete_session(self, session_id: str) -> None:
-        shutil.rmtree(self._paths.session_dir(session_id), ignore_errors=True)
-
     async def aclose(self) -> None:
         await self._writer.aclose()