@inkeep/agents-sdk 0.7.1 → 0.8.0

package/dist/index.cjs

@@ -439,7 +439,30 @@ var Tool = class {
 
 // src/utils/tool-normalization.ts
 function isAgentMcpConfig(value) {
-  return value !== null && typeof value === "object" && "server" in value && "selectedTools" in value && Array.isArray(value.selectedTools) && value.server && typeof value.server === "object";
+  return value !== null && typeof value === "object" && "server" in value && value.server && typeof value.server === "object";
+}
+function isTool(value) {
+  return value !== null && typeof value === "object" && "config" in value && (typeof value.getId === "function" || "id" in value);
+}
+function normalizeAgentCanUseType(value, fallbackName) {
+  if (isAgentMcpConfig(value)) {
+    return {
+      tool: value.server,
+      toolId: value.server.getId(),
+      selectedTools: value.selectedTools,
+      headers: value.headers,
+      isWrapped: true
+    };
+  }
+  if (isTool(value)) {
+    const toolId = value.getId?.() || value.id || fallbackName || "unknown";
+    return {
+      tool: value,
+      toolId,
+      isWrapped: false
+    };
+  }
+  throw new Error(`Invalid AgentCanUseType: expected Tool or AgentMcpConfig, got ${typeof value}`);
 }
 
 // src/agent.ts
@@ -510,6 +533,7 @@ var Agent = class {
           id = tool.server.getId();
           toolInstance = tool.server;
           toolInstance.selectedTools = tool.selectedTools;
+          toolInstance.headers = tool.headers;
         } else {
           toolInstance = tool;
           id = toolInstance.getId();
@@ -680,20 +704,13 @@ var Agent = class {
       if (tools && Array.isArray(tools)) {
         for (let i = 0; i < tools.length; i++) {
           const toolConfig = tools[i];
-          let toolId;
-          if (toolConfig instanceof Tool) {
-            toolId = toolConfig.getId();
-          } else if (toolConfig && typeof toolConfig === "object" && "server" in toolConfig) {
-            toolId = toolConfig.server.getId();
-          } else {
-            toolId = `tool-${i}`;
-          }
           try {
-            await this.createTool(toolId, toolConfig);
+            const normalizedTool = normalizeAgentCanUseType(toolConfig, `tool-${i}`);
+            await this.createTool(normalizedTool.toolId, toolConfig);
           } catch (error) {
             logger4.error(
               {
-                toolId,
+                toolId: isAgentMcpConfig(toolConfig) ? toolConfig.server.getId() : toolConfig.getId?.(),
                 error: error instanceof Error ? error.message : "Unknown error"
               },
               "Tool creation failed"
@@ -868,17 +885,15 @@ var Agent = class {
       }
       let tool;
       let selectedTools;
-      if (toolConfig && typeof toolConfig === "object" && "server" in toolConfig && "selectedTools" in toolConfig) {
-        const mcpConfig = toolConfig;
-        tool = mcpConfig.server;
-        selectedTools = mcpConfig.selectedTools;
-        tool.setContext(this.tenantId, this.projectId);
-        await tool.init();
-      } else if (toolConfig instanceof Tool) {
-        tool = toolConfig;
+      let headers;
+      try {
+        const normalizedTool = normalizeAgentCanUseType(toolConfig, toolId);
+        tool = normalizedTool.tool;
+        selectedTools = normalizedTool.selectedTools;
+        headers = normalizedTool.headers;
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
-      } else {
+      } catch (_) {
         tool = new Tool({
           id: toolId,
           name: toolConfig.name || toolId,
@@ -890,7 +905,7 @@ var Agent = class {
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
       }
-      await this.createAgentToolRelation(tool.getId(), selectedTools);
+      await this.createAgentToolRelation(tool.getId(), selectedTools, headers);
       logger4.info(
         {
           agentId: this.getId(),
@@ -1029,7 +1044,7 @@ var Agent = class {
       "Created agent-artifactComponent relation"
     );
   }
-  async createAgentToolRelation(toolId, selectedTools) {
+  async createAgentToolRelation(toolId, selectedTools, headers) {
     const relationData = {
       id: `${this.getId()}-tool-${toolId}`,
       tenantId: this.tenantId,
@@ -1040,6 +1055,9 @@ var Agent = class {
     if (selectedTools !== void 0) {
       relationData.selectedTools = selectedTools;
     }
+    if (headers !== void 0) {
+      relationData.headers = headers;
+    }
     const relationResponse = await fetch(
       `${this.baseURL}/tenants/${this.tenantId}/projects/${this.projectId}/agent-tool-relations`,
       {
@@ -1227,20 +1245,17 @@ var AgentGraph = class {
         const delegates = internalAgent.getDelegates();
         const tools = [];
         const selectedToolsMapping = {};
+        const headersMapping = {};
         const agentTools = internalAgent.getTools();
         for (const [_toolName, toolInstance] of Object.entries(agentTools)) {
-          if (toolInstance && typeof toolInstance === "object") {
-            let toolId;
-            toolId = toolInstance.getId?.() || toolInstance.id;
-            if ("selectedTools" in toolInstance && toolInstance.selectedTools !== void 0) {
-              logger6.info(
-                { toolId, selectedTools: toolInstance.selectedTools },
-                "Selected tools"
-              );
-              selectedToolsMapping[toolId] = toolInstance.selectedTools;
-            }
-            tools.push(toolId);
+          const toolId = toolInstance.getId();
+          if (toolInstance.selectedTools) {
+            selectedToolsMapping[toolId] = toolInstance.selectedTools;
+          }
+          if (toolInstance.headers) {
+            headersMapping[toolId] = toolInstance.headers;
           }
+          tools.push(toolId);
         }
         const dataComponents = [];
         const agentDataComponents = internalAgent.getDataComponents();
@@ -1260,7 +1275,8 @@ var AgentGraph = class {
         }
         const canUse = tools.map((toolId) => ({
           toolId,
-          toolSelection: selectedToolsMapping[toolId] || null
+          toolSelection: selectedToolsMapping[toolId] || null,
+          headers: headersMapping[toolId] || null
         }));
         agentsObject[internalAgent.getId()] = {
           id: internalAgent.getId(),
@@ -3116,7 +3132,8 @@ function dataComponent(config) {
 function agentMcp(config) {
   return {
     server: config.server,
-    selectedTools: config.selectedTools
+    selectedTools: config.selectedTools,
+    headers: config.headers
   };
 }
 

package/dist/index.d.cts

@@ -95,7 +95,8 @@ interface DataComponentConfig extends ComponentConfig {
 }
 type AgentMcpConfig = {
     server: Tool;
-    selectedTools: string[];
+    selectedTools?: string[];
+    headers?: Record<string, string>;
 };
 /**
  * Creates a transfer configuration for agent handoffs.
@@ -206,6 +207,7 @@ declare function externalAgents(configs: Record<string, ExternalAgentConfig>): R
  */
 type AgentTool = Tool & {
     selectedTools?: string[];
+    headers?: Record<string, string>;
 };
 interface UserMessage {
     role: 'user';

package/dist/index.d.ts

@@ -95,7 +95,8 @@ interface DataComponentConfig extends ComponentConfig {
 }
 type AgentMcpConfig = {
     server: Tool;
-    selectedTools: string[];
+    selectedTools?: string[];
+    headers?: Record<string, string>;
 };
 /**
  * Creates a transfer configuration for agent handoffs.
@@ -206,6 +207,7 @@ declare function externalAgents(configs: Record<string, ExternalAgentConfig>): R
  */
 type AgentTool = Tool & {
     selectedTools?: string[];
+    headers?: Record<string, string>;
 };
 interface UserMessage {
     role: 'user';

package/dist/index.js

@@ -437,7 +437,30 @@ var Tool = class {
 
 // src/utils/tool-normalization.ts
 function isAgentMcpConfig(value) {
-  return value !== null && typeof value === "object" && "server" in value && "selectedTools" in value && Array.isArray(value.selectedTools) && value.server && typeof value.server === "object";
+  return value !== null && typeof value === "object" && "server" in value && value.server && typeof value.server === "object";
+}
+function isTool(value) {
+  return value !== null && typeof value === "object" && "config" in value && (typeof value.getId === "function" || "id" in value);
+}
+function normalizeAgentCanUseType(value, fallbackName) {
+  if (isAgentMcpConfig(value)) {
+    return {
+      tool: value.server,
+      toolId: value.server.getId(),
+      selectedTools: value.selectedTools,
+      headers: value.headers,
+      isWrapped: true
+    };
+  }
+  if (isTool(value)) {
+    const toolId = value.getId?.() || value.id || fallbackName || "unknown";
+    return {
+      tool: value,
+      toolId,
+      isWrapped: false
+    };
+  }
+  throw new Error(`Invalid AgentCanUseType: expected Tool or AgentMcpConfig, got ${typeof value}`);
 }
 
 // src/agent.ts
@@ -508,6 +531,7 @@ var Agent = class {
           id = tool.server.getId();
           toolInstance = tool.server;
           toolInstance.selectedTools = tool.selectedTools;
+          toolInstance.headers = tool.headers;
         } else {
           toolInstance = tool;
           id = toolInstance.getId();
@@ -678,20 +702,13 @@ var Agent = class {
       if (tools && Array.isArray(tools)) {
         for (let i = 0; i < tools.length; i++) {
           const toolConfig = tools[i];
-          let toolId;
-          if (toolConfig instanceof Tool) {
-            toolId = toolConfig.getId();
-          } else if (toolConfig && typeof toolConfig === "object" && "server" in toolConfig) {
-            toolId = toolConfig.server.getId();
-          } else {
-            toolId = `tool-${i}`;
-          }
           try {
-            await this.createTool(toolId, toolConfig);
+            const normalizedTool = normalizeAgentCanUseType(toolConfig, `tool-${i}`);
+            await this.createTool(normalizedTool.toolId, toolConfig);
           } catch (error) {
             logger4.error(
               {
-                toolId,
+                toolId: isAgentMcpConfig(toolConfig) ? toolConfig.server.getId() : toolConfig.getId?.(),
                 error: error instanceof Error ? error.message : "Unknown error"
               },
               "Tool creation failed"
@@ -866,17 +883,15 @@ var Agent = class {
       }
       let tool;
       let selectedTools;
-      if (toolConfig && typeof toolConfig === "object" && "server" in toolConfig && "selectedTools" in toolConfig) {
-        const mcpConfig = toolConfig;
-        tool = mcpConfig.server;
-        selectedTools = mcpConfig.selectedTools;
-        tool.setContext(this.tenantId, this.projectId);
-        await tool.init();
-      } else if (toolConfig instanceof Tool) {
-        tool = toolConfig;
+      let headers;
+      try {
+        const normalizedTool = normalizeAgentCanUseType(toolConfig, toolId);
+        tool = normalizedTool.tool;
+        selectedTools = normalizedTool.selectedTools;
+        headers = normalizedTool.headers;
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
-      } else {
+      } catch (_) {
         tool = new Tool({
           id: toolId,
           name: toolConfig.name || toolId,
@@ -888,7 +903,7 @@ var Agent = class {
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
       }
-      await this.createAgentToolRelation(tool.getId(), selectedTools);
+      await this.createAgentToolRelation(tool.getId(), selectedTools, headers);
       logger4.info(
         {
           agentId: this.getId(),
@@ -1027,7 +1042,7 @@ var Agent = class {
       "Created agent-artifactComponent relation"
     );
   }
-  async createAgentToolRelation(toolId, selectedTools) {
+  async createAgentToolRelation(toolId, selectedTools, headers) {
     const relationData = {
       id: `${this.getId()}-tool-${toolId}`,
       tenantId: this.tenantId,
@@ -1038,6 +1053,9 @@ var Agent = class {
     if (selectedTools !== void 0) {
       relationData.selectedTools = selectedTools;
     }
+    if (headers !== void 0) {
+      relationData.headers = headers;
+    }
     const relationResponse = await fetch(
       `${this.baseURL}/tenants/${this.tenantId}/projects/${this.projectId}/agent-tool-relations`,
       {
@@ -1225,20 +1243,17 @@ var AgentGraph = class {
         const delegates = internalAgent.getDelegates();
         const tools = [];
         const selectedToolsMapping = {};
+        const headersMapping = {};
         const agentTools = internalAgent.getTools();
         for (const [_toolName, toolInstance] of Object.entries(agentTools)) {
-          if (toolInstance && typeof toolInstance === "object") {
-            let toolId;
-            toolId = toolInstance.getId?.() || toolInstance.id;
-            if ("selectedTools" in toolInstance && toolInstance.selectedTools !== void 0) {
-              logger6.info(
-                { toolId, selectedTools: toolInstance.selectedTools },
-                "Selected tools"
-              );
-              selectedToolsMapping[toolId] = toolInstance.selectedTools;
-            }
-            tools.push(toolId);
+          const toolId = toolInstance.getId();
+          if (toolInstance.selectedTools) {
+            selectedToolsMapping[toolId] = toolInstance.selectedTools;
+          }
+          if (toolInstance.headers) {
+            headersMapping[toolId] = toolInstance.headers;
           }
+          tools.push(toolId);
         }
         const dataComponents = [];
         const agentDataComponents = internalAgent.getDataComponents();
@@ -1258,7 +1273,8 @@ var AgentGraph = class {
         }
         const canUse = tools.map((toolId) => ({
           toolId,
-          toolSelection: selectedToolsMapping[toolId] || null
+          toolSelection: selectedToolsMapping[toolId] || null,
+          headers: headersMapping[toolId] || null
         }));
         agentsObject[internalAgent.getId()] = {
           id: internalAgent.getId(),
@@ -3114,7 +3130,8 @@ function dataComponent(config) {
 function agentMcp(config) {
   return {
     server: config.server,
-    selectedTools: config.selectedTools
+    selectedTools: config.selectedTools,
+    headers: config.headers
   };
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@inkeep/agents-sdk",
-  "version": "0.7.1",
+  "version": "0.8.0",
   "description": "Agents SDK for building and managing agents in the Inkeep Agent Framework",
   "main": "dist/index.js",
   "type": "module",
@@ -8,7 +8,7 @@
   "dependencies": {
     "nanoid": "^5.1.5",
     "zod": "^4.1.5",
-    "@inkeep/agents-core": "^0.7.1"
+    "@inkeep/agents-core": "^0.8.0"
   },
   "devDependencies": {
     "@types/node": "^20.11.24",