@integrity-labs/agt-cli 0.28.606 → 0.28.607

package/dist/bin/agt.js

@@ -40,7 +40,7 @@ import {
   success,
   table,
   warn
-} from "../chunk-2JB2MJAV.js";
+} from "../chunk-DYADWFBV.js";
 import {
   readDirectChatSessionState
 } from "../chunk-DESCRVVM.js";
@@ -4909,7 +4909,7 @@ import { execFileSync, execSync } from "child_process";
 import { existsSync as existsSync10, realpathSync as realpathSync2 } from "fs";
 import chalk18 from "chalk";
 import ora16 from "ora";
-var cliVersion = true ? "0.28.606" : "dev";
+var cliVersion = true ? "0.28.607" : "dev";
 async function fetchLatestVersion() {
   const host2 = getHost();
   if (!host2) return null;
@@ -6089,7 +6089,7 @@ function handleError(err) {
 }
 
 // src/bin/agt.ts
-var cliVersion2 = true ? "0.28.606" : "dev";
+var cliVersion2 = true ? "0.28.607" : "dev";
 var program = new Command();
 program.name("agt").description("Augmented CLI \u2014 agent provisioning and management").version(cliVersion2).option("--json", "Emit machine-readable JSON output (suppress spinners and colors)").option("--skip-update-check", "Skip the automatic update check on startup");
 program.hook("preAction", async (thisCommand, actionCommand) => {

package/dist/{chunk-2JB2MJAV.js → chunk-DYADWFBV.js}

@@ -6030,7 +6030,7 @@ function exchangeFailureKind(err) {
 }
 
 // src/lib/api-client.ts
-var agtCliVersion = true ? "0.28.606" : "dev";
+var agtCliVersion = true ? "0.28.607" : "dev";
 var lastConfigHash = null;
 function setConfigHash(hash) {
   lastConfigHash = hash && hash.length > 0 ? hash : null;
@@ -9505,4 +9505,4 @@ export {
   managerInstallSystemUnitCommand,
   managerUninstallSystemUnitCommand
 };
-//# sourceMappingURL=chunk-2JB2MJAV.js.map
+//# sourceMappingURL=chunk-DYADWFBV.js.map

package/dist/lib/manager-worker.js

@@ -53,7 +53,7 @@ import {
   safeWriteJsonAtomic,
   setConfigHash,
   tripClass
-} from "../chunk-2JB2MJAV.js";
+} from "../chunk-DYADWFBV.js";
 import {
   getProjectDir as getProjectDir2,
   getReadyTasks,
@@ -11880,7 +11880,7 @@ var agentRestartTimezoneInputs = /* @__PURE__ */ new Map();
 var lastVersionCheckAt = 0;
 var VERSION_CHECK_INTERVAL_MS = 5 * 60 * 1e3;
 var lastResponsivenessProbeAt = 0;
-var agtCliVersion = true ? "0.28.606" : "dev";
+var agtCliVersion = true ? "0.28.607" : "dev";
 function resolveBrewPath(execFileSync3) {
   try {
     const out = execFileSync3("which", ["brew"], { timeout: 5e3 }).toString().trim();
@@ -13823,6 +13823,7 @@ async function pollCycle() {
     const agentStates = [];
     const fastRespawn = hostFlagStore().getBoolean("fast-mcp-restart-respawn");
     const processOrder = fastRespawn ? reorderRestartedFirst(agents, new Set(restartAcks.keys())) : agents;
+    const managedToolkits = createManagedToolkitReader(agents.map((a) => a.agent_id));
     for (const agent of processOrder) {
       if (restartInFlight.has(agent.agent_id)) {
         const existing = state7.agents.find((a) => a.agentId === agent.agent_id);
@@ -13832,7 +13833,7 @@ async function pollCycle() {
         }
       }
       try {
-        await processAgent(agent, agentStates);
+        await processAgent(agent, agentStates, managedToolkits);
       } catch (err) {
         log(`Error processing agent '${agent.code_name}': ${err.message}`);
         const existing = state7.agents.find((a) => a.agentId === agent.agent_id);
@@ -14069,7 +14070,45 @@ async function getOrCacheRegisteredAgents(adapter, profile) {
   }
   return cached;
 }
-async function processAgent(agent, agentStates) {
+async function fetchManagedToolkits(agentIds) {
+  const data = await api.post(
+    "/host/managed-toolkits",
+    { agent_ids: [...agentIds] }
+  );
+  return data.toolkits ?? [];
+}
+function createManagedToolkitReader(agentIds, fetchToolkits = fetchManagedToolkits) {
+  const requested = [...new Set(agentIds)];
+  let batch = null;
+  const loadBatch = () => {
+    batch ??= (async () => {
+      if (requested.length === 0) return /* @__PURE__ */ new Map();
+      try {
+        const toolkits = await fetchToolkits(requested);
+        const byAgent = new Map(
+          requested.map((id) => [id, []])
+        );
+        for (const tk of toolkits) byAgent.get(tk.agent_id)?.push(tk);
+        return byAgent;
+      } catch (err) {
+        log(
+          `[managed-toolkit] batch fetch for ${requested.length} agent(s) failed: ${err.message} \u2014 falling back to per-agent fetches this cycle (ENG-9057)`
+        );
+        return null;
+      }
+    })();
+    return batch;
+  };
+  return {
+    async forAgent(agentId) {
+      const byAgent = await loadBatch();
+      const fromBatch = byAgent?.get(agentId);
+      if (fromBatch) return fromBatch;
+      return (await fetchToolkits([agentId])).filter((tk) => tk.agent_id === agentId);
+    }
+  };
+}
+async function processAgent(agent, agentStates, managedToolkits) {
   if (!config) return;
   const renderIntegrationsSection = shouldRenderIntegrationsSection();
   syncAccountEnforcementMarker(
@@ -15353,8 +15392,7 @@ async function processAgent(agent, agentStates) {
     const fwForMcp = agentFrameworkCache.get(agent.code_name) ?? DEFAULT_FRAMEWORK;
     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 agentToolkits = await managedToolkits.forAgent(agent.agent_id);
         const expectedServerIds = /* @__PURE__ */ new Set();
         const desiredEntries = [];
         for (const tk of agentToolkits) {
@@ -18679,6 +18717,7 @@ export {
   claudeCodeUpgradeMarkerPath,
   claudeCodeUpgradeThrottled,
   claudeManagedSettingsPath,
+  createManagedToolkitReader,
   dashboardRestartBreakerReason,
   decideDayRolloverAction,
   ensureClaudeManagedSettings,