ltcai 0.1.3 → 0.1.8

package/knowledge_graph.py

@@ -0,0 +1,802 @@
+"""
+SQLite knowledge graph for Lattice AI workspace memory.
+
+The graph keeps raw event JSON, normalized node metadata, and edges in one
+portable database so it can later migrate to Neo4j/Postgres without changing
+the ingestion contract.
+"""
+
+import hashlib
+import json
+import re
+import shutil
+import sqlite3
+import zipfile
+from datetime import datetime
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+
+GRAPH_SCHEMA_VERSION = 1
+
+
+def _now() -> str:
+    return datetime.now().isoformat()
+
+
+def _json(data: Optional[Dict[str, Any]]) -> str:
+    return json.dumps(data or {}, ensure_ascii=False, sort_keys=True)
+
+
+def _slug(text: str, max_len: int = 96) -> str:
+    value = re.sub(r"\s+", " ", str(text or "")).strip().lower()
+    value = re.sub(r"[^0-9a-zA-Z가-힣._:@/-]+", "-", value).strip("-")
+    return (value or "untitled")[:max_len]
+
+
+def _sha256_bytes(data: bytes) -> str:
+    return hashlib.sha256(data).hexdigest()
+
+
+def _sha256_text(text: str) -> str:
+    return hashlib.sha256(text.encode("utf-8", errors="replace")).hexdigest()
+
+
+def _clean_text(text: str) -> str:
+    return re.sub(r"\s+", " ", str(text or "")).strip()
+
+
+def _chunks(text: str, size: int = 1200, overlap: int = 160) -> List[str]:
+    cleaned = str(text or "").strip()
+    if not cleaned:
+        return []
+    chunks: List[str] = []
+    start = 0
+    while start < len(cleaned):
+        end = min(len(cleaned), start + size)
+        chunks.append(cleaned[start:end])
+        if end >= len(cleaned):
+            break
+        start = max(0, end - overlap)
+    return chunks
+
+
+def _topic_candidates(text: str, limit: int = 8) -> List[str]:
+    text = str(text or "")
+    candidates: Dict[str, int] = {}
+    patterns = [
+        r"[A-Za-z][A-Za-z0-9_\-./]{2,}",
+        r"[가-힣][가-힣A-Za-z0-9_\-]{1,}",
+    ]
+    stop = {
+        "the", "and", "for", "with", "this", "that", "from", "into",
+        "사용자", "내용", "파일", "채팅", "답변", "입니다", "그리고", "처럼",
+    }
+    for pattern in patterns:
+        for match in re.findall(pattern, text):
+            key = match.strip("._-/").lower()
+            if (len(key) < 3 and not re.search(r"[가-힣]", key)) or key in stop or key.isdigit():
+                continue
+            candidates[key] = candidates.get(key, 0) + 1
+    return [
+        k for k, v in sorted(candidates.items(), key=lambda item: (-item[1], item[0]))
+        if (re.search(r"[가-힣]", k) and len(k) >= 2) or (len(k) >= 4 and (v >= 2 or len(k) >= 6))
+    ][:limit]
+
+
+def _semantic_items(text: str) -> List[Dict[str, str]]:
+    """Lightweight extraction for product MVP before model-based IE is wired in."""
+    items: List[Dict[str, str]] = []
+    for raw_line in str(text or "").splitlines():
+        line = _clean_text(raw_line)
+        if len(line) < 6:
+            continue
+        lowered = line.lower()
+        if re.search(r"(결정|확정|하기로|decided|decision)", lowered):
+            items.append({"type": "Decision", "title": line[:120], "summary": line[:500]})
+        if re.search(r"(todo|해야|하자|진행|구현|수정|확인|next|task|\[ \])", lowered):
+            items.append({"type": "Task", "title": line[:120], "summary": line[:500]})
+    return items[:12]
+
+
+class KnowledgeGraphStore:
+    def __init__(self, db_path: Path, blob_dir: Path):
+        self.db_path = Path(db_path)
+        self.blob_dir = Path(blob_dir)
+        self.db_path.parent.mkdir(parents=True, exist_ok=True)
+        self.blob_dir.mkdir(parents=True, exist_ok=True)
+        self._init_db()
+
+    def _connect(self) -> sqlite3.Connection:
+        conn = sqlite3.connect(str(self.db_path))
+        conn.row_factory = sqlite3.Row
+        conn.execute("PRAGMA journal_mode=WAL")
+        conn.execute("PRAGMA foreign_keys=ON")
+        return conn
+
+    def _init_db(self) -> None:
+        with self._connect() as conn:
+            conn.executescript(
+                """
+                CREATE TABLE IF NOT EXISTS graph_meta (
+                  key TEXT PRIMARY KEY,
+                  value TEXT NOT NULL
+                );
+                CREATE TABLE IF NOT EXISTS nodes (
+                  id TEXT PRIMARY KEY,
+                  type TEXT NOT NULL,
+                  title TEXT NOT NULL,
+                  summary TEXT,
+                  metadata_json TEXT NOT NULL CHECK (json_valid(metadata_json)),
+                  raw_json TEXT NOT NULL CHECK (json_valid(raw_json)),
+                  created_at TEXT NOT NULL,
+                  updated_at TEXT NOT NULL
+                );
+                CREATE TABLE IF NOT EXISTS edges (
+                  id TEXT PRIMARY KEY,
+                  from_node TEXT NOT NULL,
+                  to_node TEXT NOT NULL,
+                  type TEXT NOT NULL,
+                  weight REAL NOT NULL DEFAULT 1.0,
+                  metadata_json TEXT NOT NULL CHECK (json_valid(metadata_json)),
+                  created_at TEXT NOT NULL,
+                  UNIQUE(from_node, to_node, type),
+                  FOREIGN KEY(from_node) REFERENCES nodes(id) ON DELETE CASCADE,
+                  FOREIGN KEY(to_node) REFERENCES nodes(id) ON DELETE CASCADE
+                );
+                CREATE TABLE IF NOT EXISTS chunks (
+                  id TEXT PRIMARY KEY,
+                  source_node TEXT NOT NULL,
+                  text TEXT NOT NULL,
+                  metadata_json TEXT NOT NULL CHECK (json_valid(metadata_json)),
+                  created_at TEXT NOT NULL,
+                  FOREIGN KEY(source_node) REFERENCES nodes(id) ON DELETE CASCADE
+                );
+                CREATE INDEX IF NOT EXISTS idx_nodes_type ON nodes(type);
+                CREATE INDEX IF NOT EXISTS idx_edges_from ON edges(from_node);
+                CREATE INDEX IF NOT EXISTS idx_edges_to ON edges(to_node);
+                CREATE INDEX IF NOT EXISTS idx_chunks_source ON chunks(source_node);
+                """
+            )
+            conn.execute(
+                "INSERT OR REPLACE INTO graph_meta(key, value) VALUES (?, ?)",
+                ("schema_version", str(GRAPH_SCHEMA_VERSION)),
+            )
+
+    def _upsert_node(
+        self,
+        conn: sqlite3.Connection,
+        node_id: str,
+        node_type: str,
+        title: str,
+        summary: str = "",
+        metadata: Optional[Dict[str, Any]] = None,
+        raw: Optional[Dict[str, Any]] = None,
+    ) -> str:
+        now = _now()
+        conn.execute(
+            """
+            INSERT INTO nodes(id, type, title, summary, metadata_json, raw_json, created_at, updated_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+            ON CONFLICT(id) DO UPDATE SET
+              title=excluded.title,
+              summary=excluded.summary,
+              metadata_json=excluded.metadata_json,
+              raw_json=excluded.raw_json,
+              updated_at=excluded.updated_at
+            """,
+            (node_id, node_type, title[:240], summary[:1000], _json(metadata), _json(raw), now, now),
+        )
+        return node_id
+
+    def _upsert_edge(
+        self,
+        conn: sqlite3.Connection,
+        from_node: str,
+        to_node: str,
+        edge_type: str,
+        weight: float = 1.0,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> str:
+        edge_id = f"edge:{_sha256_text(f'{from_node}|{edge_type}|{to_node}')[:24]}"
+        conn.execute(
+            """
+            INSERT INTO edges(id, from_node, to_node, type, weight, metadata_json, created_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?)
+            ON CONFLICT(from_node, to_node, type) DO UPDATE SET
+              weight=max(edges.weight, excluded.weight),
+              metadata_json=excluded.metadata_json
+            """,
+            (edge_id, from_node, to_node, edge_type, float(weight), _json(metadata), _now()),
+        )
+        return edge_id
+
+    def ingest_message(
+        self,
+        role: str,
+        content: str,
+        *,
+        user_email: Optional[str] = None,
+        user_nickname: Optional[str] = None,
+        source: Optional[str] = None,
+        conversation_id: Optional[str] = None,
+        raw: Optional[Dict[str, Any]] = None,
+    ) -> Dict[str, Any]:
+        content = str(content or "")
+        digest = _sha256_text("|".join([role or "", content, conversation_id or "", user_email or ""]))[:24]
+        node_type = "AIResponse" if role == "assistant" else "Message"
+        node_id = f"{node_type.lower()}:{digest}"
+        conv_id = f"conversation:{_slug(conversation_id or 'default')}"
+        metadata = {
+            "role": role,
+            "source": source,
+            "conversation_id": conversation_id,
+            "user_email": user_email,
+            "user_nickname": user_nickname,
+            "chars": len(content),
+        }
+        with self._connect() as conn:
+            self._upsert_node(conn, conv_id, "Conversation", conversation_id or "Default conversation", metadata={"source": source})
+            self._upsert_node(
+                conn,
+                node_id,
+                node_type,
+                _clean_text(content)[:80] or role,
+                summary=_clean_text(content)[:500],
+                metadata=metadata,
+                raw=raw or metadata,
+            )
+            self._upsert_edge(conn, conv_id, node_id, "contains", metadata={"source": source})
+            if user_email or user_nickname:
+                person_key = user_email or user_nickname or "unknown"
+                person_id = f"person:{_slug(person_key)}"
+                self._upsert_node(conn, person_id, "Person", user_nickname or user_email or "Unknown user", metadata={"email": user_email})
+                self._upsert_edge(conn, person_id, node_id, "authored", metadata={"role": role})
+            for index, chunk in enumerate(_chunks(content)):
+                chunk_id = f"chunk:{_sha256_text(f'{node_id}:{index}:{chunk}')[:24]}"
+                self._upsert_node(conn, chunk_id, "Chunk", f"{node_type} chunk {index + 1}", summary=chunk[:500], metadata={"index": index, "source_node": node_id})
+                conn.execute(
+                    """
+                    INSERT OR REPLACE INTO chunks(id, source_node, text, metadata_json, created_at)
+                    VALUES (?, ?, ?, ?, ?)
+                    """,
+                    (chunk_id, node_id, chunk, _json({"index": index, "source_node": node_id}), _now()),
+                )
+                self._upsert_edge(conn, node_id, chunk_id, "has_chunk")
+            for topic in _topic_candidates(content):
+                topic_id = f"topic:{_slug(topic)}"
+                self._upsert_node(conn, topic_id, "Topic", topic, metadata={"auto_extracted": True})
+                self._upsert_edge(conn, node_id, topic_id, "mentions", weight=0.5)
+            for item in _semantic_items(content):
+                semantic_type = item["type"]
+                semantic_title = item["title"]
+                semantic_id = f"{semantic_type.lower()}:{_sha256_text(f'{node_id}:{semantic_type}:{semantic_title}')[:24]}"
+                self._upsert_node(
+                    conn,
+                    semantic_id,
+                    semantic_type,
+                    semantic_title,
+                    summary=item["summary"],
+                    metadata={"auto_extracted": True, "source_node": node_id},
+                    raw=item,
+                )
+                self._upsert_edge(conn, node_id, semantic_id, "implies", weight=0.8)
+                if node_type == "AIResponse":
+                    self._upsert_edge(conn, semantic_id, node_id, "based_on", weight=0.6)
+        return {"node_id": node_id, "type": node_type}
+
+    def ingest_document(
+        self,
+        path: Path,
+        *,
+        original_filename: Optional[str] = None,
+        mime_type: Optional[str] = None,
+        uploader: Optional[str] = None,
+        conversation_id: Optional[str] = None,
+        extracted: Optional[Dict[str, Any]] = None,
+    ) -> Dict[str, Any]:
+        path = Path(path)
+        data = path.read_bytes()
+        digest = _sha256_bytes(data)
+        ext = path.suffix.lower()
+        filename = original_filename or path.name
+        blob_path = self.blob_dir / digest[:2] / f"{digest}{ext}"
+        blob_path.parent.mkdir(parents=True, exist_ok=True)
+        if not blob_path.exists():
+            shutil.copyfile(path, blob_path)
+
+        doc_meta = self._document_structure(path, ext)
+        text = str((extracted or {}).get("content") or (extracted or {}).get("preview") or "")
+        file_id = f"file:{digest[:24]}"
+        metadata = {
+            "filename": filename,
+            "ext": ext,
+            "mime_type": mime_type,
+            "bytes": len(data),
+            "sha256": digest,
+            "blob_path": str(blob_path),
+            "uploader": uploader,
+            "conversation_id": conversation_id,
+            "extracted": {k: v for k, v in (extracted or {}).items() if k != "content"},
+            "structure": doc_meta,
+        }
+        with self._connect() as conn:
+            self._upsert_node(conn, file_id, "File", filename, summary=(text or filename)[:500], metadata=metadata, raw=metadata)
+            self._ingest_structure_nodes(conn, file_id, filename, doc_meta)
+            if uploader:
+                person_id = f"person:{_slug(uploader)}"
+                self._upsert_node(conn, person_id, "Person", uploader, metadata={"email": uploader})
+                self._upsert_edge(conn, person_id, file_id, "uploaded")
+            if conversation_id:
+                conv_id = f"conversation:{_slug(conversation_id)}"
+                self._upsert_node(conn, conv_id, "Conversation", conversation_id)
+                self._upsert_edge(conn, conv_id, file_id, "contains")
+            for index, chunk in enumerate(_chunks(text)):
+                chunk_id = f"chunk:{_sha256_text(f'{file_id}:{index}:{chunk}')[:24]}"
+                self._upsert_node(conn, chunk_id, "Chunk", f"{filename} chunk {index + 1}", summary=chunk[:500], metadata={"index": index, "source_node": file_id})
+                conn.execute(
+                    "INSERT OR REPLACE INTO chunks(id, source_node, text, metadata_json, created_at) VALUES (?, ?, ?, ?, ?)",
+                    (chunk_id, file_id, chunk, _json({"index": index, "source_node": file_id}), _now()),
+                )
+                self._upsert_edge(conn, file_id, chunk_id, "has_chunk")
+            for topic in _topic_candidates(f"{filename}\n{text}"):
+                topic_id = f"topic:{_slug(topic)}"
+                self._upsert_node(conn, topic_id, "Topic", topic, metadata={"auto_extracted": True})
+                self._upsert_edge(conn, file_id, topic_id, "discusses", weight=0.7)
+            for item in _semantic_items(text):
+                semantic_type = item["type"]
+                semantic_title = item["title"]
+                semantic_id = f"{semantic_type.lower()}:{_sha256_text(f'{file_id}:{semantic_type}:{semantic_title}')[:24]}"
+                self._upsert_node(
+                    conn,
+                    semantic_id,
+                    semantic_type,
+                    semantic_title,
+                    summary=item["summary"],
+                    metadata={"auto_extracted": True, "source_node": file_id, "filename": filename},
+                    raw=item,
+                )
+                self._upsert_edge(conn, file_id, semantic_id, "contains_signal", weight=0.8)
+        return {"node_id": file_id, "sha256": digest, "metadata": metadata}
+
+    def ingest_event(
+        self,
+        event_type: str,
+        title: str,
+        *,
+        user_email: Optional[str] = None,
+        user_nickname: Optional[str] = None,
+        source: Optional[str] = None,
+        conversation_id: Optional[str] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> Dict[str, Any]:
+        event_type = str(event_type or "Event")
+        title = str(title or event_type)
+        payload = {
+            "event_type": event_type,
+            "title": title,
+            "user_email": user_email,
+            "user_nickname": user_nickname,
+            "source": source,
+            "conversation_id": conversation_id,
+            "metadata": metadata or {},
+            "timestamp": _now(),
+        }
+        event_id = f"event:{_sha256_text(_json(payload))[:24]}"
+        conv_id = f"conversation:{_slug(conversation_id or 'default')}"
+        with self._connect() as conn:
+            self._upsert_node(conn, event_id, event_type, title, summary=title, metadata=payload, raw=payload)
+            self._upsert_node(conn, conv_id, "Conversation", conversation_id or "Default conversation", metadata={"source": source})
+            self._upsert_edge(conn, conv_id, event_id, "has_event", metadata={"source": source})
+            if user_email or user_nickname:
+                person_key = user_email or user_nickname or "unknown"
+                person_id = f"person:{_slug(person_key)}"
+                self._upsert_node(conn, person_id, "Person", user_nickname or user_email or "Unknown user", metadata={"email": user_email})
+                self._upsert_edge(conn, person_id, event_id, "triggered", metadata={"event_type": event_type})
+        return {"node_id": event_id, "type": event_type}
+
+    def _ingest_structure_nodes(
+        self,
+        conn: sqlite3.Connection,
+        file_id: str,
+        filename: str,
+        structure: Dict[str, Any],
+    ) -> None:
+        for slide in structure.get("slides") or []:
+            index = slide.get("index")
+            slide_id = f"slide:{_sha256_text(f'{file_id}:slide:{index}')[:24]}"
+            title = f"{filename} slide {index}"
+            summary = "\n".join(slide.get("texts") or [])[:800]
+            self._upsert_node(conn, slide_id, "Slide", title, summary=summary, metadata=slide)
+            self._upsert_edge(conn, file_id, slide_id, "has_slide")
+            for text in slide.get("texts") or []:
+                for topic in _topic_candidates(text, limit=4):
+                    topic_id = f"topic:{_slug(topic)}"
+                    self._upsert_node(conn, topic_id, "Topic", topic, metadata={"auto_extracted": True})
+                    self._upsert_edge(conn, slide_id, topic_id, "discusses", weight=0.6)
+
+        for page in structure.get("pages") or []:
+            index = page.get("index")
+            page_id = f"page:{_sha256_text(f'{file_id}:page:{index}')[:24]}"
+            title = f"{filename} page {index}"
+            self._upsert_node(conn, page_id, "Page", title, summary=page.get("preview") or "", metadata=page)
+            self._upsert_edge(conn, file_id, page_id, "has_page")
+            for topic in _topic_candidates(page.get("preview") or "", limit=4):
+                topic_id = f"topic:{_slug(topic)}"
+                self._upsert_node(conn, topic_id, "Topic", topic, metadata={"auto_extracted": True})
+                self._upsert_edge(conn, page_id, topic_id, "discusses", weight=0.6)
+
+        for sheet in (structure.get("sheets") or []):
+            sheet_title = sheet.get("title")
+            sheet_id = f"sheet:{_sha256_text(f'{file_id}:sheet:{sheet_title}')[:24]}"
+            self._upsert_node(conn, sheet_id, "Sheet", f"{filename} / {sheet_title}", metadata=sheet)
+            self._upsert_edge(conn, file_id, sheet_id, "has_sheet")
+
+        for image in (structure.get("images") or []):
+            image_key = image.get("sha256") or _sha256_text(json.dumps(image, ensure_ascii=False, sort_keys=True))
+            image_id = f"image:{str(image_key)[:24]}"
+            title_parts = [filename, "image"]
+            if image.get("page"):
+                title_parts.append(f"page {image.get('page')}")
+            if image.get("name"):
+                title_parts.append(str(image.get("name")).split("/")[-1])
+            self._upsert_node(conn, image_id, "Image", " / ".join(title_parts), metadata=image)
+            self._upsert_edge(conn, file_id, image_id, "contains_image")
+
+    def _document_structure(self, path: Path, ext: str) -> Dict[str, Any]:
+        try:
+            if ext == ".pptx":
+                return self._pptx_structure(path)
+            if ext == ".pdf":
+                return self._pdf_structure(path)
+            if ext == ".docx":
+                return self._docx_structure(path)
+            if ext == ".xlsx":
+                return self._xlsx_structure(path)
+        except Exception as exc:
+            return {"error": str(exc)}
+        return {}
+
+    def _pptx_structure(self, path: Path) -> Dict[str, Any]:
+        result: Dict[str, Any] = {"slides": [], "images": []}
+        try:
+            from PIL import Image
+            from pptx import Presentation
+            prs = Presentation(str(path))
+            for slide_index, slide in enumerate(prs.slides, start=1):
+                slide_info = {"index": slide_index, "shapes": [], "texts": []}
+                for shape_index, shape in enumerate(slide.shapes, start=1):
+                    shape_info = {
+                        "index": shape_index,
+                        "name": getattr(shape, "name", ""),
+                        "shape_type": str(getattr(shape, "shape_type", "")),
+                        "bbox": {
+                            "left": int(getattr(shape, "left", 0) or 0),
+                            "top": int(getattr(shape, "top", 0) or 0),
+                            "width": int(getattr(shape, "width", 0) or 0),
+                            "height": int(getattr(shape, "height", 0) or 0),
+                        },
+                    }
+                    if getattr(shape, "has_text_frame", False):
+                        text = shape.text_frame.text.strip()
+                        if text:
+                            shape_info["text"] = text[:1000]
+                            slide_info["texts"].append(text)
+                    slide_info["shapes"].append(shape_info)
+                result["slides"].append(slide_info)
+            with zipfile.ZipFile(path) as zf:
+                for name in zf.namelist():
+                    if not name.startswith("ppt/media/"):
+                        continue
+                    data = zf.read(name)
+                    image_info: Dict[str, Any] = {
+                        "name": name,
+                        "bytes": len(data),
+                        "sha256": _sha256_bytes(data),
+                    }
+                    try:
+                        from io import BytesIO
+                        with Image.open(BytesIO(data)) as img:
+                            image_info.update({"width": img.width, "height": img.height, "format": img.format})
+                    except Exception:
+                        pass
+                    result["images"].append(image_info)
+        except Exception as exc:
+            result["error"] = str(exc)
+        return result
+
+    def _pdf_structure(self, path: Path) -> Dict[str, Any]:
+        result: Dict[str, Any] = {"pages": [], "images": []}
+        try:
+            import pdfplumber
+            with pdfplumber.open(str(path)) as pdf:
+                metadata = dict(pdf.metadata or {})
+                result["metadata"] = {str(k): str(v) for k, v in metadata.items()}
+                for page_index, page in enumerate(pdf.pages, start=1):
+                    text = page.extract_text() or ""
+                    page_info = {
+                        "index": page_index,
+                        "width": float(page.width or 0),
+                        "height": float(page.height or 0),
+                        "chars": len(text),
+                        "preview": _clean_text(text)[:500],
+                        "image_count": len(page.images or []),
+                    }
+                    result["pages"].append(page_info)
+                    for image_index, image in enumerate(page.images or [], start=1):
+                        result["images"].append({
+                            "page": page_index,
+                            "index": image_index,
+                            "name": image.get("name"),
+                            "width": image.get("width"),
+                            "height": image.get("height"),
+                            "bbox": {
+                                "x0": image.get("x0"),
+                                "top": image.get("top"),
+                                "x1": image.get("x1"),
+                                "bottom": image.get("bottom"),
+                            },
+                        })
+        except Exception as exc:
+            result["error"] = str(exc)
+        return result
+
+    def _docx_structure(self, path: Path) -> Dict[str, Any]:
+        from docx import Document
+        doc = Document(str(path))
+        headings = []
+        paragraphs = 0
+        for p in doc.paragraphs:
+            text = p.text.strip()
+            if not text:
+                continue
+            paragraphs += 1
+            style = getattr(p.style, "name", "")
+            if style.lower().startswith("heading"):
+                headings.append({"style": style, "text": text[:240]})
+        return {"paragraphs": paragraphs, "headings": headings[:80], "tables": len(doc.tables)}
+
+    def _xlsx_structure(self, path: Path) -> Dict[str, Any]:
+        from openpyxl import load_workbook
+        wb = load_workbook(str(path), read_only=True, data_only=True)
+        sheets = []
+        for ws in wb.worksheets:
+            sheets.append({"title": ws.title, "max_row": ws.max_row, "max_column": ws.max_column})
+        return {"sheets": sheets}
+
+    def graph(self, limit: int = 300) -> Dict[str, Any]:
+        limit = max(1, min(int(limit or 300), 2000))
+        with self._connect() as conn:
+            nodes = [
+                {
+                    "id": row["id"],
+                    "type": row["type"],
+                    "title": row["title"],
+                    "summary": row["summary"],
+                    "metadata": json.loads(row["metadata_json"] or "{}"),
+                }
+                for row in conn.execute(
+                    "SELECT id, type, title, summary, metadata_json FROM nodes WHERE type != 'Chunk' ORDER BY updated_at DESC LIMIT ?",
+                    (limit,),
+                )
+            ]
+            node_ids = {node["id"] for node in nodes}
+            edges = [
+                {
+                    "id": row["id"],
+                    "from": row["from_node"],
+                    "to": row["to_node"],
+                    "type": row["type"],
+                    "weight": row["weight"],
+                    "metadata": json.loads(row["metadata_json"] or "{}"),
+                }
+                for row in conn.execute(
+                    "SELECT id, from_node, to_node, type, weight, metadata_json FROM edges ORDER BY created_at DESC LIMIT ?",
+                    (limit * 3,),
+                )
+                if row["from_node"] in node_ids and row["to_node"] in node_ids
+            ]
+        return {"nodes": nodes, "edges": edges}
+
+    def search(self, query: str, limit: int = 30) -> Dict[str, Any]:
+        query = str(query or "").strip()
+        q = f"%{query}%"
+        limit = max(1, min(int(limit or 30), 100))
+        with self._connect() as conn:
+            rows = []
+            if query:
+                rows = conn.execute(
+                    """
+                    SELECT id, type, title, summary, metadata_json, updated_at
+                    FROM nodes
+                    WHERE title LIKE ? OR summary LIKE ? OR metadata_json LIKE ?
+                    ORDER BY updated_at DESC
+                    LIMIT ?
+                    """,
+                    (q, q, q, limit),
+                ).fetchall()
+
+            if len(rows) < limit:
+                terms = _topic_candidates(query, limit=8)
+                if terms:
+                    clauses = []
+                    params: List[str] = []
+                    for term in terms:
+                        clauses.append("(title LIKE ? OR summary LIKE ? OR metadata_json LIKE ?)")
+                        params.extend([f"%{term}%", f"%{term}%", f"%{term}%"])
+                    extra = conn.execute(
+                        f"""
+                        SELECT id, type, title, summary, metadata_json, updated_at
+                        FROM nodes
+                        WHERE {' OR '.join(clauses)}
+                        ORDER BY updated_at DESC
+                        LIMIT ?
+                        """,
+                        (*params, limit * 3),
+                    ).fetchall()
+                    by_id = {row["id"]: row for row in rows}
+                    for row in extra:
+                        by_id.setdefault(row["id"], row)
+                    rows = list(by_id.values())
+
+            terms_for_score = set(_topic_candidates(query, limit=12))
+            def score(row: sqlite3.Row) -> tuple:
+                haystack = f"{row['title']} {row['summary']} {row['metadata_json']}".lower()
+                hits = sum(1 for term in terms_for_score if term.lower() in haystack)
+                type_boost = 1 if row["type"] in {"Decision", "Task", "File", "Page", "Slide"} else 0
+                return (hits, type_boost, row["updated_at"] or "")
+
+            rows = sorted(rows, key=score, reverse=True)[:limit]
+        return {
+            "query": query,
+            "matches": [
+                {
+                    "id": row["id"],
+                    "type": row["type"],
+                    "title": row["title"],
+                    "summary": row["summary"],
+                    "metadata": json.loads(row["metadata_json"] or "{}"),
+                }
+                for row in rows
+            ],
+        }
+
+    def context_for_query(self, query: str, limit: int = 6) -> str:
+        """Return compact graph-backed RAG context for chat generation."""
+        query = str(query or "").strip()
+        if not query:
+            return ""
+        matches = self.search(query, limit).get("matches", [])
+        if not matches:
+            topics = _topic_candidates(query, limit=4)
+            if topics:
+                with self._connect() as conn:
+                    rows = []
+                    for topic in topics:
+                        rows.extend(conn.execute(
+                            """
+                            SELECT id, type, title, summary, metadata_json
+                            FROM nodes
+                            WHERE title LIKE ? OR metadata_json LIKE ?
+                            ORDER BY updated_at DESC
+                            LIMIT 3
+                            """,
+                            (f"%{topic}%", f"%{topic}%"),
+                        ).fetchall())
+                seen = set()
+                matches = []
+                for row in rows:
+                    if row["id"] in seen:
+                        continue
+                    seen.add(row["id"])
+                    matches.append({
+                        "id": row["id"],
+                        "type": row["type"],
+                        "title": row["title"],
+                        "summary": row["summary"],
+                        "metadata": json.loads(row["metadata_json"] or "{}"),
+                    })
+                    if len(matches) >= limit:
+                        break
+        lines = []
+        for match in matches[:limit]:
+            meta = match.get("metadata") or {}
+            source = meta.get("filename") or meta.get("conversation_id") or meta.get("source") or match["id"]
+            summary = _clean_text(match.get("summary") or "")[:700]
+            lines.append(f"- [{match['type']}] {match['title']} | source={source} | {summary}")
+        return "\n".join(lines)
+
+    def neighbors(self, node_id: str) -> Dict[str, Any]:
+        """Return direct neighbors (1-hop) of a node."""
+        with self._connect() as conn:
+            edge_rows = conn.execute(
+                "SELECT from_node, to_node, type, weight FROM edges WHERE from_node=? OR to_node=?",
+                (node_id, node_id),
+            ).fetchall()
+            neighbor_ids: set = set()
+            edges = []
+            for row in edge_rows:
+                neighbor_ids.add(row["from_node"])
+                neighbor_ids.add(row["to_node"])
+                edges.append({"from": row["from_node"], "to": row["to_node"], "type": row["type"], "weight": row["weight"]})
+            neighbor_ids.discard(node_id)
+            nodes = []
+            if neighbor_ids:
+                placeholders = ",".join("?" * len(neighbor_ids))
+                nodes = [
+                    {
+                        "id": row["id"],
+                        "type": row["type"],
+                        "title": row["title"],
+                        "summary": row["summary"],
+                        "metadata": json.loads(row["metadata_json"] or "{}"),
+                    }
+                    for row in conn.execute(
+                        f"SELECT id, type, title, summary, metadata_json FROM nodes WHERE id IN ({placeholders})",
+                        list(neighbor_ids),
+                    )
+                ]
+        return {"node_id": node_id, "neighbors": nodes, "edges": edges}
+
+    def delete_conversation(self, conversation_id: str) -> Dict[str, Any]:
+        conversation_id = str(conversation_id or "").strip()
+        if not conversation_id:
+            return {"status": "skipped", "removed_nodes": 0}
+        conv_id = f"conversation:{_slug(conversation_id)}"
+        with self._connect() as conn:
+            direct_ids = [
+                row["to_node"]
+                for row in conn.execute(
+                    "SELECT to_node FROM edges WHERE from_node=? AND type='contains'",
+                    (conv_id,),
+                )
+            ]
+            remove_ids = set(direct_ids)
+            for source_id in list(direct_ids):
+                for row in conn.execute(
+                    """
+                    SELECT to_node FROM edges
+                    WHERE from_node=? AND type IN ('has_chunk', 'implies', 'contains_signal', 'has_page', 'has_slide', 'has_sheet', 'contains_image')
+                    """,
+                    (source_id,),
+                ):
+                    remove_ids.add(row["to_node"])
+            remove_ids.add(conv_id)
+            for node_id in remove_ids:
+                conn.execute("DELETE FROM nodes WHERE id=?", (node_id,))
+            conn.execute(
+                """
+                DELETE FROM nodes
+                WHERE type='Topic'
+                  AND id NOT IN (SELECT to_node FROM edges)
+                  AND id NOT IN (SELECT from_node FROM edges)
+                """
+            )
+        return {"status": "ok", "conversation_id": conversation_id, "removed_nodes": len(remove_ids)}
+
+    def clear_all(self) -> Dict[str, Any]:
+        with self._connect() as conn:
+            counts = {
+                "nodes": conn.execute("SELECT COUNT(*) AS c FROM nodes").fetchone()["c"],
+                "edges": conn.execute("SELECT COUNT(*) AS c FROM edges").fetchone()["c"],
+                "chunks": conn.execute("SELECT COUNT(*) AS c FROM chunks").fetchone()["c"],
+            }
+            conn.execute("DELETE FROM chunks")
+            conn.execute("DELETE FROM edges")
+            conn.execute("DELETE FROM nodes")
+        if self.blob_dir.exists():
+            shutil.rmtree(self.blob_dir, ignore_errors=True)
+            self.blob_dir.mkdir(parents=True, exist_ok=True)
+        return {"status": "ok", "removed": counts}
+
+    def stats(self) -> Dict[str, Any]:
+        with self._connect() as conn:
+            node_counts = {
+                row["type"]: row["count"]
+                for row in conn.execute("SELECT type, COUNT(*) AS count FROM nodes GROUP BY type")
+            }
+            edge_counts = {
+                row["type"]: row["count"]
+                for row in conn.execute("SELECT type, COUNT(*) AS count FROM edges GROUP BY type")
+            }
+        return {"db_path": str(self.db_path), "nodes": node_counts, "edges": edge_counts}