@integrity-labs/agt-cli 0.14.4 → 0.14.6

package/dist/lib/manager-worker.js

@@ -18,7 +18,7 @@ import {
   resolveChannels,
   resolveDmTarget,
   wrapScheduledTaskPrompt
-} from "../chunk-UZVDDDH2.js";
+} from "../chunk-5SFAHM2Z.js";
 import {
   findTaskByTemplate,
   getProjectDir,
@@ -1002,6 +1002,7 @@ var knownChannelConfigHashes = /* @__PURE__ */ new Map();
 var knownModels = /* @__PURE__ */ new Map();
 var knownTasksHashes = /* @__PURE__ */ new Map();
 var knownIntegrationHashes = /* @__PURE__ */ new Map();
+var knownManagedMcpHashes = /* @__PURE__ */ new Map();
 var knownSkillHashes = /* @__PURE__ */ new Map();
 var lastCronRunTs = /* @__PURE__ */ new Map();
 var lastWorkTriggerAt = /* @__PURE__ */ new Map();
@@ -1042,6 +1043,7 @@ function clearAgentCaches(agentId, codeName) {
   knownModels.delete(agentId);
   knownTasksHashes.delete(agentId);
   knownIntegrationHashes.delete(agentId);
+  knownManagedMcpHashes.delete(agentId);
   agentDisplayNames.delete(codeName);
   codeNameToAgentId.delete(codeName);
   agentFrameworkCache.delete(codeName);
@@ -1221,6 +1223,7 @@ async function ensureFrameworkBinary(frameworkId) {
   agentRuntimeAuthenticated = await checkClaudeAuth();
 }
 var UPDATE_CHECK_INTERVAL_MS = 5 * 60 * 1e3;
+var selfUpdateUpToDateLogged = false;
 async function checkAndUpdateCli() {
   const cliPath = process.argv[1] ?? "";
   const isHomebrew = cliPath.includes("/Cellar/") || cliPath.includes("/homebrew/");
@@ -1267,8 +1270,9 @@ async function checkAndUpdateCli() {
       } catch (err) {
         log(`[self-update] Upgrade failed: ${err.message}`);
       }
-    } else {
+    } else if (!selfUpdateUpToDateLogged) {
       log(`[self-update] agt CLI is up to date`);
+      selfUpdateUpToDateLogged = true;
     }
   } catch (err) {
     log(`[self-update] brew outdated failed: ${err.message}`);
@@ -1717,6 +1721,7 @@ Automatic restart failed: ${err.message}`
 }
 async function pollCycle() {
   if (!config) return;
+  checkAndUpdateCli().catch((err) => log(`[self-update] Check failed: ${err.message}`));
   try {
     registeredAgentsCache.clear();
     gatewaysStartedThisCycle.clear();
@@ -2438,27 +2443,42 @@ async function processAgent(agent, agentStates) {
           });
           log(`[hot-reload] Notified '${agent.code_name}' about integration update: ${names}`);
         }
-        const managedIntegrations = integrations.filter((i) => i.auth_type === "managed");
-        if (managedIntegrations.length > 0 && frameworkAdapter.writeMcpServer) {
-          try {
-            const toolkitData = await api.post("/host/managed-toolkits", { agent_ids: [agent.agent_id] });
-            const expectedServerIds = /* @__PURE__ */ new Set();
-            for (const tk of toolkitData.toolkits) {
-              if (tk.agent_id !== agent.agent_id) continue;
-              const serverId = tk.toolkit_id.replace(/[^a-z0-9]/gi, "_").toLowerCase();
-              expectedServerIds.add(serverId);
-              const mcpUrl = tk.mcp_url;
-              const mcpHeaders = tk.mcp_headers;
-              const proxyUrl = tk.proxy_url.startsWith("/") ? `${requireHost()}${tk.proxy_url}` : tk.proxy_url;
-              const url = mcpUrl ?? proxyUrl;
-              frameworkAdapter.writeMcpServer(agent.code_name, serverId, { url, headers: mcpHeaders });
-              log(`[managed-toolkit] ${agent.code_name}: ${tk.toolkit_name} \u2192 ${url}`);
-            }
-            if (frameworkAdapter.removeMcpServer) {
+        needsGatewayRestart = true;
+      }
+    }
+    const fwForMcp = agentFrameworkCache.get(agent.code_name) ?? "openclaw";
+    if (frameworkAdapter.writeMcpServer) {
+      try {
+        const toolkitData = await api.post("/host/managed-toolkits", { agent_ids: [agent.agent_id] });
+        const agentToolkits = (toolkitData.toolkits ?? []).filter((tk) => tk.agent_id === agent.agent_id);
+        const expectedServerIds = /* @__PURE__ */ new Set();
+        const desiredEntries = [];
+        for (const tk of agentToolkits) {
+          const serverId = tk.toolkit_id.replace(/[^a-z0-9]/gi, "_").toLowerCase();
+          expectedServerIds.add(serverId);
+          const mcpUrl = tk.mcp_url;
+          const mcpHeaders = tk.mcp_headers;
+          const proxyUrl = tk.proxy_url.startsWith("/") ? `${requireHost()}${tk.proxy_url}` : tk.proxy_url;
+          const url = mcpUrl ?? proxyUrl;
+          desiredEntries.push({ serverId, url, headers: mcpHeaders, name: tk.toolkit_name });
+        }
+        const hashBasis = desiredEntries.slice().sort((a, b) => a.serverId.localeCompare(b.serverId)).map((e) => {
+          const headersHash = createHash("sha256").update(canonicalJson(e.headers ?? {})).digest("hex").slice(0, 16);
+          return `${e.serverId}|${e.url}|${headersHash}`;
+        }).join("\n");
+        const mcpHash = createHash("sha256").update(hashBasis).digest("hex").slice(0, 16);
+        const prevMcpHash = knownManagedMcpHashes.get(agent.agent_id);
+        if (mcpHash !== prevMcpHash) {
+          for (const e of desiredEntries) {
+            frameworkAdapter.writeMcpServer(agent.code_name, e.serverId, { url: e.url, headers: e.headers });
+            const urlHash = createHash("sha256").update(e.url).digest("hex").slice(0, 12);
+            log(`[managed-toolkit] ${agent.code_name}: wrote '${e.name}' (serverId=${e.serverId}, url_hash=${urlHash})`);
+          }
+          if (frameworkAdapter.removeMcpServer && frameworkAdapter.getMcpPath) {
+            const mcpPath = frameworkAdapter.getMcpPath(agent.code_name);
+            if (mcpPath) {
               try {
                 const { readFileSync: readFileSync2 } = await import("fs");
-                const { join: join3 } = await import("path");
-                const mcpPath = join3(frameworkAdapter.getAgentDir(agent.code_name), "provision", ".mcp.json");
                 const mcpConfig = JSON.parse(readFileSync2(mcpPath, "utf-8"));
                 if (mcpConfig.mcpServers) {
                   const managedPrefixes = [
@@ -2467,7 +2487,6 @@ async function processAgent(agent, agentStates) {
                     "pipedream_",
                     "nango_",
                     "paragon_",
-                    // Legacy hyphenated format
                     "composio-",
                     "one-",
                     "pipedream-",
@@ -2484,30 +2503,31 @@ async function processAgent(agent, agentStates) {
               } catch {
               }
             }
-          } catch (err) {
-            log(`[managed-toolkit] Failed to provision for '${agent.code_name}': ${err.message}`);
           }
-        }
-        needsGatewayRestart = true;
-        if (managedIntegrations.length > 0 && fw === "claude-code" && isSessionHealthy(agent.code_name)) {
-          const mcpNames = managedIntegrations.map((i) => i.display_name || i.definition_id).join(", ");
-          log(`[hot-reload] MCP servers changed for '${agent.code_name}': ${mcpNames} \u2014 restarting session`);
-          const delivered = await injectMessage(
-            agent.code_name,
-            "system",
-            `New MCP tool servers have been configured: ${mcpNames}. Note: MCP servers require a session restart to connect. Your manager will restart your session shortly.`,
-            { task_name: "mcp-update" },
-            log
-          ).catch(() => false);
-          const delay = delivered ? 8e3 : 3e3;
-          if (!delivered) {
-            log(`[hot-reload] Inject notification unconfirmed for '${agent.code_name}' \u2014 proceeding with shorter delay`);
+          knownManagedMcpHashes.set(agent.agent_id, mcpHash);
+          if (prevMcpHash !== void 0 && fwForMcp === "claude-code" && isSessionHealthy(agent.code_name)) {
+            const mcpNames = agentToolkits.map((tk) => tk.toolkit_name).join(", ") || "none (all removed)";
+            log(`[hot-reload] MCP servers changed for '${agent.code_name}': ${mcpNames} \u2014 restarting session`);
+            const restartNotice = agentToolkits.length > 0 ? `New MCP tool servers have been configured: ${mcpNames}. Note: MCP servers require a session restart to connect. Your manager will restart your session shortly.` : "Managed MCP tool servers were removed. Your manager will restart your session shortly so the session drops those tools.";
+            const delivered = await injectMessage(
+              agent.code_name,
+              "system",
+              restartNotice,
+              { task_name: "mcp-update" },
+              log
+            ).catch(() => false);
+            const delay = delivered ? 8e3 : 3e3;
+            if (!delivered) {
+              log(`[hot-reload] Inject notification unconfirmed for '${agent.code_name}' \u2014 proceeding with shorter delay`);
+            }
+            setTimeout(() => {
+              stopPersistentSession(agent.code_name, log);
+              log(`[hot-reload] Session stopped for '${agent.code_name}' \u2014 will respawn with new MCP servers`);
+            }, delay);
           }
-          setTimeout(() => {
-            stopPersistentSession(agent.code_name, log);
-            log(`[hot-reload] Session stopped for '${agent.code_name}' \u2014 will respawn with new MCP servers`);
-          }, delay);
         }
+      } catch (err) {
+        log(`[managed-toolkit] Failed to provision for '${agent.code_name}': ${err.message}`);
       }
     }
   } catch (err) {
@@ -2908,15 +2928,14 @@ In progress for ${age} minutes \u2014 auto-failed`).catch(() => {
   }
   const trackedFiles = frameworkAdapter.driftTrackedFiles();
   if (trackedFiles.length > 0 && existsSync(agentDir)) {
-    let hashes = writtenHashes.get(agent.agent_id);
-    if (!hashes) {
-      hashes = /* @__PURE__ */ new Map();
-      writtenHashes.set(agent.agent_id, hashes);
-    }
+    const hashes = /* @__PURE__ */ new Map();
     for (const file of trackedFiles) {
       const h = hashFile(join2(agentDir, file));
       if (h) hashes.set(file, h);
     }
+    writtenHashes.set(agent.agent_id, hashes);
+  } else {
+    writtenHashes.delete(agent.agent_id);
   }
   agentStates.push({
     agentId: agent.agent_id,
@@ -4993,7 +5012,6 @@ function startPolling() {
   running = true;
   void startCaffeinate();
   log(`Starting poll loop (interval=${config.intervalMs}ms, configDir=${config.configDir})`);
-  checkAndUpdateCli().catch((err) => log(`[self-update] Check failed: ${err.message}`));
   void killAllAgtTmuxSessions().catch(() => {
   }).then(() => migrateToProfiles()).then(() => {
     startGatewayPool();