@desplega.ai/agent-swarm 1.98.1 → 1.99.0

package/src/tests/scripts-mcp-e2e.test.ts

@@ -330,7 +330,7 @@ describe("script_ MCP HTTP proxy tools", () => {
     const tools = buildToolServer();
     const source = `
       import type { ScriptContext, SwarmSdk } from "swarm-sdk";
-      const compileOnly = (swarm: SwarmSdk) => swarm.memory_search({ query: "foo" });
+      const compileOnly = (swarm: SwarmSdk) => swarm.memory_search({ query: "foo", intent: "test" });
       export default async (_args: unknown, ctx: ScriptContext) => {
         void compileOnly;
         return { hasMemorySearch: typeof ctx.swarm.memory_search === "function" };

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/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",