kyp-mem 0.4.0 → 0.4.2

package/kyp_mem/ui.py

@@ -142,6 +142,28 @@ def create_app(vault_path: str = None) -> FastAPI:
         vault.write_note(path, content, tags, {})
         return JSONResponse({"ok": True, "path": path})
 
+    @app.get("/api/sessions/search")
+    def search_sessions(q: str = "", project: str = ""):
+        from .vector import get_session_memory
+        mem = get_session_memory()
+        if not mem or not q:
+            return JSONResponse([])
+        results = mem.search_sessions(q, project=project or None, n_results=10)
+        if not results or not results.get("ids") or not results["ids"][0]:
+            return JSONResponse([])
+        items = []
+        for i, path in enumerate(results["ids"][0]):
+            doc = results["documents"][0][i]
+            dist = results["distances"][0][i]
+            note = vault.index.notes.get(path)
+            items.append({
+                "path": path,
+                "title": note.title if note else path,
+                "distance": dist,
+                "snippet": doc[:300],
+            })
+        return JSONResponse(items)
+
     @app.get("/api/sessions")
     def list_sessions(project: str = ""):
         sessions = {}

package/kyp_mem/vault.py

@@ -200,12 +200,23 @@ class Index:
         return results[:15]
 
 
+from .vector import init_vector_db, get_session_memory
+
 class Vault:
     def __init__(self, vault_path: str):
         self.root = Path(vault_path).expanduser().resolve()
         self.root.mkdir(parents=True, exist_ok=True)
         self.index = Index()
+        init_vector_db(str(self.root))
         self._load_all()
+        self._sync_vector_db()
+
+    def _sync_vector_db(self):
+        mem = get_session_memory()
+        for path, note in self.index.notes.items():
+            if "/Sessions/" in path or path.startswith("Sessions/"):
+                folder = note.folder
+                mem.upsert_session(path, folder, note.content)
 
     def _load_all(self):
         notes = {}
@@ -250,6 +261,8 @@ class Vault:
 
         full.write_text(serialize_note(note), encoding="utf-8")
         self._load_all()
+        if "/Sessions/" in path or path.startswith("Sessions/"):
+            get_session_memory().upsert_session(path, note.folder, note.content)
 
     def delete(self, path: str) -> bool:
         full = self.root / path
@@ -261,6 +274,8 @@ class Vault:
                 parent.rmdir()
                 parent = parent.parent
             self._load_all()
+            if "/Sessions/" in path or path.startswith("Sessions/"):
+                get_session_memory().delete_session(path)
             return True
         return False
 

package/kyp_mem/vector.py

@@ -0,0 +1,43 @@
+import chromadb
+from pathlib import Path
+
+class SessionMemory:
+    def __init__(self, vault_path: str):
+        self.db_path = Path(vault_path).parent / "chroma"
+        self.db_path.mkdir(parents=True, exist_ok=True)
+        self.client = chromadb.PersistentClient(path=str(self.db_path))
+        self.collection = self.client.get_or_create_collection(name="sessions")
+
+    def upsert_session(self, path: str, project: str, content: str):
+        self.collection.upsert(
+            documents=[content],
+            metadatas=[{"project": project}],
+            ids=[path]
+        )
+
+    def delete_session(self, path: str):
+        try:
+            self.collection.delete(ids=[path])
+        except Exception:
+            pass
+
+    def search_sessions(self, query: str, project: str = None, n_results: int = 5):
+        where = {"project": project} if project else None
+        try:
+            results = self.collection.query(
+                query_texts=[query],
+                n_results=n_results,
+                where=where
+            )
+            return results
+        except Exception:
+            return {"ids": [], "documents": [], "metadatas": [], "distances": []}
+
+session_memory = None
+
+def init_vector_db(vault_path: str):
+    global session_memory
+    session_memory = SessionMemory(vault_path)
+
+def get_session_memory():
+    return session_memory

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kyp-mem",
-  "version": "0.4.0",
+  "version": "0.4.2",
   "description": "Know Your Project — Persistent knowledge base for AI agents. MCP-powered with wikilinks, backlinks, auto-learning, and neon web UI.",
   "bin": {
     "kyp-mem": "bin/cli.mjs"

package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kyp-mem"
-version = "0.4.0"
+version = "0.4.2"
 description = "Know Your Project — Persistent knowledge base for AI agents. MCP-powered with wikilinks, backlinks, auto-learning, and neon web UI."
 readme = "README.md"
 license = {text = "MIT"}
@@ -30,6 +30,7 @@ dependencies = [
     "pyyaml>=6.0",
     "fastapi>=0.100.0",
     "uvicorn>=0.20.0",
+    "chromadb>=0.4.0",
 ]
 
 [project.urls]