dsh-trivium 0.3.0 → 0.4.0

package/lib/server.js

@@ -17,6 +17,8 @@ import {
 import { embeddingPublicStatus } from "./embed.js";
 import { recallModeOf, writeUiSettings } from "./settings.js";
 import { discoverWorkbuddyFiles, exportMarkdown, importWorkbuddy } from "./markdown.js";
+import { renameEpisode, sessionMapSnapshot } from "./episode.js";
+import { applyPinsUpdate, listChips, readSessionPins } from "./pins.js";
 
 export { readUiSettings, writeUiSettings } from "./settings.js";
 
@@ -258,5 +260,76 @@ async function handle(req, res, { liveOptions, status }) {
     return;
   }
 
+  if (path === "/api/dsh-trivium/map" && method === "GET") {
+    const { cwd, db } = await dbForRequest(url);
+    const sessionId = url.searchParams.get("sessionId") || "";
+    if (!db) {
+      sendJson(res, 200, { ok: true, cwd: cwd || "", sessionId, nodes: [], edges: [] });
+      return;
+    }
+    const snap = sessionMapSnapshot(db, sessionId);
+    sendJson(res, 200, { ok: true, cwd, ...snap });
+    return;
+  }
+
+  if (path === "/api/dsh-trivium/chips" && method === "GET") {
+    const { cwd, db } = await dbForRequest(url);
+    const q = url.searchParams.get("q") || "";
+    const sessionId = url.searchParams.get("sessionId") || "";
+    if (!db) {
+      sendJson(res, 200, { ok: true, cwd: cwd || "", sessionId, q, chips: [] });
+      return;
+    }
+    sendJson(res, 200, { ok: true, cwd, sessionId, q, chips: listChips(db, { q, sessionId }) });
+    return;
+  }
+
+  if (path === "/api/dsh-trivium/pins" && method === "GET") {
+    const { cwd, db } = await dbForRequest(url);
+    const sessionId = url.searchParams.get("sessionId") || "";
+    if (!db) {
+      sendJson(res, 200, { ok: true, cwd: cwd || "", sessionId, ids: [], clipped: false, tokens: 0 });
+      return;
+    }
+    const pin = readSessionPins(db, sessionId);
+    sendJson(res, 200, { ok: true, cwd, sessionId, ids: pin.ids, clipped: pin.clipped, tokens: pin.tokens });
+    return;
+  }
+
+  if (path === "/api/dsh-trivium/pins" && method === "POST") {
+    const { cwd, db } = await dbForRequest(url);
+    if (!db) {
+      sendJson(res, 400, { ok: false, message: "no workspace" });
+      return;
+    }
+    const body = JSON.parse((await readBody(req)) || "{}");
+    const sessionId = body.sessionId || url.searchParams.get("sessionId") || "";
+    const pin = applyPinsUpdate(db, { ...body, sessionId });
+    sendJson(res, 200, { ok: true, cwd, sessionId, ids: pin.ids, clipped: pin.clipped, tokens: pin.tokens });
+    return;
+  }
+
+  const episodeMatch = path.match(/^\/api\/dsh-trivium\/episodes\/(\d+)$/);
+  if (episodeMatch && method === "POST") {
+    const { cwd, db } = await dbForRequest(url);
+    if (!db) {
+      sendJson(res, 400, { ok: false, message: "no workspace" });
+      return;
+    }
+    const body = JSON.parse((await readBody(req)) || "{}");
+    const result = renameEpisode(db, episodeMatch[1], body.name);
+    sendJson(res, result.ok ? 200 : result.status || 400, {
+      ok: result.ok,
+      cwd,
+      id: Number(episodeMatch[1]),
+      name: result.name || "",
+      summary: result.summary || "",
+      message: result.message,
+    });
+    return;
+  }
+
   sendJson(res, 404, { ok: false, message: "not found" });
 }
+
+export { handle as handleMemoryApi };

package/lib/settings.js

@@ -59,10 +59,36 @@ export function writeUiSettings(patch) {
   if (typeof patch.embeddingApiKey === "string" && patch.embeddingApiKey.trim()) {
     next.embeddingApiKey = patch.embeddingApiKey.trim();
   }
+  if (patch.pinsBySession && typeof patch.pinsBySession === "object") {
+    next.pinsBySession = patch.pinsBySession;
+  } else if (prev.pinsBySession && typeof prev.pinsBySession === "object") {
+    next.pinsBySession = prev.pinsBySession;
+  }
   writeFileSync(SETTINGS_FILE, JSON.stringify(next, null, 2), "utf8");
   return next;
 }
 
+export function pinsOf(sessionId) {
+  const raw = readUiSettings().pinsBySession;
+  if (!raw || typeof raw !== "object") return [];
+  const ids = raw[String(sessionId || "")];
+  if (!Array.isArray(ids)) return [];
+  return [...new Set(ids.map(Number).filter((id) => Number.isFinite(id)))];
+}
+
+export function writePins(sessionId, ids) {
+  const key = String(sessionId || "");
+  const prev = readUiSettings();
+  const pinsBySession = { ...(prev.pinsBySession && typeof prev.pinsBySession === "object" ? prev.pinsBySession : {}) };
+  const nextIds = [...new Set((ids || []).map(Number).filter((id) => Number.isFinite(id)))];
+  if (!key) return nextIds;
+  if (!nextIds.length) delete pinsBySession[key];
+  else pinsBySession[key] = nextIds;
+  mkdirSync(join(homedir(), ".dsh"), { recursive: true });
+  writeFileSync(SETTINGS_FILE, JSON.stringify({ ...prev, pinsBySession }, null, 2), "utf8");
+  return nextIds;
+}
+
 export function settingsFileExists() {
   return existsSync(SETTINGS_FILE);
 }

package/lib/store.js

@@ -163,7 +163,7 @@ function rebuildTextIndex(db) {
 
 export function findMergeTarget(db, candidate) {
   const type = candidate?.type;
-  if (!type) return null;
+  if (!type || type === "episode") return null;
   const name = String(candidate.name || "").trim().toLowerCase();
   const text = String(candidate.text || "").trim().toLowerCase();
   for (const n of findByType(db, type)) {
@@ -227,7 +227,7 @@ export function mergeInto(db, id, incoming) {
 
 export function listNodes(
   db,
-  { type, q, includeArchived = false, includeStale = false, aboutId } = {},
+  { type, q, includeArchived = false, includeStale = false, aboutId, includeEpisodes = false } = {},
 ) {
   const rows = [];
   const needle = String(q || "").trim().toLowerCase();
@@ -241,6 +241,7 @@ export function listNodes(
     if (!node) continue;
     const payload = node.payload || {};
     if (!includeArchived && payload.status === "archived") continue;
+    if (payload.type === "episode" && !includeEpisodes && type !== "episode") continue;
     if (type && payload.type !== type) continue;
     if (around && !around.has(Number(id))) continue;
     const stale = isStalePayload(payload);
@@ -418,7 +419,7 @@ export function exportGraph(db, cwd = "") {
   };
 }
 
-function findByUri(db, uri) {
+export function findByUri(db, uri) {
   const needle = String(uri || "").trim();
   if (!needle) return null;
   for (const id of db.allNodeIds?.() || []) {
@@ -542,6 +543,7 @@ function persistVector(db, id, vec) {
 }
 
 function queueEmbed(db, id, payload) {
+  if (payload?.type === "episode") return;
   if (!embeddingEnabled()) return;
   const text = indexTextFor(payload);
   if (!text) return;
@@ -564,7 +566,7 @@ export async function backfillEmbeddings(db, { log } = {}) {
   for (const id of db.allNodeIds?.() || []) {
     const node = db.get(id);
     const payload = node?.payload;
-    if (!payload || payload.status === "archived") continue;
+    if (!payload || payload.status === "archived" || payload.type === "episode") continue;
     const text = indexTextFor(payload);
     if (!text) continue;
     const vec = await embedText(text);
@@ -705,7 +707,14 @@ function rankBoost(db, hit, query, now) {
 
 function hitRecord(db, id) {
   const node = db.get(Number(id));
-  if (!node || node.payload?.status === "archived" || node.payload?.type === "workspace") return null;
+  if (
+    !node ||
+    node.payload?.status === "archived" ||
+    node.payload?.type === "workspace" ||
+    node.payload?.type === "episode"
+  ) {
+    return null;
+  }
   return {
     id: Number(id),
     score: 0,
@@ -749,7 +758,7 @@ function entityAnchorIds(db, query) {
   const ids = [];
   for (const id of db.allNodeIds?.() || []) {
     const node = db.get(id);
-    if (node?.payload?.status === "archived") continue;
+    if (node?.payload?.status === "archived" || node?.payload?.type === "episode") continue;
     if (entityAnchorScore(node?.payload, q) > 0) ids.push(Number(id));
   }
   return ids;
@@ -795,7 +804,7 @@ function scanBusinessNeighbors(db, id) {
   return ids;
 }
 
-function businessNeighborIds(db, id) {
+export function businessNeighborIds(db, id) {
   const ids = scanBusinessNeighbors(db, id);
   const viaTql = tqlBusinessNeighbors(db, id);
   if (viaTql) {
@@ -804,6 +813,17 @@ function businessNeighborIds(db, id) {
   return [...ids].filter((n) => Number.isFinite(n));
 }
 
+/** Chip suggestions: business-edge neighbors of entities named in `text`. Not written to pins. */
+export function suggestedNeighborIds(db, text) {
+  const ids = new Set();
+  for (const eid of entityAnchorIds(db, text)) {
+    for (const nid of businessNeighborIds(db, eid)) {
+      if (Number(nid) !== Number(eid)) ids.add(Number(nid));
+    }
+  }
+  return [...ids];
+}
+
 function tqlBusinessNeighbors(db, id) {
   if (typeof db.tql !== "function") return null;
   const nid = Number(id);
@@ -828,7 +848,14 @@ function keywordFallback(db, text, topK) {
   const hits = [];
   for (const id of ids) {
     const node = db.get(id);
-    if (!node || node.payload?.status === "archived" || node.payload?.type === "workspace") continue;
+    if (
+      !node ||
+      node.payload?.status === "archived" ||
+      node.payload?.type === "workspace" ||
+      node.payload?.type === "episode"
+    ) {
+      continue;
+    }
     const hay = indexTextFor(node.payload || {}).toLowerCase();
     if (!hay.includes(q)) continue;
     hits.push({
@@ -1000,6 +1027,7 @@ export function buildShortMapReport(db, budgetTokens = 400) {
     const nodes = findByType(db, type);
     counts[type] = nodes.length;
     for (const n of nodes) {
+      if (n.payload?.type === "episode") continue;
       if (type === "decision" && isStalePayload(n.payload)) continue;
       let label = n.payload?.name || n.payload?.text;
       if (!label) continue;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "dsh-trivium",
-  "version": "0.3.0",
-  "description": "In-process graph memory kernel for DeepSeek Harness, backed by TriviumDB.",
+  "version": "0.4.0",
+  "description": "In-process graph memory and session map for DeepSeek Harness, backed by TriviumDB.",
   "license": "MIT",
   "type": "module",
   "main": "lib/index.js",
@@ -23,7 +23,8 @@
     "client": {
       "inject": [
         "@deepseek-ai/dsh-client-runtime",
-        "@deepseek-ai/dsh-client-ui-settings"
+        "@deepseek-ai/dsh-client-ui-settings",
+        "@deepseek-ai/dsh-client-ui-conversation"
       ],
       "platform": "web"
     }
@@ -37,6 +38,9 @@
     "smoke-p4": "node scripts/smoke-p4.mjs",
     "smoke-p5": "node scripts/smoke-p5.mjs",
     "smoke-p6": "node scripts/smoke-p6.mjs",
+    "smoke-p7": "node scripts/smoke-p7.mjs",
+    "smoke-p8": "node scripts/smoke-p8.mjs",
+    "smoke-p9": "node scripts/smoke-p9.mjs",
     "verify-live-p2": "node scripts/verify-live-p2.mjs"
   },
   "engines": {