@devness/useai 0.5.4 → 0.5.5

package/dist/index.js

@@ -1713,7 +1713,7 @@ function resolveClient(server2, session2) {
   }
   session2.setClient(detectClient());
 }
-function registerTools(server2, session2) {
+function registerTools(server2, session2, opts) {
   server2.tool(
     "useai_start",
     `Start tracking an AI coding session. Call this at the beginning of every response. Generate a session title from the user's prompt: a generic public "title" (no project/file names) and a detailed "private_title" (can include specifics).`,
@@ -1723,6 +1723,9 @@ function registerTools(server2, session2) {
       private_title: z2.string().optional().describe('Detailed session title for private records. Can include project names and specifics. Example: "Fix JWT refresh in UseAI login flow"')
     },
     async ({ task_type, title, private_title }) => {
+      if (session2.sessionRecordCount > 0 && opts?.sealBeforeReset) {
+        opts.sealBeforeReset();
+      }
       session2.reset();
       resolveClient(server2, session2);
       session2.setTaskType(task_type ?? "coding");
@@ -1863,7 +1866,7 @@ function registerTools(server2, session2) {
         chain_end_hash: endRecord.hash,
         seal_signature: sealSignature
       };
-      const sessions2 = getSessions();
+      const sessions2 = getSessions().filter((s) => s.session_id !== seal.session_id);
       sessions2.push(seal);
       writeJson(SESSIONS_FILE, sessions2);
       let milestoneCount = 0;
@@ -2250,6 +2253,9 @@ function sealOrphanFile(sessionId) {
     const client = startData["client"] ?? "unknown";
     const taskType = startData["task_type"] ?? "coding";
     const startTime = firstRecord.timestamp;
+    const orphanTitle = startData["title"] ?? void 0;
+    const orphanPrivateTitle = startData["private_title"] ?? void 0;
+    const orphanProject = startData["project"] ?? void 0;
     let heartbeatCount = 0;
     for (const line of lines) {
       try {
@@ -2297,30 +2303,29 @@ function sealOrphanFile(sessionId) {
       renameSync3(filePath, join7(SEALED_DIR, `${sessionId}.jsonl`));
     } catch {
     }
-    if (!isSessionInIndex(sessionId)) {
-      const convId = startData["conversation_id"] ?? void 0;
-      const convIdx = startData["conversation_index"] ?? void 0;
-      const seal = {
-        session_id: sessionId,
-        conversation_id: convId,
-        conversation_index: convIdx,
-        client,
-        task_type: taskType,
-        languages: [],
-        files_touched: 0,
-        started_at: startTime,
-        ended_at: now,
-        duration_seconds: duration,
-        heartbeat_count: heartbeatCount,
-        record_count: lines.length + 2,
-        chain_start_hash: firstRecord.prev_hash,
-        chain_end_hash: endRecord.hash,
-        seal_signature: sealSignature
-      };
-      const allSessions = readJson(SESSIONS_FILE, []);
-      allSessions.push(seal);
-      writeJson(SESSIONS_FILE, allSessions);
-    }
+    const convId = startData["conversation_id"] ?? void 0;
+    const convIdx = startData["conversation_index"] ?? void 0;
+    const seal = {
+      session_id: sessionId,
+      conversation_id: convId,
+      conversation_index: convIdx,
+      client,
+      task_type: taskType,
+      languages: [],
+      files_touched: 0,
+      project: orphanProject,
+      title: orphanTitle,
+      private_title: orphanPrivateTitle,
+      started_at: startTime,
+      ended_at: now,
+      duration_seconds: duration,
+      heartbeat_count: heartbeatCount,
+      record_count: lines.length + 2,
+      chain_start_hash: firstRecord.prev_hash,
+      chain_end_hash: endRecord.hash,
+      seal_signature: sealSignature
+    };
+    upsertSessionSeal(seal);
   } catch {
   }
 }
@@ -2346,10 +2351,6 @@ function isSessionAlreadySealed(session2) {
   const activePath = join7(ACTIVE_DIR, `${session2.sessionId}.jsonl`);
   return !existsSync8(activePath);
 }
-function isSessionInIndex(sessionId) {
-  const allSessions = readJson(SESSIONS_FILE, []);
-  return allSessions.some((s) => s.session_id === sessionId);
-}
 function sealRichness(s) {
   let score = 0;
   if (s.title) score += 10;
@@ -2377,6 +2378,18 @@ function deduplicateSessionsIndex() {
     writeJson(SESSIONS_FILE, deduped);
   }
 }
+function upsertSessionSeal(seal) {
+  const allSessions = readJson(SESSIONS_FILE, []);
+  const existingIdx = allSessions.findIndex((s) => s.session_id === seal.session_id);
+  if (existingIdx >= 0) {
+    if (sealRichness(seal) >= sealRichness(allSessions[existingIdx])) {
+      allSessions[existingIdx] = seal;
+    }
+  } else {
+    allSessions.push(seal);
+  }
+  writeJson(SESSIONS_FILE, allSessions);
+}
 function autoSealSession(active) {
   const { session: session2 } = active;
   if (session2.sessionRecordCount === 0) return;
@@ -2399,6 +2412,9 @@ function autoSealSession(active) {
     task_type: session2.sessionTaskType,
     languages: [],
     files_touched: 0,
+    project: session2.project ?? void 0,
+    title: session2.sessionTitle ?? void 0,
+    private_title: session2.sessionPrivateTitle ?? void 0,
     started_at: new Date(session2.sessionStartTime).toISOString(),
     ended_at: now,
     duration_seconds: duration,
@@ -2423,29 +2439,28 @@ function autoSealSession(active) {
     }
   } catch {
   }
-  if (!isSessionInIndex(session2.sessionId)) {
-    const chainStartHash = session2.chainTipHash === "GENESIS" ? "GENESIS" : session2.chainTipHash;
-    const seal = {
-      session_id: session2.sessionId,
-      conversation_id: session2.conversationId,
-      conversation_index: session2.conversationIndex,
-      client: session2.clientName,
-      task_type: session2.sessionTaskType,
-      languages: [],
-      files_touched: 0,
-      started_at: new Date(session2.sessionStartTime).toISOString(),
-      ended_at: now,
-      duration_seconds: duration,
-      heartbeat_count: session2.heartbeatCount,
-      record_count: session2.sessionRecordCount,
-      chain_start_hash: chainStartHash,
-      chain_end_hash: endRecord.hash,
-      seal_signature: sealSignature
-    };
-    const allSessions = readJson(SESSIONS_FILE, []);
-    allSessions.push(seal);
-    writeJson(SESSIONS_FILE, allSessions);
-  }
+  const chainStartHash = session2.chainTipHash === "GENESIS" ? "GENESIS" : session2.chainTipHash;
+  const seal = {
+    session_id: session2.sessionId,
+    conversation_id: session2.conversationId,
+    conversation_index: session2.conversationIndex,
+    client: session2.clientName,
+    task_type: session2.sessionTaskType,
+    languages: [],
+    files_touched: 0,
+    project: session2.project ?? void 0,
+    title: session2.sessionTitle ?? void 0,
+    private_title: session2.sessionPrivateTitle ?? void 0,
+    started_at: new Date(session2.sessionStartTime).toISOString(),
+    ended_at: now,
+    duration_seconds: duration,
+    heartbeat_count: session2.heartbeatCount,
+    record_count: session2.sessionRecordCount,
+    chain_start_hash: chainStartHash,
+    chain_end_hash: endRecord.hash,
+    seal_signature: sealSignature
+  };
+  upsertSessionSeal(seal);
 }
 function sealSessionData(active) {
   autoSealSession(active);
@@ -2613,7 +2628,18 @@ async function startDaemon(port) {
             name: "UseAI",
             version: VERSION
           });
-          registerTools(mcpServer, sessionState);
+          registerTools(mcpServer, sessionState, {
+            sealBeforeReset: () => {
+              for (const [, active] of sessions) {
+                if (active.session === sessionState) {
+                  if (active.session.sessionRecordCount > 0 && !isSessionAlreadySealed(active.session)) {
+                    autoSealSession(active);
+                  }
+                  break;
+                }
+              }
+            }
+          });
           const transport = new StreamableHTTPServerTransport({
             sessionIdGenerator: () => randomUUID4(),
             onsessioninitialized: (newSid) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devness/useai",
-  "version": "0.5.4",
+  "version": "0.5.5",
   "description": "Track your AI-assisted development workflow. MCP server that records usage metrics across all your AI tools.",
   "keywords": [
     "mcp",