mycode-cli 0.19.1__tar.gz → 0.19.2__tar.gz

{mycode_cli-0.19.1 → mycode_cli-0.19.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: mycode-cli
-Version: 0.19.1
+Version: 0.19.2
 Summary: Minimal coding agent CLI and web UI built on mycode-sdk.
 Project-URL: Homepage, https://github.com/legibet/mycode
 Project-URL: Repository, https://github.com/legibet/mycode
@@ -25,7 +25,7 @@ Requires-Dist: beautifulsoup4>=4.15.0
 Requires-Dist: fastapi>=0.139.0
 Requires-Dist: httpx2>=2.10.0
 Requires-Dist: markdownify>=1.2.3
-Requires-Dist: mycode-sdk==0.19.1
+Requires-Dist: mycode-sdk==0.19.2
 Requires-Dist: prompt-toolkit>=3.0.52
 Requires-Dist: pydantic>=2.13.4
 Requires-Dist: pygments>=2.20.0

{mycode_cli-0.19.1 → mycode_cli-0.19.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "mycode-cli"
-version = "0.19.1"
+version = "0.19.2"
 description = "Minimal coding agent CLI and web UI built on mycode-sdk."
 readme = "README.md"
 requires-python = ">=3.12"
@@ -26,7 +26,7 @@ classifiers = [
 ]
 keywords = ["agent", "cli", "llm", "coding-agent", "fastapi"]
 dependencies = [
-    "mycode-sdk==0.19.1",
+    "mycode-sdk==0.19.2",
     "fastapi>=0.139.0",
     "uvicorn[standard]>=0.51.0",
     "rich>=15.0.0",

{mycode_cli-0.19.1 → mycode_cli-0.19.2}/src/mycode_cli/sessions.py

@@ -96,9 +96,14 @@ def derive_title(text: str) -> str:
     return title or DEFAULT_SESSION_TITLE
 
 
+@dataclass
 class SessionStore(TimelineStore):
     """Session catalog and timeline façade used by the TUI and web server."""
 
+    # Search cache derived from ``messages.jsonl``: session_id -> ((mtime_ns, size),
+    # whitespace-collapsed text of each visible user/assistant message).
+    _search_texts: dict[str, tuple[tuple[int, int], list[str]]] = field(default_factory=dict, init=False, repr=False)
+
     # ------------------------------------------------------------------
     # Catalog I/O
     # ------------------------------------------------------------------
@@ -182,6 +187,7 @@ class SessionStore(TimelineStore):
     async def delete_session(self, session_id: str) -> None:
         """Delete the whole session directory: catalog, timeline, tool output."""
 
+        self._search_texts.pop(session_id, None)
         await asyncio.to_thread(shutil.rmtree, self.session_dir(session_id), True)
 
     # ------------------------------------------------------------------
@@ -239,6 +245,27 @@ class SessionStore(TimelineStore):
 
         return await asyncio.to_thread(load)
 
+    def _searchable_texts(self, session_id: str) -> list[str]:
+        """Visible user/assistant text per message, cached until the timeline file changes."""
+
+        try:
+            stat = self.messages_path(session_id).stat()
+        except FileNotFoundError:
+            return []
+        key = (stat.st_mtime_ns, stat.st_size)
+        cached = self._search_texts.get(session_id)
+        if cached is not None and cached[0] == key:
+            return cached[1]
+        # The key was taken before the read: a concurrent append leaves a stale
+        # key behind, so the next search re-reads the file.
+        texts = [
+            " ".join(flatten_message_text(message, include_thinking=False).split())
+            for message in self.load_messages_sync(session_id)
+            if message.get("role") in {"user", "assistant"}
+        ]
+        self._search_texts[session_id] = (key, texts)
+        return texts
+
     async def search_sessions(self, query: str, *, cwd: str | None = None, limit: int = 50) -> list[SessionSearchHit]:
         """Find sessions whose title or visible user/assistant text contains
         ``query`` (case-insensitive), newest first.
@@ -257,10 +284,7 @@ class SessionStore(TimelineStore):
             hits: list[SessionSearchHit] = []
             for session in sessions:
                 snippet: SessionSnippet | None = None
-                for message in self.load_messages_sync(str(session["id"])):
-                    if message.get("role") not in {"user", "assistant"}:
-                        continue
-                    text = " ".join(flatten_message_text(message, include_thinking=False).split())
+                for text in self._searchable_texts(str(session["id"])):
                     if found := pattern.search(text):
                         start, end = found.span()
                         snippet = {