@inkeep/agents-core 0.7.1 → 0.8.0

package/dist/{chunk-Y2L44YJB.js → chunk-E4ZLV4GU.js}

@@ -436,6 +436,7 @@ var agentToolRelations = sqliteTable(
     id: text("id").notNull(),
     toolId: text("tool_id").notNull(),
     selectedTools: blob("selected_tools", { mode: "json" }).$type(),
+    headers: blob("headers", { mode: "json" }).$type(),
     createdAt: text("created_at").notNull().default(sql`CURRENT_TIMESTAMP`),
     updatedAt: text("updated_at").notNull().default(sql`CURRENT_TIMESTAMP`)
   },

package/dist/{chunk-6LGKJVQN.js → chunk-P5LTXXFH.js}

@@ -1,4 +1,4 @@
-import { agents, agentRelations, agentGraph, tasks, taskRelations, tools, conversations, messages, contextCache, dataComponents, agentDataComponents, artifactComponents, agentArtifactComponents, externalAgents, apiKeys, credentialReferences, contextConfigs, agentToolRelations, ledgerArtifacts, projects } from './chunk-Y2L44YJB.js';
+import { agents, agentRelations, agentGraph, tasks, taskRelations, tools, conversations, messages, contextCache, dataComponents, agentDataComponents, artifactComponents, agentArtifactComponents, externalAgents, apiKeys, credentialReferences, contextConfigs, agentToolRelations, ledgerArtifacts, projects } from './chunk-E4ZLV4GU.js';
 import { VALID_RELATION_TYPES, MCPTransportType, TOOL_STATUS_VALUES, CredentialStoreType, MCPServerType } from './chunk-YFHT5M2R.js';
 import { z } from '@hono/zod-openapi';
 import { createSelectSchema, createInsertSchema } from 'drizzle-zod';
@@ -416,7 +416,8 @@ var AgentToolRelationInsertSchema = createInsertSchema(agentToolRelations).exten
   id: resourceIdSchema,
   agentId: resourceIdSchema,
   toolId: resourceIdSchema,
-  selectedTools: z.array(z.string()).nullish()
+  selectedTools: z.array(z.string()).nullish(),
+  headers: z.record(z.string(), z.string()).nullish()
 });
 var AgentToolRelationUpdateSchema = AgentToolRelationInsertSchema.partial();
 var AgentToolRelationApiSelectSchema = createGraphScopedApiSchema(
@@ -452,7 +453,8 @@ var StatusUpdateSchema = z.object({
 });
 var CanUseItemSchema = z.object({
   toolId: z.string(),
-  toolSelection: z.array(z.string()).nullable().optional()
+  toolSelection: z.array(z.string()).nullish(),
+  headers: z.record(z.string(), z.string()).nullish()
 });
 var FullGraphAgentInsertSchema = AgentApiInsertSchema.extend({
   type: z.literal("internal"),

package/dist/{chunk-LLC4XKH2.js → chunk-VZ62ZJTI.js}

@@ -1,4 +1,4 @@
-import { FullGraphDefinitionSchema, resourceIdSchema, MAX_ID_LENGTH } from './chunk-6LGKJVQN.js';
+import { FullGraphDefinitionSchema, resourceIdSchema, MAX_ID_LENGTH } from './chunk-P5LTXXFH.js';
 
 // src/validation/graphFull.ts
 function isInternalAgent(agent) {

package/dist/client-exports.cjs

@@ -411,6 +411,7 @@ var agentToolRelations = sqliteCore.sqliteTable(
     id: sqliteCore.text("id").notNull(),
     toolId: sqliteCore.text("tool_id").notNull(),
     selectedTools: sqliteCore.blob("selected_tools", { mode: "json" }).$type(),
+    headers: sqliteCore.blob("headers", { 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`)
   },
@@ -1346,7 +1347,8 @@ var AgentToolRelationInsertSchema = drizzleZod.createInsertSchema(agentToolRelat
   id: resourceIdSchema,
   agentId: resourceIdSchema,
   toolId: resourceIdSchema,
-  selectedTools: zodOpenapi.z.array(zodOpenapi.z.string()).nullish()
+  selectedTools: zodOpenapi.z.array(zodOpenapi.z.string()).nullish(),
+  headers: zodOpenapi.z.record(zodOpenapi.z.string(), zodOpenapi.z.string()).nullish()
 });
 var AgentToolRelationUpdateSchema = AgentToolRelationInsertSchema.partial();
 createGraphScopedApiSchema(
@@ -1382,7 +1384,8 @@ var StatusUpdateSchema = zodOpenapi.z.object({
 });
 var CanUseItemSchema = zodOpenapi.z.object({
   toolId: zodOpenapi.z.string(),
-  toolSelection: zodOpenapi.z.array(zodOpenapi.z.string()).nullable().optional()
+  toolSelection: zodOpenapi.z.array(zodOpenapi.z.string()).nullish(),
+  headers: zodOpenapi.z.record(zodOpenapi.z.string(), zodOpenapi.z.string()).nullish()
 });
 var FullGraphAgentInsertSchema = AgentApiInsertSchema.extend({
   type: zodOpenapi.z.literal("internal"),

package/dist/client-exports.d.cts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
-import { C as ConversationHistoryConfig, F as FullGraphAgentInsertSchema } from './utility-DPZ_yTkY.cjs';
-export { c as AgentStopWhen, A as AgentStopWhenSchema, d as CredentialStoreType, b as GraphStopWhen, G as GraphStopWhenSchema, M as MCPTransportType, a as StopWhen, S as StopWhenSchema } from './utility-DPZ_yTkY.cjs';
+import { C as ConversationHistoryConfig, F as FullGraphAgentInsertSchema } from './utility-CT2pq49U.cjs';
+export { c as AgentStopWhen, A as AgentStopWhenSchema, d as CredentialStoreType, b as GraphStopWhen, G as GraphStopWhenSchema, M as MCPTransportType, a as StopWhen, S as StopWhenSchema } from './utility-CT2pq49U.cjs';
 import 'drizzle-zod';
 import 'drizzle-orm/sqlite-core';
 import '@hono/zod-openapi';
@@ -200,6 +200,7 @@ declare const FullGraphDefinitionSchema: z.ZodObject<{
         canUse: z.ZodArray<z.ZodObject<{
             toolId: z.ZodString;
             toolSelection: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
+            headers: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodString>>>;
         }, z.core.$strip>>;
         dataComponents: z.ZodOptional<z.ZodArray<z.ZodString>>;
         artifactComponents: z.ZodOptional<z.ZodArray<z.ZodString>>;

package/dist/client-exports.d.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
-import { C as ConversationHistoryConfig, F as FullGraphAgentInsertSchema } from './utility-DPZ_yTkY.js';
-export { c as AgentStopWhen, A as AgentStopWhenSchema, d as CredentialStoreType, b as GraphStopWhen, G as GraphStopWhenSchema, M as MCPTransportType, a as StopWhen, S as StopWhenSchema } from './utility-DPZ_yTkY.js';
+import { C as ConversationHistoryConfig, F as FullGraphAgentInsertSchema } from './utility-CT2pq49U.js';
+export { c as AgentStopWhen, A as AgentStopWhenSchema, d as CredentialStoreType, b as GraphStopWhen, G as GraphStopWhenSchema, M as MCPTransportType, a as StopWhen, S as StopWhenSchema } from './utility-CT2pq49U.js';
 import 'drizzle-zod';
 import 'drizzle-orm/sqlite-core';
 import '@hono/zod-openapi';
@@ -200,6 +200,7 @@ declare const FullGraphDefinitionSchema: z.ZodObject<{
         canUse: z.ZodArray<z.ZodObject<{
             toolId: z.ZodString;
             toolSelection: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
+            headers: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodString>>>;
         }, z.core.$strip>>;
         dataComponents: z.ZodOptional<z.ZodArray<z.ZodString>>;
         artifactComponents: z.ZodOptional<z.ZodArray<z.ZodString>>;

package/dist/client-exports.js

@@ -1,5 +1,5 @@
-import { FullGraphAgentInsertSchema } from './chunk-6LGKJVQN.js';
-export { AgentStopWhenSchema, GraphStopWhenSchema, StopWhenSchema } from './chunk-6LGKJVQN.js';
+import { FullGraphAgentInsertSchema } from './chunk-P5LTXXFH.js';
+export { AgentStopWhenSchema, GraphStopWhenSchema, StopWhenSchema } from './chunk-P5LTXXFH.js';
 import { CredentialStoreType } from './chunk-YFHT5M2R.js';
 export { CredentialStoreType, MCPTransportType } from './chunk-YFHT5M2R.js';
 import { z } from 'zod';

package/dist/db/schema.cjs

@@ -391,6 +391,7 @@ var agentToolRelations = sqliteCore.sqliteTable(
     id: sqliteCore.text("id").notNull(),
     toolId: sqliteCore.text("tool_id").notNull(),
     selectedTools: sqliteCore.blob("selected_tools", { mode: "json" }).$type(),
+    headers: sqliteCore.blob("headers", { 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`)
   },

package/dist/db/schema.d.cts

@@ -1,7 +1,7 @@
 import 'drizzle-orm';
 import 'drizzle-orm/sqlite-core';
-import '../utility-DPZ_yTkY.cjs';
-export { k as agentArtifactComponents, N as agentArtifactComponentsRelations, i as agentDataComponents, P as agentDataComponentsRelations, a as agentGraph, E as agentGraphRelations, e as agentRelations, R as agentRelationsRelations, m as agentToolRelations, H as agentToolRelationsRelations, d as agents, D as agentsRelations, r as apiKeys, G as apiKeysRelations, j as artifactComponents, M as artifactComponentsRelations, b as contextCache, C as contextCacheRelations, c as contextConfigs, B as contextConfigsRelations, n as conversations, K as conversationsRelations, u as credentialReferences, I as credentialReferencesRelations, h as dataComponents, O as dataComponentsRelations, f as externalAgents, F as externalAgentsRelations, q as ledgerArtifacts, w as ledgerArtifactsContextIdIdx, Q as ledgerArtifactsRelations, x as ledgerArtifactsTaskContextNameUnique, v as ledgerArtifactsTaskIdIdx, o as messages, L as messagesRelations, p as projects, z as projectsRelations, g as taskRelations, A as taskRelationsRelations, t as tasks, y as tasksRelations, l as tools, J as toolsRelations } from '../schema-C138dGBf.cjs';
+import '../utility-CT2pq49U.cjs';
+export { k as agentArtifactComponents, N as agentArtifactComponentsRelations, i as agentDataComponents, P as agentDataComponentsRelations, a as agentGraph, E as agentGraphRelations, e as agentRelations, R as agentRelationsRelations, m as agentToolRelations, H as agentToolRelationsRelations, d as agents, D as agentsRelations, r as apiKeys, G as apiKeysRelations, j as artifactComponents, M as artifactComponentsRelations, b as contextCache, C as contextCacheRelations, c as contextConfigs, B as contextConfigsRelations, n as conversations, K as conversationsRelations, u as credentialReferences, I as credentialReferencesRelations, h as dataComponents, O as dataComponentsRelations, f as externalAgents, F as externalAgentsRelations, q as ledgerArtifacts, w as ledgerArtifactsContextIdIdx, Q as ledgerArtifactsRelations, x as ledgerArtifactsTaskContextNameUnique, v as ledgerArtifactsTaskIdIdx, o as messages, L as messagesRelations, p as projects, z as projectsRelations, g as taskRelations, A as taskRelationsRelations, t as tasks, y as tasksRelations, l as tools, J as toolsRelations } from '../schema-BF9NpAv5.cjs';
 import 'zod';
 import 'drizzle-zod';
 import '@hono/zod-openapi';

package/dist/db/schema.d.ts

@@ -1,7 +1,7 @@
 import 'drizzle-orm';
 import 'drizzle-orm/sqlite-core';
-import '../utility-DPZ_yTkY.js';
-export { k as agentArtifactComponents, N as agentArtifactComponentsRelations, i as agentDataComponents, P as agentDataComponentsRelations, a as agentGraph, E as agentGraphRelations, e as agentRelations, R as agentRelationsRelations, m as agentToolRelations, H as agentToolRelationsRelations, d as agents, D as agentsRelations, r as apiKeys, G as apiKeysRelations, j as artifactComponents, M as artifactComponentsRelations, b as contextCache, C as contextCacheRelations, c as contextConfigs, B as contextConfigsRelations, n as conversations, K as conversationsRelations, u as credentialReferences, I as credentialReferencesRelations, h as dataComponents, O as dataComponentsRelations, f as externalAgents, F as externalAgentsRelations, q as ledgerArtifacts, w as ledgerArtifactsContextIdIdx, Q as ledgerArtifactsRelations, x as ledgerArtifactsTaskContextNameUnique, v as ledgerArtifactsTaskIdIdx, o as messages, L as messagesRelations, p as projects, z as projectsRelations, g as taskRelations, A as taskRelationsRelations, t as tasks, y as tasksRelations, l as tools, J as toolsRelations } from '../schema-DbxoPUgw.js';
+import '../utility-CT2pq49U.js';
+export { k as agentArtifactComponents, N as agentArtifactComponentsRelations, i as agentDataComponents, P as agentDataComponentsRelations, a as agentGraph, E as agentGraphRelations, e as agentRelations, R as agentRelationsRelations, m as agentToolRelations, H as agentToolRelationsRelations, d as agents, D as agentsRelations, r as apiKeys, G as apiKeysRelations, j as artifactComponents, M as artifactComponentsRelations, b as contextCache, C as contextCacheRelations, c as contextConfigs, B as contextConfigsRelations, n as conversations, K as conversationsRelations, u as credentialReferences, I as credentialReferencesRelations, h as dataComponents, O as dataComponentsRelations, f as externalAgents, F as externalAgentsRelations, q as ledgerArtifacts, w as ledgerArtifactsContextIdIdx, Q as ledgerArtifactsRelations, x as ledgerArtifactsTaskContextNameUnique, v as ledgerArtifactsTaskIdIdx, o as messages, L as messagesRelations, p as projects, z as projectsRelations, g as taskRelations, A as taskRelationsRelations, t as tasks, y as tasksRelations, l as tools, J as toolsRelations } from '../schema-78jVngGh.js';
 import 'zod';
 import 'drizzle-zod';
 import '@hono/zod-openapi';

package/dist/db/schema.js

@@ -1 +1 @@
-export { agentArtifactComponents, agentArtifactComponentsRelations, agentDataComponents, agentDataComponentsRelations, agentGraph, agentGraphRelations, agentRelations, agentRelationsRelations, agentToolRelations, agentToolRelationsRelations, agents, agentsRelations, apiKeys, apiKeysRelations, artifactComponents, artifactComponentsRelations, contextCache, contextCacheRelations, contextConfigs, contextConfigsRelations, conversations, conversationsRelations, credentialReferences, credentialReferencesRelations, dataComponents, dataComponentsRelations, externalAgents, externalAgentsRelations, ledgerArtifacts, ledgerArtifactsContextIdIdx, ledgerArtifactsRelations, ledgerArtifactsTaskContextNameUnique, ledgerArtifactsTaskIdIdx, messages, messagesRelations, projects, projectsRelations, taskRelations, taskRelationsRelations, tasks, tasksRelations, tools, toolsRelations } from '../chunk-Y2L44YJB.js';
+export { agentArtifactComponents, agentArtifactComponentsRelations, agentDataComponents, agentDataComponentsRelations, agentGraph, agentGraphRelations, agentRelations, agentRelationsRelations, agentToolRelations, agentToolRelationsRelations, agents, agentsRelations, apiKeys, apiKeysRelations, artifactComponents, artifactComponentsRelations, contextCache, contextCacheRelations, contextConfigs, contextConfigsRelations, conversations, conversationsRelations, credentialReferences, credentialReferencesRelations, dataComponents, dataComponentsRelations, externalAgents, externalAgentsRelations, ledgerArtifacts, ledgerArtifactsContextIdIdx, ledgerArtifactsRelations, ledgerArtifactsTaskContextNameUnique, ledgerArtifactsTaskIdIdx, messages, messagesRelations, projects, projectsRelations, taskRelations, taskRelationsRelations, tasks, tasksRelations, tools, toolsRelations } from '../chunk-E4ZLV4GU.js';

package/dist/index.cjs

@@ -642,6 +642,7 @@ var agentToolRelations = sqliteCore.sqliteTable(
     id: sqliteCore.text("id").notNull(),
     toolId: sqliteCore.text("tool_id").notNull(),
     selectedTools: sqliteCore.blob("selected_tools", { mode: "json" }).$type(),
+    headers: sqliteCore.blob("headers", { 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`)
   },
@@ -1595,7 +1596,8 @@ var AgentToolRelationInsertSchema = drizzleZod.createInsertSchema(agentToolRelat
   id: resourceIdSchema,
   agentId: resourceIdSchema,
   toolId: resourceIdSchema,
-  selectedTools: zodOpenapi.z.array(zodOpenapi.z.string()).nullish()
+  selectedTools: zodOpenapi.z.array(zodOpenapi.z.string()).nullish(),
+  headers: zodOpenapi.z.record(zodOpenapi.z.string(), zodOpenapi.z.string()).nullish()
 });
 var AgentToolRelationUpdateSchema = AgentToolRelationInsertSchema.partial();
 var AgentToolRelationApiSelectSchema = createGraphScopedApiSchema(
@@ -1631,7 +1633,8 @@ var StatusUpdateSchema = zodOpenapi.z.object({
 });
 var CanUseItemSchema = zodOpenapi.z.object({
   toolId: zodOpenapi.z.string(),
-  toolSelection: zodOpenapi.z.array(zodOpenapi.z.string()).nullable().optional()
+  toolSelection: zodOpenapi.z.array(zodOpenapi.z.string()).nullish(),
+  headers: zodOpenapi.z.record(zodOpenapi.z.string(), zodOpenapi.z.string()).nullish()
 });
 var FullGraphAgentInsertSchema = AgentApiInsertSchema.extend({
   type: zodOpenapi.z.literal("internal"),
@@ -2139,13 +2142,25 @@ var _TemplateEngine = class _TemplateEngine {
           if (options.preserveUnresolved) {
             return match2;
           }
-          logger2.warn(
-            {
-              variable: trimmedPath,
-              availableKeys: Object.keys(context)
-            },
-            "Template variable not found in context"
-          );
+          if (trimmedPath.startsWith("requestContext.")) {
+            logger2.warn(
+              {
+                variable: trimmedPath,
+                availableKeys: Object.keys(context),
+                contextStructure: JSON.stringify(context, null, 2),
+                requestContextContent: context.requestContext ? JSON.stringify(context.requestContext, null, 2) : "undefined"
+              },
+              "RequestContext template variable debugging"
+            );
+          } else {
+            logger2.warn(
+              {
+                variable: trimmedPath,
+                availableKeys: Object.keys(context)
+              },
+              "Template variable not found in context"
+            );
+          }
           return "";
         }
         if (typeof result === "object") {
@@ -2758,7 +2773,8 @@ var createAgentToolRelation = (db) => async (params) => {
     graphId: params.scopes.graphId,
     agentId: params.data.agentId,
     toolId: params.data.toolId,
-    selectedTools: params.data.selectedTools
+    selectedTools: params.data.selectedTools,
+    headers: params.data.headers
   }).returning();
   return relation[0];
 };
@@ -2903,6 +2919,7 @@ var getToolsForAgent = (db) => async (params) => {
       agentId: agentToolRelations.agentId,
       toolId: agentToolRelations.toolId,
       selectedTools: agentToolRelations.selectedTools,
+      headers: agentToolRelations.headers,
       createdAt: agentToolRelations.createdAt,
       updatedAt: agentToolRelations.updatedAt,
       tool: {
@@ -2961,6 +2978,7 @@ var getAgentsForTool = (db) => async (params) => {
       agentId: agentToolRelations.agentId,
       toolId: agentToolRelations.toolId,
       selectedTools: agentToolRelations.selectedTools,
+      headers: agentToolRelations.headers,
       createdAt: agentToolRelations.createdAt,
       updatedAt: agentToolRelations.updatedAt,
       agent: {
@@ -3668,9 +3686,9 @@ var getFullGraphDefinition = (db) => async ({
         credentialReferenceId: tools.credentialReferenceId,
         tenantId: tools.tenantId,
         projectId: tools.projectId,
-        headers: tools.headers,
         imageUrl: tools.imageUrl,
-        selectedTools: agentToolRelations.selectedTools
+        selectedTools: agentToolRelations.selectedTools,
+        headers: agentToolRelations.headers
       }).from(agentToolRelations).innerJoin(
         tools,
         drizzleOrm.and(
@@ -3704,7 +3722,8 @@ var getFullGraphDefinition = (db) => async ({
       );
       const canUse = agentTools.map((tool2) => ({
         toolId: tool2.id,
-        toolSelection: tool2.selectedTools || null
+        toolSelection: tool2.selectedTools || null,
+        headers: tool2.headers || null
       }));
       return {
         id: agent.id,
@@ -5250,9 +5269,8 @@ var detectAuthenticationRequired = async ({
   error,
   logger: logger12
 }) => {
-  let hasOAuthEndpoints = false;
   try {
-    hasOAuthEndpoints = await checkForOAuthEndpoints(serverUrl, logger12);
+    const hasOAuthEndpoints = await checkForOAuthEndpoints(serverUrl, logger12);
     if (hasOAuthEndpoints) {
       logger12?.info(
         { toolId, serverUrl },
@@ -5276,27 +5294,30 @@ var detectAuthenticationRequired = async ({
     });
     if (response.status === 401) {
       const wwwAuth = response.headers.get("WWW-Authenticate");
-      if (wwwAuth && (wwwAuth.toLowerCase().includes("bearer") || wwwAuth.toLowerCase().includes("oauth") || wwwAuth.toLowerCase().includes("authorization_uri"))) {
-        logger12?.info(
-          { toolId, wwwAuth },
-          "OAuth authentication detected via WWW-Authenticate header"
-        );
-        return true;
+      if (wwwAuth) {
+        const authLower = wwwAuth.toLowerCase();
+        const hasActiveOAuthFlow = authLower.includes("authorization_uri") || authLower.includes("as_uri=") || authLower.includes("bearer") && (authLower.includes("scope=") || authLower.includes("error_uri="));
+        if (hasActiveOAuthFlow) {
+          logger12?.info(
+            { toolId, wwwAuth },
+            "Active OAuth flow detected via WWW-Authenticate parameters"
+          );
+          return true;
+        } else {
+          logger12?.debug(
+            { toolId, wwwAuth },
+            "Bearer authentication detected - likely simple token auth, not OAuth"
+          );
+        }
       }
     }
   } catch (fetchError) {
     logger12?.debug({ toolId, fetchError }, "Direct fetch authentication check failed");
   }
-  if (error.message.includes("401") || error.message.toLowerCase().includes("unauthorized")) {
-    if (hasOAuthEndpoints) {
-      logger12?.info(
-        { toolId, error: error.message },
-        "OAuth required: 401 error + OAuth endpoints detected"
-      );
-      return true;
-    }
-  }
-  logger12?.debug({ toolId, error: error.message }, "No OAuth authentication requirement detected");
+  logger12?.debug(
+    { toolId, error: error.message },
+    "No OAuth 2.1/PKCE authentication requirement detected"
+  );
   return false;
 };
 var ErrorCode = zodOpenapi.z.enum([
@@ -6006,6 +6027,7 @@ var addToolToAgent = (db) => async (params) => {
     agentId: params.agentId,
     toolId: params.toolId,
     selectedTools: params.selectedTools,
+    headers: params.headers,
     createdAt: now,
     updatedAt: now
   }).returning();
@@ -6042,7 +6064,8 @@ var upsertAgentToolRelation = (db) => async (params) => {
     data: {
       agentId: params.agentId,
       toolId: params.toolId,
-      selectedTools: params.selectedTools
+      selectedTools: params.selectedTools,
+      headers: params.headers
     }
   });
 };
@@ -6303,13 +6326,14 @@ var createFullGraphServerSide = (db, logger12 = defaultLogger) => async (scopes,
           agentToolPromises.push(
             (async () => {
               try {
-                const { toolId, toolSelection } = canUseItem;
+                const { toolId, toolSelection, headers } = canUseItem;
                 logger12.info({ agentId, toolId }, "Processing agent-tool relation");
                 await upsertAgentToolRelation(db)({
                   scopes: { tenantId, projectId, graphId: finalGraphId },
                   agentId,
                   toolId,
-                  selectedTools: toolSelection || void 0
+                  selectedTools: toolSelection || void 0,
+                  headers: headers || void 0
                 });
                 logger12.info({ agentId, toolId }, "Agent-tool relation processed successfully");
               } catch (error) {
@@ -6699,13 +6723,14 @@ var updateFullGraphServerSide = (db, logger12 = defaultLogger) => async (scopes,
           agentToolPromises.push(
             (async () => {
               try {
-                const { toolId, toolSelection } = canUseItem;
+                const { toolId, toolSelection, headers } = canUseItem;
                 await createAgentToolRelation(db)({
                   scopes: { tenantId, projectId, graphId: finalGraphId },
                   data: {
                     agentId,
                     toolId,
-                    selectedTools: toolSelection || void 0
+                    selectedTools: toolSelection || void 0,
+                    headers: headers || void 0
                   }
                 });
                 logger12.info({ agentId, toolId }, "Agent-tool relation created");