@devness/useai 0.4.10 → 0.4.12

package/dist/index.js

@@ -112,7 +112,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.4.10";
+    VERSION = "0.4.12";
   }
 });
 
@@ -947,12 +947,20 @@ process.stdin.on('end', () => {
     if (!hook.transcript_path) process.exit(0);
     let transcript;
     try { transcript = readFileSync(hook.transcript_path, 'utf-8'); } catch { process.exit(0); }
-    if (!transcript.includes('useai_session_start')) process.exit(0);
-    if (transcript.includes('useai_session_end')) process.exit(0);
-    process.stdout.write(JSON.stringify({
-      decision: 'block',
-      reason: 'Before ending, call useai_session_end with milestones (generic descriptions of accomplishments), languages used, and files_touched_count.'
-    }));
+    var starts = (transcript.match(/useai_session_start/g) || []).length;
+    if (starts === 0) process.exit(0);
+    var ends = (transcript.match(/useai_session_end/g) || []).length;
+    if (starts > ends) {
+      process.stdout.write(JSON.stringify({
+        decision: 'block',
+        reason: 'Before ending, call useai_session_end with milestones (generic descriptions of accomplishments), languages used, and files_touched_count.'
+      }));
+    } else {
+      process.stdout.write(JSON.stringify({
+        decision: 'block',
+        reason: 'You did work without UseAI tracking. Call useai_session_start first, then useai_session_end with milestones (generic descriptions of accomplishments), languages used, and files_touched_count.'
+      }));
+    }
   } catch { process.exit(0); }
 });
 `;
@@ -1093,7 +1101,7 @@ function resolveClient(server2, session2) {
 function registerTools(server2, session2) {
   server2.tool(
     "useai_session_start",
-    "Start tracking an AI coding session. Call this at the beginning of a conversation.",
+    "Start tracking an AI coding session. Call this at the beginning of every response.",
     {
       task_type: z2.enum(["coding", "debugging", "testing", "planning", "reviewing", "documenting", "learning", "other"]).optional().describe("What kind of task is the developer working on?")
     },
@@ -1485,11 +1493,17 @@ function injectInstructions(config) {
     writeFileSync4(config.path, USEAI_INSTRUCTIONS + "\n");
     return;
   }
-  if (hasInstructionsBlock(config.path)) return;
   let existing = "";
   if (existsSync8(config.path)) {
     existing = readFileSync4(config.path, "utf-8");
   }
+  if (hasInstructionsBlock(config.path)) {
+    const pattern = new RegExp(
+      `${INSTRUCTIONS_START}[\\s\\S]*?${INSTRUCTIONS_END}`
+    );
+    writeFileSync4(config.path, existing.replace(pattern, USEAI_INSTRUCTIONS_BLOCK));
+    return;
+  }
   const separator = existing && !existing.endsWith("\n") ? "\n\n" : existing ? "\n" : "";
   writeFileSync4(config.path, existing + separator + USEAI_INSTRUCTIONS_BLOCK + "\n");
 }
@@ -1579,10 +1593,9 @@ var init_tools = __esm({
     init_dist();
     USEAI_INSTRUCTIONS_TEXT = [
       "## UseAI Session Tracking",
-      "- At the START of every conversation, call `useai_session_start` with the appropriate task_type.",
+      "- At the START of every response, call `useai_session_start` with the appropriate task_type.",
       "- For LONG sessions (15+ min), call `useai_heartbeat` periodically.",
-      "- At the END of every conversation, call `useai_session_end` with languages used, files_touched_count, and milestones.",
-      '- MILESTONES PRIVACY: Milestone titles must be COMPLETELY GENERIC. NEVER mention project names, repository names, product names, file names, file paths, class names, or company names. Write as if describing the work to a stranger who knows nothing about the codebase. GOOD: "Fixed race condition in background worker". BAD: "Fixed bug in ProjectX worker service".'
+      "- At the END of every response, call `useai_session_end` with languages used, files_touched_count, and milestones (generic descriptions only \u2014 no project names, file paths, or company names)."
     ].join("\n");
     MCP_ENTRY = {
       command: "npx",
@@ -1595,10 +1608,9 @@ var init_tools = __esm({
     INSTRUCTIONS_END = "<!-- useai:end -->";
     USEAI_INSTRUCTIONS = [
       "## UseAI Session Tracking",
-      "- At the START of every conversation, call `useai_session_start` with the appropriate task_type.",
+      "- At the START of every response, call `useai_session_start` with the appropriate task_type.",
       "- For LONG sessions (15+ min), call `useai_heartbeat` periodically.",
-      "- At the END of every conversation, call `useai_session_end` with languages used, files_touched_count, and milestones.",
-      '- MILESTONES PRIVACY: Milestone titles must be COMPLETELY GENERIC. NEVER mention project names, repository names, product names, file names, file paths, class names, or company names. Write as if describing the work to a stranger who knows nothing about the codebase. GOOD: "Fixed race condition in background worker". BAD: "Fixed bug in ProjectX worker service".'
+      "- At the END of every response, call `useai_session_end` with languages used, files_touched_count, and milestones (generic descriptions only \u2014 no project names, file paths, or company names)."
     ].join("\n");
     USEAI_INSTRUCTIONS_BLOCK = `${INSTRUCTIONS_START}
 ${USEAI_INSTRUCTIONS}

package/package.json

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