@desplega.ai/agent-swarm 1.98.0 → 1.99.0

package/src/tests/slack-blocks.test.ts

@@ -3,6 +3,7 @@ import {
   buildAssignmentSummaryBlocks,
   buildBufferFlushBlocks,
   buildCancelledBlocks,
+  buildCompletedBlockBatches,
   buildCompletedBlocks,
   buildFailedBlocks,
   buildProgressBlocks,
@@ -20,6 +21,7 @@ describe("markdownToSlack", () => {
     // **hello** → *hello* (Slack bold)
     expect(markdownToSlack("**hello**")).toBe("*hello*");
     expect(markdownToSlack("**hello world**")).toBe("*hello world*");
+    expect(markdownToSlack("__hello world__")).toBe("*hello world*");
   });
 
   test("converts italic", () => {
@@ -31,7 +33,11 @@ describe("markdownToSlack", () => {
   });
 
   test("converts links", () => {
-    expect(markdownToSlack("[click](https://example.com)")).toBe("<https://example.com|click>");
+    expect(markdownToSlack("[click](https://example.com)")).toBe("click (https://example.com)");
+    expect(markdownToSlack("![diagram](https://example.com/a.png)")).toBe(
+      "diagram (https://example.com/a.png)",
+    );
+    expect(markdownToSlack("[click](https://example.com)")).not.toContain("<https://");
   });
 
   test("converts headers to bold", () => {
@@ -163,6 +169,27 @@ describe("buildCompletedBlocks", () => {
     const totalText = bodySections.map((s) => s.text.text).join("");
     expect(totalText).toBe(longBody);
   });
+
+  test("batches very long body across multiple Slack messages without losing text", () => {
+    const longBody = "x".repeat(140_000);
+    const batches = buildCompletedBlockBatches({
+      agentName: "Alpha",
+      taskId: "abcdef12-3456-7890-abcd-ef1234567890",
+      body: longBody,
+    });
+
+    expect(batches.length).toBeGreaterThan(1);
+    for (const batch of batches) {
+      expect(batch.length).toBeLessThanOrEqual(45);
+    }
+
+    const bodyText = batches
+      .flatMap((batch) => batch.slice(1))
+      .map((block) => block.text.text)
+      .join("");
+    expect(bodyText).toBe(longBody);
+    expect(batches[1][0].text.text).toContain("continued · part 2");
+  });
 });
 
 describe("buildFailedBlocks", () => {
@@ -908,6 +935,26 @@ describe("buildTreeBlocks", () => {
     expect(text).toContain("Result: 42");
   });
 
+  test("completed root preview converts markdown and marks truncation explicitly", () => {
+    const root: TreeNode = {
+      taskId: makeTaskId("oooo0002"),
+      agentName: "Solo",
+      status: "completed",
+      slackReplySent: false,
+      output: `### Summary\n\n**Not broken** — ${"Automatic review on plain PR creation ".repeat(8)}`,
+      children: [],
+    };
+
+    const blocks = buildTreeBlocks([root]);
+    const text = blocks[0].text.text;
+
+    expect(text).toContain("*Summary*");
+    expect(text).toContain("*Not broken*");
+    expect(text).not.toContain("###");
+    expect(text).not.toContain("**Not broken**");
+    expect(text).toContain("more chars; full output in thread");
+  });
+
   test("completed root with slackReplySent suppresses output", () => {
     const root: TreeNode = {
       taskId: makeTaskId("pppp0001"),

package/src/tools/memory-get.ts

@@ -1,6 +1,7 @@
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import * as z from "zod";
 import { getMemoryStore } from "@/be/memory";
+import { recordRetrievals } from "@/be/memory/raters/retrieval";
 import { createToolRegistrar } from "@/tools/utils";
 import type { AgentMemorySource } from "@/types";
 import { AgentMemorySchema } from "@/types";
@@ -18,6 +19,12 @@ export const registerMemoryGetTool = (server: McpServer) => {
 
       inputSchema: z.object({
         memoryId: z.uuid().describe("The ID of the memory to retrieve."),
+        intent: z
+          .string()
+          .min(1)
+          .describe(
+            "Why you are retrieving this memory. Required. E.g. 'need full details of the auth fix pattern'.",
+          ),
       }),
       outputSchema: z.object({
         yourAgentId: z.string().uuid().optional(),
@@ -27,7 +34,7 @@ export const registerMemoryGetTool = (server: McpServer) => {
         rateHint: z.string().optional(),
       }),
     },
-    async ({ memoryId }, requestInfo, _meta) => {
+    async ({ memoryId, intent }, requestInfo, _meta) => {
       const memory = getMemoryStore().get(memoryId);
 
       if (!memory) {
@@ -41,6 +48,20 @@ export const registerMemoryGetTool = (server: McpServer) => {
         };
       }
 
+      if (requestInfo.sourceTaskId && requestInfo.agentId) {
+        try {
+          recordRetrievals(
+            requestInfo.sourceTaskId,
+            requestInfo.agentId,
+            [{ memoryId: memory.id, similarity: 1.0 }],
+            requestInfo.sessionId,
+            { intent, contextKey: requestInfo.contextKey, eventType: "get" },
+          );
+        } catch (err) {
+          console.error("[memory-get] recordRetrievals failed:", (err as Error).message);
+        }
+      }
+
       const inTaskContext = !!requestInfo.sourceTaskId;
       const rateHint =
         inTaskContext && NUDGE_ELIGIBLE_SOURCES.has(memory.source as AgentMemorySource)

package/src/tools/memory-search.ts

@@ -26,6 +26,12 @@ export const registerMemorySearchTool = (server: McpServer) => {
 
       inputSchema: z.object({
         query: z.string().min(1).describe("Natural language search query."),
+        intent: z
+          .string()
+          .min(1)
+          .describe(
+            "Why you are searching for this memory. Required. E.g. 'looking for auth pattern to fix login bug'.",
+          ),
         scope: z
           .enum(["all", "agent", "swarm"])
           .default("all")
@@ -56,7 +62,7 @@ export const registerMemorySearchTool = (server: McpServer) => {
         _ratingNudge: z.string().optional(),
       }),
     },
-    async ({ query, scope, limit, source }, requestInfo, _meta) => {
+    async ({ query, intent, scope, limit, source }, requestInfo, _meta) => {
       if (!requestInfo.agentId) {
         return {
           content: [{ type: "text", text: "Agent ID required for memory search." }],
@@ -97,6 +103,7 @@ export const registerMemorySearchTool = (server: McpServer) => {
               requestInfo.agentId,
               ranked.map((r) => ({ memoryId: r.id, similarity: r.similarity })),
               requestInfo.sessionId,
+              { intent, contextKey: requestInfo.contextKey, eventType: "search" },
             );
           } catch (err) {
             console.error("[memory-search] recordRetrievals failed:", (err as Error).message);

package/src/tools/utils.ts

@@ -21,11 +21,13 @@ export type RequestInfo = {
   sessionId: string | undefined;
   agentId: string | undefined;
   sourceTaskId: string | undefined;
+  contextKey: string | undefined;
 };
 
 export const getRequestInfo = (req: Meta): RequestInfo => {
   const agentIdHeader = req.requestInfo?.headers?.["x-agent-id"];
   const sourceTaskIdHeader = req.requestInfo?.headers?.["x-source-task-id"];
+  const contextKeyHeader = req.requestInfo?.headers?.["x-context-key"];
 
   let agentId: string | undefined;
   if (Array.isArray(agentIdHeader)) {
@@ -41,10 +43,18 @@ export const getRequestInfo = (req: Meta): RequestInfo => {
     sourceTaskId = sourceTaskIdHeader;
   }
 
+  let contextKey: string | undefined;
+  if (Array.isArray(contextKeyHeader)) {
+    contextKey = contextKeyHeader?.[0];
+  } else if (typeof contextKeyHeader === "string") {
+    contextKey = contextKeyHeader;
+  }
+
   return {
     sessionId: req.sessionId || undefined,
     agentId,
     sourceTaskId,
+    contextKey,
   };
 };
 

package/src/types.ts

@@ -725,6 +725,8 @@ export const AgentLogEventTypeSchema = z.enum([
   "budget.deleted",
   "pricing.inserted",
   "pricing.deleted",
+  "pricing.refresh",
+  "pricing.refresh.failed",
   // Graceful pause/resume via follow-up
   "task_superseded",
 ]);

package/src/utils/internal-ai/complete-structured.ts

@@ -78,7 +78,8 @@ function stripJsonFences(raw: string): string {
   return fenced?.[1] ? fenced[1].trim() : trimmed;
 }
 
-async function defaultSpawnClaudeCli(
+/** Exported for tests only — production callers go through `completeStructured`. */
+export async function defaultSpawnClaudeCli(
   prompt: string,
   model: string,
   signal?: AbortSignal,
@@ -105,6 +106,14 @@ async function defaultSpawnClaudeCli(
   if (!env.CLAUDE_CODE_OAUTH_TOKEN && env.AGENT_SWARM_CLAUDE_OAUTH_TOKEN) {
     env.CLAUDE_CODE_OAUTH_TOKEN = env.AGENT_SWARM_CLAUDE_OAUTH_TOKEN;
   }
+  // Recursion guard: this shellout is itself a full claude session — on exit it
+  // fires the same global Stop hook, whose session-summary path would spawn
+  // another `claude -p`, recursively (each level holds a ~0.5-1GB node process;
+  // observed OOM-wedging 8GB E2B worker sandboxes within ~90s, leaving 10+
+  // near-identical summarizer transcripts in ~/.claude/projects). The hook's
+  // `runStopHookSessionSummary` honors this flag — same convention as the
+  // `claude -p` shellout in `src/be/memory/raters/llm-client.ts`.
+  env.SKIP_SESSION_SUMMARY = "1";
   const proc = Bun.spawn({
     cmd,
     env,

package/tsconfig.json

@@ -34,6 +34,7 @@
     "noPropertyAccessFromIndexSignature": false
   },
   "exclude": [
+    "evals",
     "ui",
     "templates-ui",
     "node_modules",