@devness/useai 0.6.10 → 0.6.12

package/dist/index.js

@@ -684,7 +684,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.6.10";
+    VERSION = "0.6.12";
   }
 });
 
@@ -34589,6 +34589,12 @@ var init_session_state = __esm({
       modelId;
       /** Token estimates for the useai_start tool call. */
       startCallTokensEst;
+      /** Whether a UseAI session is actively in-progress (between useai_start and useai_end). */
+      inProgress;
+      /** Timestamp when the current session was marked in-progress. */
+      inProgressSince;
+      /** Session ID that was auto-sealed by seal-active hook (for useai_end fallback). */
+      autoSealedSessionId;
       constructor() {
         this.sessionId = generateSessionId();
         this.conversationId = generateSessionId();
@@ -34596,6 +34602,9 @@ var init_session_state = __esm({
         this.mcpSessionId = null;
         this.modelId = null;
         this.startCallTokensEst = null;
+        this.inProgress = false;
+        this.inProgressSince = null;
+        this.autoSealedSessionId = null;
         this.sessionStartTime = Date.now();
         this.heartbeatCount = 0;
         this.sessionRecordCount = 0;
@@ -34622,6 +34631,8 @@ var init_session_state = __esm({
         this.sessionPromptWordCount = null;
         this.modelId = null;
         this.startCallTokensEst = null;
+        this.inProgress = false;
+        this.inProgressSince = null;
         this.detectProject();
       }
       detectProject() {
@@ -34715,8 +34726,20 @@ __export(register_tools_exports, {
   registerTools: () => registerTools
 });
 import { createHash as createHash3, randomUUID as randomUUID3 } from "crypto";
-import { existsSync as existsSync8, renameSync as renameSync2 } from "fs";
+import { existsSync as existsSync8, readFileSync as readFileSync5, renameSync as renameSync2 } from "fs";
 import { join as join7 } from "path";
+function coerceJsonString(schema) {
+  return external_exports.preprocess((val) => {
+    if (typeof val === "string") {
+      try {
+        return JSON.parse(val);
+      } catch {
+        return val;
+      }
+    }
+    return val;
+  }, schema);
+}
 function getConfig() {
   return readJson(CONFIG_FILE, {
     milestone_tracking: true,
@@ -34738,6 +34761,130 @@ function resolveClient2(server2, session2) {
   }
   session2.setClient(detectClient());
 }
+function enrichAutoSealedSession(sealedSessionId, session2, args) {
+  const sealedPath = join7(SEALED_DIR, `${sealedSessionId}.jsonl`);
+  const activePath = join7(ACTIVE_DIR, `${sealedSessionId}.jsonl`);
+  const chainPath = existsSync8(sealedPath) ? sealedPath : existsSync8(activePath) ? activePath : null;
+  if (!chainPath) {
+    return "No active session to end (already sealed or never started).";
+  }
+  let startData = {};
+  let duration3 = 0;
+  let endedAt = (/* @__PURE__ */ new Date()).toISOString();
+  let startedAt2 = endedAt;
+  try {
+    const content = readFileSync5(chainPath, "utf-8").trim();
+    const lines = content.split("\n").filter(Boolean);
+    if (lines.length > 0) {
+      const firstRecord = JSON.parse(lines[0]);
+      startData = firstRecord.data;
+      startedAt2 = firstRecord.timestamp;
+      const lastRecord = JSON.parse(lines[lines.length - 1]);
+      if (lastRecord.type === "session_seal" && lastRecord.data["seal"]) {
+        try {
+          const sealObj = JSON.parse(lastRecord.data["seal"]);
+          duration3 = sealObj.duration_seconds ?? Math.round((new Date(lastRecord.timestamp).getTime() - new Date(startedAt2).getTime()) / 1e3);
+          endedAt = sealObj.ended_at ?? lastRecord.timestamp;
+        } catch {
+          duration3 = Math.round((new Date(lastRecord.timestamp).getTime() - new Date(startedAt2).getTime()) / 1e3);
+          endedAt = lastRecord.timestamp;
+        }
+      } else {
+        duration3 = Math.round((new Date(lastRecord.timestamp).getTime() - new Date(startedAt2).getTime()) / 1e3);
+        endedAt = lastRecord.timestamp;
+      }
+    }
+  } catch {
+    return "No active session to end (chain file unreadable).";
+  }
+  const taskType = args.task_type ?? startData.task_type ?? "coding";
+  const languages = args.languages ?? [];
+  const filesTouched = args.files_touched_count ?? 0;
+  let milestoneCount = 0;
+  if (args.milestones && args.milestones.length > 0) {
+    const config2 = getConfig();
+    if (config2.milestone_tracking) {
+      const durationMinutes = Math.round(duration3 / 60);
+      const allMilestones = getMilestones();
+      for (const m of args.milestones) {
+        allMilestones.push({
+          id: `m_${randomUUID3().slice(0, 8)}`,
+          session_id: sealedSessionId,
+          title: m.title,
+          private_title: m.private_title,
+          project: startData.project ?? session2.project ?? void 0,
+          category: m.category,
+          complexity: m.complexity ?? "medium",
+          duration_minutes: durationMinutes,
+          languages,
+          client: startData.client ?? session2.clientName,
+          created_at: (/* @__PURE__ */ new Date()).toISOString(),
+          published: false,
+          published_at: null,
+          chain_hash: ""
+        });
+        milestoneCount++;
+      }
+      writeJson(MILESTONES_FILE, allMilestones);
+    }
+  }
+  let sessionScore;
+  let frameworkId;
+  if (args.evaluation) {
+    const config2 = getConfig();
+    const framework = getFramework(config2.evaluation_framework);
+    sessionScore = Math.round(framework.computeSessionScore(args.evaluation));
+    frameworkId = framework.id;
+  }
+  const richSeal = {
+    session_id: sealedSessionId,
+    conversation_id: startData.conversation_id,
+    conversation_index: startData.conversation_index,
+    client: startData.client ?? session2.clientName,
+    task_type: taskType,
+    languages,
+    files_touched: filesTouched,
+    project: startData.project ?? session2.project ?? void 0,
+    title: startData.title ?? void 0,
+    private_title: startData.private_title ?? void 0,
+    model: startData.model ?? session2.modelId ?? void 0,
+    evaluation: args.evaluation ?? void 0,
+    session_score: sessionScore,
+    evaluation_framework: frameworkId,
+    started_at: startedAt2,
+    ended_at: endedAt,
+    duration_seconds: duration3,
+    heartbeat_count: 0,
+    record_count: 0,
+    chain_start_hash: "",
+    chain_end_hash: "",
+    seal_signature: ""
+  };
+  const allSessions = getSessions();
+  const existingIdx = allSessions.findIndex((s) => s.session_id === sealedSessionId);
+  if (existingIdx >= 0) {
+    const existing = allSessions[existingIdx];
+    allSessions[existingIdx] = {
+      ...existing,
+      // Enrich with data from useai_end call
+      task_type: taskType,
+      languages,
+      files_touched: filesTouched,
+      evaluation: args.evaluation ?? existing.evaluation,
+      session_score: sessionScore ?? existing.session_score,
+      evaluation_framework: frameworkId ?? existing.evaluation_framework
+    };
+  } else {
+    allSessions.push(richSeal);
+  }
+  writeJson(SESSIONS_FILE, allSessions);
+  const durationStr = formatDuration(duration3);
+  const langStr = languages.length > 0 ? ` using ${languages.join(", ")}` : "";
+  const milestoneStr = milestoneCount > 0 ? ` \xB7 ${milestoneCount} milestone${milestoneCount > 1 ? "s" : ""} recorded` : "";
+  const evalStr = args.evaluation ? ` \xB7 eval: ${args.evaluation.task_outcome} (prompt: ${args.evaluation.prompt_quality}/5)` : "";
+  const scoreStr = sessionScore !== void 0 ? ` \xB7 score: ${sessionScore}/100 (${frameworkId})` : "";
+  return `Session ended (enriched auto-seal): ${durationStr} ${taskType}${langStr}${milestoneStr}${evalStr}${scoreStr}`;
+}
 function registerTools(server2, session2, opts) {
   server2.tool(
     "useai_start",
@@ -34756,6 +34903,7 @@ function registerTools(server2, session2, opts) {
         opts.sealBeforeReset();
       }
       session2.reset();
+      session2.autoSealedSessionId = null;
       resolveClient2(server2, session2);
       if (conversation_id) {
         if (conversation_id !== prevConvId) {
@@ -34783,6 +34931,8 @@ function registerTools(server2, session2, opts) {
       if (private_title) chainData.private_title = private_title;
       if (model) chainData.model = model;
       const record2 = session2.appendToChain("session_start", chainData);
+      session2.inProgress = true;
+      session2.inProgressSince = Date.now();
       writeMcpMapping(session2.mcpSessionId, session2.sessionId);
       const responseText = `useai session started \u2014 ${session2.sessionTaskType} on ${session2.clientName} \xB7 ${session2.sessionId.slice(0, 8)} \xB7 conv ${session2.conversationId.slice(0, 8)}#${session2.conversationIndex} \xB7 ${session2.signingAvailable ? "signed" : "unsigned"}`;
       const paramsJson = JSON.stringify({ task_type, title, private_title, project, model });
@@ -34825,15 +34975,15 @@ function registerTools(server2, session2, opts) {
     'End the current AI coding session and record milestones. Each milestone needs TWO titles: (1) a generic public "title" safe for public display (NEVER include project names, file names, class names, or any identifying details), and (2) an optional detailed "private_title" for the user\'s own records that CAN include project names, file names, and specific details. GOOD title: "Implemented user authentication". GOOD private_title: "Added JWT auth to UseAI API server". BAD title: "Fixed bug in Acme auth service". Also provide an `evaluation` object assessing the session: prompt_quality (1-5), context_provided (1-5), task_outcome (completed/partial/abandoned/blocked), iteration_count, independence_level (1-5), scope_quality (1-5), and tools_leveraged count. Score honestly based on the actual interaction. For any scored metric < 5 or non-completed outcome, you MUST provide a *_reason field explaining what was lacking and a concrete tip for the user to improve next time. Only skip *_reason for a perfect 5.',
     {
       task_type: taskTypeSchema.optional().describe("What kind of task was the developer working on?"),
-      languages: external_exports.array(external_exports.string()).optional().describe("Programming languages used (e.g. ['typescript', 'python'])"),
-      files_touched_count: external_exports.number().optional().describe("Approximate number of files created or modified (count only, no names)"),
-      milestones: external_exports.array(external_exports.object({
+      languages: coerceJsonString(external_exports.array(external_exports.string())).optional().describe("Programming languages used (e.g. ['typescript', 'python'])"),
+      files_touched_count: coerceJsonString(external_exports.number()).optional().describe("Approximate number of files created or modified (count only, no names)"),
+      milestones: coerceJsonString(external_exports.array(external_exports.object({
         title: external_exports.string().describe("PRIVACY-CRITICAL: Generic description of what was accomplished. NEVER include project names, repo names, product names, package names, file names, file paths, class names, API endpoints, database names, company names, or ANY identifier that could reveal which codebase this work was done in. Write as if describing the work to a stranger. GOOD: 'Implemented user authentication', 'Fixed race condition in background worker', 'Added unit tests for data validation', 'Refactored state management layer'. BAD: 'Fixed bug in Acme auth', 'Investigated ProjectX pipeline', 'Updated UserService.ts in src/services/', 'Added tests for coverit MCP tool'"),
         private_title: external_exports.string().optional().describe("Detailed description for the user's private records. CAN include project names, file names, and specific details. Example: 'Added private/public milestone support to UseAI MCP server'"),
         category: milestoneCategorySchema.describe("Type of work completed"),
         complexity: complexitySchema.optional().describe("How complex was this task?")
-      })).optional().describe("What was accomplished this session? List each distinct piece of work completed. Provide both a generic public title and an optional detailed private_title."),
-      evaluation: external_exports.object({
+      }))).optional().describe("What was accomplished this session? List each distinct piece of work completed. Provide both a generic public title and an optional detailed private_title."),
+      evaluation: coerceJsonString(external_exports.object({
         prompt_quality: external_exports.number().min(1).max(5).describe("How clear, specific, and complete was the initial prompt? 1=vague/ambiguous, 5=crystal clear with acceptance criteria"),
         prompt_quality_reason: external_exports.string().optional().describe("Required if prompt_quality < 5. Explain what was vague/missing and how the user could phrase it better next time."),
         context_provided: external_exports.number().min(1).max(5).describe("Did the user provide relevant context (files, errors, constraints)? 1=no context, 5=comprehensive context"),
@@ -34846,10 +34996,21 @@ function registerTools(server2, session2, opts) {
         scope_quality: external_exports.number().min(1).max(5).describe("Was the task well-scoped? 1=vague or impossibly broad, 5=precise and achievable"),
         scope_quality_reason: external_exports.string().optional().describe("Required if scope_quality < 5. How was the scope too broad/vague and how could it be better defined?"),
         tools_leveraged: external_exports.number().min(0).describe("Count of distinct AI capabilities used (code gen, debugging, refactoring, testing, docs, etc.)")
-      }).optional().describe("AI-assessed evaluation of this session. Score honestly based on the actual interaction.")
+      })).optional().describe("AI-assessed evaluation of this session. Score honestly based on the actual interaction.")
     },
     async ({ task_type, languages, files_touched_count, milestones: milestonesInput, evaluation }) => {
       if (session2.sessionRecordCount === 0) {
+        if (session2.autoSealedSessionId) {
+          const enrichResult = enrichAutoSealedSession(
+            session2.autoSealedSessionId,
+            session2,
+            { task_type, languages, files_touched_count, milestones: milestonesInput, evaluation }
+          );
+          session2.autoSealedSessionId = null;
+          session2.inProgress = false;
+          session2.inProgressSince = null;
+          return { content: [{ type: "text", text: enrichResult }] };
+        }
         return {
           content: [{ type: "text", text: "No active session to end (already sealed or never started)." }]
         };
@@ -34997,6 +35158,8 @@ function registerTools(server2, session2, opts) {
       const sessions2 = getSessions().filter((s) => s.session_id !== seal.session_id);
       sessions2.push(seal);
       writeJson(SESSIONS_FILE, sessions2);
+      session2.inProgress = false;
+      session2.inProgressSince = null;
       return {
         content: [
           {
@@ -35509,7 +35672,7 @@ __export(daemon_exports, {
 });
 import { createServer } from "http";
 import { createHash as createHash4, randomUUID as randomUUID4 } from "crypto";
-import { existsSync as existsSync10, readdirSync, readFileSync as readFileSync5, appendFileSync as appendFileSync2, renameSync as renameSync3, writeFileSync as writeFileSync5, unlinkSync as unlinkSync5, statSync } from "fs";
+import { existsSync as existsSync10, readdirSync, readFileSync as readFileSync6, appendFileSync as appendFileSync2, renameSync as renameSync3, writeFileSync as writeFileSync5, unlinkSync as unlinkSync5, statSync } from "fs";
 import { join as join9 } from "path";
 function getActiveUseaiSessionIds() {
   const ids = /* @__PURE__ */ new Set();
@@ -35522,7 +35685,7 @@ function sealOrphanFile(sessionId) {
   const filePath = join9(ACTIVE_DIR, `${sessionId}.jsonl`);
   if (!existsSync10(filePath)) return;
   try {
-    const content = readFileSync5(filePath, "utf-8").trim();
+    const content = readFileSync6(filePath, "utf-8").trim();
     if (!content) return;
     const lines = content.split("\n").filter(Boolean);
     if (lines.length === 0) return;
@@ -35760,8 +35923,10 @@ function autoSealSession(active) {
   upsertSessionSeal(seal);
 }
 function sealSessionData(active) {
+  const sealedId = active.session.sessionId;
   autoSealSession(active);
   active.session.reset();
+  active.session.autoSealedSessionId = sealedId;
 }
 function resetIdleTimer(sessionId) {
   const active = sessions.get(sessionId);
@@ -35865,7 +36030,7 @@ function readChainMetadata(useaiSessionId) {
   const chainPath = existsSync10(activePath) ? activePath : existsSync10(sealedPath) ? sealedPath : null;
   if (!chainPath) return null;
   try {
-    const firstLine = readFileSync5(chainPath, "utf-8").split("\n")[0];
+    const firstLine = readFileSync6(chainPath, "utf-8").split("\n")[0];
     if (!firstLine) return null;
     const record2 = JSON.parse(firstLine);
     const d = record2.data;
@@ -35934,7 +36099,7 @@ function recoverHeartbeat(staleMcpSessionId, rpcId, res) {
     return true;
   }
   try {
-    const content = readFileSync5(chainPath, "utf-8").trim();
+    const content = readFileSync6(chainPath, "utf-8").trim();
     const lines = content.split("\n").filter(Boolean);
     if (lines.length === 0) return false;
     const firstRecord = JSON.parse(lines[0]);
@@ -35969,7 +36134,7 @@ function recoverEndSession(staleMcpSessionId, args, rpcId, res) {
   const chainPath = existsSync10(activePath) ? activePath : existsSync10(sealedPath) ? sealedPath : null;
   if (!chainPath) return false;
   const isAlreadySealed = chainPath === sealedPath;
-  const content = readFileSync5(chainPath, "utf-8").trim();
+  const content = readFileSync6(chainPath, "utf-8").trim();
   if (!content) return false;
   const lines = content.split("\n").filter(Boolean);
   if (lines.length === 0) return false;
@@ -36375,8 +36540,14 @@ async function startDaemon(port) {
     }
     if (url.pathname === "/api/seal-active" && req.method === "POST") {
       let sealed = 0;
+      let skipped = 0;
       for (const [, active] of sessions) {
         if (active.session.sessionRecordCount > 0 && !isSessionAlreadySealed(active.session)) {
+          const { inProgress, inProgressSince } = active.session;
+          if (inProgress && inProgressSince && Date.now() - inProgressSince < SEAL_GRACE_MS) {
+            skipped++;
+            continue;
+          }
           sealSessionData(active);
           sealed++;
         }
@@ -36385,7 +36556,7 @@ async function startDaemon(port) {
         "Content-Type": "application/json",
         "Access-Control-Allow-Origin": "*"
       });
-      res.end(JSON.stringify({ sealed }));
+      res.end(JSON.stringify({ sealed, skipped }));
       return;
     }
     if ((url.pathname.startsWith("/api/local/") || url.pathname === "/api/seal-active") && req.method === "OPTIONS") {
@@ -36542,7 +36713,7 @@ async function startDaemon(port) {
   console.log(`UseAI daemon listening on http://127.0.0.1:${listenPort}`);
   console.log(`PID: ${process.pid}`);
 }
-var IDLE_TIMEOUT_MS, sessions, daemonSigningKey, ORPHAN_SWEEP_INTERVAL_MS, startedAt, updateCheckCache, UPDATE_CHECK_TTL_MS;
+var IDLE_TIMEOUT_MS, SEAL_GRACE_MS, sessions, daemonSigningKey, ORPHAN_SWEEP_INTERVAL_MS, startedAt, updateCheckCache, UPDATE_CHECK_TTL_MS;
 var init_daemon2 = __esm({
   "src/daemon.ts"() {
     "use strict";
@@ -36556,6 +36727,7 @@ var init_daemon2 = __esm({
     init_html();
     init_local_api();
     IDLE_TIMEOUT_MS = 30 * 60 * 1e3;
+    SEAL_GRACE_MS = 30 * 60 * 1e3;
     sessions = /* @__PURE__ */ new Map();
     daemonSigningKey = null;
     ORPHAN_SWEEP_INTERVAL_MS = 15 * 60 * 1e3;

package/package.json

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