@inkeep/agents-run-api 0.19.6 → 0.19.8

package/dist/index.js

@@ -2250,8 +2250,14 @@ var AgentSession = class {
         return `Agent ${event.subAgentId} generating response`;
       case "agent_reasoning":
         return `Agent ${event.subAgentId} reasoning through request`;
-      case "tool_execution":
-        return `Tool execution: ${event.data.toolName || "unknown"}`;
+      case "tool_call":
+        return `Tool call: ${event.data.toolName || "unknown"}`;
+      case "tool_result": {
+        const status = event.data.error ? "failed" : "completed";
+        return `Tool result: ${event.data.toolName || "unknown"} (${status})`;
+      }
+      case "error":
+        return `Error: ${event.data.message}`;
       case "transfer":
         return `Agent transfer: ${event.data.fromSubAgent} \u2192 ${event.data.targetSubAgent}`;
       case "delegation_sent":
@@ -2972,15 +2978,29 @@ ${this.statusUpdateState?.config.prompt?.trim() || ""}`;
     const activities = [];
     for (const event of events) {
       switch (event.eventType) {
-        case "tool_execution": {
-          const resultStr = JSON.stringify(event.data.result);
+        case "tool_call": {
+          activities.push(
+            `\u{1F527} **${event.data.toolName}** (called)
+   \u{1F4E5} Input: ${JSON.stringify(event.data.args)}`
+          );
+          break;
+        }
+        case "tool_result": {
+          const resultStr = event.data.error ? `\u274C Error: ${event.data.error}` : JSON.stringify(event.data.result);
           activities.push(
             `\u{1F527} **${event.data.toolName}** ${event.data.duration ? `(${event.data.duration}ms)` : ""}
-   \u{1F4E5} Input: ${JSON.stringify(event.data.args)}
    \u{1F4E4} Output: ${resultStr}`
           );
           break;
         }
+        case "error": {
+          activities.push(
+            `\u274C **Error**: ${event.data.message}
+   \u{1F50D} Code: ${event.data.code || "unknown"}
+   \u{1F4CA} Severity: ${event.data.severity || "error"}`
+          );
+          break;
+        }
         // INTERNAL OPERATIONS - DO NOT EXPOSE TO STATUS UPDATES
         case "transfer":
         case "delegation_sent":
@@ -3065,7 +3085,7 @@ ${this.statusUpdateState?.config.prompt?.trim() || ""}`;
             }
           });
           const toolCallEvent = this.events.find(
-            (event) => event.eventType === "tool_execution" && event.data && "toolId" in event.data && event.data.toolId === artifactData.metadata?.toolCallId
+            (event) => event.eventType === "tool_result" && event.data && "toolCallId" in event.data && event.data.toolCallId === artifactData.metadata?.toolCallId
           );
           const toolContext = toolCallEvent ? {
             toolName: toolCallEvent.data.toolName,
@@ -6663,14 +6683,22 @@ var Agent = class {
           });
         }
         const isInternalTool = toolName.includes("save_tool_result") || toolName.includes("thinking_complete") || toolName.startsWith("transfer_to_") || toolName.startsWith("delegate_to_");
+        if (streamRequestId && !isInternalTool) {
+          agentSessionManager.recordEvent(streamRequestId, "tool_call", this.config.id, {
+            toolName,
+            args,
+            toolCallId: context?.toolCallId,
+            toolId
+          });
+        }
         try {
           const result = await originalExecute(args, context);
           const duration = Date.now() - startTime;
           if (streamRequestId && !isInternalTool) {
-            agentSessionManager.recordEvent(streamRequestId, "tool_execution", this.config.id, {
+            agentSessionManager.recordEvent(streamRequestId, "tool_result", this.config.id, {
               toolName,
-              args,
               result,
+              toolCallId: context?.toolCallId,
               toolId,
               duration
             });
@@ -6680,12 +6708,13 @@ var Agent = class {
           const duration = Date.now() - startTime;
           const errorMessage = error instanceof Error ? error.message : "Unknown error";
           if (streamRequestId && !isInternalTool) {
-            agentSessionManager.recordEvent(streamRequestId, "tool_execution", this.config.id, {
+            agentSessionManager.recordEvent(streamRequestId, "tool_result", this.config.id, {
               toolName,
-              args,
-              result: { error: errorMessage },
+              result: null,
+              toolCallId: context?.toolCallId,
               toolId,
-              duration
+              duration,
+              error: errorMessage
             });
           }
           throw error;
@@ -6954,13 +6983,14 @@ var Agent = class {
           },
           (span) => {
             setSpanWithError(span, new Error(`0 effective tools available for ${tool3.name}`));
-            agentSessionManager.recordEvent(streamRequestId, "tool_execution", this.config.id, {
-              toolName: tool3.name,
-              args: { operation: "mcp_tool_discovery" },
-              result: {
-                status: "no_tools_available",
-                message: `MCP server has 0 effective tools. Double check the selected tools in your agent and the active tools in the MCP server configuration.`,
-                serverUrl: tool3.config.type === "mcp" ? tool3.config.mcp.server.url : "unknown"
+            agentSessionManager.recordEvent(streamRequestId, "error", this.config.id, {
+              message: `MCP server has 0 effective tools. Double check the selected tools in your graph and the active tools in the MCP server configuration.`,
+              code: "no_tools_available",
+              severity: "error",
+              context: {
+                toolName: tool3.name,
+                serverUrl: tool3.config.type === "mcp" ? tool3.config.mcp.server.url : "unknown",
+                operation: "mcp_tool_discovery"
               }
             });
             span.end();
@@ -7016,7 +7046,7 @@ var Agent = class {
       if (functionToolsData.length === 0) {
         return functionTools;
       }
-      const { LocalSandboxExecutor } = await import('./LocalSandboxExecutor-2UQ32ZZH.js');
+      const { LocalSandboxExecutor } = await import('./LocalSandboxExecutor-S6BPJMJV.js');
       const sandboxExecutor = LocalSandboxExecutor.getInstance();
       for (const functionToolDef of functionToolsData) {
         const functionId = functionToolDef.functionId;
@@ -7785,7 +7815,9 @@ var Agent = class {
               if (steps.length >= 2) {
                 const previousStep = steps[steps.length - 2];
                 if (previousStep && "toolCalls" in previousStep && previousStep.toolCalls) {
-                  const hasTransferCall = previousStep.toolCalls.some((tc) => tc.toolName.startsWith("transfer_to_"));
+                  const hasTransferCall = previousStep.toolCalls.some(
+                    (tc) => tc.toolName.startsWith("transfer_to_")
+                  );
                   if (hasTransferCall && "toolResults" in previousStep && previousStep.toolResults) {
                     return true;
                   }
@@ -10084,174 +10116,191 @@ app2.openapi(chatCompletionsRoute, async (c) => {
     );
     const body = c.get("requestBody") || {};
     const conversationId = body.conversationId || getConversationId();
-    const fullAgent = await getFullAgent(dbClient_default)({
-      scopes: { tenantId, projectId, agentId }
-    });
-    let agent;
-    let defaultSubAgentId;
-    if (fullAgent) {
-      agent = {
-        id: fullAgent.id,
-        name: fullAgent.name,
-        tenantId,
-        projectId,
-        defaultSubAgentId: fullAgent.defaultSubAgentId
-      };
-      const agentKeys = Object.keys(fullAgent.subAgents || {});
-      const firstAgentId = agentKeys.length > 0 ? agentKeys[0] : "";
-      defaultSubAgentId = fullAgent.defaultSubAgentId || firstAgentId;
-    } else {
-      agent = await getAgentWithDefaultSubAgent(dbClient_default)({
+    const activeSpan = trace.getActiveSpan();
+    if (activeSpan) {
+      activeSpan.setAttributes({
+        "conversation.id": conversationId,
+        "tenant.id": tenantId,
+        "agent.id": agentId,
+        "project.id": projectId
+      });
+    }
+    let currentBag = propagation.getBaggage(context.active());
+    if (!currentBag) {
+      currentBag = propagation.createBaggage();
+    }
+    currentBag = currentBag.setEntry("conversation.id", { value: conversationId });
+    const ctxWithBaggage = propagation.setBaggage(context.active(), currentBag);
+    return await context.with(ctxWithBaggage, async () => {
+      const fullAgent = await getFullAgent(dbClient_default)({
         scopes: { tenantId, projectId, agentId }
       });
-      if (!agent) {
+      let agent;
+      let defaultSubAgentId;
+      if (fullAgent) {
+        agent = {
+          id: fullAgent.id,
+          name: fullAgent.name,
+          tenantId,
+          projectId,
+          defaultSubAgentId: fullAgent.defaultSubAgentId
+        };
+        const agentKeys = Object.keys(fullAgent.subAgents || {});
+        const firstAgentId = agentKeys.length > 0 ? agentKeys[0] : "";
+        defaultSubAgentId = fullAgent.defaultSubAgentId || firstAgentId;
+      } else {
+        agent = await getAgentWithDefaultSubAgent(dbClient_default)({
+          scopes: { tenantId, projectId, agentId }
+        });
+        if (!agent) {
+          throw createApiError({
+            code: "not_found",
+            message: "Agent not found"
+          });
+        }
+        defaultSubAgentId = agent.defaultSubAgentId || "";
+      }
+      if (!defaultSubAgentId) {
         throw createApiError({
           code: "not_found",
-          message: "Agent not found"
+          message: "No default agent found in agent"
         });
       }
-      defaultSubAgentId = agent.defaultSubAgentId || "";
-    }
-    if (!defaultSubAgentId) {
-      throw createApiError({
-        code: "not_found",
-        message: "No default agent found in agent"
+      await createOrGetConversation(dbClient_default)({
+        tenantId,
+        projectId,
+        id: conversationId,
+        activeSubAgentId: defaultSubAgentId
       });
-    }
-    await createOrGetConversation(dbClient_default)({
-      tenantId,
-      projectId,
-      id: conversationId,
-      activeSubAgentId: defaultSubAgentId
-    });
-    const activeAgent = await getActiveAgentForConversation(dbClient_default)({
-      scopes: { tenantId, projectId },
-      conversationId
-    });
-    if (!activeAgent) {
-      setActiveAgentForConversation(dbClient_default)({
+      const activeAgent = await getActiveAgentForConversation(dbClient_default)({
         scopes: { tenantId, projectId },
-        conversationId,
-        subAgentId: defaultSubAgentId
+        conversationId
       });
-    }
-    const subAgentId = activeAgent?.activeSubAgentId || defaultSubAgentId;
-    const agentInfo = await getSubAgentById(dbClient_default)({
-      scopes: { tenantId, projectId, agentId },
-      subAgentId
-    });
-    if (!agentInfo) {
-      throw createApiError({
-        code: "not_found",
-        message: "Agent not found"
+      if (!activeAgent) {
+        setActiveAgentForConversation(dbClient_default)({
+          scopes: { tenantId, projectId },
+          conversationId,
+          subAgentId: defaultSubAgentId
+        });
+      }
+      const subAgentId = activeAgent?.activeSubAgentId || defaultSubAgentId;
+      const agentInfo = await getSubAgentById(dbClient_default)({
+        scopes: { tenantId, projectId, agentId },
+        subAgentId
       });
-    }
-    const validatedContext = c.get("validatedContext") || body.headers || {};
-    const credentialStores = c.get("credentialStores");
-    await handleContextResolution({
-      tenantId,
-      projectId,
-      agentId,
-      conversationId,
-      headers: validatedContext,
-      dbClient: dbClient_default,
-      credentialStores
-    });
-    logger21.info(
-      {
+      if (!agentInfo) {
+        throw createApiError({
+          code: "not_found",
+          message: "Agent not found"
+        });
+      }
+      const validatedContext = c.get("validatedContext") || body.headers || {};
+      const credentialStores = c.get("credentialStores");
+      await handleContextResolution({
         tenantId,
         projectId,
         agentId,
         conversationId,
-        defaultSubAgentId,
-        activeSubAgentId: activeAgent?.activeSubAgentId || "none",
-        hasContextConfig: !!agent.contextConfigId,
-        hasHeaders: !!body.headers,
-        hasValidatedContext: !!validatedContext,
-        validatedContextKeys: Object.keys(validatedContext)
-      },
-      "parameters"
-    );
-    const requestId2 = `chatcmpl-${Date.now()}`;
-    const timestamp = Math.floor(Date.now() / 1e3);
-    const lastUserMessage = body.messages.filter((msg) => msg.role === "user").slice(-1)[0];
-    const userMessage = lastUserMessage ? getMessageText(lastUserMessage.content) : "";
-    const messageSpan = trace.getActiveSpan();
-    if (messageSpan) {
-      messageSpan.setAttributes({
-        "message.content": userMessage,
-        "message.timestamp": Date.now()
-      });
-    }
-    await createMessage(dbClient_default)({
-      id: nanoid(),
-      tenantId,
-      projectId,
-      conversationId,
-      role: "user",
-      content: {
-        text: userMessage
-      },
-      visibility: "user-facing",
-      messageType: "chat"
-    });
-    if (messageSpan) {
-      messageSpan.addEvent("user.message.stored", {
-        "message.id": conversationId,
-        "database.operation": "insert"
+        headers: validatedContext,
+        dbClient: dbClient_default,
+        credentialStores
       });
-    }
-    return streamSSE(c, async (stream2) => {
-      try {
-        const sseHelper = createSSEStreamHelper(stream2, requestId2, timestamp);
-        await sseHelper.writeRole();
-        logger21.info({ subAgentId }, "Starting execution");
-        const emitOperationsHeader = c.req.header("x-emit-operations");
-        const emitOperations = emitOperationsHeader === "true";
-        const executionHandler = new ExecutionHandler();
-        const result = await executionHandler.execute({
-          executionContext,
+      logger21.info(
+        {
+          tenantId,
+          projectId,
+          agentId,
           conversationId,
-          userMessage,
-          initialAgentId: subAgentId,
-          requestId: requestId2,
-          sseHelper,
-          emitOperations
+          defaultSubAgentId,
+          activeSubAgentId: activeAgent?.activeSubAgentId || "none",
+          hasContextConfig: !!agent.contextConfigId,
+          hasHeaders: !!body.headers,
+          hasValidatedContext: !!validatedContext,
+          validatedContextKeys: Object.keys(validatedContext)
+        },
+        "parameters"
+      );
+      const requestId2 = `chatcmpl-${Date.now()}`;
+      const timestamp = Math.floor(Date.now() / 1e3);
+      const lastUserMessage = body.messages.filter((msg) => msg.role === "user").slice(-1)[0];
+      const userMessage = lastUserMessage ? getMessageText(lastUserMessage.content) : "";
+      const messageSpan = trace.getActiveSpan();
+      if (messageSpan) {
+        messageSpan.setAttributes({
+          "message.content": userMessage,
+          "message.timestamp": Date.now()
         });
-        logger21.info(
-          { result },
-          `Execution completed: ${result.success ? "success" : "failed"} after ${result.iterations} iterations`
-        );
-        if (!result.success) {
-          await sseHelper.writeOperation(
-            errorOp(
-              "Sorry, I was unable to process your request at this time. Please try again.",
-              "system"
-            )
-          );
-        }
-        await sseHelper.complete();
-      } catch (error) {
-        logger21.error(
-          {
-            error: error instanceof Error ? error.message : error,
-            stack: error instanceof Error ? error.stack : void 0
-          },
-          "Error during streaming execution"
-        );
+      }
+      await createMessage(dbClient_default)({
+        id: nanoid(),
+        tenantId,
+        projectId,
+        conversationId,
+        role: "user",
+        content: {
+          text: userMessage
+        },
+        visibility: "user-facing",
+        messageType: "chat"
+      });
+      if (messageSpan) {
+        messageSpan.addEvent("user.message.stored", {
+          "message.id": conversationId,
+          "database.operation": "insert"
+        });
+      }
+      return streamSSE(c, async (stream2) => {
         try {
           const sseHelper = createSSEStreamHelper(stream2, requestId2, timestamp);
-          await sseHelper.writeOperation(
-            errorOp(
-              "Sorry, I was unable to process your request at this time. Please try again.",
-              "system"
-            )
+          await sseHelper.writeRole();
+          logger21.info({ subAgentId }, "Starting execution");
+          const emitOperationsHeader = c.req.header("x-emit-operations");
+          const emitOperations = emitOperationsHeader === "true";
+          const executionHandler = new ExecutionHandler();
+          const result = await executionHandler.execute({
+            executionContext,
+            conversationId,
+            userMessage,
+            initialAgentId: subAgentId,
+            requestId: requestId2,
+            sseHelper,
+            emitOperations
+          });
+          logger21.info(
+            { result },
+            `Execution completed: ${result.success ? "success" : "failed"} after ${result.iterations} iterations`
           );
+          if (!result.success) {
+            await sseHelper.writeOperation(
+              errorOp(
+                "Sorry, I was unable to process your request at this time. Please try again.",
+                "system"
+              )
+            );
+          }
           await sseHelper.complete();
-        } catch (streamError) {
-          logger21.error({ streamError }, "Failed to write error to stream");
+        } catch (error) {
+          logger21.error(
+            {
+              error: error instanceof Error ? error.message : error,
+              stack: error instanceof Error ? error.stack : void 0
+            },
+            "Error during streaming execution"
+          );
+          try {
+            const sseHelper = createSSEStreamHelper(stream2, requestId2, timestamp);
+            await sseHelper.writeOperation(
+              errorOp(
+                "Sorry, I was unable to process your request at this time. Please try again.",
+                "system"
+              )
+            );
+            await sseHelper.complete();
+          } catch (streamError) {
+            logger21.error({ streamError }, "Failed to write error to stream");
+          }
         }
-      }
+      });
     });
   } catch (error) {
     logger21.error(
@@ -10343,123 +10392,131 @@ app3.openapi(chatDataStreamRoute, async (c) => {
         "project.id": projectId
       });
     }
-    const agent = await getAgentWithDefaultSubAgent(dbClient_default)({
-      scopes: { tenantId, projectId, agentId }
-    });
-    if (!agent) {
-      throw createApiError({
-        code: "not_found",
-        message: "Agent not found"
-      });
+    let currentBag = propagation.getBaggage(context.active());
+    if (!currentBag) {
+      currentBag = propagation.createBaggage();
     }
-    const defaultSubAgentId = agent.defaultSubAgentId;
-    const agentName = agent.name;
-    if (!defaultSubAgentId) {
-      throw createApiError({
-        code: "bad_request",
-        message: "Agent does not have a default agent configured"
+    currentBag = currentBag.setEntry("conversation.id", { value: conversationId });
+    const ctxWithBaggage = propagation.setBaggage(context.active(), currentBag);
+    return await context.with(ctxWithBaggage, async () => {
+      const agent = await getAgentWithDefaultSubAgent(dbClient_default)({
+        scopes: { tenantId, projectId, agentId }
       });
-    }
-    const activeAgent = await getActiveAgentForConversation(dbClient_default)({
-      scopes: { tenantId, projectId },
-      conversationId
-    });
-    if (!activeAgent) {
-      setActiveAgentForConversation(dbClient_default)({
+      if (!agent) {
+        throw createApiError({
+          code: "not_found",
+          message: "Agent not found"
+        });
+      }
+      const defaultSubAgentId = agent.defaultSubAgentId;
+      const agentName = agent.name;
+      if (!defaultSubAgentId) {
+        throw createApiError({
+          code: "bad_request",
+          message: "Agent does not have a default agent configured"
+        });
+      }
+      const activeAgent = await getActiveAgentForConversation(dbClient_default)({
         scopes: { tenantId, projectId },
-        conversationId,
-        subAgentId: defaultSubAgentId
+        conversationId
       });
-    }
-    const subAgentId = activeAgent?.activeSubAgentId || defaultSubAgentId;
-    const agentInfo = await getSubAgentById(dbClient_default)({
-      scopes: { tenantId, projectId, agentId },
-      subAgentId
-    });
-    if (!agentInfo) {
-      throw createApiError({
-        code: "not_found",
-        message: "Agent not found"
+      if (!activeAgent) {
+        setActiveAgentForConversation(dbClient_default)({
+          scopes: { tenantId, projectId },
+          conversationId,
+          subAgentId: defaultSubAgentId
+        });
+      }
+      const subAgentId = activeAgent?.activeSubAgentId || defaultSubAgentId;
+      const agentInfo = await getSubAgentById(dbClient_default)({
+        scopes: { tenantId, projectId, agentId },
+        subAgentId
       });
-    }
-    const validatedContext = c.get("validatedContext") || body.headers || {};
-    const credentialStores = c.get("credentialStores");
-    await handleContextResolution({
-      tenantId,
-      projectId,
-      agentId,
-      conversationId,
-      headers: validatedContext,
-      dbClient: dbClient_default,
-      credentialStores
-    });
-    const lastUserMessage = body.messages.filter((m) => m.role === "user").slice(-1)[0];
-    const userText = typeof lastUserMessage?.content === "string" ? lastUserMessage.content : lastUserMessage?.parts?.map((p) => p.text).join("") || "";
-    logger22.info({ userText, lastUserMessage }, "userText");
-    const messageSpan = trace.getActiveSpan();
-    if (messageSpan) {
-      messageSpan.setAttributes({
-        "message.timestamp": (/* @__PURE__ */ new Date()).toISOString(),
-        "message.content": userText,
-        "agent.name": agentName
+      if (!agentInfo) {
+        throw createApiError({
+          code: "not_found",
+          message: "Agent not found"
+        });
+      }
+      const validatedContext = c.get("validatedContext") || body.headers || {};
+      const credentialStores = c.get("credentialStores");
+      await handleContextResolution({
+        tenantId,
+        projectId,
+        agentId,
+        conversationId,
+        headers: validatedContext,
+        dbClient: dbClient_default,
+        credentialStores
       });
-    }
-    await createMessage(dbClient_default)({
-      id: nanoid(),
-      tenantId,
-      projectId,
-      conversationId,
-      role: "user",
-      content: { text: userText },
-      visibility: "user-facing",
-      messageType: "chat"
-    });
-    if (messageSpan) {
-      messageSpan.addEvent("user.message.stored", {
-        "message.id": conversationId,
-        "database.operation": "insert"
+      const lastUserMessage = body.messages.filter((m) => m.role === "user").slice(-1)[0];
+      const userText = typeof lastUserMessage?.content === "string" ? lastUserMessage.content : lastUserMessage?.parts?.map((p) => p.text).join("") || "";
+      logger22.info({ userText, lastUserMessage }, "userText");
+      const messageSpan = trace.getActiveSpan();
+      if (messageSpan) {
+        messageSpan.setAttributes({
+          "message.timestamp": (/* @__PURE__ */ new Date()).toISOString(),
+          "message.content": userText,
+          "agent.name": agentName
+        });
+      }
+      await createMessage(dbClient_default)({
+        id: nanoid(),
+        tenantId,
+        projectId,
+        conversationId,
+        role: "user",
+        content: { text: userText },
+        visibility: "user-facing",
+        messageType: "chat"
       });
-    }
-    const dataStream = createUIMessageStream({
-      execute: async ({ writer }) => {
-        const streamHelper = createVercelStreamHelper(writer);
-        try {
-          const emitOperationsHeader = c.req.header("x-emit-operations");
-          const emitOperations = emitOperationsHeader === "true";
-          const executionHandler = new ExecutionHandler();
-          const result = await executionHandler.execute({
-            executionContext,
-            conversationId,
-            userMessage: userText,
-            initialAgentId: subAgentId,
-            requestId: `chatds-${Date.now()}`,
-            sseHelper: streamHelper,
-            emitOperations
-          });
-          if (!result.success) {
-            await streamHelper.writeOperation(errorOp("Unable to process request", "system"));
-          }
-        } catch (err) {
-          logger22.error({ err }, "Streaming error");
-          await streamHelper.writeOperation(errorOp("Internal server error", "system"));
-        } finally {
-          if ("cleanup" in streamHelper && typeof streamHelper.cleanup === "function") {
-            streamHelper.cleanup();
+      if (messageSpan) {
+        messageSpan.addEvent("user.message.stored", {
+          "message.id": conversationId,
+          "database.operation": "insert"
+        });
+      }
+      const dataStream = createUIMessageStream({
+        execute: async ({ writer }) => {
+          const streamHelper = createVercelStreamHelper(writer);
+          try {
+            const emitOperationsHeader = c.req.header("x-emit-operations");
+            const emitOperations = emitOperationsHeader === "true";
+            const executionHandler = new ExecutionHandler();
+            const result = await executionHandler.execute({
+              executionContext,
+              conversationId,
+              userMessage: userText,
+              initialAgentId: subAgentId,
+              requestId: `chatds-${Date.now()}`,
+              sseHelper: streamHelper,
+              emitOperations
+            });
+            if (!result.success) {
+              await streamHelper.writeOperation(errorOp("Unable to process request", "system"));
+            }
+          } catch (err) {
+            logger22.error({ err }, "Streaming error");
+            await streamHelper.writeOperation(errorOp("Internal server error", "system"));
+          } finally {
+            if ("cleanup" in streamHelper && typeof streamHelper.cleanup === "function") {
+              streamHelper.cleanup();
+            }
           }
         }
-      }
+      });
+      c.header("content-type", "text/event-stream");
+      c.header("cache-control", "no-cache");
+      c.header("connection", "keep-alive");
+      c.header("x-vercel-ai-data-stream", "v2");
+      c.header("x-accel-buffering", "no");
+      return stream(
+        c,
+        (stream2) => stream2.pipe(
+          dataStream.pipeThrough(new JsonToSseTransformStream()).pipeThrough(new TextEncoderStream())
+        )
+      );
     });
-    c.header("content-type", "text/event-stream");
-    c.header("cache-control", "no-cache");
-    c.header("connection", "keep-alive");
-    c.header("x-vercel-ai-data-stream", "v2");
-    c.header("x-accel-buffering", "no");
-    return stream(
-      c,
-      (stream2) => stream2.pipe(
-        dataStream.pipeThrough(new JsonToSseTransformStream()).pipeThrough(new TextEncoderStream())
-      )
-    );
   } catch (error) {
     logger22.error({ error }, "chatDataStream error");
     throw createApiError({
@@ -10787,66 +10844,83 @@ var handleInitializationRequest = async (body, executionContext, validatedContex
   const { tenantId, projectId, agentId } = executionContext;
   logger23.info({ body }, "Received initialization request");
   const sessionId = getConversationId();
-  const agent = await getAgentWithDefaultSubAgent(dbClient_default)({
-    scopes: { tenantId, projectId, agentId }
-  });
-  if (!agent) {
-    return c.json(
-      {
-        jsonrpc: "2.0",
-        error: { code: -32001, message: "Agent not found" },
-        id: body.id || null
-      },
-      { status: 404 }
-    );
+  const activeSpan = trace.getActiveSpan();
+  if (activeSpan) {
+    activeSpan.setAttributes({
+      "conversation.id": sessionId,
+      "tenant.id": tenantId,
+      "agent.id": agentId,
+      "project.id": projectId
+    });
   }
-  if (!agent.defaultSubAgentId) {
-    return c.json(
+  let currentBag = propagation.getBaggage(context.active());
+  if (!currentBag) {
+    currentBag = propagation.createBaggage();
+  }
+  currentBag = currentBag.setEntry("conversation.id", { value: sessionId });
+  const ctxWithBaggage = propagation.setBaggage(context.active(), currentBag);
+  return await context.with(ctxWithBaggage, async () => {
+    const agent = await getAgentWithDefaultSubAgent(dbClient_default)({
+      scopes: { tenantId, projectId, agentId }
+    });
+    if (!agent) {
+      return c.json(
+        {
+          jsonrpc: "2.0",
+          error: { code: -32001, message: "Agent not found" },
+          id: body.id || null
+        },
+        { status: 404 }
+      );
+    }
+    if (!agent.defaultSubAgentId) {
+      return c.json(
+        {
+          jsonrpc: "2.0",
+          error: { code: -32001, message: "Agent does not have a default agent configured" },
+          id: body.id || null
+        },
+        { status: 400 }
+      );
+    }
+    const conversation = await createOrGetConversation(dbClient_default)({
+      id: sessionId,
+      tenantId,
+      projectId,
+      activeSubAgentId: agent.defaultSubAgentId,
+      metadata: {
+        sessionData: {
+          agentId,
+          sessionType: "mcp",
+          mcpProtocolVersion: c.req.header("mcp-protocol-version"),
+          initialized: false
+          // Track initialization state
+        }
+      }
+    });
+    logger23.info(
+      { sessionId, conversationId: conversation.id },
+      "Created MCP session as conversation"
+    );
+    const transport = new StreamableHTTPServerTransport({
+      sessionIdGenerator: () => sessionId
+    });
+    const server = await getServer(validatedContext, executionContext, sessionId, credentialStores);
+    await server.connect(transport);
+    logger23.info({ sessionId }, "Server connected for initialization");
+    res.setHeader("Mcp-Session-Id", sessionId);
+    logger23.info(
       {
-        jsonrpc: "2.0",
-        error: { code: -32001, message: "Agent does not have a default agent configured" },
-        id: body.id || null
+        sessionId,
+        bodyMethod: body?.method,
+        bodyId: body?.id
       },
-      { status: 400 }
+      "About to handle initialization request"
     );
-  }
-  const conversation = await createOrGetConversation(dbClient_default)({
-    id: sessionId,
-    tenantId,
-    projectId,
-    activeSubAgentId: agent.defaultSubAgentId,
-    metadata: {
-      sessionData: {
-        agentId,
-        sessionType: "mcp",
-        mcpProtocolVersion: c.req.header("mcp-protocol-version"),
-        initialized: false
-        // Track initialization state
-      }
-    }
-  });
-  logger23.info(
-    { sessionId, conversationId: conversation.id },
-    "Created MCP session as conversation"
-  );
-  const transport = new StreamableHTTPServerTransport({
-    sessionIdGenerator: () => sessionId
+    await transport.handleRequest(req, res, body);
+    logger23.info({ sessionId }, "Successfully handled initialization request");
+    return toFetchResponse(res);
   });
-  const server = await getServer(validatedContext, executionContext, sessionId, credentialStores);
-  await server.connect(transport);
-  logger23.info({ sessionId }, "Server connected for initialization");
-  res.setHeader("Mcp-Session-Id", sessionId);
-  logger23.info(
-    {
-      sessionId,
-      bodyMethod: body?.method,
-      bodyId: body?.id
-    },
-    "About to handle initialization request"
-  );
-  await transport.handleRequest(req, res, body);
-  logger23.info({ sessionId }, "Successfully handled initialization request");
-  return toFetchResponse(res);
 };
 var handleExistingSessionRequest = async (body, executionContext, validatedContext, req, res, credentialStores) => {
   const { tenantId, projectId, agentId } = executionContext;