@inkeep/agents-run-api 0.14.2 → 0.14.5

package/dist/index.js

@@ -1,7 +1,8 @@
-import { defaultBatchProcessor } from './chunk-Z76LOSTO.js';
+import { flushBatchProcessor } from './chunk-XO4RG36I.js';
 import { getLogger } from './chunk-A2S7GSHL.js';
-import { getFormattedConversationHistory, createDefaultConversationHistoryConfig, saveA2AMessageResponse } from './chunk-F46V23RK.js';
-import { dbClient_default, env } from './chunk-TWVQBIUL.js';
+import { getFormattedConversationHistory, createDefaultConversationHistoryConfig, saveA2AMessageResponse } from './chunk-5GUNCN5X.js';
+import { dbClient_default } from './chunk-7IMXW4RD.js';
+import { env } from './chunk-NZHNG4A3.js';
 import { __publicField } from './chunk-PKBMQBKP.js';
 import { getLogger as getLogger$1, getTracer, HeadersScopeSchema, getRequestExecutionContext, createApiError, getAgentGraphWithDefaultAgent, contextValidationMiddleware, getFullGraph, createOrGetConversation, getActiveAgentForConversation, setActiveAgentForConversation, getAgentById, handleContextResolution, createMessage, commonGetErrorResponses, loggerFactory, createDefaultCredentialStores, CredentialStoreRegistry, listTaskIdsByContextId, getTask, getLedgerArtifacts, upsertLedgerArtifact, getAgentGraphById, createTask, updateTask, setSpanWithError, updateConversation, handleApiError, TaskState, setActiveAgentForThread, getConversation, getRelatedAgentsForGraph, getToolsForAgent, getDataComponentsForAgent, getArtifactComponentsForAgent, dbResultToMcpTool, validateAndGetApiKey, getProject, ContextResolver, CredentialStuffer, MCPServerType, getCredentialReference, McpClient, getContextConfigById, getFullGraphDefinition, TemplateEngine, graphHasArtifactComponents, MCPTransportType, getExternalAgent } from '@inkeep/agents-core';
 import { otel } from '@hono/otel';
@@ -2709,7 +2710,6 @@ ${this.statusUpdateState?.config.prompt?.trim() || ""}`;
           return { summaries: [] };
         } finally {
           span.end();
-          await defaultBatchProcessor.forceFlush();
         }
       }
     );
@@ -3128,7 +3128,6 @@ Make it specific and relevant.`;
           );
         } finally {
           span.end();
-          await defaultBatchProcessor.forceFlush();
         }
       }
     );
@@ -3684,7 +3683,6 @@ var ResponseFormatter = class {
         };
       } finally {
         span.end();
-        await defaultBatchProcessor.forceFlush();
       }
     });
   }
@@ -3735,7 +3733,6 @@ var ResponseFormatter = class {
         return { text: responseText };
       } finally {
         span.end();
-        await defaultBatchProcessor.forceFlush();
       }
     });
   }
@@ -6166,6 +6163,8 @@ var Agent = class {
     __publicField(this, "isDelegatedAgent", false);
     __publicField(this, "contextResolver");
     __publicField(this, "credentialStoreRegistry");
+    __publicField(this, "mcpClientCache", /* @__PURE__ */ new Map());
+    __publicField(this, "mcpConnectionLocks", /* @__PURE__ */ new Map());
     this.artifactComponents = config.artifactComponents || [];
     let processedDataComponents = config.dataComponents || [];
     if (processedDataComponents.length > 0) {
@@ -6488,6 +6487,7 @@ var Agent = class {
     };
   }
   async getMcpTool(tool3) {
+    const cacheKey = `${this.config.tenantId}-${this.config.projectId}-${tool3.id}-${tool3.credentialReferenceId || "no-cred"}`;
     const credentialReferenceId = tool3.credentialReferenceId;
     const toolsForAgent = await getToolsForAgent(dbClient_default)({
       scopes: {
@@ -6556,12 +6556,56 @@ var Agent = class {
       },
       "Built MCP server config with credentials"
     );
+    let client = this.mcpClientCache.get(cacheKey);
+    if (client && !client.isConnected()) {
+      this.mcpClientCache.delete(cacheKey);
+      client = void 0;
+    }
+    if (!client) {
+      let connectionPromise = this.mcpConnectionLocks.get(cacheKey);
+      if (!connectionPromise) {
+        connectionPromise = this.createMcpConnection(tool3, serverConfig);
+        this.mcpConnectionLocks.set(cacheKey, connectionPromise);
+      }
+      try {
+        client = await connectionPromise;
+        this.mcpClientCache.set(cacheKey, client);
+      } catch (error) {
+        this.mcpConnectionLocks.delete(cacheKey);
+        logger16.error(
+          {
+            toolName: tool3.name,
+            agentId: this.config.id,
+            cacheKey,
+            error: error instanceof Error ? error.message : String(error)
+          },
+          "MCP connection failed"
+        );
+        throw error;
+      }
+    }
+    const tools = await client.tools();
+    return tools;
+  }
+  async createMcpConnection(tool3, serverConfig) {
     const client = new McpClient({
       name: tool3.name,
       server: serverConfig
     });
-    await client.connect();
-    return client.tools();
+    try {
+      await client.connect();
+      return client;
+    } catch (error) {
+      logger16.error(
+        {
+          toolName: tool3.name,
+          agentId: this.config.id,
+          error: error instanceof Error ? error.message : String(error)
+        },
+        "Agent failed to connect to MCP server"
+      );
+      throw error;
+    }
   }
   getFunctionTools(streamRequestId) {
     if (!this.config.functionTools) return {};
@@ -6607,7 +6651,6 @@ var Agent = class {
         requestContext: requestContext || {},
         tenantId: this.config.tenantId
       });
-      await defaultBatchProcessor.forceFlush();
       const contextWithBuiltins = {
         ...result.resolvedContext,
         $now: (/* @__PURE__ */ new Date()).toISOString(),
@@ -6777,7 +6820,7 @@ var Agent = class {
       inputSchema: tool3.inputSchema || tool3.parameters || {},
       usageGuidelines: name.startsWith("transfer_to_") || name.startsWith("delegate_to_") ? `Use this tool to ${name.startsWith("transfer_to_") ? "transfer" : "delegate"} to another agent when appropriate.` : "Use this tool when appropriate for the task at hand."
     }));
-    const { getConversationScopedArtifacts } = await import('./conversations-AP5WZGWA.js');
+    const { getConversationScopedArtifacts } = await import('./conversations-6HJ4FX5F.js');
     const historyConfig = this.config.conversationHistoryConfig ?? createDefaultConversationHistoryConfig();
     const referenceArtifacts = await getConversationScopedArtifacts({
       tenantId: this.config.tenantId,
@@ -7146,7 +7189,6 @@ var Agent = class {
               throw err;
             } finally {
               childSpan.end();
-              await defaultBatchProcessor.forceFlush();
             }
           }
         );
@@ -7577,7 +7619,6 @@ ${output}${structureHintsFormatted}`;
         }
         span.setStatus({ code: SpanStatusCode.OK });
         span.end();
-        await defaultBatchProcessor.forceFlush();
         let formattedContent = response.formattedContent || null;
         if (!formattedContent) {
           const session = toolSessionManager.getSession(sessionId);
@@ -9129,7 +9170,6 @@ var ExecutionHandler = class {
               throw error;
             } finally {
               span.end();
-              await defaultBatchProcessor.forceFlush();
             }
           });
         }
@@ -9651,7 +9691,6 @@ app3.openapi(chatDataStreamRoute, async (c) => {
       dbClient: dbClient_default,
       credentialStores
     });
-    await defaultBatchProcessor.forceFlush();
     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");
@@ -10455,19 +10494,19 @@ function createExecutionHono(serverConfig, credentialStores) {
   setupOpenAPIRoutes(app6);
   app6.use("/tenants/*", async (_c, next) => {
     await next();
-    await defaultBatchProcessor.forceFlush();
+    await flushBatchProcessor();
   });
   app6.use("/agents/*", async (_c, next) => {
     await next();
-    await defaultBatchProcessor.forceFlush();
+    await flushBatchProcessor();
   });
   app6.use("/v1/*", async (_c, next) => {
     await next();
-    await defaultBatchProcessor.forceFlush();
+    await flushBatchProcessor();
   });
   app6.use("/api/*", async (_c, next) => {
     await next();
-    await defaultBatchProcessor.forceFlush();
+    await flushBatchProcessor();
   });
   const baseApp = new Hono();
   baseApp.route("/", app6);

package/dist/instrumentation.cjs

@@ -9,12 +9,58 @@ var resources = require('@opentelemetry/resources');
 var sdkNode = require('@opentelemetry/sdk-node');
 var sdkTraceBase = require('@opentelemetry/sdk-trace-base');
 var semanticConventions = require('@opentelemetry/semantic-conventions');
+var agentsCore = require('@inkeep/agents-core');
+var zod = require('zod');
 
 // src/instrumentation.ts
-var otlpExporter = new exporterTraceOtlpHttp.OTLPTraceExporter();
-var defaultBatchProcessor = new sdkTraceBase.BatchSpanProcessor(otlpExporter, {
-  scheduledDelayMillis: 1e3
+agentsCore.loadEnvironmentFiles();
+var envSchema = zod.z.object({
+  NODE_ENV: zod.z.enum(["development", "production", "test"]).optional(),
+  ENVIRONMENT: zod.z.enum(["development", "production", "pentest", "test"]).optional().default("development"),
+  DB_FILE_NAME: zod.z.string().optional(),
+  TURSO_DATABASE_URL: zod.z.string().optional(),
+  TURSO_AUTH_TOKEN: zod.z.string().optional(),
+  AGENTS_RUN_API_URL: zod.z.string().optional().default("http://localhost:3003"),
+  LOG_LEVEL: zod.z.enum(["trace", "debug", "info", "warn", "error"]).optional().default("debug"),
+  NANGO_SECRET_KEY: zod.z.string().optional(),
+  OPENAI_API_KEY: zod.z.string().optional(),
+  ANTHROPIC_API_KEY: zod.z.string(),
+  INKEEP_AGENTS_RUN_API_BYPASS_SECRET: zod.z.string().optional(),
+  OTEL_BSP_SCHEDULE_DELAY: zod.z.coerce.number().optional().default(500),
+  OTEL_BSP_MAX_EXPORT_BATCH_SIZE: zod.z.coerce.number().optional().default(64)
 });
+var parseEnv = () => {
+  try {
+    const parsedEnv = envSchema.parse(process.env);
+    return parsedEnv;
+  } catch (error) {
+    if (error instanceof zod.z.ZodError) {
+      const missingVars = error.issues.map((issue) => issue.path.join("."));
+      throw new Error(
+        `\u274C Invalid environment variables: ${missingVars.join(", ")}
+${error.message}`
+      );
+    }
+    throw error;
+  }
+};
+var env = parseEnv();
+
+// src/instrumentation.ts
+var otlpExporter = new exporterTraceOtlpHttp.OTLPTraceExporter();
+var logger = agentsCore.getLogger("instrumentation");
+function createSafeBatchProcessor() {
+  try {
+    return new sdkTraceBase.BatchSpanProcessor(otlpExporter, {
+      scheduledDelayMillis: env.OTEL_BSP_SCHEDULE_DELAY,
+      maxExportBatchSize: env.OTEL_BSP_MAX_EXPORT_BATCH_SIZE
+    });
+  } catch (error) {
+    logger.warn({ error }, "Failed to create batch processor");
+    return new sdkTraceBase.NoopSpanProcessor();
+  }
+}
+var defaultBatchProcessor = createSafeBatchProcessor();
 var defaultResource = resources.resourceFromAttributes({
   [semanticConventions.ATTR_SERVICE_NAME]: "inkeep-agents-run-api"
 });
@@ -55,6 +101,13 @@ var defaultSDK = new sdkNode.NodeSDK({
   spanProcessors: defaultSpanProcessors,
   instrumentations: defaultInstrumentations
 });
+async function flushBatchProcessor() {
+  try {
+    await defaultBatchProcessor.forceFlush();
+  } catch (error) {
+    logger.warn({ error }, "Failed to flush batch processor");
+  }
+}
 
 exports.defaultBatchProcessor = defaultBatchProcessor;
 exports.defaultContextManager = defaultContextManager;
@@ -63,3 +116,4 @@ exports.defaultResource = defaultResource;
 exports.defaultSDK = defaultSDK;
 exports.defaultSpanProcessors = defaultSpanProcessors;
 exports.defaultTextMapPropagator = defaultTextMapPropagator;
+exports.flushBatchProcessor = flushBatchProcessor;

package/dist/instrumentation.d.cts

@@ -2,14 +2,15 @@ import * as _opentelemetry_resources from '@opentelemetry/resources';
 import { AsyncLocalStorageContextManager } from '@opentelemetry/context-async-hooks';
 import { CompositePropagator } from '@opentelemetry/core';
 import { NodeSDKConfiguration, NodeSDK } from '@opentelemetry/sdk-node';
-import { BatchSpanProcessor, SpanProcessor } from '@opentelemetry/sdk-trace-base';
+import { SpanProcessor } from '@opentelemetry/sdk-trace-base';
 
-declare const defaultBatchProcessor: BatchSpanProcessor;
+declare const defaultBatchProcessor: SpanProcessor;
 declare const defaultResource: _opentelemetry_resources.Resource;
 declare const defaultInstrumentations: NonNullable<NodeSDKConfiguration['instrumentations']>;
 declare const defaultSpanProcessors: SpanProcessor[];
 declare const defaultContextManager: AsyncLocalStorageContextManager;
 declare const defaultTextMapPropagator: CompositePropagator;
 declare const defaultSDK: NodeSDK;
+declare function flushBatchProcessor(): Promise<void>;
 
-export { defaultBatchProcessor, defaultContextManager, defaultInstrumentations, defaultResource, defaultSDK, defaultSpanProcessors, defaultTextMapPropagator };
+export { defaultBatchProcessor, defaultContextManager, defaultInstrumentations, defaultResource, defaultSDK, defaultSpanProcessors, defaultTextMapPropagator, flushBatchProcessor };

package/dist/instrumentation.d.ts

@@ -2,14 +2,15 @@ import * as _opentelemetry_resources from '@opentelemetry/resources';
 import { AsyncLocalStorageContextManager } from '@opentelemetry/context-async-hooks';
 import { CompositePropagator } from '@opentelemetry/core';
 import { NodeSDKConfiguration, NodeSDK } from '@opentelemetry/sdk-node';
-import { BatchSpanProcessor, SpanProcessor } from '@opentelemetry/sdk-trace-base';
+import { SpanProcessor } from '@opentelemetry/sdk-trace-base';
 
-declare const defaultBatchProcessor: BatchSpanProcessor;
+declare const defaultBatchProcessor: SpanProcessor;
 declare const defaultResource: _opentelemetry_resources.Resource;
 declare const defaultInstrumentations: NonNullable<NodeSDKConfiguration['instrumentations']>;
 declare const defaultSpanProcessors: SpanProcessor[];
 declare const defaultContextManager: AsyncLocalStorageContextManager;
 declare const defaultTextMapPropagator: CompositePropagator;
 declare const defaultSDK: NodeSDK;
+declare function flushBatchProcessor(): Promise<void>;
 
-export { defaultBatchProcessor, defaultContextManager, defaultInstrumentations, defaultResource, defaultSDK, defaultSpanProcessors, defaultTextMapPropagator };
+export { defaultBatchProcessor, defaultContextManager, defaultInstrumentations, defaultResource, defaultSDK, defaultSpanProcessors, defaultTextMapPropagator, flushBatchProcessor };

package/dist/instrumentation.js

@@ -1 +1 @@
-export { defaultBatchProcessor, defaultContextManager, defaultInstrumentations, defaultResource, defaultSDK, defaultSpanProcessors, defaultTextMapPropagator } from './chunk-Z76LOSTO.js';
+export { defaultBatchProcessor, defaultContextManager, defaultInstrumentations, defaultResource, defaultSDK, defaultSpanProcessors, defaultTextMapPropagator, flushBatchProcessor } from './chunk-XO4RG36I.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@inkeep/agents-run-api",
-  "version": "0.14.2",
+  "version": "0.14.5",
   "description": "Agents Run API for Inkeep Agent Framework - handles chat, agent execution, and streaming",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -51,7 +51,7 @@
     "traverse": "^0.6.11",
     "ts-pattern": "^5.7.1",
     "zod": "^4.1.11",
-    "@inkeep/agents-core": "^0.14.2"
+    "@inkeep/agents-core": "^0.14.5"
   },
   "optionalDependencies": {
     "keytar": "^7.9.0"

package/dist/dbClient-GGR4KMBD.js

@@ -1 +0,0 @@
-export { dbClient_default as default } from './chunk-TWVQBIUL.js';