@inkeep/agents-run-api 0.0.0-dev-20250915105321 → 0.0.0-dev-20250915190940

package/dist/index.cjs

@@ -461,6 +461,7 @@ var apiKeyAuth = () => factory.createMiddleware(async (c, next) => {
       return;
     } else if (apiKey) {
       const executionContext = await extractContextFromApiKey(apiKey);
+      executionContext.agentId = agentId;
       c.set("executionContext", executionContext);
       logger.info({}, "API key authenticated successfully");
       await next();
@@ -478,12 +479,14 @@ var apiKeyAuth = () => factory.createMiddleware(async (c, next) => {
   }
   try {
     const executionContext = await extractContextFromApiKey(apiKey);
+    executionContext.agentId = agentId;
     c.set("executionContext", executionContext);
     logger.debug(
       {
         tenantId: executionContext.tenantId,
         projectId: executionContext.projectId,
-        graphId: executionContext.graphId
+        graphId: executionContext.graphId,
+        agentId: executionContext.agentId
       },
       "API key authenticated successfully"
     );
@@ -7707,69 +7710,75 @@ var ExecutionHandler = class {
               textContent += part.text;
             }
           }
-          const activeSpan = api.trace.getActiveSpan();
-          if (activeSpan) {
-            activeSpan.setAttributes({
-              "ai.response.content": textContent || "No response content",
-              "ai.response.timestamp": (/* @__PURE__ */ new Date()).toISOString(),
-              "ai.agent.name": currentAgentId
-            });
-          }
-          await agentsCore.createMessage(dbClient_default)({
-            id: nanoid.nanoid(),
-            tenantId,
-            projectId,
-            conversationId,
-            role: "agent",
-            content: {
-              text: textContent || void 0,
-              parts: responseParts.map((part) => ({
-                type: part.kind === "text" ? "text" : "data",
-                text: part.kind === "text" ? part.text : void 0,
-                data: part.kind === "data" ? JSON.stringify(part.data) : void 0
-              }))
-            },
-            visibility: "user-facing",
-            messageType: "chat",
-            agentId: currentAgentId,
-            fromAgentId: currentAgentId,
-            taskId: task.id
-          });
-          const updateTaskStart = Date.now();
-          await agentsCore.updateTask(dbClient_default)({
-            taskId: task.id,
-            data: {
-              status: "completed",
-              metadata: {
-                ...task.metadata,
-                completed_at: (/* @__PURE__ */ new Date()).toISOString(),
-                response: {
-                  text: textContent,
-                  parts: responseParts,
-                  hasText: !!textContent,
-                  hasData: responseParts.some((p) => p.kind === "data")
+          return tracer.startActiveSpan("execution_handler.execute", {}, async (span) => {
+            try {
+              span.setAttributes({
+                "ai.response.content": textContent || "No response content",
+                "ai.response.timestamp": (/* @__PURE__ */ new Date()).toISOString(),
+                "ai.agent.name": currentAgentId
+              });
+              await agentsCore.createMessage(dbClient_default)({
+                id: nanoid.nanoid(),
+                tenantId,
+                projectId,
+                conversationId,
+                role: "agent",
+                content: {
+                  text: textContent || void 0,
+                  parts: responseParts.map((part) => ({
+                    type: part.kind === "text" ? "text" : "data",
+                    text: part.kind === "text" ? part.text : void 0,
+                    data: part.kind === "data" ? JSON.stringify(part.data) : void 0
+                  }))
+                },
+                visibility: "user-facing",
+                messageType: "chat",
+                agentId: currentAgentId,
+                fromAgentId: currentAgentId,
+                taskId: task.id
+              });
+              const updateTaskStart = Date.now();
+              await agentsCore.updateTask(dbClient_default)({
+                taskId: task.id,
+                data: {
+                  status: "completed",
+                  metadata: {
+                    ...task.metadata,
+                    completed_at: (/* @__PURE__ */ new Date()).toISOString(),
+                    response: {
+                      text: textContent,
+                      parts: responseParts,
+                      hasText: !!textContent,
+                      hasData: responseParts.some((p) => p.kind === "data")
+                    }
+                  }
                 }
+              });
+              const updateTaskEnd = Date.now();
+              logger18.info(
+                { duration: updateTaskEnd - updateTaskStart },
+                "Completed updateTask operation"
+              );
+              await sseHelper.writeOperation(completionOp(currentAgentId, iterations));
+              await sseHelper.complete();
+              logger18.info({}, "Ending GraphSession and cleaning up");
+              graphSessionManager.endSession(requestId2);
+              logger18.info({}, "Cleaning up streamHelper");
+              unregisterStreamHelper(requestId2);
+              let response;
+              if (sseHelper instanceof MCPStreamHelper) {
+                const captured = sseHelper.getCapturedResponse();
+                response = captured.text || "No response content";
               }
+              logger18.info({}, "ExecutionHandler returning success");
+              return { success: true, iterations, response };
+            } catch (error) {
+              agentsCore.setSpanWithError(span, error);
+              throw error;
+            } finally {
+              span.end();
             }
           });
-          const updateTaskEnd = Date.now();
-          logger18.info(
-            { duration: updateTaskEnd - updateTaskStart },
-            "Completed updateTask operation"
-          );
-          await sseHelper.writeOperation(completionOp(currentAgentId, iterations));
-          await sseHelper.complete();
-          logger18.info({}, "Ending GraphSession and cleaning up");
-          graphSessionManager.endSession(requestId2);
-          logger18.info({}, "Cleaning up streamHelper");
-          unregisterStreamHelper(requestId2);
-          let response;
-          if (sseHelper instanceof MCPStreamHelper) {
-            const captured = sseHelper.getCapturedResponse();
-            response = captured.text || "No response content";
-          }
-          logger18.info({}, "ExecutionHandler returning success");
-          return { success: true, iterations, response };
         }
         errorCount++;
         logger18.warn(

package/dist/index.js

@@ -8,7 +8,7 @@ import { resourceFromAttributes } from '@opentelemetry/resources';
 import { NodeSDK } from '@opentelemetry/sdk-node';
 import { BatchSpanProcessor } from '@opentelemetry/sdk-trace-base';
 import { ATTR_SERVICE_NAME } from '@opentelemetry/semantic-conventions';
-import { getLogger, getTracer, HeadersScopeSchema, getRequestExecutionContext, getAgentGraphWithDefaultAgent, contextValidationMiddleware, getFullGraph, createOrGetConversation, getActiveAgentForConversation, setActiveAgentForConversation, getAgentById, handleContextResolution, createMessage, commonGetErrorResponses, createDefaultCredentialStores, CredentialStoreRegistry, listTaskIdsByContextId, getTask, getLedgerArtifacts, getAgentGraph, createTask, updateTask, updateConversation, handleApiError, setSpanWithError, TaskState, setActiveAgentForThread, getConversation, getRelatedAgentsForGraph, getToolsForAgent, getDataComponentsForAgent, getArtifactComponentsForAgent, validateAndGetApiKey, getProject, ContextResolver, CredentialStuffer, MCPServerType, getCredentialReference, McpClient, getContextConfigById, getFullGraphDefinition, TemplateEngine, graphHasArtifactComponents, MCPTransportType, getExternalAgent } from '@inkeep/agents-core';
+import { getLogger, getTracer, HeadersScopeSchema, getRequestExecutionContext, getAgentGraphWithDefaultAgent, contextValidationMiddleware, getFullGraph, createOrGetConversation, getActiveAgentForConversation, setActiveAgentForConversation, getAgentById, handleContextResolution, createMessage, commonGetErrorResponses, createDefaultCredentialStores, CredentialStoreRegistry, listTaskIdsByContextId, getTask, getLedgerArtifacts, getAgentGraph, createTask, updateTask, setSpanWithError, updateConversation, handleApiError, TaskState, setActiveAgentForThread, getConversation, getRelatedAgentsForGraph, getToolsForAgent, getDataComponentsForAgent, getArtifactComponentsForAgent, validateAndGetApiKey, getProject, ContextResolver, CredentialStuffer, MCPServerType, getCredentialReference, McpClient, getContextConfigById, getFullGraphDefinition, TemplateEngine, graphHasArtifactComponents, MCPTransportType, getExternalAgent } from '@inkeep/agents-core';
 import { OpenAPIHono, createRoute, z as z$1 } from '@hono/zod-openapi';
 import { trace, propagation, context, SpanStatusCode } from '@opentelemetry/api';
 import { Hono } from 'hono';
@@ -166,6 +166,7 @@ var apiKeyAuth = () => createMiddleware(async (c, next) => {
       return;
     } else if (apiKey) {
       const executionContext = await extractContextFromApiKey(apiKey);
+      executionContext.agentId = agentId;
       c.set("executionContext", executionContext);
       logger.info({}, "API key authenticated successfully");
       await next();
@@ -183,12 +184,14 @@ var apiKeyAuth = () => createMiddleware(async (c, next) => {
   }
   try {
     const executionContext = await extractContextFromApiKey(apiKey);
+    executionContext.agentId = agentId;
     c.set("executionContext", executionContext);
     logger.debug(
       {
         tenantId: executionContext.tenantId,
         projectId: executionContext.projectId,
-        graphId: executionContext.graphId
+        graphId: executionContext.graphId,
+        agentId: executionContext.agentId
       },
       "API key authenticated successfully"
     );
@@ -7375,69 +7378,75 @@ var ExecutionHandler = class {
               textContent += part.text;
             }
           }
-          const activeSpan = trace.getActiveSpan();
-          if (activeSpan) {
-            activeSpan.setAttributes({
-              "ai.response.content": textContent || "No response content",
-              "ai.response.timestamp": (/* @__PURE__ */ new Date()).toISOString(),
-              "ai.agent.name": currentAgentId
-            });
-          }
-          await createMessage(dbClient_default)({
-            id: nanoid(),
-            tenantId,
-            projectId,
-            conversationId,
-            role: "agent",
-            content: {
-              text: textContent || void 0,
-              parts: responseParts.map((part) => ({
-                type: part.kind === "text" ? "text" : "data",
-                text: part.kind === "text" ? part.text : void 0,
-                data: part.kind === "data" ? JSON.stringify(part.data) : void 0
-              }))
-            },
-            visibility: "user-facing",
-            messageType: "chat",
-            agentId: currentAgentId,
-            fromAgentId: currentAgentId,
-            taskId: task.id
-          });
-          const updateTaskStart = Date.now();
-          await updateTask(dbClient_default)({
-            taskId: task.id,
-            data: {
-              status: "completed",
-              metadata: {
-                ...task.metadata,
-                completed_at: (/* @__PURE__ */ new Date()).toISOString(),
-                response: {
-                  text: textContent,
-                  parts: responseParts,
-                  hasText: !!textContent,
-                  hasData: responseParts.some((p) => p.kind === "data")
+          return tracer.startActiveSpan("execution_handler.execute", {}, async (span) => {
+            try {
+              span.setAttributes({
+                "ai.response.content": textContent || "No response content",
+                "ai.response.timestamp": (/* @__PURE__ */ new Date()).toISOString(),
+                "ai.agent.name": currentAgentId
+              });
+              await createMessage(dbClient_default)({
+                id: nanoid(),
+                tenantId,
+                projectId,
+                conversationId,
+                role: "agent",
+                content: {
+                  text: textContent || void 0,
+                  parts: responseParts.map((part) => ({
+                    type: part.kind === "text" ? "text" : "data",
+                    text: part.kind === "text" ? part.text : void 0,
+                    data: part.kind === "data" ? JSON.stringify(part.data) : void 0
+                  }))
+                },
+                visibility: "user-facing",
+                messageType: "chat",
+                agentId: currentAgentId,
+                fromAgentId: currentAgentId,
+                taskId: task.id
+              });
+              const updateTaskStart = Date.now();
+              await updateTask(dbClient_default)({
+                taskId: task.id,
+                data: {
+                  status: "completed",
+                  metadata: {
+                    ...task.metadata,
+                    completed_at: (/* @__PURE__ */ new Date()).toISOString(),
+                    response: {
+                      text: textContent,
+                      parts: responseParts,
+                      hasText: !!textContent,
+                      hasData: responseParts.some((p) => p.kind === "data")
+                    }
+                  }
                 }
+              });
+              const updateTaskEnd = Date.now();
+              logger18.info(
+                { duration: updateTaskEnd - updateTaskStart },
+                "Completed updateTask operation"
+              );
+              await sseHelper.writeOperation(completionOp(currentAgentId, iterations));
+              await sseHelper.complete();
+              logger18.info({}, "Ending GraphSession and cleaning up");
+              graphSessionManager.endSession(requestId2);
+              logger18.info({}, "Cleaning up streamHelper");
+              unregisterStreamHelper(requestId2);
+              let response;
+              if (sseHelper instanceof MCPStreamHelper) {
+                const captured = sseHelper.getCapturedResponse();
+                response = captured.text || "No response content";
               }
+              logger18.info({}, "ExecutionHandler returning success");
+              return { success: true, iterations, response };
+            } catch (error) {
+              setSpanWithError(span, error);
+              throw error;
+            } finally {
+              span.end();
             }
           });
-          const updateTaskEnd = Date.now();
-          logger18.info(
-            { duration: updateTaskEnd - updateTaskStart },
-            "Completed updateTask operation"
-          );
-          await sseHelper.writeOperation(completionOp(currentAgentId, iterations));
-          await sseHelper.complete();
-          logger18.info({}, "Ending GraphSession and cleaning up");
-          graphSessionManager.endSession(requestId2);
-          logger18.info({}, "Cleaning up streamHelper");
-          unregisterStreamHelper(requestId2);
-          let response;
-          if (sseHelper instanceof MCPStreamHelper) {
-            const captured = sseHelper.getCapturedResponse();
-            response = captured.text || "No response content";
-          }
-          logger18.info({}, "ExecutionHandler returning success");
-          return { success: true, iterations, response };
         }
         errorCount++;
         logger18.warn(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@inkeep/agents-run-api",
-  "version": "0.0.0-dev-20250915105321",
+  "version": "0.0.0-dev-20250915190940",
   "description": "Agents Run API for Inkeep Agent Framework - handles chat, agent execution, and streaming",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -44,7 +44,7 @@
     "traverse": "^0.6.11",
     "ts-pattern": "^5.7.1",
     "zod": "^4.1.5",
-    "@inkeep/agents-core": "^0.0.0-dev-20250915105321"
+    "@inkeep/agents-core": "^0.0.0-dev-20250915190940"
   },
   "devDependencies": {
     "@hono/vite-dev-server": "^0.20.1",