@inkeep/agents-run-api 0.33.2 → 0.34.1

package/dist/index.cjs

@@ -34,6 +34,7 @@ var anthropic = require('@ai-sdk/anthropic');
 var gateway = require('@ai-sdk/gateway');
 var google = require('@ai-sdk/google');
 var openai = require('@ai-sdk/openai');
+var openaiCompatible = require('@ai-sdk/openai-compatible');
 var aiSdkProvider = require('@openrouter/ai-sdk-provider');
 var jmespath = require('jmespath');
 var Ajv = require('ajv');
@@ -2578,6 +2579,13 @@ init_logger();
 // src/agents/ModelFactory.ts
 init_logger();
 var logger4 = agentsCore.getLogger("ModelFactory");
+var nimDefault = openaiCompatible.createOpenAICompatible({
+  name: "nim",
+  baseURL: "https://integrate.api.nvidia.com/v1",
+  headers: {
+    Authorization: `Bearer ${process.env.NIM_API_KEY}`
+  }
+});
 var _ModelFactory = class _ModelFactory {
   /**
    * Create a provider instance with custom configuration
@@ -2602,6 +2610,47 @@ var _ModelFactory = class _ModelFactory {
         };
       case "gateway":
         return gateway.createGateway(config);
+      case "nim": {
+        const nimConfig = {
+          name: "nim",
+          baseURL: "https://integrate.api.nvidia.com/v1",
+          headers: {
+            Authorization: `Bearer ${process.env.NIM_API_KEY}`
+          },
+          ...config
+        };
+        return openaiCompatible.createOpenAICompatible(nimConfig);
+      }
+      case "custom": {
+        if (!config.baseURL && !config.baseUrl) {
+          throw new Error(
+            "Custom provider requires baseURL. Please provide it in providerOptions.baseURL or providerOptions.baseUrl"
+          );
+        }
+        const customConfig = {
+          name: "custom",
+          baseURL: config.baseURL || config.baseUrl,
+          headers: {
+            ...process.env.CUSTOM_LLM_API_KEY && {
+              Authorization: `Bearer ${process.env.CUSTOM_LLM_API_KEY}`
+            },
+            ...config.headers || {}
+          },
+          ...config
+        };
+        logger4.info(
+          {
+            config: {
+              baseURL: customConfig.baseURL,
+              hasApiKey: !!process.env.CUSTOM_LLM_API_KEY,
+              apiKeyPrefix: process.env.CUSTOM_LLM_API_KEY?.substring(0, 10) + "...",
+              headers: Object.keys(customConfig.headers || {})
+            }
+          },
+          "Creating custom OpenAI-compatible provider"
+        );
+        return openaiCompatible.createOpenAICompatible(customConfig);
+      }
       default:
         throw new Error(`Unsupported provider: ${provider}`);
     }
@@ -2618,9 +2667,18 @@ var _ModelFactory = class _ModelFactory {
     if (providerOptions.baseUrl || providerOptions.baseURL) {
       providerConfig.baseURL = providerOptions.baseUrl || providerOptions.baseURL;
     }
+    if (providerOptions.headers) {
+      providerConfig.headers = providerOptions.headers;
+    }
     if (providerOptions.gateway) {
       Object.assign(providerConfig, providerOptions.gateway);
     }
+    if (providerOptions.nim) {
+      Object.assign(providerConfig, providerOptions.nim);
+    }
+    if (providerOptions.custom) {
+      Object.assign(providerConfig, providerOptions.custom);
+    }
     return providerConfig;
   }
   /**
@@ -2665,9 +2723,15 @@ var _ModelFactory = class _ModelFactory {
         return aiSdkProvider.openrouter(modelName);
       case "gateway":
         return gateway.gateway(modelName);
+      case "nim":
+        return nimDefault(modelName);
+      case "custom":
+        throw new Error(
+          "Custom provider requires configuration. Please provide baseURL in providerOptions.custom.baseURL or providerOptions.baseURL"
+        );
       default:
         throw new Error(
-          `Unsupported provider: ${provider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id) or Vercel AI Gateway (gateway/model-id).`
+          `Unsupported provider: ${provider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id), Vercel AI Gateway (gateway/model-id), NVIDIA NIM (nim/model-id), or Custom OpenAI-compatible (custom/model-id).`
         );
     }
   }
@@ -2683,7 +2747,7 @@ var _ModelFactory = class _ModelFactory {
       const normalizedProvider = provider.toLowerCase();
       if (!_ModelFactory.BUILT_IN_PROVIDERS.includes(normalizedProvider)) {
         throw new Error(
-          `Unsupported provider: ${normalizedProvider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id) or Vercel AI Gateway (gateway/model-id).`
+          `Unsupported provider: ${normalizedProvider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id), Vercel AI Gateway (gateway/model-id), NVIDIA NIM (nim/model-id), or Custom OpenAI-compatible (custom/model-id).`
         );
       }
       return {
@@ -2702,7 +2766,16 @@ var _ModelFactory = class _ModelFactory {
     if (!providerOptions) {
       return {};
     }
-    const excludedKeys = ["apiKey", "baseURL", "baseUrl", "maxDuration"];
+    const excludedKeys = [
+      "apiKey",
+      "baseURL",
+      "baseUrl",
+      "maxDuration",
+      "headers",
+      "gateway",
+      "nim",
+      "custom"
+    ];
     const params = {};
     for (const [key, value] of Object.entries(providerOptions)) {
       if (!excludedKeys.includes(key) && value !== void 0) {
@@ -2763,7 +2836,9 @@ __publicField(_ModelFactory, "BUILT_IN_PROVIDERS", [
   "openai",
   "google",
   "openrouter",
-  "gateway"
+  "gateway",
+  "nim",
+  "custom"
 ]);
 var ModelFactory = _ModelFactory;
 
@@ -4806,8 +4881,10 @@ ${this.statusUpdateState?.config.prompt?.trim() || ""}`;
             }
           });
           const result = object;
+          logger9.info({ result: JSON.stringify(result) }, "DEBUG: Result");
           const summaries = [];
           for (const [componentId, data] of Object.entries(result)) {
+            logger9.info({ componentId, data: JSON.stringify(data) }, "DEBUG: Component data");
             if (componentId === "no_relevant_updates") {
               continue;
             }
@@ -5374,10 +5451,10 @@ var AgentSessionManager = class {
   /**
    * Initialize status updates for a session
    */
-  initializeStatusUpdates(sessionId, config, summarizerModel) {
+  initializeStatusUpdates(sessionId, config, summarizerModel, baseModel) {
     const session = this.sessions.get(sessionId);
     if (session) {
-      session.initializeStatusUpdates(config, summarizerModel);
+      session.initializeStatusUpdates(config, summarizerModel, baseModel);
     } else {
       logger9.error(
         {
@@ -8767,7 +8844,7 @@ var Agent = class {
   /**
    * Wraps a tool with streaming lifecycle tracking (start, complete, error) and AgentSession recording
    */
-  wrapToolWithStreaming(toolName, toolDefinition, streamRequestId, toolType) {
+  wrapToolWithStreaming(toolName, toolDefinition, streamRequestId, toolType, relationshipId) {
     if (!toolDefinition || typeof toolDefinition !== "object" || !("execute" in toolDefinition)) {
       return toolDefinition;
     }
@@ -8793,7 +8870,8 @@ var Agent = class {
           agentSessionManager.recordEvent(streamRequestId, "tool_call", this.config.id, {
             toolName,
             input: args,
-            toolCallId
+            toolCallId,
+            relationshipId
           });
         }
         try {
@@ -8838,7 +8916,8 @@ var Agent = class {
               toolName,
               output: result,
               toolCallId,
-              duration
+              duration,
+              relationshipId
             });
           }
           return result;
@@ -8851,7 +8930,8 @@ var Agent = class {
               output: null,
               toolCallId,
               duration,
-              error: errorMessage
+              error: errorMessage,
+              relationshipId
             });
           }
           throw error;
@@ -8917,22 +8997,24 @@ var Agent = class {
     }) || [];
     const tools = await Promise.all(mcpTools.map((tool3) => this.getMcpTool(tool3)) || []) || [];
     if (!sessionId) {
-      const combinedTools = tools.reduce((acc, tool3) => {
-        return Object.assign(acc, tool3);
-      }, {});
       const wrappedTools2 = {};
-      for (const [toolName, toolDef] of Object.entries(combinedTools)) {
-        wrappedTools2[toolName] = this.wrapToolWithStreaming(
-          toolName,
-          toolDef,
-          streamRequestId,
-          "mcp"
-        );
+      for (const [index, toolSet] of tools.entries()) {
+        const relationshipId = mcpTools[index]?.relationshipId;
+        for (const [toolName, toolDef] of Object.entries(toolSet)) {
+          wrappedTools2[toolName] = this.wrapToolWithStreaming(
+            toolName,
+            toolDef,
+            streamRequestId,
+            "mcp",
+            relationshipId
+          );
+        }
       }
       return wrappedTools2;
     }
     const wrappedTools = {};
-    for (const toolSet of tools) {
+    for (const [index, toolSet] of tools.entries()) {
+      const relationshipId = mcpTools[index]?.relationshipId;
       for (const [toolName, originalTool] of Object.entries(toolSet)) {
         if (!isValidTool(originalTool)) {
           logger19.error({ toolName }, "Invalid MCP tool structure - missing required properties");
@@ -8945,7 +9027,7 @@ var Agent = class {
             logger19.debug({ toolName, toolCallId }, "MCP Tool Called");
             try {
               const rawResult = await originalTool.execute(args, { toolCallId });
-              if (rawResult && typeof rawResult === "object" && "isError" in rawResult && rawResult.isError) {
+              if (rawResult && typeof rawResult === "object" && rawResult.isError) {
                 const errorMessage = rawResult.content?.[0]?.text || "MCP tool returned an error";
                 logger19.error(
                   { toolName, toolCallId, errorMessage, rawResult },
@@ -8966,7 +9048,8 @@ var Agent = class {
                     context: {
                       toolName,
                       toolCallId,
-                      errorMessage
+                      errorMessage,
+                      relationshipId
                     }
                   });
                 }
@@ -9005,7 +9088,8 @@ var Agent = class {
           toolName,
           sessionWrappedTool,
           streamRequestId,
-          "mcp"
+          "mcp",
+          relationshipId
         );
       }
     }
@@ -10677,7 +10761,12 @@ var createTaskHandler = (config, credentialStoreRegistry) => {
       const stopWhen = "stopWhen" in config.agentSchema ? config.agentSchema.stopWhen : void 0;
       const toolsForAgentResult = await Promise.all(
         toolsForAgent.data.map(async (item) => {
-          const mcpTool = await agentsCore.dbResultToMcpTool(item.tool, dbClient_default, credentialStoreRegistry);
+          const mcpTool = await agentsCore.dbResultToMcpTool(
+            item.tool,
+            dbClient_default,
+            credentialStoreRegistry,
+            item.id
+          );
           if (item.selectedTools && item.selectedTools.length > 0) {
             const selectedToolsSet = new Set(item.selectedTools);
             mcpTool.availableTools = mcpTool.availableTools?.filter((tool3) => selectedToolsSet.has(tool3.name)) || [];
@@ -10759,7 +10848,8 @@ var createTaskHandler = (config, credentialStoreRegistry) => {
                   const mcpTool = await agentsCore.dbResultToMcpTool(
                     item.tool,
                     dbClient_default,
-                    credentialStoreRegistry
+                    credentialStoreRegistry,
+                    item.id
                   );
                   if (item.selectedTools && item.selectedTools.length > 0) {
                     const selectedToolsSet = new Set(item.selectedTools);
@@ -12083,11 +12173,42 @@ var ExecutionHandler = class {
         scopes: { tenantId, projectId, agentId }
       });
       if (agentConfig?.statusUpdates && agentConfig.statusUpdates.enabled !== false) {
-        agentSessionManager.initializeStatusUpdates(
-          requestId2,
-          agentConfig.statusUpdates,
-          agentConfig.models?.summarizer
-        );
+        try {
+          const agentWithDefault = await agentsCore.getAgentWithDefaultSubAgent(dbClient_default)({
+            scopes: { tenantId, projectId, agentId }
+          });
+          if (agentWithDefault?.defaultSubAgent) {
+            const resolvedModels = await resolveModelConfig(
+              agentId,
+              agentWithDefault.defaultSubAgent
+            );
+            agentSessionManager.initializeStatusUpdates(
+              requestId2,
+              agentConfig.statusUpdates,
+              resolvedModels.summarizer,
+              resolvedModels.base
+            );
+          } else {
+            agentSessionManager.initializeStatusUpdates(
+              requestId2,
+              agentConfig.statusUpdates,
+              agentConfig.models?.summarizer
+            );
+          }
+        } catch (modelError) {
+          logger24.warn(
+            {
+              error: modelError instanceof Error ? modelError.message : "Unknown error",
+              agentId
+            },
+            "Failed to resolve models for status updates, using agent-level config"
+          );
+          agentSessionManager.initializeStatusUpdates(
+            requestId2,
+            agentConfig.statusUpdates,
+            agentConfig.models?.summarizer
+          );
+        }
       }
     } catch (error) {
       logger24.error(

package/dist/index.js

@@ -5,7 +5,7 @@ import { env } from './chunk-UZXC6MEO.js';
 import { getLogger } from './chunk-A2S7GSHL.js';
 import { STREAM_PARSER_MAX_SNAPSHOT_SIZE, STREAM_PARSER_MAX_STREAMED_SIZE, STREAM_PARSER_MAX_COLLECTED_PARTS, STREAM_BUFFER_MAX_SIZE_BYTES, SESSION_CLEANUP_INTERVAL_MS, STREAM_MAX_LIFETIME_MS, STREAM_TEXT_GAP_THRESHOLD_MS, AGENT_EXECUTION_MAX_CONSECUTIVE_ERRORS, SESSION_TOOL_RESULT_CACHE_TIMEOUT_MS, ARTIFACT_GENERATION_MAX_RETRIES, ARTIFACT_SESSION_MAX_PENDING, STATUS_UPDATE_DEFAULT_INTERVAL_SECONDS, STATUS_UPDATE_DEFAULT_NUM_EVENTS, ARTIFACT_SESSION_MAX_PREVIOUS_SUMMARIES, ARTIFACT_GENERATION_BACKOFF_INITIAL_MS, ARTIFACT_GENERATION_BACKOFF_MAX_MS, AGENT_EXECUTION_MAX_GENERATION_STEPS, FUNCTION_TOOL_SANDBOX_VCPUS_DEFAULT, FUNCTION_TOOL_EXECUTION_TIMEOUT_MS_DEFAULT, LLM_GENERATION_MAX_ALLOWED_TIMEOUT_MS, LLM_GENERATION_FIRST_CALL_TIMEOUT_MS_STREAMING, LLM_GENERATION_FIRST_CALL_TIMEOUT_MS_NON_STREAMING, LLM_GENERATION_SUBSEQUENT_CALL_TIMEOUT_MS, DELEGATION_TOOL_BACKOFF_MAX_ELAPSED_TIME_MS, DELEGATION_TOOL_BACKOFF_EXPONENT, DELEGATION_TOOL_BACKOFF_MAX_INTERVAL_MS, DELEGATION_TOOL_BACKOFF_INITIAL_INTERVAL_MS } from './chunk-IVALDC72.js';
 import { __publicField } from './chunk-PKBMQBKP.js';
-import { getTracer, HeadersScopeSchema, getRequestExecutionContext, createApiError, getAgentWithDefaultSubAgent, contextValidationMiddleware, getConversationId, getFullAgent, createOrGetConversation, getActiveAgentForConversation, setActiveAgentForConversation, getSubAgentById, handleContextResolution, createMessage, generateId, commonGetErrorResponses, loggerFactory, createDefaultCredentialStores, CredentialStoreRegistry, listTaskIdsByContextId, getTask, getLedgerArtifacts, upsertLedgerArtifact, createTask, updateTask, setSpanWithError, AGENT_EXECUTION_TRANSFER_COUNT_DEFAULT, updateConversation, handleApiError, CONVERSATION_HISTORY_MAX_OUTPUT_TOKENS_DEFAULT, CONVERSATION_HISTORY_DEFAULT_LIMIT, TaskState, setActiveAgentForThread, getConversation, getAgentById, getRelatedAgentsForAgent, getExternalAgentsForSubAgent, getTeamAgentsForSubAgent, getToolsForAgent, getDataComponentsForAgent, getArtifactComponentsForAgent, dbResultToMcpTool, validateAndGetApiKey, verifyServiceToken, validateTargetAgent, getProject, ContextResolver, CredentialStuffer, MCPServerType, getCredentialReference, McpClient, getFunctionToolsForSubAgent, getFunction, getContextConfigById, getFullAgentDefinition, TemplateEngine, agentHasArtifactComponents, MCPTransportType, SPAN_KEYS, headers, generateServiceToken } from '@inkeep/agents-core';
+import { getTracer, HeadersScopeSchema, getRequestExecutionContext, createApiError, getAgentWithDefaultSubAgent, contextValidationMiddleware, getConversationId, getFullAgent, createOrGetConversation, getActiveAgentForConversation, setActiveAgentForConversation, getSubAgentById, handleContextResolution, createMessage, generateId, commonGetErrorResponses, loggerFactory, createDefaultCredentialStores, CredentialStoreRegistry, listTaskIdsByContextId, getTask, getLedgerArtifacts, upsertLedgerArtifact, createTask, updateTask, setSpanWithError, AGENT_EXECUTION_TRANSFER_COUNT_DEFAULT, updateConversation, handleApiError, CONVERSATION_HISTORY_MAX_OUTPUT_TOKENS_DEFAULT, CONVERSATION_HISTORY_DEFAULT_LIMIT, TaskState, getAgentById, getProject, setActiveAgentForThread, getConversation, getRelatedAgentsForAgent, getExternalAgentsForSubAgent, getTeamAgentsForSubAgent, getToolsForAgent, getDataComponentsForAgent, getArtifactComponentsForAgent, dbResultToMcpTool, validateAndGetApiKey, verifyServiceToken, validateTargetAgent, ContextResolver, CredentialStuffer, MCPServerType, getCredentialReference, McpClient, getFunctionToolsForSubAgent, getFunction, getContextConfigById, getFullAgentDefinition, TemplateEngine, agentHasArtifactComponents, MCPTransportType, SPAN_KEYS, headers, generateServiceToken } from '@inkeep/agents-core';
 import { otel } from '@hono/otel';
 import { OpenAPIHono, createRoute, z as z$1 } from '@hono/zod-openapi';
 import { trace, propagation, context, SpanStatusCode } from '@opentelemetry/api';
@@ -22,6 +22,7 @@ import { createAnthropic, anthropic } from '@ai-sdk/anthropic';
 import { createGateway, gateway } from '@ai-sdk/gateway';
 import { createGoogleGenerativeAI, google } from '@ai-sdk/google';
 import { createOpenAI, openai } from '@ai-sdk/openai';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
 import { createOpenRouter, openrouter } from '@openrouter/ai-sdk-provider';
 import jmespath from 'jmespath';
 import Ajv from 'ajv';
@@ -932,6 +933,13 @@ async function handleTasksResubscribe(c, agent, request) {
   }
 }
 var logger3 = getLogger("ModelFactory");
+var nimDefault = createOpenAICompatible({
+  name: "nim",
+  baseURL: "https://integrate.api.nvidia.com/v1",
+  headers: {
+    Authorization: `Bearer ${process.env.NIM_API_KEY}`
+  }
+});
 var _ModelFactory = class _ModelFactory {
   /**
    * Create a provider instance with custom configuration
@@ -956,6 +964,47 @@ var _ModelFactory = class _ModelFactory {
         };
       case "gateway":
         return createGateway(config);
+      case "nim": {
+        const nimConfig = {
+          name: "nim",
+          baseURL: "https://integrate.api.nvidia.com/v1",
+          headers: {
+            Authorization: `Bearer ${process.env.NIM_API_KEY}`
+          },
+          ...config
+        };
+        return createOpenAICompatible(nimConfig);
+      }
+      case "custom": {
+        if (!config.baseURL && !config.baseUrl) {
+          throw new Error(
+            "Custom provider requires baseURL. Please provide it in providerOptions.baseURL or providerOptions.baseUrl"
+          );
+        }
+        const customConfig = {
+          name: "custom",
+          baseURL: config.baseURL || config.baseUrl,
+          headers: {
+            ...process.env.CUSTOM_LLM_API_KEY && {
+              Authorization: `Bearer ${process.env.CUSTOM_LLM_API_KEY}`
+            },
+            ...config.headers || {}
+          },
+          ...config
+        };
+        logger3.info(
+          {
+            config: {
+              baseURL: customConfig.baseURL,
+              hasApiKey: !!process.env.CUSTOM_LLM_API_KEY,
+              apiKeyPrefix: process.env.CUSTOM_LLM_API_KEY?.substring(0, 10) + "...",
+              headers: Object.keys(customConfig.headers || {})
+            }
+          },
+          "Creating custom OpenAI-compatible provider"
+        );
+        return createOpenAICompatible(customConfig);
+      }
       default:
         throw new Error(`Unsupported provider: ${provider}`);
     }
@@ -972,9 +1021,18 @@ var _ModelFactory = class _ModelFactory {
     if (providerOptions.baseUrl || providerOptions.baseURL) {
       providerConfig.baseURL = providerOptions.baseUrl || providerOptions.baseURL;
     }
+    if (providerOptions.headers) {
+      providerConfig.headers = providerOptions.headers;
+    }
     if (providerOptions.gateway) {
       Object.assign(providerConfig, providerOptions.gateway);
     }
+    if (providerOptions.nim) {
+      Object.assign(providerConfig, providerOptions.nim);
+    }
+    if (providerOptions.custom) {
+      Object.assign(providerConfig, providerOptions.custom);
+    }
     return providerConfig;
   }
   /**
@@ -1019,9 +1077,15 @@ var _ModelFactory = class _ModelFactory {
         return openrouter(modelName);
       case "gateway":
         return gateway(modelName);
+      case "nim":
+        return nimDefault(modelName);
+      case "custom":
+        throw new Error(
+          "Custom provider requires configuration. Please provide baseURL in providerOptions.custom.baseURL or providerOptions.baseURL"
+        );
       default:
         throw new Error(
-          `Unsupported provider: ${provider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id) or Vercel AI Gateway (gateway/model-id).`
+          `Unsupported provider: ${provider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id), Vercel AI Gateway (gateway/model-id), NVIDIA NIM (nim/model-id), or Custom OpenAI-compatible (custom/model-id).`
         );
     }
   }
@@ -1037,7 +1101,7 @@ var _ModelFactory = class _ModelFactory {
       const normalizedProvider = provider.toLowerCase();
       if (!_ModelFactory.BUILT_IN_PROVIDERS.includes(normalizedProvider)) {
         throw new Error(
-          `Unsupported provider: ${normalizedProvider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id) or Vercel AI Gateway (gateway/model-id).`
+          `Unsupported provider: ${normalizedProvider}. Supported providers are: ${_ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id), Vercel AI Gateway (gateway/model-id), NVIDIA NIM (nim/model-id), or Custom OpenAI-compatible (custom/model-id).`
         );
       }
       return {
@@ -1056,7 +1120,16 @@ var _ModelFactory = class _ModelFactory {
     if (!providerOptions) {
       return {};
     }
-    const excludedKeys = ["apiKey", "baseURL", "baseUrl", "maxDuration"];
+    const excludedKeys = [
+      "apiKey",
+      "baseURL",
+      "baseUrl",
+      "maxDuration",
+      "headers",
+      "gateway",
+      "nim",
+      "custom"
+    ];
     const params = {};
     for (const [key, value] of Object.entries(providerOptions)) {
       if (!excludedKeys.includes(key) && value !== void 0) {
@@ -1117,7 +1190,9 @@ __publicField(_ModelFactory, "BUILT_IN_PROVIDERS", [
   "openai",
   "google",
   "openrouter",
-  "gateway"
+  "gateway",
+  "nim",
+  "custom"
 ]);
 var ModelFactory = _ModelFactory;
 var logger4 = getLogger("ToolSessionManager");
@@ -3142,8 +3217,10 @@ ${this.statusUpdateState?.config.prompt?.trim() || ""}`;
             }
           });
           const result = object;
+          logger8.info({ result: JSON.stringify(result) }, "DEBUG: Result");
           const summaries = [];
           for (const [componentId, data] of Object.entries(result)) {
+            logger8.info({ componentId, data: JSON.stringify(data) }, "DEBUG: Component data");
             if (componentId === "no_relevant_updates") {
               continue;
             }
@@ -3710,10 +3787,10 @@ var AgentSessionManager = class {
   /**
    * Initialize status updates for a session
    */
-  initializeStatusUpdates(sessionId, config, summarizerModel) {
+  initializeStatusUpdates(sessionId, config, summarizerModel, baseModel) {
     const session = this.sessions.get(sessionId);
     if (session) {
-      session.initializeStatusUpdates(config, summarizerModel);
+      session.initializeStatusUpdates(config, summarizerModel, baseModel);
     } else {
       logger8.error(
         {
@@ -7075,7 +7152,7 @@ var Agent = class {
   /**
    * Wraps a tool with streaming lifecycle tracking (start, complete, error) and AgentSession recording
    */
-  wrapToolWithStreaming(toolName, toolDefinition, streamRequestId, toolType) {
+  wrapToolWithStreaming(toolName, toolDefinition, streamRequestId, toolType, relationshipId) {
     if (!toolDefinition || typeof toolDefinition !== "object" || !("execute" in toolDefinition)) {
       return toolDefinition;
     }
@@ -7101,7 +7178,8 @@ var Agent = class {
           agentSessionManager.recordEvent(streamRequestId, "tool_call", this.config.id, {
             toolName,
             input: args,
-            toolCallId
+            toolCallId,
+            relationshipId
           });
         }
         try {
@@ -7146,7 +7224,8 @@ var Agent = class {
               toolName,
               output: result,
               toolCallId,
-              duration
+              duration,
+              relationshipId
             });
           }
           return result;
@@ -7159,7 +7238,8 @@ var Agent = class {
               output: null,
               toolCallId,
               duration,
-              error: errorMessage
+              error: errorMessage,
+              relationshipId
             });
           }
           throw error;
@@ -7225,22 +7305,24 @@ var Agent = class {
     }) || [];
     const tools = await Promise.all(mcpTools.map((tool3) => this.getMcpTool(tool3)) || []) || [];
     if (!sessionId) {
-      const combinedTools = tools.reduce((acc, tool3) => {
-        return Object.assign(acc, tool3);
-      }, {});
       const wrappedTools2 = {};
-      for (const [toolName, toolDef] of Object.entries(combinedTools)) {
-        wrappedTools2[toolName] = this.wrapToolWithStreaming(
-          toolName,
-          toolDef,
-          streamRequestId,
-          "mcp"
-        );
+      for (const [index, toolSet] of tools.entries()) {
+        const relationshipId = mcpTools[index]?.relationshipId;
+        for (const [toolName, toolDef] of Object.entries(toolSet)) {
+          wrappedTools2[toolName] = this.wrapToolWithStreaming(
+            toolName,
+            toolDef,
+            streamRequestId,
+            "mcp",
+            relationshipId
+          );
+        }
       }
       return wrappedTools2;
     }
     const wrappedTools = {};
-    for (const toolSet of tools) {
+    for (const [index, toolSet] of tools.entries()) {
+      const relationshipId = mcpTools[index]?.relationshipId;
       for (const [toolName, originalTool] of Object.entries(toolSet)) {
         if (!isValidTool(originalTool)) {
           logger15.error({ toolName }, "Invalid MCP tool structure - missing required properties");
@@ -7253,7 +7335,7 @@ var Agent = class {
             logger15.debug({ toolName, toolCallId }, "MCP Tool Called");
             try {
               const rawResult = await originalTool.execute(args, { toolCallId });
-              if (rawResult && typeof rawResult === "object" && "isError" in rawResult && rawResult.isError) {
+              if (rawResult && typeof rawResult === "object" && rawResult.isError) {
                 const errorMessage = rawResult.content?.[0]?.text || "MCP tool returned an error";
                 logger15.error(
                   { toolName, toolCallId, errorMessage, rawResult },
@@ -7274,7 +7356,8 @@ var Agent = class {
                     context: {
                       toolName,
                       toolCallId,
-                      errorMessage
+                      errorMessage,
+                      relationshipId
                     }
                   });
                 }
@@ -7313,7 +7396,8 @@ var Agent = class {
           toolName,
           sessionWrappedTool,
           streamRequestId,
-          "mcp"
+          "mcp",
+          relationshipId
         );
       }
     }
@@ -8985,7 +9069,12 @@ var createTaskHandler = (config, credentialStoreRegistry) => {
       const stopWhen = "stopWhen" in config.agentSchema ? config.agentSchema.stopWhen : void 0;
       const toolsForAgentResult = await Promise.all(
         toolsForAgent.data.map(async (item) => {
-          const mcpTool = await dbResultToMcpTool(item.tool, dbClient_default, credentialStoreRegistry);
+          const mcpTool = await dbResultToMcpTool(
+            item.tool,
+            dbClient_default,
+            credentialStoreRegistry,
+            item.id
+          );
           if (item.selectedTools && item.selectedTools.length > 0) {
             const selectedToolsSet = new Set(item.selectedTools);
             mcpTool.availableTools = mcpTool.availableTools?.filter((tool3) => selectedToolsSet.has(tool3.name)) || [];
@@ -9067,7 +9156,8 @@ var createTaskHandler = (config, credentialStoreRegistry) => {
                   const mcpTool = await dbResultToMcpTool(
                     item.tool,
                     dbClient_default,
-                    credentialStoreRegistry
+                    credentialStoreRegistry,
+                    item.id
                   );
                   if (item.selectedTools && item.selectedTools.length > 0) {
                     const selectedToolsSet = new Set(item.selectedTools);
@@ -10372,11 +10462,42 @@ var ExecutionHandler = class {
         scopes: { tenantId, projectId, agentId }
       });
       if (agentConfig?.statusUpdates && agentConfig.statusUpdates.enabled !== false) {
-        agentSessionManager.initializeStatusUpdates(
-          requestId2,
-          agentConfig.statusUpdates,
-          agentConfig.models?.summarizer
-        );
+        try {
+          const agentWithDefault = await getAgentWithDefaultSubAgent(dbClient_default)({
+            scopes: { tenantId, projectId, agentId }
+          });
+          if (agentWithDefault?.defaultSubAgent) {
+            const resolvedModels = await resolveModelConfig(
+              agentId,
+              agentWithDefault.defaultSubAgent
+            );
+            agentSessionManager.initializeStatusUpdates(
+              requestId2,
+              agentConfig.statusUpdates,
+              resolvedModels.summarizer,
+              resolvedModels.base
+            );
+          } else {
+            agentSessionManager.initializeStatusUpdates(
+              requestId2,
+              agentConfig.statusUpdates,
+              agentConfig.models?.summarizer
+            );
+          }
+        } catch (modelError) {
+          logger20.warn(
+            {
+              error: modelError instanceof Error ? modelError.message : "Unknown error",
+              agentId
+            },
+            "Failed to resolve models for status updates, using agent-level config"
+          );
+          agentSessionManager.initializeStatusUpdates(
+            requestId2,
+            agentConfig.statusUpdates,
+            agentConfig.models?.summarizer
+          );
+        }
       }
     } catch (error) {
       logger20.error(

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@inkeep/agents-run-api",
-  "version": "0.33.2",
+  "version": "0.34.1",
   "description": "Agents Run API for Inkeep Agent Framework - handles chat, agent execution, and streaming",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -18,6 +18,7 @@
     "@ai-sdk/gateway": "^1.0.23",
     "@ai-sdk/google": "^2.0.14",
     "@ai-sdk/openai": "2.0.11",
+    "@ai-sdk/openai-compatible": "^1.0.27",
     "@ai-sdk/react": "2.0.11",
     "@alcyone-labs/modelcontextprotocol-sdk": "^1.16.0",
     "@hono/node-server": "^1.14.3",
@@ -53,7 +54,7 @@
     "traverse": "^0.6.11",
     "ts-pattern": "^5.7.1",
     "zod": "4.1.5",
-    "@inkeep/agents-core": "^0.33.2"
+    "@inkeep/agents-core": "^0.34.1"
   },
   "optionalDependencies": {
     "keytar": "^7.9.0"