@inkeep/agents-core 0.0.0-dev-20250911192304 → 0.0.0-dev-20250911210702

package/dist/index.cjs

@@ -23,37 +23,13 @@ var ai = require('ai');
 var exitHook = require('exit-hook');
 var tsPattern = require('ts-pattern');
 var zod = require('zod');
-var fs = require('fs');
-var path = require('path');
-var dotenv = require('dotenv');
 var node = require('@nangohq/node');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
-function _interopNamespace(e) {
-  if (e && e.__esModule) return e;
-  var n = Object.create(null);
-  if (e) {
-    Object.keys(e).forEach(function (k) {
-      if (k !== 'default') {
-        var d = Object.getOwnPropertyDescriptor(e, k);
-        Object.defineProperty(n, k, d.get ? d : {
-          enumerable: true,
-          get: function () { return e[k]; }
-        });
-      }
-    });
-  }
-  n.default = e;
-  return Object.freeze(n);
-}
-
 var jmespath__default = /*#__PURE__*/_interopDefault(jmespath);
 var crypto__default = /*#__PURE__*/_interopDefault(crypto);
 var Ajv__default = /*#__PURE__*/_interopDefault(Ajv);
-var fs__default = /*#__PURE__*/_interopDefault(fs);
-var path__default = /*#__PURE__*/_interopDefault(path);
-var dotenv__namespace = /*#__PURE__*/_interopNamespace(dotenv);
 
 var __defProp = Object.defineProperty;
 var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
@@ -99,8 +75,8 @@ var LoggerFactory = class {
   /**
    * Configure the logger factory
    */
-  configure(config2) {
-    this.config = config2;
+  configure(config) {
+    this.config = config;
     this.loggers.clear();
   }
   /**
@@ -133,8 +109,8 @@ var loggerFactory = new LoggerFactory();
 function getLogger(name) {
   return loggerFactory.getLogger(name);
 }
-function configureLogging(config2) {
-  loggerFactory.configure(config2);
+function configureLogging(config) {
+  loggerFactory.configure(config);
 }
 
 // src/db/schema.ts
@@ -512,6 +488,7 @@ var agentToolRelations = sqliteCore.sqliteTable(
     id: sqliteCore.text("id").notNull(),
     agentId: sqliteCore.text("agent_id").notNull(),
     toolId: sqliteCore.text("tool_id").notNull(),
+    selectedTools: sqliteCore.blob("selected_tools", { mode: "json" }).$type(),
     createdAt: sqliteCore.text("created_at").notNull().default(drizzleOrm.sql`CURRENT_TIMESTAMP`),
     updatedAt: sqliteCore.text("updated_at").notNull().default(drizzleOrm.sql`CURRENT_TIMESTAMP`)
   },
@@ -1378,7 +1355,8 @@ var AgentToolRelationSelectSchema = drizzleZod.createSelectSchema(agentToolRelat
 var AgentToolRelationInsertSchema = drizzleZod.createInsertSchema(agentToolRelations).extend({
   id: resourceIdSchema,
   agentId: resourceIdSchema,
-  toolId: resourceIdSchema
+  toolId: resourceIdSchema,
+  selectedTools: zodOpenapi.z.array(zodOpenapi.z.string()).nullish()
 });
 var AgentToolRelationUpdateSchema = AgentToolRelationInsertSchema.partial();
 var AgentToolRelationApiSelectSchema = createApiSchema(AgentToolRelationSelectSchema);
@@ -1412,6 +1390,7 @@ var StatusUpdateSchema = zodOpenapi.z.object({
 });
 var FullGraphAgentInsertSchema = AgentApiInsertSchema.extend({
   tools: zodOpenapi.z.array(zodOpenapi.z.string()),
+  selectedTools: zodOpenapi.z.record(zodOpenapi.z.string(), zodOpenapi.z.array(zodOpenapi.z.string())).optional(),
   dataComponents: zodOpenapi.z.array(zodOpenapi.z.string()).optional(),
   artifactComponents: zodOpenapi.z.array(zodOpenapi.z.string()).optional(),
   canTransferTo: zodOpenapi.z.array(zodOpenapi.z.string()).optional(),
@@ -1833,8 +1812,8 @@ var _TemplateEngine = class _TemplateEngine {
    * Process variable substitutions {{variable.path}} using JMESPath
    */
   static processVariables(template, context, options) {
-    return template.replace(/\{\{([^}]+)\}\}/g, (match2, path2) => {
-      const trimmedPath = path2.trim();
+    return template.replace(/\{\{([^}]+)\}\}/g, (match2, path) => {
+      const trimmedPath = path.trim();
       try {
         if (trimmedPath.startsWith("$")) {
           return _TemplateEngine.processBuiltinVariable(trimmedPath);
@@ -2164,7 +2143,7 @@ var CredentialStuffer = class {
   /**
    * Build MCP server configuration with credentials
    */
-  async buildMcpServerConfig(context, tool2, storeReference) {
+  async buildMcpServerConfig(context, tool2, storeReference, selectedTools) {
     let credentialHeaders = {};
     if (storeReference || tool2.headers) {
       credentialHeaders = await this.getCredentialHeaders({
@@ -2177,7 +2156,8 @@ var CredentialStuffer = class {
     const baseConfig = {
       type: tool2.transport?.type || MCPTransportType.streamableHttp,
       url: tool2.serverUrl,
-      activeTools: tool2.activeTools
+      activeTools: tool2.activeTools,
+      selectedTools
     };
     if (baseConfig.type === MCPTransportType.streamableHttp || baseConfig.type === MCPTransportType.sse) {
       const httpConfig = {
@@ -2193,14 +2173,14 @@ var CredentialStuffer = class {
     return baseConfig;
   }
 };
-function createDatabaseClient(config2) {
+function createDatabaseClient(config) {
   const client$1 = client.createClient({
-    url: config2.url,
-    authToken: config2.authToken
+    url: config.url,
+    authToken: config.authToken
   });
   return libsql.drizzle(client$1, {
     schema: schema_exports,
-    logger: config2.logger
+    logger: config.logger
   });
 }
 function createInMemoryDatabaseClient() {
@@ -2445,7 +2425,8 @@ var createAgentToolRelation = (db) => async (params) => {
     tenantId: params.scopes.tenantId,
     projectId: params.scopes.projectId,
     agentId: params.data.agentId,
-    toolId: params.data.toolId
+    toolId: params.data.toolId,
+    selectedTools: params.data.selectedTools
   }).returning();
   return relation[0];
 };
@@ -2600,6 +2581,7 @@ var getToolsForAgent = (db) => async (params) => {
       tenantId: agentToolRelations.tenantId,
       agentId: agentToolRelations.agentId,
       toolId: agentToolRelations.toolId,
+      selectedTools: agentToolRelations.selectedTools,
       createdAt: agentToolRelations.createdAt,
       updatedAt: agentToolRelations.updatedAt,
       tool: {
@@ -2647,6 +2629,7 @@ var getAgentsForTool = (db) => async (params) => {
       tenantId: agentToolRelations.tenantId,
       agentId: agentToolRelations.agentId,
       toolId: agentToolRelations.toolId,
+      selectedTools: agentToolRelations.selectedTools,
       createdAt: agentToolRelations.createdAt,
       updatedAt: agentToolRelations.updatedAt,
       agent: {
@@ -3231,14 +3214,14 @@ var deleteAgentGraph = (db) => async (params) => {
   ).returning();
   return result.length > 0;
 };
-var fetchComponentRelationships = (db) => async (scopes, agentIds, config2) => {
+var fetchComponentRelationships = (db) => async (scopes, agentIds, config) => {
   const componentsObject = {};
   if (agentIds.length > 0) {
-    const results = await db.select(config2.selectFields).from(config2.relationTable).innerJoin(config2.componentTable, drizzleOrm.eq(config2.relationIdField, config2.componentIdField)).where(
+    const results = await db.select(config.selectFields).from(config.relationTable).innerJoin(config.componentTable, drizzleOrm.eq(config.relationIdField, config.componentIdField)).where(
       drizzleOrm.and(
-        drizzleOrm.eq(config2.relationTable.tenantId, scopes.tenantId),
-        drizzleOrm.eq(config2.relationTable.projectId, scopes.projectId),
-        drizzleOrm.inArray(config2.relationTable.agentId, agentIds)
+        drizzleOrm.eq(config.relationTable.tenantId, scopes.tenantId),
+        drizzleOrm.eq(config.relationTable.projectId, scopes.projectId),
+        drizzleOrm.inArray(config.relationTable.agentId, agentIds)
       )
     );
     for (const component of results) {
@@ -3346,7 +3329,8 @@ var getFullGraphDefinition = (db) => async ({
         lastHealthCheck: tools.lastHealthCheck,
         lastError: tools.lastError,
         availableTools: tools.availableTools,
-        lastToolsSync: tools.lastToolsSync
+        lastToolsSync: tools.lastToolsSync,
+        selectedTools: agentToolRelations.selectedTools
       }).from(agentToolRelations).innerJoin(tools, drizzleOrm.eq(agentToolRelations.toolId, tools.id)).where(
         drizzleOrm.and(drizzleOrm.eq(agentToolRelations.tenantId, tenantId), drizzleOrm.eq(agentToolRelations.agentId, agentId))
       );
@@ -3366,6 +3350,12 @@ var getFullGraphDefinition = (db) => async ({
       const agentArtifactComponentIds = agentArtifactComponentRelations.map(
         (rel) => rel.artifactComponentId
       );
+      const selectedTools = {};
+      agentTools.forEach((tool2) => {
+        if (tool2.selectedTools && Array.isArray(tool2.selectedTools)) {
+          selectedTools[tool2.id] = tool2.selectedTools;
+        }
+      });
       return {
         id: agent.id,
         name: agent.name,
@@ -3377,6 +3367,7 @@ var getFullGraphDefinition = (db) => async ({
         canDelegateTo,
         dataComponents: agentDataComponentIds,
         artifactComponents: agentArtifactComponentIds,
+        ...Object.keys(selectedTools).length > 0 && { selectedTools },
         tools: agentTools.map((tool2) => ({
           id: tool2.id,
           name: tool2.name,
@@ -3425,14 +3416,19 @@ var getFullGraphDefinition = (db) => async ({
     } else {
       const toolsData = agent.tools || [];
       const toolIds = [];
+      const agentSelectedTools = {};
       for (const tool2 of toolsData) {
         toolsObject[tool2.id] = tool2;
         toolIds.push(tool2.id);
+        if (tool2.selectedTools !== null && tool2.selectedTools !== void 0) {
+          agentSelectedTools[tool2.id] = tool2.selectedTools;
+        }
       }
       agentsObject[agent.id] = {
         ...agent,
-        tools: toolIds
+        tools: toolIds,
         // Replace tool objects with tool IDs
+        ...Object.keys(agentSelectedTools).length > 0 && { selectedTools: agentSelectedTools }
       };
     }
   }
@@ -4958,6 +4954,7 @@ var addToolToAgent = (db) => async (params) => {
     projectId: params.scopes.projectId,
     agentId: params.agentId,
     toolId: params.toolId,
+    selectedTools: params.selectedTools,
     createdAt: now,
     updatedAt: now
   }).returning();
@@ -4986,7 +4983,15 @@ var upsertAgentToolRelation = (db) => async (params) => {
   if (!existing) {
     return await addToolToAgent(db)(params);
   }
-  return existing;
+  return await updateAgentToolRelation(db)({
+    scopes: params.scopes,
+    relationId: existing.id,
+    data: {
+      agentId: params.agentId,
+      toolId: params.toolId,
+      selectedTools: params.selectedTools
+    }
+  });
 };
 var updateToolStatus = (db) => async (params) => {
   return updateTool(db)({
@@ -5344,11 +5349,13 @@ var createFullGraphServerSide = (db, logger11 = defaultLogger) => async (scopes,
           agentToolPromises.push(
             (async () => {
               try {
+                const selectedTools = agentData.selectedTools?.[toolId];
                 logger11.info({ agentId, toolId }, "Processing agent-tool relation");
                 await upsertAgentToolRelation(db)({
                   scopes: { tenantId, projectId },
                   agentId,
-                  toolId
+                  toolId,
+                  selectedTools
                 });
                 logger11.info({ agentId, toolId }, "Agent-tool relation processed successfully");
               } catch (error) {
@@ -5816,11 +5823,13 @@ var updateFullGraphServerSide = (db, logger11 = defaultLogger) => async (scopes,
           agentToolPromises.push(
             (async () => {
               try {
+                const selectedTools = agentData.selectedTools?.[toolId];
                 await createAgentToolRelation(db)({
                   scopes: { tenantId, projectId },
                   data: {
                     agentId,
-                    toolId
+                    toolId,
+                    selectedTools
                   }
                 });
                 logger11.info({ agentId, toolId }, "Agent-tool relation created");
@@ -6593,8 +6602,8 @@ var tryWellKnownEndpoints = async (baseUrl, logger11) => {
   return null;
 };
 var checkForOAuthEndpoints = async (serverUrl, logger11) => {
-  const config2 = await discoverOAuthEndpoints(serverUrl, logger11);
-  return config2 !== null;
+  const config = await discoverOAuthEndpoints(serverUrl, logger11);
+  return config !== null;
 };
 var discoverOAuthEndpoints = async (serverUrl, logger11) => {
   try {
@@ -6974,7 +6983,7 @@ var McpClient = class {
   }
   async connect() {
     if (this.connected) return;
-    await tsPattern.match(this.serverConfig).with({ type: MCPTransportType.streamableHttp }, (config2) => this.connectHttp(config2)).with({ type: MCPTransportType.sse }, (config2) => this.connectSSE(config2)).exhaustive();
+    await tsPattern.match(this.serverConfig).with({ type: MCPTransportType.streamableHttp }, (config) => this.connectHttp(config)).with({ type: MCPTransportType.sse }, (config) => this.connectSSE(config)).exhaustive();
     this.connected = true;
     const close = this.client.onclose;
     this.client.onclose = () => {
@@ -6986,25 +6995,25 @@ var McpClient = class {
     exitHook.asyncExitHook(() => this.disconnect(), { wait: 5e3 });
     process.on("SIGTERM", () => exitHook.gracefulExit());
   }
-  async connectSSE(config2) {
-    const url = typeof config2.url === "string" ? config2.url : config2.url.toString();
+  async connectSSE(config) {
+    const url = typeof config.url === "string" ? config.url : config.url.toString();
     this.transport = new sse_js.SSEClientTransport(new URL(url), {
-      eventSourceInit: config2.eventSourceInit,
+      eventSourceInit: config.eventSourceInit,
       requestInit: {
-        headers: config2.headers || {}
+        headers: config.headers || {}
       }
     });
     await this.client.connect(this.transport, {
-      timeout: config2.timeout ?? this.timeout
+      timeout: config.timeout ?? this.timeout
     });
   }
-  async connectHttp(config2) {
-    const { url, requestInit } = config2;
+  async connectHttp(config) {
+    const { url, requestInit } = config;
     const mergedRequestInit = {
       ...requestInit,
       headers: {
         ...requestInit?.headers || {},
-        ...config2.headers || {}
+        ...config.headers || {}
       }
     };
     const urlObj = new URL(url);
@@ -7015,9 +7024,9 @@ var McpClient = class {
         maxReconnectionDelay: 3e4,
         initialReconnectionDelay: 1e3,
         reconnectionDelayGrowFactor: 1.5,
-        ...config2.reconnectionOptions
+        ...config.reconnectionOptions
       },
-      sessionId: config2.sessionId
+      sessionId: config.sessionId
     });
     await this.client.connect(this.transport, { timeout: 3e3 });
   }
@@ -7044,11 +7053,18 @@ var McpClient = class {
   }
   async selectTools() {
     const { tools: tools2 } = await this.client.listTools({ timeout: this.timeout });
-    const { activeTools } = this.serverConfig;
-    if (!activeTools) return tools2;
+    const { selectedTools, activeTools } = this.serverConfig;
+    let toolsToFilter;
+    if (selectedTools && selectedTools.length > 0) {
+      toolsToFilter = selectedTools;
+    } else if (activeTools && activeTools.length > 0) {
+      toolsToFilter = activeTools;
+    } else {
+      return tools2;
+    }
     const toolNames = tools2.map((tool2) => tool2.name);
-    this.validateSelectedTools(toolNames, activeTools);
-    return tools2.filter((tool2) => activeTools.includes(tool2.name));
+    this.validateSelectedTools(toolNames, toolsToFilter);
+    return tools2.filter((tool2) => toolsToFilter.includes(tool2.name));
   }
   async tools() {
     const tools2 = await this.selectTools();
@@ -7114,52 +7130,7 @@ var McpClient = class {
     return results;
   }
 };
-dotenv__namespace.config({ quiet: true });
-var environmentSchema = zod.z.enum(["development", "pentest", "production", "test"]);
-var criticalEnv = zod.z.object({
-  ENVIRONMENT: environmentSchema
-}).parse(process.env);
-var loadEnvFile = () => {
-  const envPath = path__default.default.resolve(process.cwd(), `.env.${criticalEnv.ENVIRONMENT}.nonsecret`);
-  if (fs__default.default.existsSync(envPath)) {
-    const envConfig = dotenv__namespace.parse(fs__default.default.readFileSync(envPath));
-    for (const k in envConfig) {
-      if (!(k in process.env)) {
-        process.env[k] = envConfig[k];
-      }
-    }
-  }
-};
-loadEnvFile();
-var envSchema = zod.z.object({
-  ENVIRONMENT: zod.z.enum(["development", "production", "pentest", "test"]).optional(),
-  DB_FILE_NAME: zod.z.string().default("file:../../local.db"),
-  OTEL_TRACES_FORCE_FLUSH_ENABLED: zod.z.stringbool().optional()
-});
-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/utils/tracer.ts
 var logger4 = getLogger("tracer");
-var FORCE_FLUSH_ENVIRONMENTS = ["development"];
-var BASE = "inkeep-chat";
-var SERVICE_NAME = "inkeep-chat";
-var SERVICE_VERSION = "1.0.0";
-var createSpanName = (suffix) => `${BASE}.${suffix}`;
 var createNoOpSpan = () => ({
   setAttributes: () => ({}),
   recordException: () => ({}),
@@ -7188,8 +7159,7 @@ var noopTracer = {
     return createNoOpSpan();
   }
 };
-var globalTracerInstance = null;
-function handleSpanError(span, error, logger11, logMessage) {
+function setSpanWithError(span, error, logger11, logMessage) {
   const errorMessage = error instanceof Error ? error.message : String(error);
   span.recordException(error);
   span.setStatus({
@@ -7200,45 +7170,21 @@ function handleSpanError(span, error, logger11, logMessage) {
     logger11.error({ error: errorMessage }, logMessage);
   }
 }
-function getGlobalTracer() {
-  if (!globalTracerInstance) {
-    try {
-      globalTracerInstance = api.trace.getTracer(SERVICE_NAME, SERVICE_VERSION);
-    } catch (error) {
-      logger4.debug(
-        { error: error instanceof Error ? error.message : "Unknown error" },
-        "OpenTelemetry tracer not available, using no-op tracer"
-      );
-      globalTracerInstance = noopTracer;
-    }
-  }
-  return globalTracerInstance;
-}
-async function forceFlushTracer() {
-  const isOtelTracesForceFlushEnabled = env.OTEL_TRACES_FORCE_FLUSH_ENABLED;
-  const isForceFlushEnvironment = env.ENVIRONMENT && FORCE_FLUSH_ENVIRONMENTS.includes(env.ENVIRONMENT);
-  const shouldForceFlush = isOtelTracesForceFlushEnabled === true || isOtelTracesForceFlushEnabled == null && isForceFlushEnvironment;
-  if (!shouldForceFlush) {
-    return;
-  }
+function getTracer(serviceName, serviceVersion) {
   try {
-    const tracerProvider = api.trace.getTracerProvider();
-    if (tracerProvider && "forceFlush" in tracerProvider && typeof tracerProvider.forceFlush === "function") {
-      await tracerProvider.forceFlush();
-      logger4.debug(
-        { message: "Tracer provider force flush completed" },
-        "Tracer provider force flush completed"
-      );
-    } else {
-      logger4.debug(
-        { message: "Tracer provider does not support force flush or is not available" },
-        "Tracer provider does not support force flush or is not available"
-      );
-    }
-  } catch (error) {
-    logger4.warn({ error }, "Failed to force flush tracer");
+    return api.trace.getTracer(serviceName, serviceVersion);
+  } catch (_error) {
+    logger4.debug({}, "OpenTelemetry tracer not available, using no-op tracer");
+    return noopTracer;
   }
 }
+
+// package.json
+var package_default = {
+  version: "0.0.0-dev-20250911210702"};
+
+// src/utils/tracer.ts
+var tracer = getTracer("agents-core", package_default.version);
 var logger5 = getLogger("context-cache");
 var ContextCache = class {
   constructor(tenantId, projectId, dbClient) {
@@ -7434,7 +7380,6 @@ var ContextCache = class {
 
 // src/context/ContextResolver.ts
 var logger6 = getLogger("context-resolver");
-var tracer = getGlobalTracer();
 var ContextResolver = class {
   constructor(tenantId, projectId, dbClient, credentialStoreRegistry) {
     __publicField(this, "fetcher");
@@ -7467,7 +7412,7 @@ var ContextResolver = class {
       "Starting context resolution"
     );
     return tracer.startActiveSpan(
-      createSpanName("context.resolve"),
+      "context.resolve",
       {
         attributes: {
           "context.config_id": contextConfig2.id,
@@ -7619,7 +7564,7 @@ var ContextResolver = class {
           return result;
         } catch (error) {
           const durationMs = Date.now() - startTime;
-          handleSpanError(parentSpan, error);
+          setSpanWithError(parentSpan, error);
           logger6.error(
             {
               contextConfigId: contextConfig2.id,
@@ -7640,7 +7585,7 @@ var ContextResolver = class {
    */
   async resolveSingleFetchDefinition(contextConfig2, definition, templateKey, options, requestHash, result) {
     return tracer.startActiveSpan(
-      createSpanName("context-resolver.resolve_single_fetch_definition"),
+      "context-resolver.resolve_single_fetch_definition",
       {
         attributes: {
           "context.definition_id": definition.id,
@@ -7725,7 +7670,7 @@ var ContextResolver = class {
             "Context variable resolved and cached"
           );
         } catch (error) {
-          handleSpanError(parentSpan, error);
+          setSpanWithError(parentSpan, error);
           throw error;
         } finally {
           parentSpan.end();
@@ -8316,21 +8261,21 @@ var ContextFetcher = class {
   /**
    * Perform HTTP request
    */
-  async performRequest(config2) {
+  async performRequest(config) {
     const startTime = Date.now();
     try {
       logger8.debug(
         {
-          url: config2.url,
-          method: config2.method
+          url: config.url,
+          method: config.method
         },
         "Performing HTTP request"
       );
-      const response = await fetch(config2.url, {
-        method: config2.method,
-        headers: config2.headers,
-        body: config2.body ? JSON.stringify(config2.body) : void 0,
-        signal: AbortSignal.timeout(config2.timeout || this.defaultTimeout)
+      const response = await fetch(config.url, {
+        method: config.method,
+        headers: config.headers,
+        body: config.body ? JSON.stringify(config.body) : void 0,
+        signal: AbortSignal.timeout(config.timeout || this.defaultTimeout)
       });
       if (!response.ok) {
         const errorText = await response.text();
@@ -8350,7 +8295,7 @@ var ContextFetcher = class {
       }
       return {
         data,
-        source: config2.url,
+        source: config.url,
         durationMs
       };
     } catch (error) {
@@ -8358,7 +8303,7 @@ var ContextFetcher = class {
       const requestError = error instanceof Error ? error : new Error("Unknown error");
       logger8.warn(
         {
-          url: config2.url,
+          url: config.url,
           error: requestError.message,
           durationMs
         },
@@ -8446,7 +8391,6 @@ var ContextFetcher = class {
   }
 };
 var logger9 = getLogger("context");
-var tracer2 = getGlobalTracer();
 async function determineContextTrigger(tenantId, projectId, conversationId, dbClient) {
   const conversation = await getConversation(dbClient)({
     scopes: { tenantId, projectId },
@@ -8477,8 +8421,8 @@ async function handleContextConfigChange(tenantId, projectId, conversationId, gr
   }
 }
 async function handleContextResolution(tenantId, projectId, conversationId, graphId, requestContext, dbClient, credentialStores) {
-  return tracer2.startActiveSpan(
-    createSpanName("context.handle_context_resolution"),
+  return tracer.startActiveSpan(
+    "context.handle_context_resolution",
     {
       attributes: {
         "context.request_context_keys": Object.keys(requestContext)
@@ -8553,7 +8497,6 @@ async function handleContextResolution(tenantId, projectId, conversationId, grap
         } else {
           parentSpan.setStatus({ code: api.SpanStatusCode.OK });
         }
-        await forceFlushTracer();
         logger9.info(
           {
             conversationId,
@@ -8571,15 +8514,11 @@ async function handleContextResolution(tenantId, projectId, conversationId, grap
         return resolvedContext;
       } catch (error) {
         const errorMessage = error instanceof Error ? error.message : "Unknown error";
-        parentSpan.recordException(error);
         parentSpan.setAttributes({
           "context.final_status": "failed",
           "context.error_message": errorMessage
         });
-        parentSpan.setStatus({
-          code: api.SpanStatusCode.ERROR,
-          message: errorMessage
-        });
+        setSpanWithError(parentSpan, error);
         logger9.error(
           {
             error: errorMessage,
@@ -8593,7 +8532,6 @@ async function handleContextResolution(tenantId, projectId, conversationId, grap
           },
           "Failed to resolve context, proceeding without context resolution"
         );
-        await forceFlushTracer();
         return null;
       } finally {
         parentSpan.end();
@@ -8943,13 +8881,13 @@ function isSupportedAuthMode(mode) {
   return SUPPORTED_AUTH_MODES.includes(mode);
 }
 var NangoCredentialStore = class {
-  constructor(id, config2) {
+  constructor(id, config) {
     __publicField(this, "id");
     __publicField(this, "type", CredentialStoreType.nango);
     __publicField(this, "nangoConfig");
     __publicField(this, "nangoClient");
     this.id = id;
-    this.nangoConfig = config2;
+    this.nangoConfig = config;
     this.nangoClient = new node.Nango({
       secretKey: this.nangoConfig.secretKey,
       host: this.nangoConfig.apiUrl
@@ -9176,8 +9114,8 @@ var NangoCredentialStore = class {
     }
   }
 };
-function createNangoCredentialStore(id, config2) {
-  const nangoSecretKey = config2?.secretKey || process.env.NANGO_SECRET_KEY;
+function createNangoCredentialStore(id, config) {
+  const nangoSecretKey = config?.secretKey || process.env.NANGO_SECRET_KEY;
   if (!nangoSecretKey || nangoSecretKey === "your_nango_secret_key" || nangoSecretKey.includes("mock")) {
     throw new Error(
       "NANGO_SECRET_KEY environment variable is required and must be a real Nango secret key (not mock/placeholder)"
@@ -9185,7 +9123,7 @@ function createNangoCredentialStore(id, config2) {
   }
   return new NangoCredentialStore(id, {
     apiUrl: "https://api.nango.dev",
-    ...config2,
+    ...config,
     secretKey: nangoSecretKey
   });
 }
@@ -9275,7 +9213,6 @@ exports.ArtifactComponentApiUpdateSchema = ArtifactComponentApiUpdateSchema;
 exports.ArtifactComponentInsertSchema = ArtifactComponentInsertSchema;
 exports.ArtifactComponentSelectSchema = ArtifactComponentSelectSchema;
 exports.ArtifactComponentUpdateSchema = ArtifactComponentUpdateSchema;
-exports.BASE = BASE;
 exports.ConsoleLogger = ConsoleLogger;
 exports.ContextCache = ContextCache;
 exports.ContextCacheApiInsertSchema = ContextCacheApiInsertSchema;
@@ -9372,8 +9309,6 @@ exports.ProjectModelSchema = ProjectModelSchema;
 exports.ProjectSelectSchema = ProjectSelectSchema;
 exports.ProjectUpdateSchema = ProjectUpdateSchema;
 exports.RemovedResponseSchema = RemovedResponseSchema;
-exports.SERVICE_NAME = SERVICE_NAME;
-exports.SERVICE_VERSION = SERVICE_VERSION;
 exports.SingleResponseSchema = SingleResponseSchema;
 exports.StatusComponentSchema = StatusComponentSchema;
 exports.StatusUpdateSchema = StatusUpdateSchema;
@@ -9473,7 +9408,6 @@ exports.createMessage = createMessage;
 exports.createNangoCredentialStore = createNangoCredentialStore;
 exports.createOrGetConversation = createOrGetConversation;
 exports.createProject = createProject;
-exports.createSpanName = createSpanName;
 exports.createTask = createTask;
 exports.createTool = createTool;
 exports.createValidatedDataAccess = createValidatedDataAccess;
@@ -9514,7 +9448,6 @@ exports.externalAgentsRelations = externalAgentsRelations;
 exports.extractPublicId = extractPublicId;
 exports.fetchComponentRelationships = fetchComponentRelationships;
 exports.fetchDefinition = fetchDefinition;
-exports.forceFlushTracer = forceFlushTracer;
 exports.generateAndCreateApiKey = generateAndCreateApiKey;
 exports.generateApiKey = generateApiKey;
 exports.generateIdFromName = generateIdFromName;
@@ -9560,7 +9493,6 @@ exports.getExternalAgentByUrl = getExternalAgentByUrl;
 exports.getExternalAgentRelations = getExternalAgentRelations;
 exports.getFullGraph = getFullGraph;
 exports.getFullGraphDefinition = getFullGraphDefinition;
-exports.getGlobalTracer = getGlobalTracer;
 exports.getGraphAgentInfos = getGraphAgentInfos;
 exports.getHealthyToolsForAgent = getHealthyToolsForAgent;
 exports.getLedgerArtifacts = getLedgerArtifacts;
@@ -9577,12 +9509,12 @@ exports.getTask = getTask;
 exports.getToolById = getToolById;
 exports.getToolsByStatus = getToolsByStatus;
 exports.getToolsForAgent = getToolsForAgent;
+exports.getTracer = getTracer;
 exports.getVisibleMessages = getVisibleMessages;
 exports.graphHasArtifactComponents = graphHasArtifactComponents;
 exports.handleApiError = handleApiError;
 exports.handleContextConfigChange = handleContextConfigChange;
 exports.handleContextResolution = handleContextResolution;
-exports.handleSpanError = handleSpanError;
 exports.hasApiKey = hasApiKey;
 exports.hasContextConfig = hasContextConfig;
 exports.hasCredentialReference = hasCredentialReference;
@@ -9643,6 +9575,7 @@ exports.resourceIdSchema = resourceIdSchema;
 exports.setActiveAgentForConversation = setActiveAgentForConversation;
 exports.setActiveAgentForThread = setActiveAgentForThread;
 exports.setCacheEntry = setCacheEntry;
+exports.setSpanWithError = setSpanWithError;
 exports.taskRelations = taskRelations;
 exports.taskRelationsRelations = taskRelationsRelations;
 exports.tasks = tasks;