@devness/useai 0.6.12 → 0.6.14

package/dist/index.js

@@ -684,7 +684,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.6.12";
+    VERSION = "0.6.14";
   }
 });
 
@@ -34595,6 +34595,8 @@ var init_session_state = __esm({
       inProgressSince;
       /** Session ID that was auto-sealed by seal-active hook (for useai_end fallback). */
       autoSealedSessionId;
+      /** Saved parent session state when a child (subagent) session is active. */
+      parentState;
       constructor() {
         this.sessionId = generateSessionId();
         this.conversationId = generateSessionId();
@@ -34605,6 +34607,7 @@ var init_session_state = __esm({
         this.inProgress = false;
         this.inProgressSince = null;
         this.autoSealedSessionId = null;
+        this.parentState = null;
         this.sessionStartTime = Date.now();
         this.heartbeatCount = 0;
         this.sessionRecordCount = 0;
@@ -34665,6 +34668,56 @@ var init_session_state = __esm({
       getSessionDuration() {
         return Math.round((Date.now() - this.sessionStartTime) / 1e3);
       }
+      /**
+       * Save the current session state as the parent, so it can be restored
+       * after a child (subagent) session finishes.
+       */
+      saveParentState() {
+        this.parentState = {
+          sessionId: this.sessionId,
+          sessionStartTime: this.sessionStartTime,
+          heartbeatCount: this.heartbeatCount,
+          sessionRecordCount: this.sessionRecordCount,
+          chainTipHash: this.chainTipHash,
+          conversationId: this.conversationId,
+          conversationIndex: this.conversationIndex,
+          sessionTaskType: this.sessionTaskType,
+          sessionTitle: this.sessionTitle,
+          sessionPrivateTitle: this.sessionPrivateTitle,
+          sessionPromptWordCount: this.sessionPromptWordCount,
+          project: this.project,
+          modelId: this.modelId,
+          startCallTokensEst: this.startCallTokensEst,
+          inProgress: this.inProgress,
+          inProgressSince: this.inProgressSince
+        };
+      }
+      /**
+       * Restore the parent session state after a child session ends.
+       * Returns true if parent state was restored, false if no parent state was saved.
+       */
+      restoreParentState() {
+        if (!this.parentState) return false;
+        const p = this.parentState;
+        this.sessionId = p.sessionId;
+        this.sessionStartTime = p.sessionStartTime;
+        this.heartbeatCount = p.heartbeatCount;
+        this.sessionRecordCount = p.sessionRecordCount;
+        this.chainTipHash = p.chainTipHash;
+        this.conversationId = p.conversationId;
+        this.conversationIndex = p.conversationIndex;
+        this.sessionTaskType = p.sessionTaskType;
+        this.sessionTitle = p.sessionTitle;
+        this.sessionPrivateTitle = p.sessionPrivateTitle;
+        this.sessionPromptWordCount = p.sessionPromptWordCount;
+        this.project = p.project;
+        this.modelId = p.modelId;
+        this.startCallTokensEst = p.startCallTokensEst;
+        this.inProgress = p.inProgress;
+        this.inProgressSince = p.inProgressSince;
+        this.parentState = null;
+        return true;
+      }
       initializeKeystore() {
         ensureDir();
         if (existsSync7(KEYSTORE_FILE)) {
@@ -34899,8 +34952,14 @@ function registerTools(server2, session2, opts) {
     },
     async ({ task_type, title, private_title, project, model, conversation_id }) => {
       const prevConvId = session2.conversationId;
-      if (session2.sessionRecordCount > 0 && opts?.sealBeforeReset) {
-        opts.sealBeforeReset();
+      const isChildSession = session2.inProgress && session2.sessionRecordCount > 0;
+      const parentSessionId = isChildSession ? session2.sessionId : null;
+      if (isChildSession) {
+        session2.saveParentState();
+      } else {
+        if (session2.sessionRecordCount > 0 && opts?.sealBeforeReset) {
+          opts.sealBeforeReset();
+        }
       }
       session2.reset();
       session2.autoSealedSessionId = null;
@@ -34910,7 +34969,7 @@ function registerTools(server2, session2, opts) {
           session2.conversationId = conversation_id;
           session2.conversationIndex = 0;
         }
-      } else {
+      } else if (!isChildSession) {
         session2.conversationId = generateSessionId();
         session2.conversationIndex = 0;
       }
@@ -34930,11 +34989,13 @@ function registerTools(server2, session2, opts) {
       if (title) chainData.title = title;
       if (private_title) chainData.private_title = private_title;
       if (model) chainData.model = model;
+      if (parentSessionId) chainData.parent_session_id = parentSessionId;
       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 childSuffix = parentSessionId ? ` \xB7 child of ${parentSessionId.slice(0, 8)}` : "";
+      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}${childSuffix} \xB7 ${session2.signingAvailable ? "signed" : "unsigned"}`;
       const paramsJson = JSON.stringify({ task_type, title, private_title, project, model });
       session2.startCallTokensEst = {
         output: Math.ceil(paramsJson.length / 4),
@@ -34972,17 +35033,18 @@ function registerTools(server2, session2, opts) {
   );
   server2.tool(
     "useai_end",
-    '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.',
+    'End the current AI coding session and record milestones. Each milestone is an object with required fields: "title" (generic, no project names), "category" (e.g. "feature", "bugfix", "investigation", "analysis", "refactor", "test", "docs"), and optional "private_title" and "complexity". Example: [{"title": "Implemented auth flow", "category": "feature"}]. 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.',
     {
+      session_id: external_exports.string().optional().describe("Session ID to end. If omitted, ends the current active session. Pass the session_id from your useai_start response to explicitly target your own session (important when subagents may have started their own sessions on the same connection)."),
       task_type: taskTypeSchema.optional().describe("What kind of task was the developer working on?"),
       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."),
+        title: external_exports.string().describe("PRIVACY-CRITICAL: Generic description of what was accomplished. NEVER include project names, file paths, class names, or identifying details. GOOD: 'Implemented user authentication'. BAD: 'Fixed bug in Acme auth'."),
+        private_title: external_exports.string().optional().describe("Detailed description for the user's private records. CAN include project names and specifics."),
+        category: milestoneCategorySchema.describe("Required. Type of work: feature, bugfix, refactor, test, docs, investigation, analysis, research, setup, deployment, performance, cleanup, chore, security, migration, design, devops, config, other"),
+        complexity: complexitySchema.optional().describe("Optional. simple, medium, or complex. Defaults to medium.")
+      }))).optional().describe('Array of milestone objects. Each MUST have "title" (string) and "category" (string). Example: [{"title": "Implemented auth flow", "category": "feature"}, {"title": "Fixed race condition", "category": "bugfix"}]'),
       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."),
@@ -34998,7 +35060,79 @@ function registerTools(server2, session2, opts) {
         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.")
     },
-    async ({ task_type, languages, files_touched_count, milestones: milestonesInput, evaluation }) => {
+    async ({ session_id: targetSessionId, task_type, languages, files_touched_count, milestones: milestonesInput, evaluation }) => {
+      if (targetSessionId && session2.parentState && session2.parentState.sessionId.startsWith(targetSessionId)) {
+        if (session2.sessionRecordCount > 0) {
+          const childDuration = session2.getSessionDuration();
+          const childNow = (/* @__PURE__ */ new Date()).toISOString();
+          const childEndRecord = session2.appendToChain("session_end", {
+            duration_seconds: childDuration,
+            task_type: session2.sessionTaskType,
+            languages: [],
+            files_touched: 0,
+            heartbeat_count: session2.heartbeatCount,
+            auto_sealed: true,
+            parent_ended: true
+          });
+          const childSealData = JSON.stringify({
+            session_id: session2.sessionId,
+            parent_session_id: session2.parentState.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,
+            model: session2.modelId ?? void 0,
+            started_at: new Date(session2.sessionStartTime).toISOString(),
+            ended_at: childNow,
+            duration_seconds: childDuration,
+            heartbeat_count: session2.heartbeatCount,
+            record_count: session2.sessionRecordCount,
+            chain_end_hash: childEndRecord.hash
+          });
+          const childSealSig = signHash(
+            createHash3("sha256").update(childSealData).digest("hex"),
+            session2.signingKey
+          );
+          session2.appendToChain("session_seal", { seal: childSealData, seal_signature: childSealSig });
+          const childActivePath = join7(ACTIVE_DIR, `${session2.sessionId}.jsonl`);
+          const childSealedPath = join7(SEALED_DIR, `${session2.sessionId}.jsonl`);
+          try {
+            if (existsSync8(childActivePath)) renameSync2(childActivePath, childSealedPath);
+          } catch {
+          }
+          const childSeal = {
+            session_id: session2.sessionId,
+            parent_session_id: session2.parentState.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,
+            model: session2.modelId ?? void 0,
+            started_at: new Date(session2.sessionStartTime).toISOString(),
+            ended_at: childNow,
+            duration_seconds: childDuration,
+            heartbeat_count: session2.heartbeatCount,
+            record_count: session2.sessionRecordCount,
+            chain_start_hash: "GENESIS",
+            chain_end_hash: childEndRecord.hash,
+            seal_signature: childSealSig
+          };
+          const childSessions = getSessions().filter((s) => s.session_id !== childSeal.session_id);
+          childSessions.push(childSeal);
+          writeJson(SESSIONS_FILE, childSessions);
+        }
+        session2.restoreParentState();
+      }
       if (session2.sessionRecordCount === 0) {
         if (session2.autoSealedSessionId) {
           const enrichResult = enrichAutoSealedSession(
@@ -35078,8 +35212,10 @@ function registerTools(server2, session2, opts) {
         ...session2.modelId ? { model: session2.modelId } : {}
       });
       const startEst = session2.startCallTokensEst ?? { input: 0, output: 0 };
+      const parentId = session2.parentState?.sessionId;
       const sealData = JSON.stringify({
         session_id: session2.sessionId,
+        ...parentId ? { parent_session_id: parentId } : {},
         conversation_id: session2.conversationId,
         conversation_index: session2.conversationIndex,
         client: session2.clientName,
@@ -35131,6 +35267,7 @@ function registerTools(server2, session2, opts) {
       };
       const seal = {
         session_id: session2.sessionId,
+        ...parentId ? { parent_session_id: parentId } : {},
         conversation_id: session2.conversationId,
         conversation_index: session2.conversationIndex,
         client: session2.clientName,
@@ -35160,11 +35297,13 @@ function registerTools(server2, session2, opts) {
       writeJson(SESSIONS_FILE, sessions2);
       session2.inProgress = false;
       session2.inProgressSince = null;
+      const restoredParent = session2.restoreParentState();
+      const parentRestoredStr = restoredParent ? ` \xB7 parent ${session2.sessionId.slice(0, 8)} restored` : "";
       return {
         content: [
           {
             type: "text",
-            text: responseText
+            text: responseText + parentRestoredStr
           }
         ]
       };

package/package.json

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