@alpic80/rivet-core 1.24.0-aidon.5 → 1.24.2-aidon.1

package/dist/cjs/bundle.cjs

@@ -89,6 +89,10 @@ __export(src_exports, {
   LoadDatasetNodeImpl: () => LoadDatasetNodeImpl,
   LoopControllerNodeImpl: () => LoopControllerNodeImpl,
   LoopUntilNodeImpl: () => LoopUntilNodeImpl,
+  MCPError: () => MCPError,
+  MCPErrorType: () => MCPErrorType,
+  MCPGetPromptNodeImpl: () => MCPGetPromptNodeImpl,
+  MCPToolCallNodeImpl: () => MCPToolCallNodeImpl,
   MatchNodeImpl: () => MatchNodeImpl,
   NodeImpl: () => NodeImpl,
   NodeRegistration: () => NodeRegistration,
@@ -161,6 +165,7 @@ __export(src_exports, {
   coreCreateProcessor: () => coreCreateProcessor,
   coreRunGraph: () => coreRunGraph,
   createDatasetNode: () => createDatasetNode,
+  createOnStreamUserEvents: () => createOnStreamUserEvents,
   cronNode: () => cronNode,
   dataTypeDisplayNames: () => dataTypeDisplayNames,
   dataTypes: () => dataTypes,
@@ -238,6 +243,9 @@ __export(src_exports, {
   looseDataValueToDataValue: () => looseDataValueToDataValue,
   looseDataValuesToDataValues: () => looseDataValuesToDataValues,
   matchNode: () => matchNode,
+  mcpDiscoveryNode: () => mcpDiscoveryNode,
+  mcpGetPromptNode: () => mcpGetPromptNode,
+  mcpToolCallNode: () => mcpToolCallNode,
   newId: () => newId,
   nodeDefinition: () => nodeDefinition,
   numberNode: () => numberNode,
@@ -352,6 +360,10 @@ __export(exports_exports, {
   LoadDatasetNodeImpl: () => LoadDatasetNodeImpl,
   LoopControllerNodeImpl: () => LoopControllerNodeImpl,
   LoopUntilNodeImpl: () => LoopUntilNodeImpl,
+  MCPError: () => MCPError,
+  MCPErrorType: () => MCPErrorType,
+  MCPGetPromptNodeImpl: () => MCPGetPromptNodeImpl,
+  MCPToolCallNodeImpl: () => MCPToolCallNodeImpl,
   MatchNodeImpl: () => MatchNodeImpl,
   NodeImpl: () => NodeImpl,
   NodeRegistration: () => NodeRegistration,
@@ -423,6 +435,7 @@ __export(exports_exports, {
   coreCreateProcessor: () => coreCreateProcessor,
   coreRunGraph: () => coreRunGraph,
   createDatasetNode: () => createDatasetNode,
+  createOnStreamUserEvents: () => createOnStreamUserEvents,
   cronNode: () => cronNode,
   dataTypeDisplayNames: () => dataTypeDisplayNames,
   dataTypes: () => dataTypes,
@@ -500,6 +513,9 @@ __export(exports_exports, {
   looseDataValueToDataValue: () => looseDataValueToDataValue,
   looseDataValuesToDataValues: () => looseDataValuesToDataValues,
   matchNode: () => matchNode,
+  mcpDiscoveryNode: () => mcpDiscoveryNode,
+  mcpGetPromptNode: () => mcpGetPromptNode,
+  mcpToolCallNode: () => mcpToolCallNode,
   newId: () => newId,
   nodeDefinition: () => nodeDefinition,
   numberNode: () => numberNode,
@@ -1782,10 +1798,10 @@ var DEFAULT_CHAT_ENDPOINT = "https://api.openai.com/v1/chat/completions";
 var DEFAULT_CHAT_NODE_TIMEOUT = 3e4;
 
 // src/model/GraphProcessor.ts
-var import_lodash_es15 = require("lodash");
+var import_lodash_es16 = require("lodash");
 var import_p_queue = __toESM(require("p-queue-6"), 1);
 var import_emittery2 = __toESM(require("emittery-0-13"), 1);
-var import_non_secure76 = require("nanoid/non-secure");
+var import_non_secure78 = require("nanoid/non-secure");
 var import_ts_pattern9 = require("ts-pattern");
 
 // src/model/NodeImpl.ts
@@ -2114,8 +2130,10 @@ var import_non_secure4 = require("nanoid/non-secure");
 var import_ts_dedent3 = require("ts-dedent");
 
 // src/utils/interpolation.ts
-var TOKEN_MATCH_REGEX = /\{\{(?!\{)([^{}\s][^{}]*[^{}\s]|[^{}\s])\}\}(?!\})/g;
-var ESCAPED_TOKEN_REGEX = /\{{3}([^{}]+)\}{3}/g;
+var import_lodash_es4 = require("lodash");
+var TOKEN_MATCH_REGEX = /\{\{([^}]+?)\}\}/g;
+var ESCAPED_TOKEN_REGEX = /\{\{\{([^}]+?)\}\}\}/g;
+var ESCAPED_ESCAPED_TOKEN_REGEX = /\\\{\\\{([^}]+?)\\\}\\\}/g;
 var processingFunctions = {
   indent: (input, spaces = 0) => {
     const indent = " ".repeat(spaces);
@@ -2167,6 +2185,68 @@ var processingFunctions = {
     return lines.join("\n");
   }
 };
+function unwrapPotentialDataValue(value) {
+  if (typeof value === "object" && value !== null && typeof value.type === "string" && Object.prototype.hasOwnProperty.call(value, "value")) {
+    return value.value;
+  }
+  return value;
+}
+function resolveExpressionRawValue(source, expression, sourceType) {
+  if (!source) {
+    return void 0;
+  }
+  const match13 = expression.trim().match(/^([^[.\s]+)\s*(.*)$/);
+  let key;
+  let path;
+  if (match13 && typeof match13[1] === "string") {
+    key = match13[1];
+    const rawPath = match13[2];
+    if (rawPath) {
+      path = rawPath.trim().replace(/\s*(\.|\[|\])\s*/g, "$1");
+    } else {
+      path = void 0;
+    }
+  } else {
+    key = expression.trim();
+    path = void 0;
+  }
+  if (!key) {
+    return void 0;
+  }
+  const topLevelValue = source[key];
+  if (topLevelValue === void 0) {
+    return void 0;
+  }
+  const baseValue = unwrapPotentialDataValue(topLevelValue);
+  let finalValue;
+  if (path) {
+    try {
+      finalValue = (0, import_lodash_es4.get)(baseValue, path);
+      finalValue = unwrapPotentialDataValue(finalValue);
+    } catch (error) {
+      console.warn(`Error accessing path "${path}" in ${sourceType} value for key "${key}":`, error);
+      return void 0;
+    }
+  } else {
+    finalValue = baseValue;
+  }
+  return finalValue;
+}
+function resolveExpressionToString(source, expression, sourceType) {
+  const finalValue = resolveExpressionRawValue(source, expression, sourceType);
+  if (finalValue === void 0) {
+    return void 0;
+  }
+  if (typeof finalValue === "object" && finalValue !== null) {
+    try {
+      return JSON.stringify(finalValue);
+    } catch (error) {
+      console.warn(`Error stringifying object/array in ${sourceType} for expression "${expression}":`, error);
+      return "[object Object]";
+    }
+  }
+  return String(finalValue);
+}
 function parseProcessing(instruction) {
   const parts = instruction.trim().split(/\s+/);
   return {
@@ -2186,26 +2266,57 @@ function applyProcessing(value, processingChain) {
     return processingFunc(result, param);
   }, value);
 }
-function interpolate(baseString, values2) {
-  return baseString.replace(TOKEN_MATCH_REGEX, (_m, p1) => {
-    const [token, ...processing] = p1.split("|");
-    const value = values2[token.trim()];
-    if (value === void 0)
+function interpolate(template, variables, graphInputValues, contextValues) {
+  return template.replace(ESCAPED_TOKEN_REGEX, (_match, expression) => {
+    return `\\{\\{${expression}\\}\\}`;
+  }).replace(/\{\{((?:@graphInputs|@context)\..*?|[^}]+?)\}\}/g, (_match, expressionWithMaybeProcessing) => {
+    const parts = expressionWithMaybeProcessing.split("|").map((s) => s.trim());
+    const expression = parts[0];
+    const processingChain = parts.slice(1).join("|");
+    let resolvedValue;
+    if (expression.startsWith("@graphInputs.")) {
+      resolvedValue = resolveExpressionToString(
+        graphInputValues,
+        expression.substring("@graphInputs.".length),
+        "graphInputs"
+      );
+    } else if (expression.startsWith("@context.")) {
+      resolvedValue = resolveExpressionToString(contextValues, expression.substring("@context.".length), "context");
+    } else {
+      const simpleVar = variables[expression];
+      if (simpleVar !== void 0) {
+        resolvedValue = String(unwrapPotentialDataValue(simpleVar) ?? "");
+      } else {
+        resolvedValue = void 0;
+      }
+    }
+    if (resolvedValue === void 0) {
+      console.warn(`Interpolation variable or path "${expression}" not found or resolved to undefined.`);
       return "";
-    if (processing.length > 0) {
-      return applyProcessing(value, p1);
     }
-    return value;
-  }).replace(ESCAPED_TOKEN_REGEX, (_m, p1) => {
-    return `{{${p1}}}`;
+    if (processingChain) {
+      return applyProcessing(resolvedValue, processingChain);
+    }
+    return resolvedValue;
+  }).replace(ESCAPED_ESCAPED_TOKEN_REGEX, (_match, expression) => {
+    return `{{${expression}}}`;
   });
 }
 function extractInterpolationVariables(template) {
-  const matches = template.matchAll(TOKEN_MATCH_REGEX);
+  const matches = template.replace(ESCAPED_TOKEN_REGEX, (_match, content) => {
+    return `\\{\\{${content}\\}\\}`;
+  }).matchAll(TOKEN_MATCH_REGEX);
   const variables = /* @__PURE__ */ new Set();
   for (const match13 of matches) {
-    const [token] = match13[1].split("|");
-    variables.add(token.trim());
+    if (match13[1]) {
+      const [tokenPart] = match13[1].split("|");
+      if (tokenPart) {
+        const token = tokenPart.trim();
+        if (!token.startsWith("@graphInputs.") && !token.startsWith("@context.")) {
+          variables.add(token);
+        }
+      }
+    }
   }
   return Array.from(variables);
 }
@@ -2223,7 +2334,9 @@ var TextNodeImpl = class extends NodeImpl {
         width: 300
       },
       data: {
-        text: "{{input}}"
+        text: "{{input}}",
+        normalizeLineEndings: true
+        // Default to true for better compatibility
       }
     };
     return chartNode;
@@ -2263,6 +2376,12 @@ var TextNodeImpl = class extends NodeImpl {
         dataKey: "text",
         language: "prompt-interpolation-markdown",
         theme: "prompt-interpolation"
+      },
+      {
+        type: "toggle",
+        label: "Normalize Line Endings",
+        dataKey: "normalizeLineEndings",
+        helperMessage: "Normalize line endings to use only LF (\\n) instead of CRLF (\\r\\n)."
       }
     ];
   }
@@ -2275,7 +2394,7 @@ var TextNodeImpl = class extends NodeImpl {
       text: truncated
     };
   }
-  async process(inputs) {
+  async process(inputs, context) {
     const inputMap = Object.keys(inputs).reduce(
       (acc, key) => {
         const stringValue = coerceTypeOptional(inputs[key], "string") ?? "";
@@ -2284,7 +2403,17 @@ var TextNodeImpl = class extends NodeImpl {
       },
       {}
     );
-    const outputValue = interpolate(this.chartNode.data.text, inputMap);
+    let outputValue = interpolate(
+      this.chartNode.data.text,
+      inputMap,
+      context.graphInputNodeValues,
+      // Pass graph inputs
+      context.contextValues
+      // Pass context values
+    );
+    if (this.chartNode.data.normalizeLineEndings) {
+      outputValue = outputValue.replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+    }
     return {
       output: {
         type: "string",
@@ -2326,7 +2455,7 @@ __export(openai_exports, {
   openaiModels: () => openaiModels,
   streamChatCompletions: () => streamChatCompletions
 });
-var import_lodash_es4 = require("lodash");
+var import_lodash_es5 = require("lodash");
 
 // src/utils/fetchEventSource.ts
 var EventSourceResponse = class extends Response {
@@ -2629,6 +2758,30 @@ var openaiModels = {
     },
     displayName: "GPT-4o Audio (Preview)"
   },
+  "gpt-4.1": {
+    maxTokens: 1047576,
+    cost: {
+      prompt: 2e-6,
+      completion: 8e-6
+    },
+    displayName: "GPT-4.1"
+  },
+  o3: {
+    maxTokens: 2e5,
+    cost: {
+      prompt: 1e-5,
+      completion: 4e-5
+    },
+    displayName: "o3"
+  },
+  "o4-mini": {
+    maxTokens: 2e5,
+    cost: {
+      prompt: 11e-7,
+      completion: 44e-7
+    },
+    displayName: "o4-mini"
+  },
   "local-model": {
     maxTokens: Number.MAX_SAFE_INTEGER,
     cost: {
@@ -2638,7 +2791,7 @@ var openaiModels = {
     displayName: "Local Model"
   }
 };
-var openAiModelOptions = (0, import_lodash_es4.orderBy)(
+var openAiModelOptions = (0, import_lodash_es5.orderBy)(
   Object.entries(openaiModels).map(([id, { displayName }]) => ({
     value: id,
     label: displayName
@@ -3529,7 +3682,7 @@ var ChatNodeBase = {
       })
     );
     const { messages } = getChatNodeMessages(inputs);
-    const isReasoningModel = finalModel.startsWith("o1") || finalModel.startsWith("o3");
+    const isReasoningModel = finalModel.startsWith("o1") || finalModel.startsWith("o3") || finalModel.startsWith("o4");
     const completionMessages = await Promise.all(
       messages.map((message) => chatMessageToOpenAIChatCompletionMessage(message, { isReasoningModel }))
     );
@@ -4079,7 +4232,7 @@ var chatNode = nodeDefinition(ChatNodeImpl, "Chat");
 
 // src/model/nodes/PromptNode.ts
 var import_non_secure6 = require("nanoid/non-secure");
-var import_lodash_es5 = require("lodash");
+var import_lodash_es6 = require("lodash");
 var import_ts_dedent5 = require("ts-dedent");
 var import_ts_pattern4 = require("ts-pattern");
 var PromptNodeImpl = class extends NodeImpl {
@@ -4248,8 +4401,13 @@ var PromptNodeImpl = class extends NodeImpl {
     };
   }
   async process(inputs, context) {
-    const inputMap = (0, import_lodash_es5.mapValues)(inputs, (input) => coerceType(input, "string"));
-    const outputValue = interpolate(this.chartNode.data.promptText, inputMap);
+    const inputMap = (0, import_lodash_es6.mapValues)(inputs, (input) => coerceType(input, "string"));
+    const outputValue = interpolate(
+      this.chartNode.data.promptText,
+      inputMap,
+      context.graphInputNodeValues,
+      context.contextValues
+    );
     const type = getInputOrData(this.data, inputs, "type", "string");
     const isCacheBreakpoint = getInputOrData(this.data, inputs, "isCacheBreakpoint", "boolean");
     if (["assistant", "system", "user", "function"].includes(type) === false) {
@@ -4650,13 +4808,19 @@ var CodeNodeImpl = class extends NodeImpl {
     };
   }
   async process(inputs, context) {
-    const outputs = await context.codeRunner.runCode(this.data.code, inputs, {
-      includeFetch: this.data.allowFetch ?? false,
-      includeRequire: this.data.allowRequire ?? false,
-      includeRivet: this.data.allowRivet ?? false,
-      includeProcess: this.data.allowProcess ?? false,
-      includeConsole: this.data.allowConsole ?? false
-    });
+    const outputs = await context.codeRunner.runCode(
+      this.data.code,
+      inputs,
+      {
+        includeFetch: this.data.allowFetch ?? false,
+        includeRequire: this.data.allowRequire ?? false,
+        includeRivet: this.data.allowRivet ?? false,
+        includeProcess: this.data.allowProcess ?? false,
+        includeConsole: this.data.allowConsole ?? false
+      },
+      context.graphInputNodeValues,
+      context.contextValues
+    );
     if (outputs == null || typeof outputs !== "object" || "then" in outputs && typeof outputs.then === "function") {
       throw new Error("Code node must return an object with output values.");
     }
@@ -5820,6 +5984,7 @@ var GraphInputNodeImpl = class extends NodeImpl {
       type: this.data.dataType,
       value: inputValue
     };
+    context.graphInputNodeValues[this.data.id] = value;
     return { ["data"]: value };
   }
 };
@@ -6104,7 +6269,7 @@ var subGraphNode = nodeDefinition(SubGraphNodeImpl, "Subgraph");
 
 // src/model/nodes/ArrayNode.ts
 var import_non_secure19 = require("nanoid/non-secure");
-var import_lodash_es6 = require("lodash");
+var import_lodash_es7 = require("lodash");
 var import_ts_dedent18 = require("ts-dedent");
 var ArrayNodeImpl = class extends NodeImpl {
   static create() {
@@ -6210,7 +6375,7 @@ var ArrayNodeImpl = class extends NodeImpl {
           if (Array.isArray(input == null ? void 0 : input.value)) {
             for (const value of (input == null ? void 0 : input.value) ?? []) {
               if (this.data.flattenDeep) {
-                outputArray.push(...Array.isArray(value) ? (0, import_lodash_es6.flattenDeep)(value) : [value]);
+                outputArray.push(...Array.isArray(value) ? (0, import_lodash_es7.flattenDeep)(value) : [value]);
               } else {
                 outputArray.push(value);
               }
@@ -6350,7 +6515,7 @@ var extractJsonNode = nodeDefinition(ExtractJsonNodeImpl, "Extract JSON");
 
 // src/model/nodes/AssemblePromptNode.ts
 var import_non_secure21 = require("nanoid/non-secure");
-var import_lodash_es7 = require("lodash");
+var import_lodash_es8 = require("lodash");
 var import_ts_dedent20 = require("ts-dedent");
 var AssemblePromptNodeImpl = class extends NodeImpl {
   static create() {
@@ -6457,7 +6622,7 @@ var AssemblePromptNodeImpl = class extends NodeImpl {
     const output = {};
     const isLastMessageCacheBreakpoint = getInputOrData(this.data, inputs, "isLastMessageCacheBreakpoint", "boolean");
     const outMessages = [];
-    const inputMessages = (0, import_lodash_es7.orderBy)(
+    const inputMessages = (0, import_lodash_es8.orderBy)(
       Object.entries(inputs).filter(([key]) => key.startsWith("message")),
       ([key]) => key,
       "asc"
@@ -7011,7 +7176,7 @@ var trimChatMessagesNode = nodeDefinition(TrimChatMessagesNodeImpl, "Trim Chat M
 
 // src/model/nodes/ExternalCallNode.ts
 var import_non_secure25 = require("nanoid/non-secure");
-var import_lodash_es8 = require("lodash");
+var import_lodash_es9 = require("lodash");
 var import_ts_dedent24 = require("ts-dedent");
 var ExternalCallNodeImpl = class extends NodeImpl {
   static create() {
@@ -7106,7 +7271,7 @@ var ExternalCallNodeImpl = class extends NodeImpl {
       }
     }
     const fn = context.externalFunctions[functionName];
-    const externalContext = (0, import_lodash_es8.omit)(context, ["setGlobal"]);
+    const externalContext = (0, import_lodash_es9.omit)(context, ["setGlobal"]);
     if (!fn) {
       if (this.data.useErrorOutput) {
         return {
@@ -9110,7 +9275,7 @@ var joinNode = nodeDefinition(JoinNodeImpl, "Coalesce");
 
 // src/model/nodes/FilterNode.ts
 var import_non_secure45 = require("nanoid/non-secure");
-var import_lodash_es9 = require("lodash");
+var import_lodash_es10 = require("lodash");
 var import_ts_dedent44 = require("ts-dedent");
 var FilterNodeImpl = class extends NodeImpl {
   static create() {
@@ -9166,7 +9331,7 @@ var FilterNodeImpl = class extends NodeImpl {
     var _a;
     const array = coerceType(inputs["array"], "any[]");
     const include = coerceType(inputs["include"], "boolean[]");
-    const zipped = (0, import_lodash_es9.zip)(array, include);
+    const zipped = (0, import_lodash_es10.zip)(array, include);
     const filtered = zipped.filter(([_, include2]) => include2).map(([value, _]) => value);
     return {
       ["filtered"]: {
@@ -9202,16 +9367,18 @@ var ObjectNodeImpl = class extends NodeImpl {
     return chartNode;
   }
   getInputDefinitions() {
-    const inputNames = [...new Set(this.chartNode.data.jsonTemplate.match(/\{\{([^}]+)\}\}/g))];
-    return (inputNames == null ? void 0 : inputNames.map((inputName) => {
+    const jsonTemplate = this.chartNode.data.jsonTemplate ?? "";
+    const matches = jsonTemplate.match(/\{\{([^}]+?)\}\}/g);
+    const allTokens = matches ?? [];
+    const inputTokens = allTokens.map((token) => token.slice(2, -2).trim()).filter((tokenContent) => !tokenContent.startsWith("@graphInputs.") && !tokenContent.startsWith("@context.")).filter((token) => token !== "");
+    return [...new Set(inputTokens)].map((inputName) => {
       return {
-        // id and title should not have the {{ and }}
-        id: inputName.slice(2, -2),
-        title: inputName.slice(2, -2),
+        id: inputName,
+        title: inputName,
         dataType: "any",
         required: false
       };
-    })) ?? [];
+    });
   }
   getOutputDefinitions() {
     return [
@@ -9252,10 +9419,28 @@ var ObjectNodeImpl = class extends NodeImpl {
       group: ["Objects"]
     };
   }
-  interpolate(baseString, values2) {
-    return baseString.replace(/("?)\{\{([^}]+)\}\}("?)/g, (_m, openQuote, key, _closeQuote) => {
+  interpolate(baseString, values2, graphInputNodeValues, contextValues) {
+    return baseString.replace(/("?)\{\{([^}]+?)\}\}("?)/g, (_m, openQuote, key, _closeQuote) => {
       const isQuoted = Boolean(openQuote);
-      const value = values2[key];
+      const trimmedKey = key.trim();
+      let value;
+      const graphInputPrefix = "@graphInputs.";
+      const contextPrefix = "@context.";
+      if (trimmedKey.startsWith(graphInputPrefix) && graphInputNodeValues) {
+        value = resolveExpressionRawValue(
+          graphInputNodeValues,
+          trimmedKey.substring(graphInputPrefix.length),
+          "graphInputs"
+        );
+      } else if (trimmedKey.startsWith(contextPrefix) && contextValues) {
+        value = resolveExpressionRawValue(
+          contextValues,
+          trimmedKey.substring(contextPrefix.length),
+          "context"
+        );
+      } else {
+        value = values2[trimmedKey];
+      }
       if (value == null) {
         return "null";
       }
@@ -9268,27 +9453,32 @@ var ObjectNodeImpl = class extends NodeImpl {
       return JSON.stringify(value);
     });
   }
-  async process(inputs) {
+  async process(inputs, context) {
     const inputMap = Object.keys(inputs).reduce(
       (acc, key) => {
-        var _a;
-        acc[key] = (_a = inputs[key]) == null ? void 0 : _a.value;
+        acc[key] = unwrapPotentialDataValue(inputs[key]);
         return acc;
       },
       {}
     );
-    const outputValue = JSON.parse(this.interpolate(this.chartNode.data.jsonTemplate, inputMap));
-    if (Array.isArray(outputValue)) {
-      return {
-        output: {
-          type: "object[]",
-          value: outputValue
-        }
-      };
+    const interpolatedString = this.interpolate(
+      this.chartNode.data.jsonTemplate,
+      inputMap,
+      context.graphInputNodeValues,
+      // Pass graph inputs
+      context.contextValues
+      // Pass context values
+    );
+    let outputValue;
+    try {
+      outputValue = JSON.parse(interpolatedString);
+    } catch (err) {
+      throw new Error(`Failed to parse JSON template: ${err.message}`);
     }
+    const outputType = Array.isArray(outputValue) ? "object[]" : "object";
     return {
       output: {
-        type: "object",
+        type: outputType,
         value: outputValue
       }
     };
@@ -9364,7 +9554,7 @@ var booleanNode = nodeDefinition(BooleanNodeImpl, "Boolean");
 
 // src/model/nodes/CompareNode.ts
 var import_non_secure48 = require("nanoid/non-secure");
-var import_lodash_es10 = require("lodash");
+var import_lodash_es11 = require("lodash");
 var import_ts_pattern6 = require("ts-pattern");
 var import_ts_dedent47 = require("ts-dedent");
 var CompareNodeImpl = class extends NodeImpl {
@@ -9471,7 +9661,7 @@ var CompareNodeImpl = class extends NodeImpl {
     return {
       ["output"]: {
         type: "boolean",
-        value: (0, import_ts_pattern6.match)(comparisonFunction).with("==", () => (0, import_lodash_es10.isEqual)(value1, value2)).with("!=", () => !(0, import_lodash_es10.isEqual)(value1, value2)).with("<", () => value1 < value2).with(">", () => value1 > value2).with("<=", () => value1 <= value2).with(">=", () => value1 >= value2).with("and", () => !!(value1 && value2)).with("or", () => !!(value1 || value2)).with("xor", () => !!(value1 ? !value2 : value2)).with("nand", () => !(value1 && value2)).with("nor", () => !(value1 || value2)).with("xnor", () => !(value1 ? !value2 : value2)).exhaustive()
+        value: (0, import_ts_pattern6.match)(comparisonFunction).with("==", () => (0, import_lodash_es11.isEqual)(value1, value2)).with("!=", () => !(0, import_lodash_es11.isEqual)(value1, value2)).with("<", () => value1 < value2).with(">", () => value1 > value2).with("<=", () => value1 <= value2).with(">=", () => value1 >= value2).with("and", () => !!(value1 && value2)).with("or", () => !!(value1 || value2)).with("xor", () => !!(value1 ? !value2 : value2)).with("nand", () => !(value1 && value2)).with("nor", () => !(value1 || value2)).with("xnor", () => !(value1 ? !value2 : value2)).exhaustive()
       }
     };
   }
@@ -9780,7 +9970,7 @@ var randomNumberNode = nodeDefinition(RandomNumberNodeImpl, "Random Number");
 
 // src/model/nodes/ShuffleNode.ts
 var import_non_secure52 = require("nanoid/non-secure");
-var import_lodash_es11 = require("lodash");
+var import_lodash_es12 = require("lodash");
 var import_ts_dedent51 = require("ts-dedent");
 var ShuffleNodeImpl = class extends NodeImpl {
   static create() {
@@ -9829,7 +10019,7 @@ var ShuffleNodeImpl = class extends NodeImpl {
     var _a;
     const input = inputs["array"];
     const items = input ? isArrayDataValue(input) ? input.value : [input.value] : [];
-    const shuffled = (0, import_lodash_es11.shuffle)(items);
+    const shuffled = (0, import_lodash_es12.shuffle)(items);
     return {
       ["shuffled"]: {
         type: ((_a = inputs["array"]) == null ? void 0 : _a.type) ?? "any[]",
@@ -10871,7 +11061,7 @@ var GetAllDatasetsNodeImpl = class extends NodeImpl {
   getEditors() {
     return [];
   }
-  async process(context) {
+  async process(_inputs, context) {
     const { datasetProvider } = context;
     if (datasetProvider == null) {
       throw new Error("datasetProvider is required");
@@ -11374,7 +11564,7 @@ var extractMarkdownCodeBlocksNode = nodeDefinition(
 
 // src/model/nodes/AssembleMessageNode.ts
 var import_non_secure62 = require("nanoid/non-secure");
-var import_lodash_es12 = require("lodash");
+var import_lodash_es13 = require("lodash");
 var import_ts_pattern8 = require("ts-pattern");
 var messageTypeToTitle = {
   assistant: "Assistant",
@@ -11531,7 +11721,7 @@ var AssembleMessageNodeImpl = class extends NodeImpl {
     ).otherwise(() => {
       throw new Error(`Invalid type: ${type}`);
     });
-    const inputParts = (0, import_lodash_es12.orderBy)(
+    const inputParts = (0, import_lodash_es13.orderBy)(
       Object.entries(inputs).filter(([key]) => key.startsWith("part")),
       ([key]) => key,
       "asc"
@@ -12042,9 +12232,9 @@ var import_non_secure67 = require("nanoid/non-secure");
 var import_ts_dedent61 = require("ts-dedent");
 
 // src/api/looseDataValue.ts
-var import_lodash_es13 = require("lodash");
+var import_lodash_es14 = require("lodash");
 function looseDataValuesToDataValues(values2) {
-  return (0, import_lodash_es13.mapValues)(values2, (val) => looseDataValueToDataValue(val));
+  return (0, import_lodash_es14.mapValues)(values2, (val) => looseDataValueToDataValue(val));
 }
 function looseDataValueToDataValue(value) {
   if (typeof value === "string") {
@@ -13278,7 +13468,7 @@ var cronNode = nodeDefinition(CronNodeImpl, "Cron");
 // src/model/nodes/ToTreeNode.ts
 var import_non_secure73 = require("nanoid/non-secure");
 var import_ts_dedent67 = require("ts-dedent");
-var import_lodash_es14 = require("lodash");
+var import_lodash_es15 = require("lodash");
 var ToTreeNodeImpl = class extends NodeImpl {
   static create() {
     const chartNode = {
@@ -13364,7 +13554,7 @@ var ToTreeNodeImpl = class extends NodeImpl {
     if (!Array.isArray(objects) || objects.length === 0)
       return "";
     let result = "";
-    const sortedObjects = this.data.useSortAlphabetically ? (0, import_lodash_es14.sortBy)(objects, (obj) => String((0, import_lodash_es14.get)(obj, "path", ""))) : objects;
+    const sortedObjects = this.data.useSortAlphabetically ? (0, import_lodash_es15.sortBy)(objects, (obj) => String((0, import_lodash_es15.get)(obj, "path", ""))) : objects;
     sortedObjects.forEach((obj, index) => {
       const isLastItem = index === sortedObjects.length - 1;
       const prefix = level === 0 ? "" : isLast ? "\u2514\u2500\u2500 " : "\u251C\u2500\u2500 ";
@@ -13373,14 +13563,14 @@ var ToTreeNodeImpl = class extends NodeImpl {
       const interpolationVars = matches.reduce(
         (acc, match13) => {
           const key = match13;
-          acc[key] = String((0, import_lodash_es14.get)(obj, key, ""));
+          acc[key] = String((0, import_lodash_es15.get)(obj, key, ""));
           return acc;
         },
         {}
       );
       const formattedNode = interpolate(this.data.format, interpolationVars);
       result += indent + prefix + formattedNode + "\n";
-      const children = (0, import_lodash_es14.get)(obj, this.data.childrenProperty);
+      const children = (0, import_lodash_es15.get)(obj, this.data.childrenProperty);
       if (Array.isArray(children) && children.length > 0) {
         const newPath = parentPath ? `${parentPath}/${formattedNode}` : formattedNode;
         result += this.buildTree(children, newPath, level + 1, isLastItem);
@@ -13607,113 +13797,865 @@ until ${condition}${maxIterations}`;
 };
 var loopUntilNode = nodeDefinition(LoopUntilNodeImpl, "Loop Until");
 
-// src/model/nodes/ReferencedGraphAliasNode.ts
-var import_non_secure75 = require("nanoid/non-secure");
-var import_ts_dedent69 = require("ts-dedent");
-var ReferencedGraphAliasNodeImpl = class extends NodeImpl {
+// src/model/nodes/MCPDiscoveryNode.ts
+var import_nanoid3 = require("nanoid");
+
+// src/integrations/mcp/MCPProvider.ts
+var MCPErrorType = /* @__PURE__ */ ((MCPErrorType2) => {
+  MCPErrorType2["CONFIG_NOT_FOUND"] = "CONFIG_NOT_FOUND";
+  MCPErrorType2["SERVER_NOT_FOUND"] = "SERVER_NOT_FOUND";
+  MCPErrorType2["SERVER_COMMUNICATION_FAILED"] = "SERVER_COMMUNICATION_FAILED";
+  MCPErrorType2["INVALID_SCHEMA"] = "INVALID_SCHEMA";
+  return MCPErrorType2;
+})(MCPErrorType || {});
+var MCPError = class extends Error {
+  constructor(type, message, details) {
+    super(message);
+    this.type = type;
+    this.details = details;
+    this.name = "Error";
+  }
+};
+
+// src/integrations/mcp/MCPUtils.ts
+var loadMCPConfiguration = async (context) => {
+  if (context.executor !== "nodejs") {
+    throw new MCPError("CONFIG_NOT_FOUND" /* CONFIG_NOT_FOUND */, "MCP config is not supported in browser environment");
+  }
+  const mcpConfig = context.project.metadata.mcpServer;
+  if (!mcpConfig || mcpConfig.mcpServers == null) {
+    throw new MCPError("CONFIG_NOT_FOUND" /* CONFIG_NOT_FOUND */, "MCP configuration not defined in Project tab");
+  }
+  return mcpConfig;
+};
+var getServerOptions = async (context) => {
+  if (context.executor === "nodejs" && context.nativeApi) {
+    try {
+      const config = await loadMCPConfiguration(context);
+      return Object.entries(config.mcpServers).filter(([, config2]) => !config2.disabled).map(([id]) => ({
+        label: id,
+        value: id
+      }));
+    } catch {
+    }
+  }
+  return [];
+};
+var getServerHelperMessage = (context, optionsLength) => {
+  if (optionsLength > 0)
+    return "Select an MCP server from local configuration located in the Project tab";
+  if (context.executor !== "nodejs")
+    return "MCP nodes require Node Executor";
+  return "No MCP servers found in config";
+};
+
+// src/integrations/mcp/MCPBase.ts
+var getMCPBaseInputs = (data) => {
+  const inputs = [];
+  if (data.useNameInput) {
+    inputs.push({
+      dataType: "string",
+      id: "name",
+      title: "Name"
+    });
+  }
+  if (data.useVersionInput) {
+    inputs.push({
+      dataType: "string",
+      id: "version",
+      title: "Version"
+    });
+  }
+  if (data.transportType === "http" && data.useServerUrlInput) {
+    inputs.push({
+      dataType: "string",
+      id: "serverUrl",
+      title: "Server URL",
+      description: "The endpoint URL for the MCP server"
+    });
+  }
+  return inputs;
+};
+
+// src/model/nodes/MCPDiscoveryNode.ts
+var MCPDiscoveryNodeImpl = class extends NodeImpl {
   static create() {
     const chartNode = {
-      type: "referencedGraphAlias",
-      title: "",
-      // Always set initially by the editor
-      id: (0, import_non_secure75.nanoid)(),
+      type: "mcpDiscovery",
+      title: "MCP Discovery",
+      id: (0, import_nanoid3.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
-        width: 300
+        width: 250
       },
       data: {
-        projectId: void 0,
-        // Always set initially by the editor
-        graphId: void 0,
-        // Always set initially by the editor
-        useErrorOutput: false
+        name: "mcp-client",
+        version: "1.0.0",
+        transportType: "stdio",
+        serverUrl: "http://localhost:8080/mcp",
+        serverId: "",
+        useNameInput: false,
+        useVersionInput: false,
+        useToolsOutput: true,
+        usePromptsOutput: true
       }
     };
     return chartNode;
   }
-  getInputDefinitions(_connections, _nodes, _project, referencedProjects) {
-    const referencedProject = referencedProjects[this.data.projectId];
-    if (!referencedProject) {
-      return [];
-    }
-    const graph = referencedProject.graphs[this.data.graphId];
-    if (!graph) {
-      return [];
-    }
-    const inputNodes = graph.nodes.filter((node) => node.type === "graphInput");
-    const inputIds = [...new Set(inputNodes.map((node) => node.data.id))].sort();
-    return inputIds.map(
-      (id) => ({
-        id,
-        title: id,
-        dataType: inputNodes.find((node) => node.data.id === id).data.dataType
-      })
-    );
-  }
-  getGraphOutputs(referencedProject) {
-    const graph = referencedProject.graphs[this.data.graphId];
-    if (!graph) {
-      return [];
-    }
-    const outputNodes = graph.nodes.filter((node) => node.type === "graphOutput");
-    const outputIds = [...new Set(outputNodes.map((node) => node.data.id))].sort();
-    const outputs = outputIds.map(
-      (id) => ({
-        id,
-        title: id,
-        dataType: outputNodes.find((node) => node.data.id === id).data.dataType
-      })
-    );
-    return outputs;
+  getInputDefinitions() {
+    const inputs = getMCPBaseInputs(this.data);
+    return inputs;
   }
-  getOutputDefinitions(_connections, _nodes, _project, referencedProjects) {
-    const outputs = [];
-    const referencedProject = referencedProjects[this.data.projectId];
-    if (!referencedProject) {
-      return outputs;
+  getOutputDefinitions() {
+    const outputDefinitions = [];
+    if (this.data.useToolsOutput) {
+      outputDefinitions.push({
+        id: "tools",
+        title: "Tools",
+        dataType: "object[]",
+        description: "Tools returned from the MCP server"
+      });
     }
-    outputs.push(...this.getGraphOutputs(referencedProject));
-    if (this.data.useErrorOutput) {
-      outputs.push({
-        id: "error",
-        title: "Error",
-        dataType: "string"
+    if (this.data.usePromptsOutput) {
+      outputDefinitions.push({
+        id: "prompts",
+        title: "Prompts",
+        dataType: "object[]",
+        description: "Prompts returned from the MCP server"
       });
     }
-    return outputs;
+    return outputDefinitions;
   }
-  getEditors(context) {
-    const definitions = [
+  async getEditors(context) {
+    const editors = [
       {
         type: "toggle",
-        label: "Use Error Output",
-        dataKey: "useErrorOutput"
+        label: "Output Tools",
+        dataKey: "useToolsOutput",
+        helperMessage: "Toggle on if you want to get a Tools output"
       },
       {
         type: "toggle",
-        label: "Output Cost & Duration",
-        dataKey: "outputCostDuration"
+        label: "Output Prompts",
+        dataKey: "usePromptsOutput",
+        helperMessage: "Toggle on if you want to get a Prompts output"
+      },
+      {
+        type: "string",
+        label: "Name",
+        dataKey: "name",
+        useInputToggleDataKey: "useNameInput",
+        helperMessage: "The name for the MCP Client"
+      },
+      {
+        type: "string",
+        label: "Version",
+        dataKey: "version",
+        useInputToggleDataKey: "useVersionInput",
+        helperMessage: "A version for the MCP Client"
+      },
+      {
+        type: "dropdown",
+        label: "Transport Type",
+        dataKey: "transportType",
+        options: [
+          { label: "HTTP", value: "http" },
+          { label: "STDIO", value: "stdio" }
+        ]
       }
     ];
-    const referencedProject = context.referencedProjects[this.data.projectId];
-    if (referencedProject) {
-      const graph = referencedProject.graphs[this.data.graphId];
-      if (graph) {
-        const inputNodes = graph.nodes.filter((node) => node.type === "graphInput");
-        const inputIds = [...new Set(inputNodes.map((node) => node.data.id))].sort();
-        for (const inputId of inputIds) {
-          const inputNode = inputNodes.find((node) => node.data.id === inputId);
-          definitions.push({
-            type: "dynamic",
-            dataKey: "inputData",
-            dynamicDataKey: inputNode.data.id,
-            dataType: inputNode.data.dataType,
-            label: inputNode.data.id,
-            editor: inputNode.data.editor ?? "auto"
-          });
-        }
-      }
+    if (this.data.transportType === "http") {
+      editors.push({
+        type: "string",
+        label: "Server URL",
+        dataKey: "serverUrl",
+        useInputToggleDataKey: "useServerUrlInput",
+        helperMessage: "The base URL endpoint for the MCP server with `/mcp`"
+      });
+    } else if (this.data.transportType === "stdio") {
+      const serverOptions = await getServerOptions(context);
+      editors.push({
+        type: "dropdown",
+        label: "Server ID",
+        dataKey: "serverId",
+        helperMessage: getServerHelperMessage(context, serverOptions.length),
+        options: serverOptions
+      });
+    }
+    return editors;
+  }
+  getBody(context) {
+    let base;
+    if (this.data.transportType === "http") {
+      base = this.data.useServerUrlInput ? "(Using Server URL Input)" : this.data.serverUrl;
+    } else {
+      base = `Server ID: ${this.data.serverId || "(None)"}`;
+    }
+    const namePart = `Name: ${this.data.name}`;
+    const versionPart = `Version: ${this.data.version}`;
+    const parts = [namePart, versionPart, base];
+    if (context.executor !== "nodejs") {
+      parts.push("(Requires Node Executor)");
+    }
+    return parts.join("\n");
+  }
+  static getUIData() {
+    return {
+      infoBoxBody: import_ts_dedent.dedent`
+        Connects to an MCP (Model Context Protocol) server to discover capabilities like tools and prompts.
+      `,
+      infoBoxTitle: "MCP Discovery Node",
+      contextMenuTitle: "MCP Discovery",
+      group: ["MCP"]
+    };
+  }
+  async process(inputs, context) {
+    const name = getInputOrData(this.data, inputs, "name", "string");
+    const version = getInputOrData(this.data, inputs, "version", "string");
+    const transportType = getInputOrData(this.data, inputs, "transportType", "string");
+    let tools = [];
+    let prompts = [];
+    try {
+      if (!context.mcpProvider) {
+        throw new Error("MCP Provider not found");
+      }
+      if (transportType === "http") {
+        const serverUrl = getInputOrData(this.data, inputs, "serverUrl", "string");
+        if (!serverUrl || serverUrl === "") {
+          throw new MCPError("SERVER_NOT_FOUND" /* SERVER_NOT_FOUND */, "No server URL was provided");
+        }
+        if (!serverUrl.includes("/mcp")) {
+          throw new MCPError(
+            "SERVER_COMMUNICATION_FAILED" /* SERVER_COMMUNICATION_FAILED */,
+            "Include /mcp in your server URL. For example: http://localhost:8080/mcp"
+          );
+        }
+        tools = await context.mcpProvider.getHTTPTools({ name, version }, serverUrl);
+        prompts = await context.mcpProvider.getHTTPrompts({ name, version }, serverUrl);
+      } else if (transportType === "stdio") {
+        const serverId = this.data.serverId ?? "";
+        const mcpConfig = await loadMCPConfiguration(context);
+        if (!mcpConfig.mcpServers[serverId]) {
+          throw new MCPError("SERVER_NOT_FOUND" /* SERVER_NOT_FOUND */, `Server ${serverId} not found in MCP config`);
+        }
+        const serverConfig = {
+          config: mcpConfig.mcpServers[serverId],
+          serverId
+        };
+        tools = await context.mcpProvider.getStdioTools({ name, version }, serverConfig);
+        prompts = await context.mcpProvider.getStdioPrompts({ name, version }, serverConfig);
+      }
+      const output = {};
+      const gptFunctions = tools.map((tool) => ({
+        name: tool.name,
+        description: tool.description ?? "",
+        parameters: tool.inputSchema,
+        strict: false
+      }));
+      if (this.data.useToolsOutput) {
+        output["tools"] = {
+          type: "gpt-function[]",
+          value: gptFunctions
+        };
+      }
+      if (this.data.usePromptsOutput) {
+        output["prompts"] = {
+          type: "object[]",
+          value: prompts.map((prompt) => ({
+            name: prompt.name,
+            description: prompt.description,
+            arguments: prompt.arugments
+          }))
+        };
+      }
+      return output;
+    } catch (err) {
+      if (context.executor === "browser") {
+        throw new Error("Failed to create Client without Node Executor");
+      }
+      throw err;
+    }
+  }
+};
+var mcpDiscoveryNode = nodeDefinition(MCPDiscoveryNodeImpl, "MCP Discovery");
+
+// src/model/nodes/MCPToolCallNode.ts
+var import_non_secure75 = require("nanoid/non-secure");
+var import_ts_dedent69 = require("ts-dedent");
+var MCPToolCallNodeImpl = class extends NodeImpl {
+  static create() {
+    const chartNode = {
+      type: "mcpToolCall",
+      title: "MCP Tool Call",
+      id: (0, import_non_secure75.nanoid)(),
+      visualData: {
+        x: 0,
+        y: 0,
+        width: 250
+      },
+      data: {
+        name: "mcp-tool-call-client",
+        version: "1.0.0",
+        transportType: "stdio",
+        serverUrl: "http://localhost:8080/mcp",
+        serverId: "",
+        toolName: "",
+        toolArguments: import_ts_dedent69.dedent`
+        {
+          "key": "value"
+        }`,
+        toolCallId: "",
+        useNameInput: false,
+        useVersionInput: false,
+        useToolNameInput: true,
+        useToolArgumentsInput: true,
+        useToolCallIdInput: true
+      }
+    };
+    return chartNode;
+  }
+  getInputDefinitions() {
+    const inputs = getMCPBaseInputs(this.data);
+    if (this.data.useToolNameInput) {
+      inputs.push({
+        dataType: "string",
+        id: "toolName",
+        title: "Tool Name"
+      });
+    }
+    if (this.data.useToolArgumentsInput) {
+      inputs.push({
+        dataType: "object",
+        id: "toolArguments",
+        title: "Tool Arguments"
+      });
+    }
+    if (this.data.useToolCallIdInput) {
+      inputs.push({
+        dataType: "object",
+        id: "toolCallId",
+        title: "Tool ID"
+      });
+    }
+    return inputs;
+  }
+  getOutputDefinitions() {
+    const outputDefinitions = [];
+    outputDefinitions.push({
+      id: "response",
+      title: "Response",
+      dataType: "object",
+      description: "Response from the Tool Call"
+    });
+    outputDefinitions.push({
+      id: "toolCallId",
+      title: "Tool ID",
+      dataType: "string",
+      description: "ID associated with the Tool Call"
+    });
+    return outputDefinitions;
+  }
+  async getEditors(context) {
+    const editors = [
+      {
+        type: "string",
+        label: "Name",
+        dataKey: "name",
+        useInputToggleDataKey: "useNameInput",
+        helperMessage: "The name for the MCP Client"
+      },
+      {
+        type: "string",
+        label: "Version",
+        dataKey: "version",
+        useInputToggleDataKey: "useVersionInput",
+        helperMessage: "A version for the MCP Client"
+      },
+      {
+        type: "dropdown",
+        label: "Transport Type",
+        dataKey: "transportType",
+        options: [
+          { label: "HTTP", value: "http" },
+          { label: "STDIO", value: "stdio" }
+        ]
+      },
+      {
+        type: "string",
+        label: "Tool Name",
+        dataKey: "toolName",
+        useInputToggleDataKey: "useToolNameInput",
+        helperMessage: "The name for the MCP Tool Call"
+      },
+      {
+        type: "code",
+        label: "Tool Arguments",
+        dataKey: "toolArguments",
+        language: "json",
+        useInputToggleDataKey: "useToolArgumentsInput"
+      },
+      {
+        type: "string",
+        label: "Tool ID",
+        dataKey: "toolCallId",
+        useInputToggleDataKey: "useToolCallIdInput",
+        helperMessage: "The name for the MCP Tool Call"
+      }
+    ];
+    if (this.data.transportType === "http") {
+      editors.push({
+        type: "string",
+        label: "Server URL",
+        dataKey: "serverUrl",
+        useInputToggleDataKey: "useServerUrlInput",
+        helperMessage: "The endpoint URL for the MCP server to connect"
+      });
+    } else if (this.data.transportType === "stdio") {
+      const serverOptions = await getServerOptions(context);
+      editors.push({
+        type: "dropdown",
+        label: "Server ID",
+        dataKey: "serverId",
+        helperMessage: getServerHelperMessage(context, serverOptions.length),
+        options: serverOptions
+      });
+    }
+    return editors;
+  }
+  getBody(context) {
+    let base;
+    if (this.data.transportType === "http") {
+      base = this.data.useServerUrlInput ? "(Using Server URL Input)" : this.data.serverUrl;
+    } else {
+      base = `Server ID: ${this.data.serverId || "(None)"}`;
+    }
+    const namePart = `Name: ${this.data.name}`;
+    const versionPart = `Version: ${this.data.version}`;
+    const parts = [namePart, versionPart, base];
+    if (context.executor !== "nodejs") {
+      parts.push("(Requires Node Executor)");
+    }
+    return parts.join("\n");
+  }
+  static getUIData() {
+    return {
+      infoBoxBody: import_ts_dedent69.dedent`
+        Connects to an MCP (Model Context Protocol) server and gets a tool call response.
+      `,
+      infoBoxTitle: "MCP Tool Call Node",
+      contextMenuTitle: "MCP Tool Call",
+      group: ["MCP"]
+    };
+  }
+  async process(inputs, context) {
+    const name = getInputOrData(this.data, inputs, "name", "string");
+    const version = getInputOrData(this.data, inputs, "version", "string");
+    const toolName = getInputOrData(this.data, inputs, "toolName", "string");
+    const toolCallId = getInputOrData(this.data, inputs, "toolCallId", "string");
+    let toolArguments;
+    if (this.data.useToolArgumentsInput) {
+      toolArguments = getInputOrData(this.data, inputs, "toolArguments", "object");
+      if (toolArguments == null) {
+        throw new MCPError("INVALID_SCHEMA" /* INVALID_SCHEMA */, "Cannot parse tool argument with input toggle on");
+      }
+    } else {
+      const inputMap = keys(inputs).filter((key) => key.startsWith("input")).reduce(
+        (acc, key) => {
+          const stringValue = coerceTypeOptional(inputs[key], "string") ?? "";
+          const interpolationKey = key.slice("input-".length);
+          acc[interpolationKey] = stringValue;
+          return acc;
+        },
+        {}
+      );
+      const interpolated = interpolate(this.data.toolArguments ?? "", inputMap);
+      toolArguments = JSON.parse(interpolated);
+    }
+    const toolCall = {
+      name: toolName,
+      arguments: toolArguments
+    };
+    const transportType = getInputOrData(this.data, inputs, "transportType", "string");
+    let toolResponse = void 0;
+    try {
+      if (!context.mcpProvider) {
+        throw new Error("MCP Provider not found");
+      }
+      if (transportType === "http") {
+        const serverUrl = getInputOrData(this.data, inputs, "serverUrl", "string");
+        if (!serverUrl || serverUrl === "") {
+          throw new MCPError("SERVER_NOT_FOUND" /* SERVER_NOT_FOUND */, "No server URL was provided");
+        }
+        if (!serverUrl.includes("/mcp")) {
+          throw new MCPError(
+            "SERVER_COMMUNICATION_FAILED" /* SERVER_COMMUNICATION_FAILED */,
+            "Include /mcp in your server URL. For example: http://localhost:8080/mcp"
+          );
+        }
+        toolResponse = await context.mcpProvider.httpToolCall({ name, version }, serverUrl, toolCall);
+      } else if (transportType === "stdio") {
+        const serverId = this.data.serverId ?? "";
+        const mcpConfig = await loadMCPConfiguration(context);
+        if (!mcpConfig.mcpServers[serverId]) {
+          throw new MCPError("SERVER_NOT_FOUND" /* SERVER_NOT_FOUND */, `Server ${serverId} not found in MCP config`);
+        }
+        const serverConfig = {
+          config: mcpConfig.mcpServers[serverId],
+          serverId
+        };
+        toolResponse = await context.mcpProvider.stdioToolCall({ name, version }, serverConfig, toolCall);
+      }
+      const output = {};
+      output["response"] = {
+        type: "object[]",
+        value: toolResponse == null ? void 0 : toolResponse.content
+      };
+      output["toolCallId"] = {
+        type: "string",
+        value: toolCallId
+      };
+      return output;
+    } catch (err) {
+      const { message } = getError(err);
+      if (context.executor === "browser") {
+        throw new Error("Failed to create Client without a node executor");
+      }
+      console.log(message);
+      throw err;
+    }
+  }
+};
+var mcpToolCallNode = nodeDefinition(MCPToolCallNodeImpl, "MCP Tool Call");
+
+// src/model/nodes/MCPGetPromptNode.ts
+var import_non_secure76 = require("nanoid/non-secure");
+var MCPGetPromptNodeImpl = class extends NodeImpl {
+  static create() {
+    const chartNode = {
+      type: "mcpGetPrompt",
+      title: "MCP Get Prompt",
+      id: (0, import_non_secure76.nanoid)(),
+      visualData: {
+        x: 0,
+        y: 0,
+        width: 250
+      },
+      data: {
+        name: "mcp-get-prompt-client",
+        version: "1.0.0",
+        transportType: "stdio",
+        serverUrl: "http://localhost:8080/mcp",
+        serverId: "",
+        promptName: "",
+        promptArguments: import_ts_dedent.dedent`
+        {
+          "key": "value"
+        }`,
+        useNameInput: false,
+        useVersionInput: false,
+        usePromptNameInput: false,
+        usePromptArgumentsInput: false
+      }
+    };
+    return chartNode;
+  }
+  getInputDefinitions() {
+    const inputs = getMCPBaseInputs(this.data);
+    if (this.data.usePromptNameInput) {
+      inputs.push({
+        dataType: "string",
+        id: "promptName",
+        title: "Prompt Name"
+      });
+    }
+    if (this.data.usePromptArgumentsInput) {
+      inputs.push({
+        dataType: "object",
+        id: "promptArguments",
+        title: "Prompt Arguments"
+      });
+    }
+    return inputs;
+  }
+  getOutputDefinitions() {
+    const outputDefinitions = [];
+    outputDefinitions.push({
+      id: "prompt",
+      title: "Prompt",
+      dataType: "object",
+      description: "Prompt response result"
+    });
+    return outputDefinitions;
+  }
+  async getEditors(context) {
+    const editors = [
+      {
+        type: "string",
+        label: "Name",
+        dataKey: "name",
+        useInputToggleDataKey: "useNameInput",
+        helperMessage: "The name for the MCP Client"
+      },
+      {
+        type: "string",
+        label: "Version",
+        dataKey: "version",
+        useInputToggleDataKey: "useVersionInput",
+        helperMessage: "A version for the MCP Client"
+      },
+      {
+        type: "dropdown",
+        label: "Transport Type",
+        dataKey: "transportType",
+        options: [
+          { label: "HTTP", value: "http" },
+          { label: "STDIO", value: "stdio" }
+        ]
+      },
+      {
+        type: "string",
+        label: "Prompt Name",
+        dataKey: "promptName",
+        useInputToggleDataKey: "usePromptNameInput",
+        helperMessage: "The name for the MCP prompt"
+      },
+      {
+        type: "code",
+        label: "Prompt Arguments",
+        dataKey: "promptArguments",
+        useInputToggleDataKey: "usePromptArgumentsInput",
+        language: "json",
+        helperMessage: "Arguments to provide the prompt"
+      }
+    ];
+    if (this.data.transportType === "http") {
+      editors.push({
+        type: "string",
+        label: "Server URL",
+        dataKey: "serverUrl",
+        useInputToggleDataKey: "useServerUrlInput",
+        helperMessage: "The endpoint URL for the MCP server to connect"
+      });
+    } else if (this.data.transportType === "stdio") {
+      const serverOptions = await getServerOptions(context);
+      editors.push({
+        type: "dropdown",
+        label: "Server ID",
+        dataKey: "serverId",
+        helperMessage: getServerHelperMessage(context, serverOptions.length),
+        options: serverOptions
+      });
+    }
+    return editors;
+  }
+  getBody(context) {
+    let base;
+    if (this.data.transportType === "http") {
+      base = this.data.useServerUrlInput ? "(Using Server URL Input)" : this.data.serverUrl;
+    } else {
+      base = `Server ID: ${this.data.serverId || "(None)"}`;
+    }
+    const namePart = `Name: ${this.data.name}`;
+    const versionPart = `Version: ${this.data.version}`;
+    const parts = [namePart, versionPart, base];
+    if (context.executor !== "nodejs") {
+      parts.push("(Requires Node Executor)");
+    }
+    return parts.join("\n");
+  }
+  static getUIData() {
+    return {
+      infoBoxBody: import_ts_dedent.dedent`
+        Connects to an MCP (Model Context Protocol) server and gets a prompt response.
+      `,
+      infoBoxTitle: "MCP Get Prompt Node",
+      contextMenuTitle: "MCP Get Prompt",
+      group: ["MCP"]
+    };
+  }
+  async process(inputs, context) {
+    const name = getInputOrData(this.data, inputs, "name", "string");
+    const version = getInputOrData(this.data, inputs, "version", "string");
+    const promptName = getInputOrData(this.data, inputs, "promptName", "string");
+    let promptArguments;
+    if (this.data.usePromptArgumentsInput) {
+      promptArguments = getInputOrData(this.data, inputs, "promptArguments", "object");
+      if (promptArguments == null) {
+        throw new MCPError("INVALID_SCHEMA" /* INVALID_SCHEMA */, "Cannot parse tool argument with input toggle on");
+      }
+    } else {
+      const inputMap = keys(inputs).filter((key) => key.startsWith("input")).reduce(
+        (acc, key) => {
+          const stringValue = coerceTypeOptional(inputs[key], "string") ?? "";
+          const interpolationKey = key.slice("input-".length);
+          acc[interpolationKey] = stringValue;
+          return acc;
+        },
+        {}
+      );
+      const interpolated = interpolate(this.data.promptArguments ?? "", inputMap);
+      promptArguments = JSON.parse(interpolated);
+    }
+    const getPromptRequest = {
+      name: promptName,
+      arguments: promptArguments
+    };
+    const transportType = getInputOrData(this.data, inputs, "transportType", "string");
+    let getPromptResponse = void 0;
+    try {
+      if (!context.mcpProvider) {
+        throw new Error("MCP Provider not found");
+      }
+      if (transportType === "http") {
+        const serverUrl = getInputOrData(this.data, inputs, "serverUrl", "string");
+        if (!serverUrl || serverUrl === "") {
+          throw new MCPError("SERVER_NOT_FOUND" /* SERVER_NOT_FOUND */, "No server URL was provided");
+        }
+        if (!serverUrl.includes("/mcp")) {
+          throw new MCPError(
+            "SERVER_COMMUNICATION_FAILED" /* SERVER_COMMUNICATION_FAILED */,
+            "Include /mcp in your server URL. For example: http://localhost:8080/mcp"
+          );
+        }
+        getPromptResponse = await context.mcpProvider.getHTTPrompt({ name, version }, serverUrl, getPromptRequest);
+      } else if (transportType === "stdio") {
+        const serverId = this.data.serverId ?? "";
+        const mcpConfig = await loadMCPConfiguration(context);
+        if (!mcpConfig.mcpServers[serverId]) {
+          throw new MCPError("SERVER_NOT_FOUND" /* SERVER_NOT_FOUND */, `Server ${serverId} not found in MCP config`);
+        }
+        const serverConfig = {
+          config: mcpConfig.mcpServers[serverId],
+          serverId
+        };
+        getPromptResponse = await context.mcpProvider.getStdioPrompt({ name, version }, serverConfig, getPromptRequest);
+      }
+      const output = {};
+      output["response"] = {
+        type: "object",
+        value: getPromptResponse
+      };
+      return output;
+    } catch (err) {
+      const { message } = getError(err);
+      if (context.executor === "browser") {
+        throw new Error("Failed to create Client without a node executor");
+      }
+      console.log(message);
+      throw err;
+    }
+  }
+};
+var mcpGetPromptNode = nodeDefinition(MCPGetPromptNodeImpl, "MCP Get Prompt");
+
+// src/model/nodes/ReferencedGraphAliasNode.ts
+var import_non_secure77 = require("nanoid/non-secure");
+var import_ts_dedent70 = require("ts-dedent");
+var ReferencedGraphAliasNodeImpl = class extends NodeImpl {
+  static create() {
+    const chartNode = {
+      type: "referencedGraphAlias",
+      title: "",
+      // Always set initially by the editor
+      id: (0, import_non_secure77.nanoid)(),
+      visualData: {
+        x: 0,
+        y: 0,
+        width: 300
+      },
+      data: {
+        projectId: void 0,
+        // Always set initially by the editor
+        graphId: void 0,
+        // Always set initially by the editor
+        useErrorOutput: false
+      }
+    };
+    return chartNode;
+  }
+  getInputDefinitions(_connections, _nodes, _project, referencedProjects) {
+    const referencedProject = referencedProjects[this.data.projectId];
+    if (!referencedProject) {
+      return [];
+    }
+    const graph = referencedProject.graphs[this.data.graphId];
+    if (!graph) {
+      return [];
+    }
+    const inputNodes = graph.nodes.filter((node) => node.type === "graphInput");
+    const inputIds = [...new Set(inputNodes.map((node) => node.data.id))].sort();
+    return inputIds.map(
+      (id) => ({
+        id,
+        title: id,
+        dataType: inputNodes.find((node) => node.data.id === id).data.dataType
+      })
+    );
+  }
+  getGraphOutputs(referencedProject) {
+    const graph = referencedProject.graphs[this.data.graphId];
+    if (!graph) {
+      return [];
+    }
+    const outputNodes = graph.nodes.filter((node) => node.type === "graphOutput");
+    const outputIds = [...new Set(outputNodes.map((node) => node.data.id))].sort();
+    const outputs = outputIds.map(
+      (id) => ({
+        id,
+        title: id,
+        dataType: outputNodes.find((node) => node.data.id === id).data.dataType
+      })
+    );
+    return outputs;
+  }
+  getOutputDefinitions(_connections, _nodes, _project, referencedProjects) {
+    const outputs = [];
+    const referencedProject = referencedProjects[this.data.projectId];
+    if (!referencedProject) {
+      return outputs;
+    }
+    outputs.push(...this.getGraphOutputs(referencedProject));
+    if (this.data.useErrorOutput) {
+      outputs.push({
+        id: "error",
+        title: "Error",
+        dataType: "string"
+      });
+    }
+    return outputs;
+  }
+  getEditors(context) {
+    const definitions = [
+      {
+        type: "toggle",
+        label: "Use Error Output",
+        dataKey: "useErrorOutput"
+      },
+      {
+        type: "toggle",
+        label: "Output Cost & Duration",
+        dataKey: "outputCostDuration"
+      }
+    ];
+    const referencedProject = context.referencedProjects[this.data.projectId];
+    if (referencedProject) {
+      const graph = referencedProject.graphs[this.data.graphId];
+      if (graph) {
+        const inputNodes = graph.nodes.filter((node) => node.type === "graphInput");
+        const inputIds = [...new Set(inputNodes.map((node) => node.data.id))].sort();
+        for (const inputId of inputIds) {
+          const inputNode = inputNodes.find((node) => node.data.id === inputId);
+          definitions.push({
+            type: "dynamic",
+            dataKey: "inputData",
+            dynamicDataKey: inputNode.data.id,
+            dataType: inputNode.data.dataType,
+            label: inputNode.data.id,
+            editor: inputNode.data.editor ?? "auto"
+          });
+        }
+      }
     }
     return definitions;
   }
@@ -13723,7 +14665,7 @@ var ReferencedGraphAliasNodeImpl = class extends NodeImpl {
   }
   static getUIData() {
     return {
-      infoBoxBody: import_ts_dedent69.dedent`
+      infoBoxBody: import_ts_dedent70.dedent`
         References a graph from another project. Inputs and outputs are defined by Graph Input and Graph Output nodes within the referenced graph.
       `,
       infoBoxTitle: "Referenced Graph Alias Node",
@@ -13817,7 +14759,7 @@ var referencedGraphAliasNode = nodeDefinition(ReferencedGraphAliasNodeImpl, "Ref
 
 // src/model/Nodes.ts
 var registerBuiltInNodes = (registry2) => {
-  return registry2.register(toYamlNode).register(userInputNode).register(textNode).register(chatNode).register(promptNode).register(extractRegexNode).register(codeNode).register(matchNode).register(ifNode).register(readDirectoryNode).register(readFileNode).register(writeFileNode).register(ifElseNode).register(chunkNode).register(graphInputNode).register(graphOutputNode).register(subGraphNode).register(arrayNode).register(extractJsonNode).register(assemblePromptNode).register(loopControllerNode).register(trimChatMessagesNode).register(extractYamlNode).register(externalCallNode).register(extractObjectPathNode).register(raiseEventNode).register(contextNode).register(coalesceNode).register(passthroughNode).register(popNode).register(setGlobalNode).register(getGlobalNode).register(waitForEventNode).register(gptFunctionNode).register(getEmbeddingNode).register(vectorStoreNode).register(vectorNearestNeighborsNode).register(hashNode).register(abortGraphNode).register(raceInputsNode).register(toJsonNode).register(joinNode).register(filterNode).register(objectNode).register(booleanNode).register(compareNode).register(evaluateNode).register(numberNode).register(randomNumberNode).register(shuffleNode).register(commentNode).register(imageToMDNode).register(imageNode).register(audioNode).register(httpCallNode).register(delayNode).register(appendToDatasetNode).register(createDatasetNode).register(loadDatasetNode).register(getAllDatasetsNode).register(splitNode).register(datasetNearestNeighborsNode).register(getDatasetRowNode).register(sliceNode).register(extractMarkdownCodeBlocksNode).register(assembleMessageNode).register(urlReferenceNode).register(destructureNode).register(replaceDatasetNode).register(listGraphsNode).register(graphReferenceNode).register(callGraphNode).register(delegateFunctionCallNode).register(playAudioNode).register(documentNode).register(chatLoopNode).register(readAllFilesNode).register(toMarkdownTableNode).register(cronNode).register(toTreeNode).register(loopUntilNode).register(referencedGraphAliasNode);
+  return registry2.register(toYamlNode).register(userInputNode).register(textNode).register(chatNode).register(promptNode).register(extractRegexNode).register(codeNode).register(matchNode).register(ifNode).register(readDirectoryNode).register(readFileNode).register(writeFileNode).register(ifElseNode).register(chunkNode).register(graphInputNode).register(graphOutputNode).register(subGraphNode).register(arrayNode).register(extractJsonNode).register(assemblePromptNode).register(loopControllerNode).register(trimChatMessagesNode).register(extractYamlNode).register(externalCallNode).register(extractObjectPathNode).register(raiseEventNode).register(contextNode).register(coalesceNode).register(passthroughNode).register(popNode).register(setGlobalNode).register(getGlobalNode).register(waitForEventNode).register(gptFunctionNode).register(getEmbeddingNode).register(vectorStoreNode).register(vectorNearestNeighborsNode).register(hashNode).register(abortGraphNode).register(raceInputsNode).register(toJsonNode).register(joinNode).register(filterNode).register(objectNode).register(booleanNode).register(compareNode).register(evaluateNode).register(numberNode).register(randomNumberNode).register(shuffleNode).register(commentNode).register(imageToMDNode).register(imageNode).register(audioNode).register(httpCallNode).register(delayNode).register(appendToDatasetNode).register(createDatasetNode).register(loadDatasetNode).register(getAllDatasetsNode).register(splitNode).register(datasetNearestNeighborsNode).register(getDatasetRowNode).register(sliceNode).register(extractMarkdownCodeBlocksNode).register(assembleMessageNode).register(urlReferenceNode).register(destructureNode).register(replaceDatasetNode).register(listGraphsNode).register(graphReferenceNode).register(callGraphNode).register(delegateFunctionCallNode).register(playAudioNode).register(documentNode).register(chatLoopNode).register(readAllFilesNode).register(toMarkdownTableNode).register(cronNode).register(toTreeNode).register(loopUntilNode).register(mcpDiscoveryNode).register(mcpToolCallNode).register(mcpGetPromptNode).register(referencedGraphAliasNode);
 };
 var globalRivetNodeRegistry = registerBuiltInNodes(new NodeRegistration());
 function resetGlobalRivetNodeRegistry() {
@@ -13891,7 +14833,7 @@ ${parameterName}?: ${value == null ? void 0 : value.type}`).join("\n")}
 
 // src/integrations/CodeRunner.ts
 var IsomorphicCodeRunner = class {
-  async runCode(code, inputs, options2) {
+  async runCode(code, inputs, options2, graphInputs, contextValues) {
     const argNames = ["inputs"];
     const args = [inputs];
     if (options2.includeRequire) {
@@ -13912,6 +14854,14 @@ var IsomorphicCodeRunner = class {
       argNames.push("Rivet");
       args.push(exports_exports);
     }
+    if (graphInputs) {
+      argNames.push("graphInputs");
+      args.push(graphInputs);
+    }
+    if (contextValues) {
+      argNames.push("context");
+      args.push(contextValues);
+    }
     argNames.push(code);
     const AsyncFunction = async function() {
     }.constructor;
@@ -13921,7 +14871,7 @@ var IsomorphicCodeRunner = class {
   }
 };
 var NotAllowedCodeRunner = class {
-  async runCode(_code, _inputs, _options) {
+  async runCode(_code, _inputs, _options, _graphInputs, _contextValues) {
     throw new Error("Dynamic code execution is disabled.");
   }
 };
@@ -13946,7 +14896,7 @@ var GraphProcessor = class _GraphProcessor {
   #isPaused = false;
   #parent;
   #registry;
-  id = (0, import_non_secure76.nanoid)();
+  id = (0, import_non_secure78.nanoid)();
   #includeTrace = true;
   executor;
   /** If set, specifies the node(s) that the graph will run TO, instead of the nodes without any dependents. */
@@ -13991,6 +14941,7 @@ var GraphProcessor = class _GraphProcessor {
   // @ts-expect-error
   #nodesNotInCycle = void 0;
   #nodeAbortControllers = /* @__PURE__ */ new Map();
+  #graphInputNodeValues = {};
   /** User input nodes that are pending user input. */
   #pendingUserInputs = void 0;
   get isRunning() {
@@ -14371,6 +15322,7 @@ var GraphProcessor = class _GraphProcessor {
     this.#abortSuccessfully = false;
     this.#nodeAbortControllers = /* @__PURE__ */ new Map();
     this.#loadedProjects = {};
+    this.#graphInputNodeValues = {};
   }
   /** Main function for running a graph. Runs a graph and returns the outputs from the output nodes of the graph. */
   async processGraph(context, inputs = {}, contextValues = {}) {
@@ -14611,7 +15563,7 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
       return;
     }
     const inputValues = this.#getInputValuesForNode(node);
-    if (this.#excludedDueToControlFlow(node, inputValues, (0, import_non_secure76.nanoid)(), "loop-not-broken")) {
+    if (this.#excludedDueToControlFlow(node, inputValues, (0, import_non_secure78.nanoid)(), "loop-not-broken")) {
       this.#emitTraceEvent(`Node ${node.title} is excluded due to control flow`);
       return;
     }
@@ -14745,7 +15697,7 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
     return nodeData;
   }
   async #processNode(node) {
-    const processId = (0, import_non_secure76.nanoid)();
+    const processId = (0, import_non_secure78.nanoid)();
     if (this.#abortController.signal.aborted) {
       this.#nodeErrored(node, new Error("Processing aborted"), processId);
       return processId;
@@ -14773,7 +15725,7 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
       return;
     }
     const splittingAmount = Math.min(
-      (0, import_lodash_es15.max)(values(inputValues).map((value) => Array.isArray(value == null ? void 0 : value.value) ? value == null ? void 0 : value.value.length : 1)) ?? 1,
+      (0, import_lodash_es16.max)(values(inputValues).map((value) => Array.isArray(value == null ? void 0 : value.value) ? value == null ? void 0 : value.value.length : 1)) ?? 1,
       node.splitRunMax ?? 10
     );
     this.#emitter.emit("nodeStart", { node, inputs: inputValues, processId });
@@ -14816,7 +15768,7 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
         }
       } else {
         results = await Promise.all(
-          (0, import_lodash_es15.range)(0, splittingAmount).map(async (i) => {
+          (0, import_lodash_es16.range)(0, splittingAmount).map(async (i) => {
             var _a2, _b2, _c2;
             const inputs = fromEntries(
               entries(inputValues).map(([port, value]) => [
@@ -14866,15 +15818,15 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
       }, {});
       this.#nodeResults.set(node.id, aggregateResults);
       this.#visitedNodes.add(node.id);
-      this.#totalRequestTokens += (0, import_lodash_es15.sum)(results.map((r) => {
+      this.#totalRequestTokens += (0, import_lodash_es16.sum)(results.map((r) => {
         var _a2;
         return coerceTypeOptional((_a2 = r.output) == null ? void 0 : _a2["requestTokens"], "number") ?? 0;
       }));
-      this.#totalResponseTokens += (0, import_lodash_es15.sum)(results.map((r) => {
+      this.#totalResponseTokens += (0, import_lodash_es16.sum)(results.map((r) => {
         var _a2;
         return coerceTypeOptional((_a2 = r.output) == null ? void 0 : _a2["responseTokens"], "number") ?? 0;
       }));
-      this.#totalCost += (0, import_lodash_es15.sum)(results.map((r) => {
+      this.#totalCost += (0, import_lodash_es16.sum)(results.map((r) => {
         var _a2;
         return coerceTypeOptional((_a2 = r.output) == null ? void 0 : _a2["cost"], "number") ?? 0;
       }));
@@ -15094,7 +16046,8 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
           });
         });
         return results2;
-      }
+      },
+      graphInputNodeValues: this.#graphInputNodeValues
     };
     await this.#waitUntilUnpaused();
     const results = await instance.process(inputValues, context);
@@ -15217,7 +16170,7 @@ ${erroredNodes.map(([nodeId]) => `- ${this.#nodesById[nodeId].title} (${nodeId})
       const connectionDefinition = outputDefinitions.find((def) => def.id === connection.outputId);
       return connectionDefinition != null;
     });
-    const outputNodes = (0, import_lodash_es15.uniqBy)(
+    const outputNodes = (0, import_lodash_es16.uniqBy)(
       outputConnections.map((conn) => this.#nodesById[conn.inputNodeId]).filter(isNotNull),
       (x) => x.id
     );
@@ -15371,9 +16324,9 @@ var OpenAIEmbeddingGenerator = class {
 registerIntegration("embeddingGenerator", "openai", (context) => new OpenAIEmbeddingGenerator(context.settings));
 
 // src/recording/ExecutionRecorder.ts
-var import_non_secure77 = require("nanoid/non-secure");
+var import_non_secure79 = require("nanoid/non-secure");
 var import_emittery3 = __toESM(require("emittery-0-13"), 1);
-var import_lodash_es16 = require("lodash");
+var import_lodash_es17 = require("lodash");
 var toRecordedEventMap = {
   graphStart: ({ graph, inputs }) => ({ graphId: graph.metadata.id, inputs }),
   graphFinish: ({ graph, outputs }) => ({ graphId: graph.metadata.id, outputs }),
@@ -15470,16 +16423,16 @@ function toRecordedEvent(event, data) {
 function mapValuesDeep(obj, fn) {
   if (Array.isArray(obj)) {
     return obj.map((value) => {
-      if ((0, import_lodash_es16.isPlainObject)(value) || Array.isArray(value)) {
+      if ((0, import_lodash_es17.isPlainObject)(value) || Array.isArray(value)) {
         return mapValuesDeep(value, fn);
       }
       return fn(value);
     });
   }
-  if ((0, import_lodash_es16.isPlainObject)(obj)) {
+  if ((0, import_lodash_es17.isPlainObject)(obj)) {
     return Object.fromEntries(
       Object.entries(obj).map(([key, value]) => {
-        if ((0, import_lodash_es16.isPlainObject)(value) || Array.isArray(value)) {
+        if ((0, import_lodash_es17.isPlainObject)(value) || Array.isArray(value)) {
           return [key, mapValuesDeep(value, fn)];
         }
         return [key, fn(value)];
@@ -15505,7 +16458,7 @@ var ExecutionRecorder = class _ExecutionRecorder {
   once = void 0;
   recordSocket(channel) {
     return new Promise((resolve) => {
-      this.recordingId = (0, import_non_secure77.nanoid)();
+      this.recordingId = (0, import_non_secure79.nanoid)();
       const listener = (event) => {
         const { message, data } = JSON.parse(event.data);
         if (this.#includePartialOutputs === false && message === "partialOutput") {
@@ -15527,7 +16480,7 @@ var ExecutionRecorder = class _ExecutionRecorder {
     });
   }
   record(processor) {
-    this.recordingId = (0, import_non_secure77.nanoid)();
+    this.recordingId = (0, import_non_secure79.nanoid)();
     processor.onAny((event, data) => {
       if (this.#includePartialOutputs === false && event === "partialOutput") {
         return;
@@ -15589,7 +16542,7 @@ var ExecutionRecorder = class _ExecutionRecorder {
         const asString = uint8ArrayToBase64Sync(val);
         const existingAsset = Object.entries(serialized.assets).find(([, asset]) => asset === asString);
         if (!existingAsset) {
-          const id = (0, import_non_secure77.nanoid)();
+          const id = (0, import_non_secure79.nanoid)();
           serialized.assets[id] = asString;
           return `$ASSET:${id}`;
         } else {
@@ -15604,8 +16557,8 @@ var ExecutionRecorder = class _ExecutionRecorder {
 };
 
 // src/plugins/aidon/nodes/ChatAidonNode.ts
-var import_lodash_es17 = require("lodash");
-var import_ts_dedent70 = require("ts-dedent");
+var import_lodash_es18 = require("lodash");
+var import_ts_dedent71 = require("ts-dedent");
 var registry = globalRivetNodeRegistry;
 var ChatAidonNodeImpl = class extends ChatNodeImpl {
   create() {
@@ -15742,7 +16695,7 @@ var ChatAidonNodeImpl = class extends ChatNodeImpl {
           message: JSON.stringify(data)
         });
       }
-      inputs = (0, import_lodash_es17.omit)(inputs, ["functions", "prompt"]);
+      inputs = (0, import_lodash_es18.omit)(inputs, ["functions", "prompt"]);
       inputs["prompt"] = messages;
       outputs = await super.process(inputs, context);
     }
@@ -15786,7 +16739,7 @@ var createPluginNodeImpl = (chatNode2) => {
     },
     getUIData() {
       return {
-        infoBoxBody: import_ts_dedent70.dedent`
+        infoBoxBody: import_ts_dedent71.dedent`
 					Makes a call to an Aidon chat model. The settings contains many options for tweaking the model's behavior.
 				`,
         infoBoxTitle: "Chat (Aidon) Node",
@@ -16001,6 +16954,22 @@ var anthropicModels = {
       completion: 15e-6
     },
     displayName: "Claude 3.7 Sonnet"
+  },
+  "claude-sonnet-4-20250514": {
+    maxTokens: 2e5,
+    cost: {
+      prompt: 3e-6,
+      completion: 375e-8
+    },
+    displayName: "Claude Sonnet 4"
+  },
+  "claude-opus-4-20250514": {
+    maxTokens: 2e5,
+    cost: {
+      prompt: 15e-6,
+      completion: 1875e-8
+    },
+    displayName: "Claude Opus 4"
   }
 };
 var anthropicModelOptions = Object.entries(anthropicModels).map(([id, { displayName }]) => ({
@@ -16011,6 +16980,7 @@ async function* streamChatCompletions2({
   apiEndpoint,
   apiKey,
   signal,
+  additionalHeaders,
   ...rest
 }) {
   const defaultSignal = new AbortController().signal;
@@ -16020,7 +16990,8 @@ async function* streamChatCompletions2({
       "Content-Type": "application/json",
       "x-api-key": apiKey,
       "anthropic-version": "2023-06-01",
-      "anthropic-dangerous-direct-browser-access": "true"
+      "anthropic-dangerous-direct-browser-access": "true",
+      ...additionalHeaders
     },
     body: JSON.stringify({
       ...rest,
@@ -16055,6 +17026,7 @@ async function* streamMessageApi({
   apiKey,
   signal,
   beta,
+  additionalHeaders,
   ...rest
 }) {
   const defaultSignal = new AbortController().signal;
@@ -16065,7 +17037,8 @@ async function* streamMessageApi({
       "x-api-key": apiKey,
       "anthropic-version": "2023-06-01",
       "anthropic-dangerous-direct-browser-access": "true",
-      ...beta ? { "anthropic-beta": beta } : {}
+      ...beta ? { "anthropic-beta": beta } : {},
+      ...additionalHeaders
     },
     body: JSON.stringify({
       ...rest,
@@ -16104,8 +17077,8 @@ var AnthropicError = class extends Error {
 };
 
 // src/plugins/anthropic/nodes/ChatAnthropicNode.ts
-var import_non_secure78 = require("nanoid/non-secure");
-var import_ts_dedent71 = require("ts-dedent");
+var import_non_secure80 = require("nanoid/non-secure");
+var import_ts_dedent72 = require("ts-dedent");
 var import_p_retry2 = __toESM(require("p-retry-4"), 1);
 var import_ts_pattern10 = require("ts-pattern");
 
@@ -16121,14 +17094,14 @@ var ChatAnthropicNodeImpl = {
     const chartNode = {
       type: "chatAnthropic",
       title: "Chat (Anthropic)",
-      id: (0, import_non_secure78.nanoid)(),
+      id: (0, import_non_secure80.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
         width: 275
       },
       data: {
-        model: "claude-3-7-sonnet-latest",
+        model: "claude-sonnet-4-20250514",
         useModelInput: false,
         temperature: 0.5,
         useTemperatureInput: false,
@@ -16221,6 +17194,14 @@ var ChatAnthropicNodeImpl = {
         coerced: false
       });
     }
+    if (data.useHeadersInput) {
+      inputs.push({
+        dataType: "object",
+        id: "headers",
+        title: "Headers",
+        description: "Additional headers to send to the API."
+      });
+    }
     return inputs;
   },
   getOutputDefinitions(data) {
@@ -16257,7 +17238,7 @@ var ChatAnthropicNodeImpl = {
   getBody(data) {
     var _a;
     const modelName = data.overrideModel ? data.overrideModel : ((_a = anthropicModels[data.model]) == null ? void 0 : _a.displayName) ?? "Unknown Model";
-    return import_ts_dedent71.dedent`
+    return import_ts_dedent72.dedent`
       ${modelName}
       ${data.useTopP ? `Top P: ${data.useTopPInput ? "(Using Input)" : data.top_p}` : `Temperature: ${data.useTemperatureInput ? "(Using Input)" : data.temperature}`}
       Max Tokens: ${data.maxTokens}
@@ -16367,6 +17348,14 @@ var ChatAnthropicNodeImpl = {
             dataKey: "overrideModel",
             useInputToggleDataKey: "useOverrideModelInput",
             helperMessage: "Overrides the AI model used for the chat node to this value."
+          },
+          {
+            type: "keyValuePair",
+            label: "Headers",
+            dataKey: "headers",
+            useInputToggleDataKey: "useHeadersInput",
+            keyPlaceholder: "Header",
+            helperMessage: "Additional headers to send to the API."
           }
         ]
       }
@@ -16374,7 +17363,7 @@ var ChatAnthropicNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent71.dedent`
+      infoBoxBody: import_ts_dedent72.dedent`
         Makes a call to an Anthropic chat model. The settings contains many options for tweaking the model's behavior.
       `,
       infoBoxTitle: "Chat (Anthropic) Node",
@@ -16433,6 +17422,18 @@ Assistant: ${content}`;
       addWarning(output, message);
       maxTokens = Math.floor((modelInfo.maxTokens - tokenCountEstimate) * 0.95);
     }
+    const headersFromData = (data.headers ?? []).reduce(
+      (acc, header) => {
+        acc[header.key] = header.value;
+        return acc;
+      },
+      {}
+    );
+    const additionalHeaders = data.useHeadersInput ? coerceTypeOptional(inputs["headers"], "object") ?? headersFromData : headersFromData;
+    const allAdditionalHeaders = cleanHeaders({
+      ...context.settings.chatNodeHeaders,
+      ...additionalHeaders
+    });
     try {
       return await (0, import_p_retry2.default)(
         async () => {
@@ -16455,7 +17456,7 @@ Assistant: ${content}`;
             messages,
             tools: tools ? tools.map((tool) => ({ name: tool.name, description: tool.description, input_schema: tool.parameters })) : void 0
           };
-          const useMessageApi = model.startsWith("claude-3");
+          const useMessageApi = model.startsWith("claude-3") || model.startsWith("claude-sonnet") || model.startsWith("claude-opus");
           const cacheKey = JSON.stringify(useMessageApi ? messageOptions : completionOptions);
           if (data.cache) {
             const cached = cache2.get(cacheKey);
@@ -16474,6 +17475,7 @@ Assistant: ${content}`;
               apiKey: apiKey ?? "",
               signal: context.signal,
               beta: "prompt-caching-2024-07-31",
+              additionalHeaders: allAdditionalHeaders,
               ...messageOptions
             });
             const responseParts = [];
@@ -16587,6 +17589,7 @@ Assistant: ${content}`;
               apiEndpoint,
               apiKey: apiKey ?? "",
               signal: context.signal,
+              additionalHeaders: allAdditionalHeaders,
               ...completionOptions
             });
             const responseParts = [];
@@ -16869,8 +17872,8 @@ var anthropicPlugin = {
 var anthropic_default = anthropicPlugin;
 
 // src/plugins/autoevals/AutoEvalsNode.ts
-var import_non_secure79 = require("nanoid/non-secure");
-var import_ts_dedent72 = require("ts-dedent");
+var import_non_secure81 = require("nanoid/non-secure");
+var import_ts_dedent73 = require("ts-dedent");
 var import_autoevals = require("autoevals");
 var import_ts_pattern11 = require("ts-pattern");
 var options = [
@@ -16889,7 +17892,7 @@ var AutoEvalsNodeImpl = {
     const chartNode = {
       type: "autoevals",
       title: "Autoevals",
-      id: (0, import_non_secure79.nanoid)(),
+      id: (0, import_non_secure81.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
@@ -17004,7 +18007,7 @@ var AutoEvalsNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent72.dedent`
+      infoBoxBody: import_ts_dedent73.dedent`
         Evaluates the validity of a response using the autoevals library.
       `,
       infoBoxTitle: "Autoevals Node",
@@ -17084,8 +18087,8 @@ var autoevalsPlugin = {
 var autoevals_default = autoevalsPlugin;
 
 // src/plugins/assemblyAi/LemurQaNode.ts
-var import_non_secure80 = require("nanoid/non-secure");
-var import_ts_dedent73 = require("ts-dedent");
+var import_non_secure82 = require("nanoid/non-secure");
+var import_ts_dedent74 = require("ts-dedent");
 
 // src/plugins/assemblyAi/lemurHelpers.ts
 var import_assemblyai = require("assemblyai");
@@ -17205,7 +18208,7 @@ var LemurQaNodeImpl = {
     const chartNode = {
       type: "assemblyAiLemurQa",
       title: "LeMUR Question & Answers",
-      id: (0, import_non_secure80.nanoid)(),
+      id: (0, import_non_secure82.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
@@ -17271,7 +18274,7 @@ var LemurQaNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent73.dedent`Use AssemblyAI LeMUR to ask questions about transcripts`,
+      infoBoxBody: import_ts_dedent74.dedent`Use AssemblyAI LeMUR to ask questions about transcripts`,
       infoBoxTitle: "Use AssemblyAI LeMUR Question & Answer",
       contextMenuTitle: "LeMUR Q&A",
       group: ["AI", "AssemblyAI"]
@@ -17343,14 +18346,14 @@ function applyQuestionEditors(data, question) {
 var lemurQaNode = pluginNodeDefinition(LemurQaNodeImpl, "LeMUR Q&A");
 
 // src/plugins/assemblyAi/TranscribeAudioNode.ts
-var import_non_secure81 = require("nanoid/non-secure");
-var import_ts_dedent74 = require("ts-dedent");
+var import_non_secure83 = require("nanoid/non-secure");
+var import_ts_dedent75 = require("ts-dedent");
 var TranscribeAudioNodeImpl = {
   create() {
     const chartNode = {
       type: "assemblyAiTranscribeAudio",
       title: "Transcribe Audio",
-      id: (0, import_non_secure81.nanoid)(),
+      id: (0, import_non_secure83.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
@@ -17405,7 +18408,7 @@ var TranscribeAudioNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent74.dedent`Use AssemblyAI to transcribe audio`,
+      infoBoxBody: import_ts_dedent75.dedent`Use AssemblyAI to transcribe audio`,
       infoBoxTitle: "Transcribe Audio Node",
       contextMenuTitle: "Transcribe Audio",
       group: ["AI", "AssemblyAI"]
@@ -17464,15 +18467,15 @@ function getAdditionalParameters(data) {
 }
 
 // src/plugins/assemblyAi/LemurSummaryNode.ts
-var import_non_secure82 = require("nanoid/non-secure");
-var import_ts_dedent75 = require("ts-dedent");
+var import_non_secure84 = require("nanoid/non-secure");
+var import_ts_dedent76 = require("ts-dedent");
 var import_assemblyai3 = require("assemblyai");
 var LemurSummaryNodeImpl = {
   create() {
     const chartNode = {
       type: "assemblyAiLemurSummary",
       title: "LeMUR Summary",
-      id: (0, import_non_secure82.nanoid)(),
+      id: (0, import_non_secure84.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
@@ -17518,7 +18521,7 @@ var LemurSummaryNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent75.dedent`Use AssemblyAI LeMUR Summary to summarize transcripts`,
+      infoBoxBody: import_ts_dedent76.dedent`Use AssemblyAI LeMUR Summary to summarize transcripts`,
       infoBoxTitle: "Use AssemblyAI LeMUR Summary",
       contextMenuTitle: "LeMUR Summary",
       group: ["AI", "AssemblyAI"]
@@ -17542,15 +18545,15 @@ var LemurSummaryNodeImpl = {
 var lemurSummaryNode = pluginNodeDefinition(LemurSummaryNodeImpl, "LeMUR Summary");
 
 // src/plugins/assemblyAi/LemurTaskNode.ts
-var import_non_secure83 = require("nanoid/non-secure");
-var import_ts_dedent76 = require("ts-dedent");
+var import_non_secure85 = require("nanoid/non-secure");
+var import_ts_dedent77 = require("ts-dedent");
 var import_assemblyai4 = require("assemblyai");
 var LemurTaskNodeImpl = {
   create() {
     const chartNode = {
       type: "assemblyAiLemurTask",
       title: "LeMUR Task",
-      id: (0, import_non_secure83.nanoid)(),
+      id: (0, import_non_secure85.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
@@ -17596,7 +18599,7 @@ var LemurTaskNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent76.dedent`Use AssemblyAI LeMUR Custom Task to ask anything.`,
+      infoBoxBody: import_ts_dedent77.dedent`Use AssemblyAI LeMUR Custom Task to ask anything.`,
       infoBoxTitle: "Use AssemblyAI LeMUR Custom Task",
       contextMenuTitle: "LeMUR Custom Task",
       group: ["AI", "AssemblyAI"]
@@ -17622,14 +18625,14 @@ var LemurTaskNodeImpl = {
 var lemurTaskNode = pluginNodeDefinition(LemurTaskNodeImpl, "LeMUR Task");
 
 // src/plugins/assemblyAi/LemurActionItemsNode.ts
-var import_non_secure84 = require("nanoid/non-secure");
-var import_ts_dedent77 = require("ts-dedent");
+var import_non_secure86 = require("nanoid/non-secure");
+var import_ts_dedent78 = require("ts-dedent");
 var LemurActionItemsNodeImpl = {
   create() {
     const chartNode = {
       type: "assemblyAiLemurActionItems",
       title: "LeMUR Action Items",
-      id: (0, import_non_secure84.nanoid)(),
+      id: (0, import_non_secure86.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
@@ -17675,7 +18678,7 @@ var LemurActionItemsNodeImpl = {
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent77.dedent`Use AssemblyAI LeMUR Action Items to extract action items`,
+      infoBoxBody: import_ts_dedent78.dedent`Use AssemblyAI LeMUR Action Items to extract action items`,
       infoBoxTitle: "Use AssemblyAI LeMUR Action Items",
       contextMenuTitle: "LeMUR Action Items",
       group: ["AI", "AssemblyAI"]
@@ -17727,12 +18730,12 @@ var assemblyAiPlugin = {
 var assemblyAi_default = assemblyAiPlugin;
 
 // src/plugins/huggingface/nodes/ChatHuggingFace.ts
-var import_non_secure85 = require("nanoid/non-secure");
+var import_non_secure87 = require("nanoid/non-secure");
 var import_inference = require("@huggingface/inference");
 var ChatHuggingFaceNodeImpl = {
   create() {
     return {
-      id: (0, import_non_secure85.nanoid)(),
+      id: (0, import_non_secure87.nanoid)(),
       type: "chatHuggingFace",
       data: {
         model: "",
@@ -17962,13 +18965,13 @@ var ChatHuggingFaceNodeImpl = {
 var chatHuggingFaceNode = pluginNodeDefinition(ChatHuggingFaceNodeImpl, "Chat (Hugging Face)");
 
 // src/plugins/huggingface/nodes/TextToImageHuggingFace.ts
-var import_non_secure86 = require("nanoid/non-secure");
+var import_non_secure88 = require("nanoid/non-secure");
 var import_inference2 = require("@huggingface/inference");
-var import_ts_dedent78 = require("ts-dedent");
+var import_ts_dedent79 = require("ts-dedent");
 var TextToImageHuggingFaceNodeImpl = {
   create() {
     return {
-      id: (0, import_non_secure86.nanoid)(),
+      id: (0, import_non_secure88.nanoid)(),
       type: "textToImageHuggingFace",
       data: {
         model: "",
@@ -18109,7 +19112,7 @@ var TextToImageHuggingFaceNodeImpl = {
     ];
   },
   getBody(data) {
-    return import_ts_dedent78.dedent`
+    return import_ts_dedent79.dedent`
       Model: ${data.useModelInput ? "(Using Input)" : data.model}
     `;
   },
@@ -18285,7 +19288,7 @@ var pinecone_default = pineconePlugin;
 
 // src/plugins/gentrace/plugin.ts
 var import_core = require("@gentrace/core");
-var import_lodash_es18 = require("lodash");
+var import_lodash_es19 = require("lodash");
 var apiKeyConfigSpec = {
   type: "secret",
   label: "Gentrace API Key",
@@ -20844,7 +21847,7 @@ ${additional.join("\n")}`;
 var runThreadNode = pluginNodeDefinition(RunThreadNodeImpl, "Run Thread");
 
 // src/plugins/openai/nodes/ThreadMessageNode.ts
-var import_lodash_es19 = require("lodash");
+var import_lodash_es20 = require("lodash");
 var ThreadMessageNodeImpl = {
   create() {
     return {
@@ -20969,7 +21972,7 @@ var ThreadMessageNodeImpl = {
     if (data.useMetadataInput && inputData["metadata"]) {
       metadata = coerceTypeOptional(inputData["metadata"], "object");
     }
-    const inputMap = (0, import_lodash_es19.mapValues)(inputData, (input) => coerceType(input, "string"));
+    const inputMap = (0, import_lodash_es20.mapValues)(inputData, (input) => coerceType(input, "string"));
     const interpolated = interpolate(text, inputMap);
     return {
       ["message"]: {
@@ -21017,7 +22020,7 @@ var openAIPlugin = {
 };
 
 // src/plugins/google/google.ts
-var import_generative_ai = require("@google/generative-ai");
+var import_genai = require("@google/genai");
 var googleModelsDeprecated = {
   "gemini-pro": {
     maxTokens: 32760,
@@ -21037,12 +22040,20 @@ var googleModelsDeprecated = {
   }
 };
 var generativeAiGoogleModels = {
-  "gemini-2.0-flash-001": {
+  "gemini-2.5-flash-preview-04-17": {
     maxTokens: 1048576,
     cost: {
       prompt: 0.15 / 1e3,
       completion: 0.6 / 1e3
     },
+    displayName: "Gemini 2.5 Flash Preview"
+  },
+  "gemini-2.0-flash-001": {
+    maxTokens: 1048576,
+    cost: {
+      prompt: 0.1 / 1e3,
+      completion: 0.4 / 1e3
+    },
     displayName: "Gemini 2.0 Flash"
   },
   "gemini-2.0-pro-exp-02-05": {
@@ -21142,36 +22153,42 @@ async function* streamGenerativeAi({
   topP,
   topK,
   signal,
-  tools
+  tools,
+  thinkingBudget,
+  additionalHeaders
 }) {
   var _a, _b, _c, _d, _e;
-  const { GoogleGenerativeAI } = await import("@google/generative-ai");
-  const genAi = new GoogleGenerativeAI(apiKey);
-  const genaiModel = genAi.getGenerativeModel({
+  const { GoogleGenAI } = await import("@google/genai");
+  const genAi = new GoogleGenAI({ apiKey });
+  const result = await genAi.models.generateContentStream({
     model,
-    systemInstruction: systemPrompt,
-    generationConfig: {
+    contents: prompt,
+    config: {
+      systemInstruction: systemPrompt,
       maxOutputTokens,
       temperature,
       topP,
-      topK
-    },
-    tools
+      topK,
+      tools,
+      abortSignal: signal,
+      thinkingConfig: {
+        thinkingBudget
+      },
+      httpOptions: {
+        headers: {
+          ...additionalHeaders
+        }
+      }
+    }
   });
-  const result = await genaiModel.generateContentStream(
-    {
-      contents: prompt
-    },
-    { signal }
-  );
-  for await (const chunk of result.stream) {
+  for await (const chunk of result) {
     const outChunk = {
       completion: void 0,
       finish_reason: void 0,
       function_calls: void 0,
       model
     };
-    const functionCalls = chunk.functionCalls();
+    const functionCalls = chunk.functionCalls;
     if (functionCalls) {
       outChunk.function_calls = functionCalls;
     }
@@ -21232,26 +22249,26 @@ async function* streamChatCompletions3({
 }
 
 // src/plugins/google/nodes/ChatGoogleNode.ts
-var import_non_secure87 = require("nanoid/non-secure");
-var import_ts_dedent79 = require("ts-dedent");
+var import_non_secure89 = require("nanoid/non-secure");
+var import_ts_dedent80 = require("ts-dedent");
 var import_p_retry3 = __toESM(require("p-retry-4"), 1);
 var import_ts_pattern12 = require("ts-pattern");
-var import_generative_ai2 = require("@google/generative-ai");
-var import_lodash_es20 = require("lodash");
+var import_genai2 = require("@google/genai");
+var import_lodash_es21 = require("lodash");
 var cache3 = /* @__PURE__ */ new Map();
 var ChatGoogleNodeImpl = {
   create() {
     const chartNode = {
       type: "chatGoogle",
       title: "Chat (Google)",
-      id: (0, import_non_secure87.nanoid)(),
+      id: (0, import_non_secure89.nanoid)(),
       visualData: {
         x: 0,
         y: 0,
         width: 275
       },
       data: {
-        model: "gemini-2.0-flash-001",
+        model: "gemini-2.5-flash-preview-04-17",
         useModelInput: false,
         temperature: 0.5,
         useTemperatureInput: false,
@@ -21265,7 +22282,9 @@ var ChatGoogleNodeImpl = {
         useMaxTokensInput: false,
         cache: false,
         useAsGraphPartialOutput: true,
-        useToolCalling: false
+        useToolCalling: false,
+        thinkingBudget: void 0,
+        useThinkingBudgetInput: false
       }
     };
     return chartNode;
@@ -21323,11 +22342,27 @@ var ChatGoogleNodeImpl = {
         description: "Tools available for the model to call."
       });
     }
+    if (data.useThinkingBudgetInput) {
+      inputs.push({
+        dataType: "number",
+        id: "thinkingBudget",
+        title: "Thinking Budget",
+        description: "The token budget for the model to think before responding."
+      });
+    }
     inputs.push({
       dataType: ["chat-message", "chat-message[]"],
       id: "prompt",
       title: "Prompt"
     });
+    if (data.useHeadersInput) {
+      inputs.push({
+        dataType: "object",
+        id: "headers",
+        title: "Headers",
+        description: "Additional headers to send to the API."
+      });
+    }
     return inputs;
   },
   getOutputDefinitions(data) {
@@ -21361,10 +22396,11 @@ var ChatGoogleNodeImpl = {
   },
   getBody(data) {
     var _a;
-    return import_ts_dedent79.dedent`
+    return import_ts_dedent80.dedent`
       ${((_a = generativeAiGoogleModels[data.model]) == null ? void 0 : _a.displayName) ?? `Google (${data.model})`}
       ${data.useTopP ? `Top P: ${data.useTopPInput ? "(Using Input)" : data.top_p}` : `Temperature: ${data.useTemperatureInput ? "(Using Input)" : data.temperature}`}
       Max Tokens: ${data.maxTokens}
+      Thinking Budget: ${data.thinkingBudget ?? "Automatic"}
     `;
   },
   getEditors() {
@@ -21409,6 +22445,17 @@ var ChatGoogleNodeImpl = {
         max: Number.MAX_SAFE_INTEGER,
         step: 1
       },
+      {
+        type: "number",
+        label: "Thinking Budget",
+        dataKey: "thinkingBudget",
+        allowEmpty: true,
+        step: 1,
+        min: 0,
+        max: Number.MAX_SAFE_INTEGER,
+        useInputToggleDataKey: "useThinkingBudgetInput",
+        helperMessage: "The token budget for the model to think before responding. Leave blank for automatic budget."
+      },
       {
         type: "toggle",
         label: "Enable Tool Calling",
@@ -21423,12 +22470,20 @@ var ChatGoogleNodeImpl = {
         type: "toggle",
         label: "Use for subgraph partial output",
         dataKey: "useAsGraphPartialOutput"
+      },
+      {
+        type: "keyValuePair",
+        label: "Headers",
+        dataKey: "headers",
+        useInputToggleDataKey: "useHeadersInput",
+        keyPlaceholder: "Header",
+        helperMessage: "Additional headers to send to the API."
       }
     ];
   },
   getUIData() {
     return {
-      infoBoxBody: import_ts_dedent79.dedent`
+      infoBoxBody: import_ts_dedent80.dedent`
         Makes a call to an Google chat model. The settings contains many options for tweaking the model's behavior.
       `,
       infoBoxTitle: "Chat (Google) Node",
@@ -21444,6 +22499,7 @@ var ChatGoogleNodeImpl = {
     const temperature = getInputOrData(data, inputs, "temperature", "number");
     const topP = getInputOrData(data, inputs, "top_p", "number");
     const useTopP = getInputOrData(data, inputs, "useTopP", "boolean");
+    const thinkingBudget = getInputOrData(data, inputs, "thinkingBudget", "number");
     const { messages } = getChatGoogleNodeMessages(inputs);
     let prompt = await Promise.all(
       messages.map(async (message) => {
@@ -21542,8 +22598,8 @@ var ChatGoogleNodeImpl = {
                 name: tool.name,
                 description: tool.description,
                 parameters: Object.keys(tool.parameters.properties).length === 0 ? void 0 : {
-                  type: import_generative_ai2.SchemaType.OBJECT,
-                  properties: (0, import_lodash_es20.mapValues)(tool.parameters.properties, (p) => ({
+                  type: import_genai2.Type.OBJECT,
+                  properties: (0, import_lodash_es21.mapValues)(tool.parameters.properties, (p) => ({
                     // gemini doesn't support union property types, it uses openapi style not jsonschema, what a mess
                     type: Array.isArray(p.type) ? p.type.filter((t) => t !== "null")[0] : p.type,
                     description: p.description
@@ -21567,6 +22623,18 @@ var ChatGoogleNodeImpl = {
         throw new Error("Google Application Credentials or Google API Key is not defined.");
       }
     }
+    const headersFromData = (data.headers ?? []).reduce(
+      (acc, header) => {
+        acc[header.key] = header.value;
+        return acc;
+      },
+      {}
+    );
+    const additionalHeaders = data.useHeadersInput ? coerceTypeOptional(inputs["headers"], "object") ?? headersFromData : headersFromData;
+    const allAdditionalHeaders = cleanHeaders({
+      ...context.settings.chatNodeHeaders,
+      ...additionalHeaders
+    });
     try {
       return await (0, import_p_retry3.default)(
         async () => {
@@ -21579,7 +22647,9 @@ var ChatGoogleNodeImpl = {
             maxOutputTokens: maxTokens,
             systemPrompt,
             topK: void 0,
-            tools
+            tools,
+            thinkingBudget,
+            additionalHeaders: allAdditionalHeaders
           };
           const cacheKey = JSON.stringify(options2);
           if (data.cache) {
@@ -21604,7 +22674,9 @@ var ChatGoogleNodeImpl = {
               topK: void 0,
               apiKey,
               systemPrompt,
-              tools
+              tools,
+              thinkingBudget,
+              additionalHeaders: allAdditionalHeaders
             });
           } else {
             chunks = streamChatCompletions3({
@@ -21622,6 +22694,15 @@ var ChatGoogleNodeImpl = {
           }
           const responseParts = [];
           const functionCalls = [];
+          let throttleLastCalledTime = Date.now();
+          const onPartialOutput = (output2) => {
+            var _a2;
+            const now = Date.now();
+            if (now - throttleLastCalledTime > (context.settings.throttleChatNode ?? 100)) {
+              (_a2 = context.onPartialOutputs) == null ? void 0 : _a2.call(context, output2);
+              throttleLastCalledTime = now;
+            }
+          };
           for await (const chunk of chunks) {
             if (chunk.completion) {
               responseParts.push(chunk.completion);
@@ -21641,8 +22722,9 @@ var ChatGoogleNodeImpl = {
                 }))
               };
             }
-            (_a = context.onPartialOutputs) == null ? void 0 : _a.call(context, output);
+            onPartialOutput == null ? void 0 : onPartialOutput(output);
           }
+          (_a = context.onPartialOutputs) == null ? void 0 : _a.call(context, output);
           const endTime = Date.now();
           output["all-messages"] = {
             type: "chat-message[]",
@@ -21794,7 +22876,7 @@ var plugins = {
 };
 
 // src/integrations/DatasetProvider.ts
-var import_lodash_es21 = require("lodash");
+var import_lodash_es22 = require("lodash");
 var InMemoryDatasetProvider = class {
   #datasets;
   constructor(datasets) {
@@ -21874,7 +22956,7 @@ var InMemoryDatasetProvider = class {
     return sorted.slice(0, k).map((r) => ({ ...r.row, distance: r.similarity }));
   }
   async exportDatasetsForProject(_projectId) {
-    return (0, import_lodash_es21.cloneDeep)(this.#datasets);
+    return (0, import_lodash_es22.cloneDeep)(this.#datasets);
   }
 };
 var dotProductSimilarity = (a, b) => {
@@ -21960,6 +23042,23 @@ async function* getProcessorEvents(processor, spec) {
     }
   }
 }
+var createOnStreamUserEvents = (eventList, handleUserEvent) => {
+  if (!(eventList == null ? void 0 : eventList.trim())) {
+    return void 0;
+  }
+  const events = eventList.split(",").map((e) => e.trim()).filter(Boolean);
+  if (!events.length) {
+    return void 0;
+  }
+  return Object.fromEntries(
+    events.map((event) => [
+      event,
+      async (data) => {
+        await handleUserEvent(event, data);
+      }
+    ])
+  );
+};
 function getProcessorSSEStream(processor, spec) {
   const encoder = new TextEncoder();
   function sendEvent(controller, type, data) {
@@ -21971,6 +23070,18 @@ data: ${JSON.stringify(data)}
   }
   return new ReadableStream({
     async start(controller) {
+      const userEventHandler = async (eventName, data) => {
+        sendEvent(controller, "event", {
+          name: eventName,
+          message: coerceType(data, "string")
+        });
+      };
+      const streamEvents = createOnStreamUserEvents(spec.userStreamEvents, userEventHandler);
+      if (streamEvents) {
+        for (const [name, fn] of Object.entries(streamEvents)) {
+          processor.onUserEvent(name, fn);
+        }
+      }
       try {
         for await (const event of getProcessorEvents(processor, spec)) {
           sendEvent(controller, event.type, event);
@@ -22125,6 +23236,7 @@ function coreCreateProcessor(project, options2) {
           nativeApi: options2.nativeApi,
           datasetProvider: options2.datasetProvider,
           audioProvider: options2.audioProvider,
+          mcpProvider: options2.mcpProvider,
           codeRunner: options2.codeRunner,
           projectPath: options2.projectPath,
           projectReferenceLoader: options2.projectReferenceLoader,
@@ -22223,6 +23335,10 @@ var Rivet = void 0;
   LoadDatasetNodeImpl,
   LoopControllerNodeImpl,
   LoopUntilNodeImpl,
+  MCPError,
+  MCPErrorType,
+  MCPGetPromptNodeImpl,
+  MCPToolCallNodeImpl,
   MatchNodeImpl,
   NodeImpl,
   NodeRegistration,
@@ -22295,6 +23411,7 @@ var Rivet = void 0;
   coreCreateProcessor,
   coreRunGraph,
   createDatasetNode,
+  createOnStreamUserEvents,
   cronNode,
   dataTypeDisplayNames,
   dataTypes,
@@ -22372,6 +23489,9 @@ var Rivet = void 0;
   looseDataValueToDataValue,
   looseDataValuesToDataValues,
   matchNode,
+  mcpDiscoveryNode,
+  mcpGetPromptNode,
+  mcpToolCallNode,
   newId,
   nodeDefinition,
   numberNode,