@integrity-labs/agt-cli 0.14.7 → 0.14.10

package/dist/lib/manager-worker.js

@@ -1,6 +1,8 @@
 import {
+  SUPERVISOR_RESTART_EXIT_CODE,
   api,
   appendDmFooter,
+  classifyOutput,
   exchangeApiKey,
   extractFrontmatter,
   getApiKey,
@@ -9,7 +11,6 @@ import {
   getIntegration,
   isParseError,
   isResolveError,
-  isSuppressOutput,
   parseDeliveryTarget,
   provision,
   provisionIsolationHook,
@@ -18,7 +19,7 @@ import {
   resolveChannels,
   resolveDmTarget,
   wrapScheduledTaskPrompt
-} from "../chunk-5SFAHM2Z.js";
+} from "../chunk-RMTMETCK.js";
 import {
   findTaskByTemplate,
   getProjectDir,
@@ -28,6 +29,7 @@ import {
   syncTasksToScheduler
 } from "../chunk-M6FSTVGG.js";
 import {
+  buildAllowedTools,
   getProjectDir as getProjectDir2,
   injectMessage,
   isSessionHealthy,
@@ -37,7 +39,7 @@ import {
   startPersistentSession,
   stopAllSessionsAndWait,
   stopPersistentSession
-} from "../chunk-JL4XSENX.js";
+} from "../chunk-4CZUEGNQ.js";
 
 // src/lib/manager-worker.ts
 import { createHash } from "crypto";
@@ -1224,6 +1226,8 @@ async function ensureFrameworkBinary(frameworkId) {
 }
 var UPDATE_CHECK_INTERVAL_MS = 5 * 60 * 1e3;
 var selfUpdateUpToDateLogged = false;
+var restartAfterUpgrade = false;
+var pendingUpgradeVersion = null;
 async function checkAndUpdateCli() {
   const cliPath = process.argv[1] ?? "";
   const isHomebrew = cliPath.includes("/Cellar/") || cliPath.includes("/homebrew/");
@@ -1266,7 +1270,9 @@ async function checkAndUpdateCli() {
           timeout: 12e4,
           stdio: "pipe"
         });
-        log(`[self-update] agt CLI upgraded to ${latest}. Restart the manager to use the new version.`);
+        log(`[self-update] agt CLI upgraded to ${latest}. Scheduling manager restart so the new binary takes effect.`);
+        restartAfterUpgrade = true;
+        pendingUpgradeVersion = latest;
       } catch (err) {
         log(`[self-update] Upgrade failed: ${err.message}`);
       }
@@ -1721,6 +1727,7 @@ Automatic restart failed: ${err.message}`
 }
 async function pollCycle() {
   if (!config) return;
+  if (restartAfterUpgrade) return;
   checkAndUpdateCli().catch((err) => log(`[self-update] Check failed: ${err.message}`));
   try {
     registeredAgentsCache.clear();
@@ -1744,7 +1751,7 @@ async function pollCycle() {
     }
     try {
       const { detectHostSecurity } = await import("../host-security-6PDFG7F5.js");
-      const { collectDiagnostics } = await import("../persistent-session-SRUW7AWU.js");
+      const { collectDiagnostics } = await import("../persistent-session-GBQ3VQK3.js");
       const diagCodeNames = [...persistentSessionAgents];
       const agentDiagnostics = diagCodeNames.length > 0 ? collectDiagnostics(diagCodeNames) : void 0;
       let tailscaleHostname;
@@ -3179,8 +3186,7 @@ async function executeAndProcessClaudeTask(codeName, agentId, task, prompt) {
       } catch {
       }
     }
-    const mcpPatterns = serverNames.map((n) => `mcp__${n.replace(/-/g, "_")}__*`);
-    const allowedTools = [...mcpPatterns, "Bash", "Read", "Write", "Edit", "Grep", "Glob"].join(",");
+    const allowedTools = buildAllowedTools(serverNames);
     const claudeArgs = [
       "-p",
       prompt,
@@ -3254,12 +3260,17 @@ async function executeAndProcessClaudeTask(codeName, agentId, task, prompt) {
     claudeSchedulerStates.set(codeName, updated);
   }
 }
-async function processClaudeTaskResult(codeName, agentId, templateId, output, delivery) {
+async function processClaudeTaskResult(codeName, agentId, templateId, rawOutput, delivery) {
   try {
-    if (isSuppressOutput(output)) {
-      const trimmed = (output ?? "").trim();
-      const outputSummary = trimmed.length === 0 ? "<empty>" : trimmed.length > 80 ? `${trimmed.slice(0, 77)}...` : trimmed;
-      log(`[claude-scheduler] Suppressing delivery for '${codeName}' (template=${templateId}, task=${delivery?.taskId ?? "n/a"}) \u2014 output was: ${outputSummary}`);
+    const classification = classifyOutput(rawOutput);
+    if (classification.action === "suppress") {
+      const trimmed = (rawOutput ?? "").trim();
+      const outputHash = trimmed.length === 0 ? "empty" : createHash("sha256").update(trimmed).digest("hex").slice(0, 12);
+      log(`[claude-scheduler] Suppressing delivery for '${codeName}' (template=${templateId}, task=${delivery?.taskId ?? "n/a"}) \u2014 output_len=${trimmed.length} output_hash=${outputHash}`);
+      if (classification.suppressedNotes) {
+        const notesHash = createHash("sha256").update(classification.suppressedNotes).digest("hex").slice(0, 12);
+        log(`[claude-scheduler] Suppressed notes for '${codeName}' (task=${delivery?.taskId ?? "n/a"}) \u2014 notes_len=${classification.suppressedNotes.length} notes_hash=${notesHash}`);
+      }
       if (delivery?.mode === "announce" && delivery.to) {
         await reportDeliveryStatus(agentId, delivery.taskId, {
           status: "skipped",
@@ -3268,6 +3279,10 @@ async function processClaudeTaskResult(codeName, agentId, templateId, output, de
       }
       return;
     }
+    const output = classification.deliverable;
+    if (classification.action === "strip") {
+      log(`[claude-scheduler] Stripped '<no-delivery/>' sentinel from '${codeName}' output (template=${templateId}, task=${delivery?.taskId ?? "n/a"}) \u2014 agent mixed it with real content; delivering the rest.`);
+    }
     if (STANDUP_TEMPLATES.has(templateId)) {
       const standup = parseStandupSummary(output);
       await api.post("/host/agent-status", {
@@ -3789,8 +3804,7 @@ async function processDirectChatMessage(agent, msg) {
         } catch {
         }
       }
-      const mcpPatterns = serverNames.map((n) => `mcp__${n.replace(/-/g, "_")}__*`);
-      const allowedTools = [...mcpPatterns, "Bash", "Read", "Write", "Edit", "Grep", "Glob"].join(",");
+      const allowedTools = buildAllowedTools(serverNames);
       const chatArgs = [
         "-p",
         msg.content,
@@ -5022,7 +5036,21 @@ function startPolling() {
 }
 function scheduleNext() {
   if (!running || !config) return;
+  const restartNow = () => {
+    log(`[self-update] Restarting manager to load upgraded binary (${pendingUpgradeVersion ?? "unknown version"})`);
+    void stopPolling({ forcedExitCode: SUPERVISOR_RESTART_EXIT_CODE }).then(() => {
+      process.exit(SUPERVISOR_RESTART_EXIT_CODE);
+    });
+  };
+  if (restartAfterUpgrade) {
+    restartNow();
+    return;
+  }
   pollTimer = setTimeout(() => {
+    if (restartAfterUpgrade) {
+      restartNow();
+      return;
+    }
     void pollCycle().then(scheduleNext);
   }, config.intervalMs);
 }
@@ -5050,7 +5078,7 @@ async function killAllAgtTmuxSessions() {
   } catch {
   }
 }
-async function stopPolling() {
+async function stopPolling(opts = {}) {
   running = false;
   if (pollTimer) {
     clearTimeout(pollTimer);
@@ -5058,7 +5086,7 @@ async function stopPolling() {
   }
   const shutdownTimer = setTimeout(() => {
     log("Shutdown timeout exceeded (15s), forcing exit");
-    process.exit(1);
+    process.exit(opts.forcedExitCode ?? 1);
   }, 15e3);
   shutdownTimer.unref();
   stopCaffeinate();