@inkeep/agents-sdk 0.37.0 → 0.37.1

package/dist/index.cjs

@@ -778,6 +778,7 @@ var Agent = class {
       const tools = [];
       const selectedToolsMapping = {};
       const headersMapping = {};
+      const toolPoliciesMapping = {};
       const subAgentTools = subAgent2.getTools();
       for (const [_toolName, toolInstance] of Object.entries(subAgentTools)) {
         const toolId = toolInstance.getId();
@@ -787,6 +788,9 @@ var Agent = class {
         if (toolInstance.headers) {
           headersMapping[toolId] = toolInstance.headers;
         }
+        if (toolInstance.toolPolicies) {
+          toolPoliciesMapping[toolId] = toolInstance.toolPolicies;
+        }
         tools.push(toolId);
         if (toolInstance.constructor.name === "FunctionTool" && toolInstance instanceof FunctionTool) {
           if (!functionsObject[toolId]) {
@@ -837,7 +841,8 @@ var Agent = class {
       const canUse = tools.map((toolId) => ({
         toolId,
         toolSelection: selectedToolsMapping[toolId] || null,
-        headers: headersMapping[toolId] || null
+        headers: headersMapping[toolId] || null,
+        toolPolicies: toolPoliciesMapping[toolId] || null
       }));
       subAgentsObject[subAgent2.getId()] = {
         id: subAgent2.getId(),
@@ -2772,10 +2777,12 @@ var Tool = class {
    * ```
    */
   with(config) {
+    const { selectedTools, toolPolicies } = agentsCore.normalizeToolSelections(config.selectedTools);
     return {
       server: this,
-      selectedTools: config.selectedTools,
-      headers: config.headers
+      selectedTools,
+      headers: config.headers,
+      toolPolicies
     };
   }
 };
@@ -2794,6 +2801,7 @@ function normalizeAgentCanUseType(value, fallbackName) {
       toolId: value.server.getId(),
       selectedTools: value.selectedTools,
       headers: value.headers,
+      toolPolicies: value.toolPolicies,
       isWrapped: true
     };
   }
@@ -2880,6 +2888,7 @@ var SubAgent = class {
           toolInstance = tool.server;
           toolInstance.selectedTools = tool.selectedTools;
           toolInstance.headers = tool.headers;
+          toolInstance.toolPolicies = tool.toolPolicies;
         } else {
           toolInstance = tool;
           id = toolInstance.getId();
@@ -3433,11 +3442,13 @@ var SubAgent = class {
       let tool;
       let selectedTools;
       let headers;
+      let toolPolicies;
       try {
         const normalizedTool = normalizeAgentCanUseType(toolConfig, toolId);
         tool = normalizedTool.tool;
         selectedTools = normalizedTool.selectedTools;
         headers = normalizedTool.headers;
+        toolPolicies = normalizedTool.toolPolicies;
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
       } catch (_) {
@@ -3452,7 +3463,7 @@ var SubAgent = class {
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
       }
-      await this.createAgentToolRelation(tool.getId(), selectedTools, headers);
+      await this.createAgentToolRelation(tool.getId(), selectedTools, headers, toolPolicies);
       logger11.info(
         {
           subAgentId: this.getId(),
@@ -3591,7 +3602,7 @@ var SubAgent = class {
       "Created agent-artifactComponent relation"
     );
   }
-  async createAgentToolRelation(toolId, selectedTools, headers) {
+  async createAgentToolRelation(toolId, selectedTools, headers, toolPolicies) {
     const relationData = {
       id: `${this.getId()}-tool-${toolId}`,
       tenantId: this.tenantId,
@@ -3605,6 +3616,9 @@ var SubAgent = class {
     if (headers !== void 0) {
       relationData.headers = headers;
     }
+    if (toolPolicies !== void 0 && Object.keys(toolPolicies).length > 0) {
+      relationData.toolPolicies = toolPolicies;
+    }
     const relationResponse = await fetch(
       `${this.baseURL}/tenants/${this.tenantId}/projects/${this.projectId}/agent-tool-relations`,
       {

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { ArtifactComponentInsert, MCPToolConfig as MCPToolConfig$1, DataComponentApiInsert, ArtifactComponentApiInsert, CredentialReferenceApiInsert, DataComponentInsert, FunctionToolConfig, SubAgentApiInsert, FullAgentDefinition, AgentConversationHistoryConfig, AgentStopWhen, ModelSettings, StatusUpdateSettings, ToolInsert, McpTransportConfig, StopWhen, FullProjectDefinition, StatusComponent as StatusComponent$1 } from '@inkeep/agents-core';
+import { ArtifactComponentInsert, MCPToolConfig as MCPToolConfig$1, DataComponentApiInsert, ArtifactComponentApiInsert, McpToolSelection, ToolPolicy, CredentialReferenceApiInsert, DataComponentInsert, FunctionToolConfig, SubAgentApiInsert, FullAgentDefinition, AgentConversationHistoryConfig, AgentStopWhen, ModelSettings, StatusUpdateSettings, ToolInsert, McpTransportConfig, StopWhen, FullProjectDefinition, StatusComponent as StatusComponent$1 } from '@inkeep/agents-core';
 export { ANTHROPIC_MODELS, FunctionToolConfig, GOOGLE_MODELS, ModelSettings, OPENAI_MODELS } from '@inkeep/agents-core';
 import { z } from 'zod';
 
@@ -170,13 +170,15 @@ type AgentMcpConfig = {
     server: Tool;
     selectedTools?: string[];
     headers?: Record<string, string>;
+    toolPolicies?: Record<string, ToolPolicy>;
 };
 /**
  * Input configuration for MCP tool customization
+ * Supports flexible tool selection with per-tool policies
  */
 type AgentMcpConfigInput = {
-    /** Specific tools to enable from the MCP server */
-    selectedTools?: string[];
+    /** Tools to enable from the MCP server - can be strings or objects with policies */
+    selectedTools?: McpToolSelection[];
     /** Custom headers for MCP server requests */
     headers?: Record<string, string>;
 };
@@ -347,9 +349,11 @@ interface DataComponentWithZodProps {
 type AgentTool = (Tool & {
     selectedTools?: string[];
     headers?: Record<string, string>;
+    toolPolicies?: Record<string, ToolPolicy>;
 }) | (FunctionTool & {
     selectedTools?: string[];
     headers?: Record<string, string>;
+    toolPolicies?: Record<string, ToolPolicy>;
 });
 interface UserMessage {
     role: 'user';

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { ArtifactComponentInsert, MCPToolConfig as MCPToolConfig$1, DataComponentApiInsert, ArtifactComponentApiInsert, CredentialReferenceApiInsert, DataComponentInsert, FunctionToolConfig, SubAgentApiInsert, FullAgentDefinition, AgentConversationHistoryConfig, AgentStopWhen, ModelSettings, StatusUpdateSettings, ToolInsert, McpTransportConfig, StopWhen, FullProjectDefinition, StatusComponent as StatusComponent$1 } from '@inkeep/agents-core';
+import { ArtifactComponentInsert, MCPToolConfig as MCPToolConfig$1, DataComponentApiInsert, ArtifactComponentApiInsert, McpToolSelection, ToolPolicy, CredentialReferenceApiInsert, DataComponentInsert, FunctionToolConfig, SubAgentApiInsert, FullAgentDefinition, AgentConversationHistoryConfig, AgentStopWhen, ModelSettings, StatusUpdateSettings, ToolInsert, McpTransportConfig, StopWhen, FullProjectDefinition, StatusComponent as StatusComponent$1 } from '@inkeep/agents-core';
 export { ANTHROPIC_MODELS, FunctionToolConfig, GOOGLE_MODELS, ModelSettings, OPENAI_MODELS } from '@inkeep/agents-core';
 import { z } from 'zod';
 
@@ -170,13 +170,15 @@ type AgentMcpConfig = {
     server: Tool;
     selectedTools?: string[];
     headers?: Record<string, string>;
+    toolPolicies?: Record<string, ToolPolicy>;
 };
 /**
  * Input configuration for MCP tool customization
+ * Supports flexible tool selection with per-tool policies
  */
 type AgentMcpConfigInput = {
-    /** Specific tools to enable from the MCP server */
-    selectedTools?: string[];
+    /** Tools to enable from the MCP server - can be strings or objects with policies */
+    selectedTools?: McpToolSelection[];
     /** Custom headers for MCP server requests */
     headers?: Record<string, string>;
 };
@@ -347,9 +349,11 @@ interface DataComponentWithZodProps {
 type AgentTool = (Tool & {
     selectedTools?: string[];
     headers?: Record<string, string>;
+    toolPolicies?: Record<string, ToolPolicy>;
 }) | (FunctionTool & {
     selectedTools?: string[];
     headers?: Record<string, string>;
+    toolPolicies?: Record<string, ToolPolicy>;
 });
 interface UserMessage {
     role: 'user';

package/dist/index.js

@@ -1,4 +1,4 @@
-import { getLogger, apiFetch, CredentialReferenceApiInsertSchema, MCPToolConfigSchema } from '@inkeep/agents-core';
+import { getLogger, normalizeToolSelections, apiFetch, CredentialReferenceApiInsertSchema, MCPToolConfigSchema } from '@inkeep/agents-core';
 export { ANTHROPIC_MODELS, GOOGLE_MODELS, OPENAI_MODELS } from '@inkeep/agents-core';
 import { isZodSchema, convertZodToJsonSchemaWithPreview, convertZodToJsonSchema } from '@inkeep/agents-core/utils/schema-conversion';
 import fs from 'fs';
@@ -751,6 +751,7 @@ var Agent = class {
       const tools = [];
       const selectedToolsMapping = {};
       const headersMapping = {};
+      const toolPoliciesMapping = {};
       const subAgentTools = subAgent2.getTools();
       for (const [_toolName, toolInstance] of Object.entries(subAgentTools)) {
         const toolId = toolInstance.getId();
@@ -760,6 +761,9 @@ var Agent = class {
         if (toolInstance.headers) {
           headersMapping[toolId] = toolInstance.headers;
         }
+        if (toolInstance.toolPolicies) {
+          toolPoliciesMapping[toolId] = toolInstance.toolPolicies;
+        }
         tools.push(toolId);
         if (toolInstance.constructor.name === "FunctionTool" && toolInstance instanceof FunctionTool) {
           if (!functionsObject[toolId]) {
@@ -810,7 +814,8 @@ var Agent = class {
       const canUse = tools.map((toolId) => ({
         toolId,
         toolSelection: selectedToolsMapping[toolId] || null,
-        headers: headersMapping[toolId] || null
+        headers: headersMapping[toolId] || null,
+        toolPolicies: toolPoliciesMapping[toolId] || null
       }));
       subAgentsObject[subAgent2.getId()] = {
         id: subAgent2.getId(),
@@ -2745,10 +2750,12 @@ var Tool = class {
    * ```
    */
   with(config) {
+    const { selectedTools, toolPolicies } = normalizeToolSelections(config.selectedTools);
     return {
       server: this,
-      selectedTools: config.selectedTools,
-      headers: config.headers
+      selectedTools,
+      headers: config.headers,
+      toolPolicies
     };
   }
 };
@@ -2767,6 +2774,7 @@ function normalizeAgentCanUseType(value, fallbackName) {
       toolId: value.server.getId(),
       selectedTools: value.selectedTools,
       headers: value.headers,
+      toolPolicies: value.toolPolicies,
       isWrapped: true
     };
   }
@@ -2853,6 +2861,7 @@ var SubAgent = class {
           toolInstance = tool.server;
           toolInstance.selectedTools = tool.selectedTools;
           toolInstance.headers = tool.headers;
+          toolInstance.toolPolicies = tool.toolPolicies;
         } else {
           toolInstance = tool;
           id = toolInstance.getId();
@@ -3406,11 +3415,13 @@ var SubAgent = class {
       let tool;
       let selectedTools;
       let headers;
+      let toolPolicies;
       try {
         const normalizedTool = normalizeAgentCanUseType(toolConfig, toolId);
         tool = normalizedTool.tool;
         selectedTools = normalizedTool.selectedTools;
         headers = normalizedTool.headers;
+        toolPolicies = normalizedTool.toolPolicies;
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
       } catch (_) {
@@ -3425,7 +3436,7 @@ var SubAgent = class {
         tool.setContext(this.tenantId, this.projectId);
         await tool.init();
       }
-      await this.createAgentToolRelation(tool.getId(), selectedTools, headers);
+      await this.createAgentToolRelation(tool.getId(), selectedTools, headers, toolPolicies);
       logger11.info(
         {
           subAgentId: this.getId(),
@@ -3564,7 +3575,7 @@ var SubAgent = class {
       "Created agent-artifactComponent relation"
     );
   }
-  async createAgentToolRelation(toolId, selectedTools, headers) {
+  async createAgentToolRelation(toolId, selectedTools, headers, toolPolicies) {
     const relationData = {
       id: `${this.getId()}-tool-${toolId}`,
       tenantId: this.tenantId,
@@ -3578,6 +3589,9 @@ var SubAgent = class {
     if (headers !== void 0) {
       relationData.headers = headers;
     }
+    if (toolPolicies !== void 0 && Object.keys(toolPolicies).length > 0) {
+      relationData.toolPolicies = toolPolicies;
+    }
     const relationResponse = await fetch(
       `${this.baseURL}/tenants/${this.tenantId}/projects/${this.projectId}/agent-tool-relations`,
       {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@inkeep/agents-sdk",
-  "version": "0.37.0",
+  "version": "0.37.1",
   "description": "Agents SDK for building and managing agents in the Inkeep Agent Framework",
   "main": "dist/index.js",
   "type": "module",
@@ -12,7 +12,7 @@
     "nanoid": "^5.1.5",
     "typescript": "^5.3.3",
     "zod": "^4.1.11",
-    "@inkeep/agents-core": "^0.37.0"
+    "@inkeep/agents-core": "^0.37.1"
   },
   "devDependencies": {
     "@types/js-yaml": "^4.0.9",