@cognigy/rest-api-client 4.98.0 → 4.99.0

package/build/shared/charts/descriptors/service/aiAgent/aiAgentJob.js

@@ -30,7 +30,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
         collapsable: true,
         placement: {
             children: {
-                whitelist: ["aiAgentJobDefault", "aiAgentJobTool"],
+                whitelist: ["aiAgentJobDefault", "aiAgentJobTool", "aiAgentJobMCPTool"],
             },
         },
     },
@@ -835,7 +835,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
     ],
     tags: ["ai", "aiAgent"],
     function: async ({ cognigy, config, childConfigs, nodeId }) => {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12;
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, _16, _17, _18, _19;
         const { api, context, input, profile, flowReferenceId } = cognigy;
         const { aiAgent, llmProviderReferenceId, name: jobName, description: jobDescription, instructions: jobInstructions, outputImmediately, toolChoice, useStrict, memoryType, selectedProfileFields, memoryContextInjection, knowledgeSearchBehavior, knowledgeSearchTags, knowledgeSearchTagsFilterOp, knowledgeSearchAiAgentKnowledge, knowledgeSearchJobKnowledge, knowledgeSearchJobStore, knowledgeSearchGenerateSearchPrompt, knowledgeSearchTopK, timeoutInMs, maxTokens, temperature, logErrorToSystem, storeErrorInInput, errorHandling, errorHandlingGotoTarget, errorMessage, debugConfig, debugLogTokenCount, debugResult, storeLocation, contextKey, inputKey, streamStoreCopyInInput, streamStopTokens, processImages, transcriptImageHandling, sessionParams } = config;
         try {
@@ -875,7 +875,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                     const voiceSettings = (0, setSessionConfig_mapper_2.voiceConfigParamsToVoiceSettings)(config, api);
                     const payload = setSessionConfig_mapper_1.setSessionConfig.handleVGInput(voiceSettings, sessionParams || {}, api);
                     if (payload && Object.keys(payload).length > 0) {
-                        api.say(null, {
+                        (_b = api.say) === null || _b === void 0 ? void 0 : _b.call(api, null, {
                             _cognigy: payload,
                         });
                     }
@@ -885,16 +885,16 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                     throw new Error(`[VG2] Error on AI Agent Job node. Error message: ${error.message}`);
                 }
             }
-            const _13 = profile, { profileId, accepted_gdpr, prevent_data_collection, privacy_policy } = _13, cleanedProfile = __rest(_13, ["profileId", "accepted_gdpr", "prevent_data_collection", "privacy_policy"]);
+            const _20 = profile, { profileId, accepted_gdpr, prevent_data_collection, privacy_policy } = _20, cleanedProfile = __rest(_20, ["profileId", "accepted_gdpr", "prevent_data_collection", "privacy_policy"]);
             const userMemory = (0, getUserMemory_1.getUserMemory)(memoryType, selectedProfileFields, aiAgent, cleanedProfile);
             /**
              *  ----- Knowledge Search Section -----
             */
             let knowledgeSearchResponseData;
             const sessionState = await api.loadSessionState();
-            const lastToolCall = (_b = sessionState.lastToolCall) === null || _b === void 0 ? void 0 : _b.toolCall;
+            const lastToolCall = (_c = sessionState.lastToolCall) === null || _c === void 0 ? void 0 : _c.toolCall;
             if (knowledgeSearchBehavior === "always" ||
-                (knowledgeSearchBehavior === "onDemand" && ((_c = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _c === void 0 ? void 0 : _c.name) === "retrieve_knowledge")) {
+                (knowledgeSearchBehavior === "onDemand" && ((_d = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _d === void 0 ? void 0 : _d.name) === "retrieve_knowledge")) {
                 const knowledgeStoreIds = [];
                 if (knowledgeSearchAiAgentKnowledge && aiAgent.knowledgeReferenceId) {
                     knowledgeStoreIds.push(aiAgent.knowledgeReferenceId);
@@ -902,8 +902,8 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                 if (knowledgeSearchJobKnowledge && knowledgeSearchJobStore) {
                     knowledgeStoreIds.push(knowledgeSearchJobStore);
                 }
-                if (knowledgeStoreIds.length > 0 && (input.text || ((_e = (_d = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _d === void 0 ? void 0 : _d.arguments) === null || _e === void 0 ? void 0 : _e.generated_prompt))) {
-                    let query = ((_g = (_f = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _f === void 0 ? void 0 : _f.arguments) === null || _g === void 0 ? void 0 : _g.generated_prompt) || input.text;
+                if (knowledgeStoreIds.length > 0 && (input.text || ((_f = (_e = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _e === void 0 ? void 0 : _e.arguments) === null || _f === void 0 ? void 0 : _f.generated_prompt))) {
+                    let query = ((_h = (_g = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _g === void 0 ? void 0 : _g.arguments) === null || _h === void 0 ? void 0 : _h.generated_prompt) || input.text;
                     if (knowledgeSearchBehavior === "always" && knowledgeSearchGenerateSearchPrompt) {
                         const generatedSearchPrompt = await (0, generateSearchPrompt_1.generateSearchPrompt)({ api, userMemory, llmProviderReferenceId, debugLogTokenCount, memoryContextInjection });
                         if (generatedSearchPrompt === null || generatedSearchPrompt === void 0 ? void 0 : generatedSearchPrompt.generated_prompt) {
@@ -919,10 +919,10 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                         knowledgeStoreIds,
                     };
                     if (knowledgeSearchBehavior === "onDemand") {
-                        const generated_buffer_phrase = (_j = (_h = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _h === void 0 ? void 0 : _h.arguments) === null || _j === void 0 ? void 0 : _j.generated_buffer_phrase;
+                        const generated_buffer_phrase = (_k = (_j = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _j === void 0 ? void 0 : _j.arguments) === null || _k === void 0 ? void 0 : _k.generated_buffer_phrase;
                         if (generated_buffer_phrase) {
                             // output the generated buffer phrase. Don't add it to the transcript, else the LLM will repeat it next time.
-                            await ((_k = api.output) === null || _k === void 0 ? void 0 : _k.call(api, generated_buffer_phrase, {
+                            await ((_l = api.output) === null || _l === void 0 ? void 0 : _l.call(api, generated_buffer_phrase, {
                                 _cognigy: {
                                     _preventTranscript: true
                                 }
@@ -952,7 +952,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                         if (query) {
                             messageLines.push(`\n<b>UI__DEBUG_MODE__AI_AGENT_JOB__KNOWLEDGE_SEARCH__SEARCH_PROMPT</b> ${query}`);
                         }
-                        if ((_l = knowledgeSearchResponseData === null || knowledgeSearchResponseData === void 0 ? void 0 : knowledgeSearchResponseData.topK) === null || _l === void 0 ? void 0 : _l.length) {
+                        if ((_m = knowledgeSearchResponseData === null || knowledgeSearchResponseData === void 0 ? void 0 : knowledgeSearchResponseData.topK) === null || _m === void 0 ? void 0 : _m.length) {
                             knowledgeSearchResponseData === null || knowledgeSearchResponseData === void 0 ? void 0 : knowledgeSearchResponseData.topK.forEach((result, index) => {
                                 var _a;
                                 messageLines.push(`\nTop ${index + 1}:`);
@@ -964,7 +964,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                         else {
                             messageLines.push("UI__DEBUG_MODE__AI_AGENT_JOB__KNOWLEDGE_SEARCH__NO_RESULTS");
                         }
-                        (_m = api.logDebugMessage) === null || _m === void 0 ? void 0 : _m.call(api, messageLines.join("\n"), "UI__DEBUG_MODE__AI_AGENT_JOB__KNOWLEDGE_SEARCH__HEADER");
+                        (_o = api.logDebugMessage) === null || _o === void 0 ? void 0 : _o.call(api, messageLines.join("\n"), "UI__DEBUG_MODE__AI_AGENT_JOB__KNOWLEDGE_SEARCH__HEADER");
                     }
                     // Knowledge Search "onDemand" responds to a tool call
                     if (knowledgeSearchBehavior === "onDemand") {
@@ -993,7 +993,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                         await api.addTranscriptStep(toolAnswer);
                     }
                 }
-                if (((_o = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _o === void 0 ? void 0 : _o.name) === "retrieve_knowledge") {
+                if (((_p = lastToolCall === null || lastToolCall === void 0 ? void 0 : lastToolCall.function) === null || _p === void 0 ? void 0 : _p.name) === "retrieve_knowledge") {
                     // remove the retrieve_knowledge toolCall from session state to avoid infinite loops
                     api.updateSessionStateValues({
                         lastToolCall: undefined
@@ -1005,32 +1005,39 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
             // create the system Message from the AI Agent resource and this Node's config storage
             const systemMessage = (0, createSystemMessage_1.createSystemMessage)(aiAgent, input, jobName, jobDescription, jobInstructions, userMemory, memoryContextInjection, isOnDemandKnowledgeStoreConfigured ? "onDemand" : "none");
             // Create Tools JSON
+            /** This is the list of tools that are used in the AI Agent Job */
             const tools = [];
+            /** Array of tool IDs for deduping */
             const toolIds = [];
-            childConfigs.forEach(child => {
+            /** Map of MCP tool IDs to their respective node IDs they were loaded from */
+            const toolMap = new Map();
+            /** Array of tool names for listing in the debug message */
+            const toolNames = [];
+            for (const child of childConfigs) {
                 if (child.type === "aiAgentJobDefault") {
-                    return;
+                    continue;
                 }
-                ;
                 const toolId = child.config.toolId;
-                if (!toolId) {
-                    throw new Error(`Tool ID is missing in Tool Node configuration.`);
-                }
-                const parsedToolId = api.parseCognigyScriptText(toolId);
-                if (!(0, createSystemMessage_1.validateToolId)(parsedToolId)) {
-                    throw new Error(`Tool ID ${parsedToolId} is not valid. Please use only alphanumeric characters, dashes and underscores.`);
-                }
-                if (toolIds.includes(parsedToolId)) {
-                    throw new Error(`Tool ID ${parsedToolId} is not unique. Please ensure each tool has a unique id.`);
-                }
-                toolIds.push(parsedToolId);
-                if (child.type === "aiAgentJobTool" && (!child.config.condition || !!api.parseCognigyScriptCondition(child.config.condition))) {
+                if (child.type === "aiAgentJobTool" &&
+                    (!child.config.condition || !!api.parseCognigyScriptCondition(child.config.condition))) {
+                    if (!toolId) {
+                        throw new Error(`Tool ID is missing in Tool Node configuration.`);
+                    }
+                    const parsedToolId = api.parseCognigyScriptText(toolId);
+                    if (!(0, createSystemMessage_1.validateToolId)(parsedToolId)) {
+                        throw new Error(`Tool ID ${parsedToolId} is not valid. Please use only alphanumeric characters, dashes and underscores.`);
+                    }
+                    if (toolIds.includes(parsedToolId)) {
+                        throw new Error(`Tool ID ${parsedToolId} is not unique. Please ensure each tool has a unique id.`);
+                    }
+                    toolIds.push(parsedToolId);
+                    toolNames.push(parsedToolId);
                     const tool = {
                         type: "function",
                         function: {
                             name: parsedToolId,
                             description: api.parseCognigyScriptText(child.config.description),
-                        }
+                        },
                     };
                     if (useStrict) {
                         tool.function.strict = true;
@@ -1040,7 +1047,106 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                     }
                     tools.push(tool);
                 }
-            });
+                if (child.type === "aiAgentJobMCPTool" &&
+                    (!child.config.condition || !!api.parseCognigyScriptCondition(child.config.condition))) {
+                    if (!child.config.mcpServerUrl) {
+                        throw new Error(`MCP Server URL is missing in Tool Node configuration.`);
+                    }
+                    const mcpServerUrl = child.config.mcpServerUrl;
+                    const timeout = child.config.timeout;
+                    const cacheTools = child.config.cacheTools;
+                    const sendDebug = child.config.debugMessageFetchedTools;
+                    const toolFilter = child.config.toolFilter;
+                    let mcpTools = null;
+                    try {
+                        mcpTools = await api.fetchMcpTools({
+                            mcpServerUrl,
+                            timeout,
+                            cacheTools,
+                        });
+                    }
+                    catch (error) {
+                        const errorDetails = error instanceof Error
+                            ? {
+                                name: error.name,
+                                message: error.message,
+                            }
+                            : error;
+                        (_q = api.logDebugError) === null || _q === void 0 ? void 0 : _q.call(api, `Unable to connect to MCP Server:<br>${JSON.stringify(errorDetails, null, 2)}`, child.config.name);
+                    }
+                    if (mcpTools) {
+                        if (sendDebug) {
+                            if (mcpTools.length === 0) {
+                                (_r = api.logDebugMessage) === null || _r === void 0 ? void 0 : _r.call(api, `No tools fetched from MCP Tool "${child.config.name}".`, "MCP Tool");
+                            }
+                            if (mcpTools.length > 0) {
+                                const messageLines = [`Fetched tools from MCP Tool "${child.config.name}"`];
+                                mcpTools.forEach((tool) => {
+                                    messageLines.push(`<br>- <b>${tool.name}</b>: ${tool.description}`);
+                                    if (child.config.debugMessageParameters && tool.inputSchema) {
+                                        messageLines.push(`   <b>Parameters</b>:`);
+                                        Object.keys(tool.inputSchema.properties).forEach((key) => {
+                                            const parameter = tool.inputSchema.properties[key];
+                                            const requiredText = tool.inputSchema.required && !tool.inputSchema.required.includes(key) ? " (optional)" : "";
+                                            if (parameter.description) {
+                                                messageLines.push(`   - ${key} (${parameter.type}): ${parameter.description}${requiredText}`);
+                                            }
+                                            else {
+                                                messageLines.push(`   - ${key}: ${parameter.type}${requiredText}`);
+                                            }
+                                        });
+                                    }
+                                });
+                                (_s = api.logDebugMessage) === null || _s === void 0 ? void 0 : _s.call(api, messageLines.join("\n"), "MCP Tool");
+                            }
+                        }
+                        const filteredMcpTools = mcpTools.filter((tool) => {
+                            if (toolFilter && toolFilter !== "none") {
+                                if (toolFilter === "whitelist" && child.config.whitelist) {
+                                    const whitelist = child.config.whitelist.map((item) => item.trim());
+                                    return whitelist.includes(tool.name);
+                                }
+                                else if (toolFilter === "blacklist") {
+                                    // If the blacklist is falsy, all tools are allowed
+                                    if (!child.config.blacklist) {
+                                        return true;
+                                    }
+                                    const blacklist = child.config.blacklist.map((item) => item.trim());
+                                    return !blacklist.includes(tool.name);
+                                }
+                            }
+                            else {
+                                return true;
+                            }
+                        });
+                        const structuredMcpTools = [];
+                        filteredMcpTools.forEach((tool) => {
+                            var _a;
+                            if (toolIds.includes(tool.name)) {
+                                (_a = api.logDebugError) === null || _a === void 0 ? void 0 : _a.call(api, `Tool "${tool.name}" from MCP Tool "${child.config.name}" is not unique and will not be added. Please ensure each tool has a unique id.`);
+                                return;
+                            }
+                            // add tool to the list of tool ids to prevent duplicates
+                            toolIds.push(tool.name);
+                            toolNames.push(`${tool.name} (${child.config.name})`);
+                            toolMap.set(tool.name, child.id);
+                            const structuredTool = {
+                                type: "function",
+                                function: {
+                                    name: tool.name,
+                                    description: tool.description,
+                                },
+                            };
+                            if (tool.inputSchema) {
+                                structuredTool.function.parameters = tool.inputSchema;
+                            }
+                            structuredMcpTools.push(structuredTool);
+                        });
+                        tools.push(...structuredMcpTools);
+                    }
+                }
+            }
+            ;
             // Optional Debug Message with the config
             if (debugConfig) {
                 const messageLines = [];
@@ -1048,14 +1154,14 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                 messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__AI_AGENT_NAME__LABEL</b> ${aiAgent.name}`);
                 messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__JOB_NAME__LABEL</b> ${jobName}`);
                 // Safety settings
-                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_AVOID_HARMFUL_CONTENT</b> ${(_p = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _p === void 0 ? void 0 : _p.avoidHarmfulContent}`);
-                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_AVOID_UNGROUNDED_CONTENT</b> ${(_q = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _q === void 0 ? void 0 : _q.avoidUngroundedContent}`);
-                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_AVOID_COPYRIGHT_INFRINGEMENTS</b> ${(_r = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _r === void 0 ? void 0 : _r.avoidCopyrightInfringements}`);
-                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_PREVENT_JAILBREAK_AND_MANIPULATION</b> ${(_s = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _s === void 0 ? void 0 : _s.preventJailbreakAndManipulation}`);
+                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_AVOID_HARMFUL_CONTENT</b> ${(_t = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _t === void 0 ? void 0 : _t.avoidHarmfulContent}`);
+                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_AVOID_UNGROUNDED_CONTENT</b> ${(_u = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _u === void 0 ? void 0 : _u.avoidUngroundedContent}`);
+                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_AVOID_COPYRIGHT_INFRINGEMENTS</b> ${(_v = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _v === void 0 ? void 0 : _v.avoidCopyrightInfringements}`);
+                messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__SAFETY_SETTINGS_PREVENT_JAILBREAK_AND_MANIPULATION</b> ${(_w = aiAgent === null || aiAgent === void 0 ? void 0 : aiAgent.safetySettings) === null || _w === void 0 ? void 0 : _w.preventJailbreakAndManipulation}`);
                 // Tools
-                if (tools.length > 0) {
+                if (toolNames.length > 0) {
                     messageLines.push("<b>UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__TOOLS__LABEL</b>");
-                    tools.forEach(tool => messageLines.push(`- ${tool.function.name}`));
+                    toolNames.forEach(toolName => messageLines.push(`- ${toolName}`));
                 }
                 // Memory
                 if (memoryType !== "none") {
@@ -1107,7 +1213,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                 messageLines.push(`UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__VOICE_SETTING__TTS_VOICE ${config.ttsVoice || 'UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__VOICE_SETTING__NOT_SET'}`);
                 messageLines.push(`UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__VOICE_SETTING__TTS_LABEL ${config.ttsLabel || 'UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__VOICE_SETTING__NOT_SET'}`);
                 messageLines.push(`UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__VOICE_SETTING__TTS_DISABLE_CACHE ${config.ttsDisableCache || 'UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__VOICE_SETTING__NOT_SET'}`);
-                (_t = api.logDebugMessage) === null || _t === void 0 ? void 0 : _t.call(api, messageLines.join("\n"), "UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__HEADER");
+                (_x = api.logDebugMessage) === null || _x === void 0 ? void 0 : _x.call(api, messageLines.join("\n"), "UI__DEBUG_MODE__AI_AGENT_JOB__CONFIGURATION__HEADER");
             }
             // keep this after the debug message since the "retrieve_knowledge" tool is implicit
             // we only add this tool if at least one knowledge source is enabled
@@ -1151,14 +1257,14 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                 transcript.length > 0 &&
                 transcript[transcript.length - 1].role === transcripts_1.TranscriptRole.USER) {
                 const userInput = transcript[transcript.length - 1];
-                const enhancedInput = `## Knowledge Source Context\nAdditional Context from the knowledge source: \n${JSON.stringify(knowledgeSearchResponseData)}\n\n\n${((_u = userInput === null || userInput === void 0 ? void 0 : userInput.payload) === null || _u === void 0 ? void 0 : _u.text) || input.text}`;
+                const enhancedInput = `## Knowledge Source Context\nAdditional Context from the knowledge source: \n${JSON.stringify(knowledgeSearchResponseData)}\n\n\n${((_y = userInput === null || userInput === void 0 ? void 0 : userInput.payload) === null || _y === void 0 ? void 0 : _y.text) || input.text}`;
                 transcript[transcript.length - 1].payload.text = enhancedInput;
             }
             const isStreamingChannel = input.channel === "webchat3" || input.channel === "adminconsole";
             const _messageId = (0, crypto_1.randomUUID)();
             const llmPromptOptions = Object.assign(Object.assign({ prompt: "", chat: systemMessage, 
                 // Temp fix to override the transcript if needed
-                transcript: ((_v = context === null || context === void 0 ? void 0 : context._cognigy) === null || _v === void 0 ? void 0 : _v.transcript) ? [...context._cognigy.transcript] : transcript, detailedResults: true, timeoutInMs: timeoutInMs !== null && timeoutInMs !== void 0 ? timeoutInMs : 8000, maxTokens: maxTokens !== null && maxTokens !== void 0 ? maxTokens : 4000, temperature: temperature !== null && temperature !== void 0 ? temperature : 0.7, topP: 1, frequencyPenalty: 0, presencePenalty: 0, responseFormat: "text", stream: storeLocation === "stream", streamOnDataHandler: (text) => {
+                transcript: ((_z = context === null || context === void 0 ? void 0 : context._cognigy) === null || _z === void 0 ? void 0 : _z.transcript) ? [...context._cognigy.transcript] : transcript, detailedResults: true, timeoutInMs: timeoutInMs !== null && timeoutInMs !== void 0 ? timeoutInMs : 8000, maxTokens: maxTokens !== null && maxTokens !== void 0 ? maxTokens : 4000, temperature: temperature !== null && temperature !== void 0 ? temperature : 0.7, topP: 1, frequencyPenalty: 0, presencePenalty: 0, responseFormat: "text", stream: storeLocation === "stream", streamOnDataHandler: (text) => {
                     var _a;
                     text = isStreamingChannel ? text : text.trim();
                     if (text) {
@@ -1182,45 +1288,51 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                 };
             }
             // Set understood to true so that an AI Agent interaction doesn't look false in our analytics
-            (_w = api.setAnalyticsData) === null || _w === void 0 ? void 0 : _w.call(api, "understood", "true");
+            (_0 = api.setAnalyticsData) === null || _0 === void 0 ? void 0 : _0.call(api, "understood", "true");
             input.understood = true;
-            const fullLlmResult = await ((_x = api.runGenerativeAIPrompt) === null || _x === void 0 ? void 0 : _x.call(api, llmPromptOptions, "aiAgent"));
+            const fullLlmResult = await ((_1 = api.runGenerativeAIPrompt) === null || _1 === void 0 ? void 0 : _1.call(api, llmPromptOptions, "aiAgent"));
             const { messages } = fullLlmResult, llmResult = __rest(fullLlmResult, ["messages"]);
             const llmProvider = llmResult === null || llmResult === void 0 ? void 0 : llmResult.provider;
             const tokenUsage = fullLlmResult.tokenUsage;
             // Send optional debug message with token usage
             if (debugLogTokenCount && tokenUsage) {
-                (_y = api.logDebugMessage) === null || _y === void 0 ? void 0 : _y.call(api, tokenUsage, "UI__DEBUG_MODE__AI_AGENT_JOB__TOKEN_USAGE__HEADER");
+                (_2 = api.logDebugMessage) === null || _2 === void 0 ? void 0 : _2.call(api, tokenUsage, "UI__DEBUG_MODE__AI_AGENT_JOB__TOKEN_USAGE__HEADER");
             }
             // Identify if the result is a tool call
             // If response is a tool call, set next node for Tools
             if (llmResult.finishReason === "tool_calls" && llmResult.toolCalls.length > 0) {
                 const mainToolCall = llmResult.toolCalls[0];
+                let isMcpToolCall = false;
                 // Find the child node with the toolId of the tool call
-                const toolChild = childConfigs.find(child => { var _a, _b; return child.type === "aiAgentJobTool" && ((_a = child.config) === null || _a === void 0 ? void 0 : _a.toolId) && api.parseCognigyScriptText((_b = child.config) === null || _b === void 0 ? void 0 : _b.toolId) === mainToolCall.function.name; });
+                let toolChild = childConfigs.find(child => { var _a, _b; return child.type === "aiAgentJobTool" && ((_a = child.config) === null || _a === void 0 ? void 0 : _a.toolId) && api.parseCognigyScriptText((_b = child.config) === null || _b === void 0 ? void 0 : _b.toolId) === mainToolCall.function.name; });
+                if (!toolChild && toolMap.has(mainToolCall.function.name)) {
+                    // If the tool call is from an MCP tool, set the next node to the corresponding child node
+                    toolChild = childConfigs.find(child => child.id === toolMap.get(mainToolCall.function.name));
+                    isMcpToolCall = true;
+                }
                 if (mainToolCall.function.name !== "retrieve_knowledge" && toolChild === undefined) {
-                    (_z = api.logDebugError) === null || _z === void 0 ? void 0 : _z.call(api, `UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__ERROR__BODY <b>${mainToolCall.function.name}</b>`, "UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__ERROR__HEADER");
+                    (_3 = api.logDebugError) === null || _3 === void 0 ? void 0 : _3.call(api, `UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__ERROR__BODY <b>${mainToolCall.function.name}</b>`, "UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__ERROR__HEADER");
                 }
                 // Add last tool call to session state for loading it from Tool Answer Node
                 api.updateSessionStateValues({
-                    lastToolCall: {
-                        llmProvider,
-                        aiAgentJobNode: {
+                    lastToolCall: Object.assign(Object.assign({ llmProvider, aiAgentJobNode: {
                             flow: flowReferenceId,
-                            node: nodeId
-                        },
-                        toolCall: mainToolCall,
-                    }
+                            node: nodeId,
+                        } }, (isMcpToolCall && {
+                        mcpServerUrl: (_4 = toolChild === null || toolChild === void 0 ? void 0 : toolChild.config) === null || _4 === void 0 ? void 0 : _4.mcpServerUrl,
+                        timeout: (_5 = toolChild === null || toolChild === void 0 ? void 0 : toolChild.config) === null || _5 === void 0 ? void 0 : _5.timeout,
+                        mcpToolNode: toolChild === null || toolChild === void 0 ? void 0 : toolChild.id,
+                    })), { toolCall: mainToolCall }),
                 });
                 // if there are any parameters/arguments, add them to the input slots
                 if (mainToolCall.function.arguments) {
-                    input.aiAgent = Object.assign(Object.assign({}, input.aiAgent), { toolArgs: Object.assign(Object.assign({}, (_1 = (_0 = input.aiAgent) === null || _0 === void 0 ? void 0 : _0.toolArgs) !== null && _1 !== void 0 ? _1 : {}), mainToolCall.function.arguments) });
+                    input.aiAgent = Object.assign(Object.assign({}, input.aiAgent), { toolArgs: Object.assign(Object.assign({}, (_7 = (_6 = input.aiAgent) === null || _6 === void 0 ? void 0 : _6.toolArgs) !== null && _7 !== void 0 ? _7 : {}), mainToolCall.function.arguments) });
                 }
                 // Debug Message for Tool Calls, configured in the Tool Node
-                if ((_2 = toolChild === null || toolChild === void 0 ? void 0 : toolChild.config) === null || _2 === void 0 ? void 0 : _2.debugMessage) {
+                if ((_8 = toolChild === null || toolChild === void 0 ? void 0 : toolChild.config) === null || _8 === void 0 ? void 0 : _8.debugMessage) {
                     const messageLines = [`<b>UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__DEBUG_MESSAGE__HEADER:</b> ${api.parseCognigyScriptText(toolChild.config.toolId)}`];
                     // Arguments / Parameters Slots
-                    const slots = ((_3 = mainToolCall === null || mainToolCall === void 0 ? void 0 : mainToolCall.function) === null || _3 === void 0 ? void 0 : _3.arguments) && Object.keys(mainToolCall.function.arguments);
+                    const slots = ((_9 = mainToolCall === null || mainToolCall === void 0 ? void 0 : mainToolCall.function) === null || _9 === void 0 ? void 0 : _9.arguments) && Object.keys(mainToolCall.function.arguments);
                     const hasSlots = slots && slots.length > 0;
                     messageLines.push(`<b>UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__DEBUG_MESSAGE__SLOTS</b>${hasSlots ? "" : " -"}`);
                     if (hasSlots) {
@@ -1235,7 +1347,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                             messageLines.push(`- ${slot}: ${slotValueAsString}`);
                         });
                     }
-                    (_4 = api.logDebugMessage) === null || _4 === void 0 ? void 0 : _4.call(api, messageLines.join("\n"), "UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__DEBUG_MESSAGE__HEADER");
+                    (_10 = api.logDebugMessage) === null || _10 === void 0 ? void 0 : _10.call(api, messageLines.join("\n"), "UI__DEBUG_MODE__AI_AGENT_JOB__TOOL_CALL__DEBUG_MESSAGE__HEADER");
                 }
                 if (toolChild) {
                     api.setNextNode(toolChild.id);
@@ -1260,7 +1372,17 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
             }
             // Optionally output the result immediately
             if (llmResult.result && outputImmediately && !llmPromptOptions.stream) {
-                await ((_5 = api.output) === null || _5 === void 0 ? void 0 : _5.call(api, llmResult.result, {}));
+                await ((_11 = api.output) === null || _11 === void 0 ? void 0 : _11.call(api, llmResult.result, {}));
+            }
+            else if (llmResult.finishReason && llmPromptOptions.stream) {
+                // send the finishReason as last output for a stream
+                (_12 = api.output) === null || _12 === void 0 ? void 0 : _12.call(api, "", {
+                    _cognigy: {
+                        _preventTranscript: true,
+                        _messageId,
+                        _finishReason: llmResult.finishReason,
+                    }
+                });
             }
             // If we are streaming and we got a result, also store it into the transcript, since streamed chunks are not stored there
             if (llmResult.result && llmPromptOptions.stream) {
@@ -1277,7 +1399,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
             }
             // Add response to Cognigy Input/Context for further usage
             if (storeLocation === "context") {
-                (_6 = api.addToContext) === null || _6 === void 0 ? void 0 : _6.call(api, contextKey, llmResult, "simple");
+                (_13 = api.addToContext) === null || _13 === void 0 ? void 0 : _13.call(api, contextKey, llmResult, "simple");
             }
             else if (storeLocation === "input") {
                 api.addToInput(inputKey, llmResult);
@@ -1290,14 +1412,14 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
             const errorDetails = {
                 name: (error === null || error === void 0 ? void 0 : error.name) || "Error",
                 code: (error === null || error === void 0 ? void 0 : error.code) || (error === null || error === void 0 ? void 0 : error.httpStatusCode),
-                message: (error === null || error === void 0 ? void 0 : error.message) || ((_7 = error.originalErrorDetails) === null || _7 === void 0 ? void 0 : _7.message),
+                message: (error === null || error === void 0 ? void 0 : error.message) || ((_14 = error.originalErrorDetails) === null || _14 === void 0 ? void 0 : _14.message),
             };
-            (_8 = api.emitEvent) === null || _8 === void 0 ? void 0 : _8.call(api, "nodeError", { nodeId, flowId: flowReferenceId, errorMessage: error });
+            (_15 = api.emitEvent) === null || _15 === void 0 ? void 0 : _15.call(api, "nodeError", { nodeId, flowId: flowReferenceId, errorMessage: error });
             if (logErrorToSystem) {
-                (_9 = api.log) === null || _9 === void 0 ? void 0 : _9.call(api, "error", JSON.stringify(errorDetails));
+                (_16 = api.log) === null || _16 === void 0 ? void 0 : _16.call(api, "error", JSON.stringify(errorDetails));
             }
             if (errorHandling !== "stop") {
-                (_10 = api.logDebugError) === null || _10 === void 0 ? void 0 : _10.call(api, errorDetails.message + (errorDetails.code ? ` (error code: ${errorDetails.code})` : ""), errorDetails.name);
+                (_17 = api.logDebugError) === null || _17 === void 0 ? void 0 : _17.call(api, errorDetails.message + (errorDetails.code ? ` (error code: ${errorDetails.code})` : ""), errorDetails.name);
             }
             if (storeErrorInInput) {
                 input.aiAgent = input.aiAgent || {};
@@ -1306,7 +1428,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
             if (errorHandling === "continue") {
                 // output the timeout message
                 if (errorMessage) {
-                    await ((_11 = api.output) === null || _11 === void 0 ? void 0 : _11.call(api, errorMessage, null));
+                    await ((_18 = api.output) === null || _18 === void 0 ? void 0 : _18.call(api, errorMessage, null));
                 }
                 // Set default node as next node
                 const defaultChild = childConfigs.find(child => child.type === "aiAgentJobDefault");
@@ -1318,7 +1440,7 @@ exports.AI_AGENT_JOB = (0, createNodeDescriptor_1.createNodeDescriptor)({
                 if (!errorHandlingGotoTarget) {
                     throw new Error("GoTo Target is required");
                 }
-                if (!((_12 = api.checkThink) === null || _12 === void 0 ? void 0 : _12.call(api, nodeId))) {
+                if (!((_19 = api.checkThink) === null || _19 === void 0 ? void 0 : _19.call(api, nodeId))) {
                     api.resetNextNodes();
                     await api.executeFlow({
                         flowNode: {