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

package/dist/types/integrations/mcp/MCPProvider.d.ts

@@ -0,0 +1,153 @@
+/**
+ * Derived from types here - https://github.com/modelcontextprotocol/typescript-sdk/blob/main/src/types.ts
+ */
+export declare namespace MCP {
+    export type TransportType = 'stdio' | 'http';
+    export interface ServerConfig {
+        command: string;
+        args?: string[];
+        env?: Record<string, string>;
+        disabled?: boolean;
+        alwaysAllow?: string[];
+    }
+    export type ServerConfigWithId = {
+        config: ServerConfig;
+        serverId: string;
+    };
+    export interface Config {
+        mcpServers: Record<string, ServerConfig>;
+    }
+    /**
+     * Tools
+     */
+    interface ToolAnnotations {
+        title?: string;
+        readOnlyHint?: boolean;
+        destructiveHint?: boolean;
+        idempotentHint?: boolean;
+        openWorldHint?: boolean;
+    }
+    export interface Tool {
+        name: string;
+        description?: string;
+        inputSchema: {
+            [x: string]: unknown;
+        };
+        annotations?: ToolAnnotations;
+    }
+    export interface ToolCallRequest {
+        [key: string]: unknown;
+        name: string;
+        arguments?: {
+            [x: string]: unknown;
+        };
+        id?: string;
+    }
+    export interface ToolCallResponse {
+        content: ResponseContent[];
+        isError?: boolean;
+    }
+    /**
+     * Prompts
+     */
+    interface PromptArgument {
+        name: string;
+        description?: string;
+        required?: boolean;
+    }
+    export interface Prompt {
+        name: string;
+        description?: string;
+        arugments?: PromptArgument[];
+    }
+    export interface GetPromptRequest {
+        [x: string]: unknown;
+        name: string;
+        arguments?: {
+            [x: string]: string;
+        };
+    }
+    export interface PromptMessage {
+        role: 'user' | 'assistant';
+        content: ResponseContent;
+    }
+    export interface GetPromptResponse {
+        messages: PromptMessage[];
+        description?: string;
+    }
+    /**
+     * Content
+     */
+    interface BaseContent {
+        type: 'text' | 'image' | 'audio';
+    }
+    interface TextContent extends BaseContent {
+        type: 'text';
+        text: string;
+    }
+    interface ImageContent extends BaseContent {
+        type: 'image';
+        data: string;
+        mimeType: string;
+    }
+    interface AudioContent extends BaseContent {
+        type: 'audio';
+        data: string;
+        mimeType: string;
+    }
+    type ResponseContent = TextContent | ImageContent | AudioContent | {
+        [key: string]: unknown;
+    };
+    export {};
+}
+/**
+ * Errors
+ */
+export declare enum MCPErrorType {
+    CONFIG_NOT_FOUND = "CONFIG_NOT_FOUND",
+    SERVER_NOT_FOUND = "SERVER_NOT_FOUND",
+    SERVER_COMMUNICATION_FAILED = "SERVER_COMMUNICATION_FAILED",
+    INVALID_SCHEMA = "INVALID_SCHEMA"
+}
+export declare class MCPError extends Error {
+    type: MCPErrorType;
+    details?: unknown | undefined;
+    constructor(type: MCPErrorType, message: string, details?: unknown | undefined);
+}
+/**
+ * MCP Provider for Node Implementation
+ */
+export interface MCPProvider {
+    getHTTPTools(clientConfig: {
+        name: string;
+        version: string;
+    }, serverUrl: string, headers: Record<string, string> | undefined): Promise<MCP.Tool[]>;
+    getStdioTools(clientConfig: {
+        name: string;
+        version: string;
+    }, serverConfig: MCP.ServerConfigWithId): Promise<MCP.Tool[]>;
+    getHTTPrompts(clientConfig: {
+        name: string;
+        version: string;
+    }, serverUrl: string, headers: Record<string, string> | undefined): Promise<MCP.Prompt[]>;
+    getStdioPrompts(clientConfig: {
+        name: string;
+        version: string;
+    }, serverConfig: MCP.ServerConfigWithId): Promise<MCP.Prompt[]>;
+    httpToolCall(clientConfig: {
+        name: string;
+        version: string;
+    }, serverUrl: string, headers: Record<string, string> | undefined, toolCall: MCP.ToolCallRequest): Promise<MCP.ToolCallResponse>;
+    stdioToolCall(clientConfig: {
+        name: string;
+        version: string;
+    }, serverConfig: MCP.ServerConfigWithId, toolCall: MCP.ToolCallRequest): Promise<MCP.ToolCallResponse>;
+    getHTTPrompt(clientConfig: {
+        name: string;
+        version: string;
+    }, serverUrl: string, headers: Record<string, string> | undefined, getPromptRequest: MCP.GetPromptRequest): Promise<MCP.GetPromptResponse>;
+    getStdioPrompt(clientConfig: {
+        name: string;
+        version: string;
+    }, serverConfig: MCP.ServerConfigWithId, getPromptRequest: MCP.GetPromptRequest): Promise<MCP.GetPromptResponse>;
+}

package/dist/types/integrations/mcp/MCPUtils.d.ts

@@ -0,0 +1,9 @@
+import type { InternalProcessContext } from '../../model/ProcessContext.js';
+import type { RivetUIContext } from '../../model/RivetUIContext.js';
+import { type MCP } from './MCPProvider.js';
+export declare const loadMCPConfiguration: (context: InternalProcessContext | RivetUIContext) => Promise<MCP.Config>;
+export declare const getServerOptions: (context: RivetUIContext) => Promise<{
+    label: string;
+    value: string;
+}[]>;
+export declare const getServerHelperMessage: (context: RivetUIContext, optionsLength: number) => string;

package/dist/types/model/GraphProcessor.d.ts

@@ -126,6 +126,10 @@ export type ProcessEvent = {
 }[keyof ProcessEvents];
 export type GraphOutputs = Record<string, DataValue>;
 export type GraphInputs = Record<string, DataValue>;
+export type GraphEvents = {
+    name: string;
+    message: string;
+};
 export type NodeResults = Map<NodeId, Outputs>;
 export type Inputs = Record<PortId, DataValue | undefined>;
 export type Outputs = Record<PortId, DataValue | undefined>;
@@ -166,7 +170,7 @@ export declare class GraphProcessor {
     recordingPlaybackChatLatency: number;
     warnOnInvalidGraph: boolean;
     get isRunning(): boolean;
-    constructor(project: Project, graphId?: GraphId, registry?: NodeRegistration, includeTrace?: boolean);
+    constructor(project: Project, graphId?: GraphId, registry?: NodeRegistration<any, any>, includeTrace?: boolean);
     on: Emittery<ProcessEvents>["on"];
     off: Emittery<ProcessEvents>["off"];
     once: Emittery<ProcessEvents>["once"];

package/dist/types/model/Nodes.d.ts

@@ -81,9 +81,20 @@ export * from './nodes/ToMarkdownTableNode.js';
 export * from './nodes/CronNode.js';
 export * from './nodes/ToTreeNode.js';
 export * from './nodes/LoopUntilNode.js';
+export * from './nodes/MCPDiscoveryNode.js';
+export * from './nodes/MCPToolCallNode.js';
+export * from './nodes/MCPGetPromptNode.js';
 export * from './nodes/ReferencedGraphAliasNode.js';
-export declare const registerBuiltInNodes: (registry: NodeRegistration) => NodeRegistration<"number" | "boolean" | "object" | "pop" | "join" | "slice" | "filter" | "match" | "split" | "image" | "audio" | "document" | "if" | "code" | "context" | "array" | "text" | "userInput" | "prompt" | "chat" | "extractRegex" | "readDirectory" | "readFile" | "writeFile" | "ifElse" | "chunk" | "graphInput" | "graphOutput" | "subGraph" | "extractJson" | "assemblePrompt" | "extractYaml" | "loopController" | "trimChatMessages" | "externalCall" | "setGlobal" | "raiseEvent" | "getGlobal" | "abortGraph" | "extractObjectPath" | "coalesce" | "passthrough" | "waitForEvent" | "gptFunction" | "toYaml" | "getEmbedding" | "vectorStore" | "vectorNearestNeighbors" | "hash" | "raceInputs" | "toJson" | "compare" | "evaluate" | "randomNumber" | "shuffle" | "comment" | "imagetoMD" | "httpCall" | "delay" | "appendToDataset" | "createDataset" | "loadDataset" | "getAllDatasets" | "datasetNearestNeighbors" | "getDatasetRow" | "extractMarkdownCodeBlocks" | "assembleMessage" | "urlReference" | "destructure" | "replaceDataset" | "listGraphs" | "graphReference" | "callGraph" | "delegateFunctionCall" | "playAudio" | "chatLoop" | "readAllFiles" | "toMarkdownTable" | "cron" | "toTree" | "loopUntil" | "referencedGraphAlias", import("./nodes/UserInputNode.js").UserInputNode | import("./nodes/TextNode.js").TextNode | import("./nodes/ChatNode.js").ChatNode | import("./nodes/PromptNode.js").PromptNode | import("./nodes/ExtractRegexNode.js").ExtractRegexNode | import("./nodes/CodeNode.js").CodeNode | import("./nodes/MatchNode.js").MatchNode | import("./nodes/IfNode.js").IfNode | import("./nodes/ReadDirectoryNode.js").ReadDirectoryNode | import("./nodes/ReadFileNode.js").ReadFileNode | import("./nodes/WriteFileNode.js").WriteFileNode | import("./nodes/IfElseNode.js").IfElseNode | import("./nodes/ChunkNode.js").ChunkNode | import("./nodes/GraphInputNode.js").GraphInputNode | import("./nodes/GraphOutputNode.js").GraphOutputNode | import("./nodes/SubGraphNode.js").SubGraphNode | import("./nodes/ArrayNode.js").ArrayNode | import("./nodes/ExtractJsonNode.js").ExtractJsonNode | import("./nodes/AssemblePromptNode.js").AssemblePromptNode | import("./nodes/ExtractYamlNode.js").ExtractYamlNode | import("./nodes/LoopControllerNode.js").LoopControllerNode | import("./nodes/TrimChatMessagesNode.js").TrimChatMessagesNode | import("./nodes/ExternalCallNode.js").ExternalCallNode | import("./nodes/ExtractObjectPathNode.js").ExtractObjectPathNode | import("./nodes/RaiseEventNode.js").RaiseEventNode | import("./nodes/ContextNode.js").ContextNode | import("./nodes/CoalesceNode.js").CoalesceNode | import("./nodes/PassthroughNode.js").PassthroughNode | import("./nodes/PopNode.js").PopNode | import("./nodes/SetGlobalNode.js").SetGlobalNode | import("./nodes/GetGlobalNode.js").GetGlobalNode | import("./nodes/WaitForEventNode.js").WaitForEventNode | import("./nodes/ToolNode.js").GptFunctionNode | import("./nodes/ToYamlNode.js").ToYamlNode | import("./nodes/GetEmbeddingNode.js").GetEmbeddingNode | import("./nodes/VectorStoreNode.js").VectorStoreNode | import("./nodes/VectorNearestNeighborsNode.js").VectorNearestNeighborsNode | import("./nodes/HashNode.js").HashNode | import("./nodes/AbortGraphNode.js").AbortGraphNode | import("./nodes/RaceInputsNode.js").RaceInputsNode | import("./nodes/ToJsonNode.js").ToJsonNode | import("./nodes/JoinNode.js").JoinNode | import("./nodes/FilterNode.js").FilterNode | import("./nodes/ObjectNode.js").ObjectNode | import("./nodes/BooleanNode.js").BooleanNode | import("./nodes/CompareNode.js").CompareNode | import("./nodes/EvaluateNode.js").EvaluateNode | import("./nodes/NumberNode.js").NumberNode | import("./nodes/RandomNumberNode.js").RandomNumberNode | import("./nodes/ShuffleNode.js").ShuffleNode | import("./nodes/CommentNode.js").CommentNode | import("./nodes/ImageToMDNode.js").ImageToMDNode | import("./nodes/ImageNode.js").ImageNode | import("./nodes/AudioNode.js").AudioNode | import("./nodes/HttpCallNode.js").HttpCallNode | import("./nodes/DelayNode.js").DelayNode | import("./nodes/AppendToDatasetNode.js").AppendToDatasetNode | import("./nodes/CreateDatasetNode.js").CreateDatasetNode | import("./nodes/LoadDatasetNode.js").LoadDatasetNode | import("./nodes/GetAllDatasetsNode.js").GetAllDatasetsNode | import("./nodes/SplitNode.js").SplitNode | import("./nodes/DatasetNearestNeigborsNode.js").DatasetNearestNeighborsNode | import("./nodes/GetDatasetRowNode.js").GetDatasetRowNode | import("./nodes/SliceNode.js").SliceNode | import("./nodes/ExtractMarkdownCodeBlocksNode.js").ExtractMarkdownCodeBlocksNode | import("./nodes/AssembleMessageNode.js").AssembleMessageNode | import("./nodes/URLReferenceNode.js").UrlReferenceNode | import("./nodes/DestructureNode.js").DestructureNode | import("./nodes/ReplaceDatasetNode.js").ReplaceDatasetNode | import("./nodes/ListGraphsNode.js").ListGraphsNode | import("./nodes/GraphReferenceNode.js").GraphReferenceNode | import("./nodes/CallGraphNode.js").CallGraphNode | import("./nodes/DelegateFunctionCallNode.js").DelegateFunctionCallNode | import("./nodes/PlayAudioNode.js").PlayAudioNode | import("./nodes/DocumentNode.js").DocumentNode | import("./nodes/ChatLoopNode.js").ChatLoopNode | import("./nodes/ReadAllFilesNode.js").ReadAllFilesNode | import("./nodes/ToMarkdownTableNode.js").ToMarkdownTableNode | import("./nodes/CronNode.js").CronNode | import("./nodes/ToTreeNode.js").ToTreeNode | import("./nodes/LoopUntilNode.js").LoopUntilNode | import("./nodes/ReferencedGraphAliasNode.js").ReferencedGraphAliasNode>;
-declare let globalRivetNodeRegistry: NodeRegistration<"number" | "boolean" | "object" | "pop" | "join" | "slice" | "filter" | "match" | "split" | "image" | "audio" | "document" | "if" | "code" | "context" | "array" | "text" | "userInput" | "prompt" | "chat" | "extractRegex" | "readDirectory" | "readFile" | "writeFile" | "ifElse" | "chunk" | "graphInput" | "graphOutput" | "subGraph" | "extractJson" | "assemblePrompt" | "extractYaml" | "loopController" | "trimChatMessages" | "externalCall" | "setGlobal" | "raiseEvent" | "getGlobal" | "abortGraph" | "extractObjectPath" | "coalesce" | "passthrough" | "waitForEvent" | "gptFunction" | "toYaml" | "getEmbedding" | "vectorStore" | "vectorNearestNeighbors" | "hash" | "raceInputs" | "toJson" | "compare" | "evaluate" | "randomNumber" | "shuffle" | "comment" | "imagetoMD" | "httpCall" | "delay" | "appendToDataset" | "createDataset" | "loadDataset" | "getAllDatasets" | "datasetNearestNeighbors" | "getDatasetRow" | "extractMarkdownCodeBlocks" | "assembleMessage" | "urlReference" | "destructure" | "replaceDataset" | "listGraphs" | "graphReference" | "callGraph" | "delegateFunctionCall" | "playAudio" | "chatLoop" | "readAllFiles" | "toMarkdownTable" | "cron" | "toTree" | "loopUntil" | "referencedGraphAlias", import("./nodes/UserInputNode.js").UserInputNode | import("./nodes/TextNode.js").TextNode | import("./nodes/ChatNode.js").ChatNode | import("./nodes/PromptNode.js").PromptNode | import("./nodes/ExtractRegexNode.js").ExtractRegexNode | import("./nodes/CodeNode.js").CodeNode | import("./nodes/MatchNode.js").MatchNode | import("./nodes/IfNode.js").IfNode | import("./nodes/ReadDirectoryNode.js").ReadDirectoryNode | import("./nodes/ReadFileNode.js").ReadFileNode | import("./nodes/WriteFileNode.js").WriteFileNode | import("./nodes/IfElseNode.js").IfElseNode | import("./nodes/ChunkNode.js").ChunkNode | import("./nodes/GraphInputNode.js").GraphInputNode | import("./nodes/GraphOutputNode.js").GraphOutputNode | import("./nodes/SubGraphNode.js").SubGraphNode | import("./nodes/ArrayNode.js").ArrayNode | import("./nodes/ExtractJsonNode.js").ExtractJsonNode | import("./nodes/AssemblePromptNode.js").AssemblePromptNode | import("./nodes/ExtractYamlNode.js").ExtractYamlNode | import("./nodes/LoopControllerNode.js").LoopControllerNode | import("./nodes/TrimChatMessagesNode.js").TrimChatMessagesNode | import("./nodes/ExternalCallNode.js").ExternalCallNode | import("./nodes/ExtractObjectPathNode.js").ExtractObjectPathNode | import("./nodes/RaiseEventNode.js").RaiseEventNode | import("./nodes/ContextNode.js").ContextNode | import("./nodes/CoalesceNode.js").CoalesceNode | import("./nodes/PassthroughNode.js").PassthroughNode | import("./nodes/PopNode.js").PopNode | import("./nodes/SetGlobalNode.js").SetGlobalNode | import("./nodes/GetGlobalNode.js").GetGlobalNode | import("./nodes/WaitForEventNode.js").WaitForEventNode | import("./nodes/ToolNode.js").GptFunctionNode | import("./nodes/ToYamlNode.js").ToYamlNode | import("./nodes/GetEmbeddingNode.js").GetEmbeddingNode | import("./nodes/VectorStoreNode.js").VectorStoreNode | import("./nodes/VectorNearestNeighborsNode.js").VectorNearestNeighborsNode | import("./nodes/HashNode.js").HashNode | import("./nodes/AbortGraphNode.js").AbortGraphNode | import("./nodes/RaceInputsNode.js").RaceInputsNode | import("./nodes/ToJsonNode.js").ToJsonNode | import("./nodes/JoinNode.js").JoinNode | import("./nodes/FilterNode.js").FilterNode | import("./nodes/ObjectNode.js").ObjectNode | import("./nodes/BooleanNode.js").BooleanNode | import("./nodes/CompareNode.js").CompareNode | import("./nodes/EvaluateNode.js").EvaluateNode | import("./nodes/NumberNode.js").NumberNode | import("./nodes/RandomNumberNode.js").RandomNumberNode | import("./nodes/ShuffleNode.js").ShuffleNode | import("./nodes/CommentNode.js").CommentNode | import("./nodes/ImageToMDNode.js").ImageToMDNode | import("./nodes/ImageNode.js").ImageNode | import("./nodes/AudioNode.js").AudioNode | import("./nodes/HttpCallNode.js").HttpCallNode | import("./nodes/DelayNode.js").DelayNode | import("./nodes/AppendToDatasetNode.js").AppendToDatasetNode | import("./nodes/CreateDatasetNode.js").CreateDatasetNode | import("./nodes/LoadDatasetNode.js").LoadDatasetNode | import("./nodes/GetAllDatasetsNode.js").GetAllDatasetsNode | import("./nodes/SplitNode.js").SplitNode | import("./nodes/DatasetNearestNeigborsNode.js").DatasetNearestNeighborsNode | import("./nodes/GetDatasetRowNode.js").GetDatasetRowNode | import("./nodes/SliceNode.js").SliceNode | import("./nodes/ExtractMarkdownCodeBlocksNode.js").ExtractMarkdownCodeBlocksNode | import("./nodes/AssembleMessageNode.js").AssembleMessageNode | import("./nodes/URLReferenceNode.js").UrlReferenceNode | import("./nodes/DestructureNode.js").DestructureNode | import("./nodes/ReplaceDatasetNode.js").ReplaceDatasetNode | import("./nodes/ListGraphsNode.js").ListGraphsNode | import("./nodes/GraphReferenceNode.js").GraphReferenceNode | import("./nodes/CallGraphNode.js").CallGraphNode | import("./nodes/DelegateFunctionCallNode.js").DelegateFunctionCallNode | import("./nodes/PlayAudioNode.js").PlayAudioNode | import("./nodes/DocumentNode.js").DocumentNode | import("./nodes/ChatLoopNode.js").ChatLoopNode | import("./nodes/ReadAllFilesNode.js").ReadAllFilesNode | import("./nodes/ToMarkdownTableNode.js").ToMarkdownTableNode | import("./nodes/CronNode.js").CronNode | import("./nodes/ToTreeNode.js").ToTreeNode | import("./nodes/LoopUntilNode.js").LoopUntilNode | import("./nodes/ReferencedGraphAliasNode.js").ReferencedGraphAliasNode>;
+export declare const registerBuiltInNodes: (registry: NodeRegistration) => NodeRegistration<"number" | "boolean" | "object" | "pop" | "join" | "slice" | "filter" | "match" | "split" | "image" | "audio" | "document" | "if" | "code" | "text" | "context" | "array" | "userInput" | "prompt" | "chat" | "extractRegex" | "readDirectory" | "readFile" | "writeFile" | "ifElse" | "chunk" | "graphInput" | "graphOutput" | "subGraph" | "extractJson" | "assemblePrompt" | "extractYaml" | "loopController" | "trimChatMessages" | "externalCall" | "setGlobal" | "raiseEvent" | "getGlobal" | "abortGraph" | "extractObjectPath" | "coalesce" | "passthrough" | "waitForEvent" | "gptFunction" | "toYaml" | "getEmbedding" | "vectorStore" | "vectorNearestNeighbors" | "hash" | "raceInputs" | "toJson" | "compare" | "evaluate" | "randomNumber" | "shuffle" | "comment" | "imagetoMD" | "httpCall" | "delay" | "appendToDataset" | "createDataset" | "loadDataset" | "getAllDatasets" | "datasetNearestNeighbors" | "getDatasetRow" | "extractMarkdownCodeBlocks" | "assembleMessage" | "urlReference" | "destructure" | "replaceDataset" | "listGraphs" | "graphReference" | "callGraph" | "delegateFunctionCall" | "playAudio" | "chatLoop" | "readAllFiles" | "toMarkdownTable" | "cron" | "toTree" | "loopUntil" | "mcpDiscovery" | "mcpToolCall" | "mcpGetPrompt" | "referencedGraphAlias", import("./nodes/UserInputNode.js").UserInputNode | import("./nodes/TextNode.js").TextNode | import("./nodes/ChatNode.js").ChatNode | import("./nodes/PromptNode.js").PromptNode | import("./nodes/ExtractRegexNode.js").ExtractRegexNode | import("./nodes/CodeNode.js").CodeNode | import("./nodes/MatchNode.js").MatchNode | import("./nodes/IfNode.js").IfNode | import("./nodes/ReadDirectoryNode.js").ReadDirectoryNode | import("./nodes/ReadFileNode.js").ReadFileNode | import("./nodes/WriteFileNode.js").WriteFileNode | import("./nodes/IfElseNode.js").IfElseNode | import("./nodes/ChunkNode.js").ChunkNode | import("./nodes/GraphInputNode.js").GraphInputNode | import("./nodes/GraphOutputNode.js").GraphOutputNode | import("./nodes/SubGraphNode.js").SubGraphNode | import("./nodes/ArrayNode.js").ArrayNode | import("./nodes/ExtractJsonNode.js").ExtractJsonNode | import("./nodes/AssemblePromptNode.js").AssemblePromptNode | import("./nodes/ExtractYamlNode.js").ExtractYamlNode | import("./nodes/LoopControllerNode.js").LoopControllerNode | import("./nodes/TrimChatMessagesNode.js").TrimChatMessagesNode | import("./nodes/ExternalCallNode.js").ExternalCallNode | import("./nodes/ExtractObjectPathNode.js").ExtractObjectPathNode | import("./nodes/RaiseEventNode.js").RaiseEventNode | import("./nodes/ContextNode.js").ContextNode | import("./nodes/CoalesceNode.js").CoalesceNode | import("./nodes/PassthroughNode.js").PassthroughNode | import("./nodes/PopNode.js").PopNode | import("./nodes/SetGlobalNode.js").SetGlobalNode | import("./nodes/GetGlobalNode.js").GetGlobalNode | import("./nodes/WaitForEventNode.js").WaitForEventNode | import("./nodes/ToolNode.js").GptFunctionNode | import("./nodes/ToYamlNode.js").ToYamlNode | import("./nodes/GetEmbeddingNode.js").GetEmbeddingNode | import("./nodes/VectorStoreNode.js").VectorStoreNode | import("./nodes/VectorNearestNeighborsNode.js").VectorNearestNeighborsNode | import("./nodes/HashNode.js").HashNode | import("./nodes/AbortGraphNode.js").AbortGraphNode | import("./nodes/RaceInputsNode.js").RaceInputsNode | import("./nodes/ToJsonNode.js").ToJsonNode | import("./nodes/JoinNode.js").JoinNode | import("./nodes/FilterNode.js").FilterNode | import("./nodes/ObjectNode.js").ObjectNode | import("./nodes/BooleanNode.js").BooleanNode | import("./nodes/CompareNode.js").CompareNode | import("./nodes/EvaluateNode.js").EvaluateNode | import("./nodes/NumberNode.js").NumberNode | import("./nodes/RandomNumberNode.js").RandomNumberNode | import("./nodes/ShuffleNode.js").ShuffleNode | import("./nodes/CommentNode.js").CommentNode | import("./nodes/ImageToMDNode.js").ImageToMDNode | import("./nodes/ImageNode.js").ImageNode | import("./nodes/AudioNode.js").AudioNode | import("./nodes/HttpCallNode.js").HttpCallNode | import("./nodes/DelayNode.js").DelayNode | import("./nodes/AppendToDatasetNode.js").AppendToDatasetNode | import("./nodes/CreateDatasetNode.js").CreateDatasetNode | import("./nodes/LoadDatasetNode.js").LoadDatasetNode | import("./nodes/GetAllDatasetsNode.js").GetAllDatasetsNode | import("./nodes/SplitNode.js").SplitNode | import("./nodes/DatasetNearestNeigborsNode.js").DatasetNearestNeighborsNode | import("./nodes/GetDatasetRowNode.js").GetDatasetRowNode | import("./nodes/SliceNode.js").SliceNode | import("./nodes/ExtractMarkdownCodeBlocksNode.js").ExtractMarkdownCodeBlocksNode | import("./nodes/AssembleMessageNode.js").AssembleMessageNode | import("./nodes/URLReferenceNode.js").UrlReferenceNode | import("./nodes/DestructureNode.js").DestructureNode | import("./nodes/ReplaceDatasetNode.js").ReplaceDatasetNode | import("./nodes/ListGraphsNode.js").ListGraphsNode | import("./nodes/GraphReferenceNode.js").GraphReferenceNode | import("./nodes/CallGraphNode.js").CallGraphNode | import("./nodes/DelegateFunctionCallNode.js").DelegateFunctionCallNode | import("./nodes/PlayAudioNode.js").PlayAudioNode | import("./nodes/DocumentNode.js").DocumentNode | import("./nodes/ChatLoopNode.js").ChatLoopNode | import("./nodes/ReadAllFilesNode.js").ReadAllFilesNode | import("./nodes/ToMarkdownTableNode.js").ToMarkdownTableNode | import("./nodes/CronNode.js").CronNode | import("./nodes/ToTreeNode.js").ToTreeNode | import("./nodes/LoopUntilNode.js").LoopUntilNode | (import("./NodeBase.js").NodeBase & {
+    type: "mcpDiscovery";
+    data: import("./nodes/MCPDiscoveryNode.js").MCPDiscoveryNodeData;
+    variants?: import("./NodeBase.js").ChartNodeVariant<import("./nodes/MCPDiscoveryNode.js").MCPDiscoveryNodeData>[] | undefined;
+}) | import("./nodes/MCPToolCallNode.js").MCPToolCallNode | import("./nodes/MCPGetPromptNode.js").MCPGetPromptNode | import("./nodes/ReferencedGraphAliasNode.js").ReferencedGraphAliasNode>;
+declare let globalRivetNodeRegistry: NodeRegistration<"number" | "boolean" | "object" | "pop" | "join" | "slice" | "filter" | "match" | "split" | "image" | "audio" | "document" | "if" | "code" | "text" | "context" | "array" | "userInput" | "prompt" | "chat" | "extractRegex" | "readDirectory" | "readFile" | "writeFile" | "ifElse" | "chunk" | "graphInput" | "graphOutput" | "subGraph" | "extractJson" | "assemblePrompt" | "extractYaml" | "loopController" | "trimChatMessages" | "externalCall" | "setGlobal" | "raiseEvent" | "getGlobal" | "abortGraph" | "extractObjectPath" | "coalesce" | "passthrough" | "waitForEvent" | "gptFunction" | "toYaml" | "getEmbedding" | "vectorStore" | "vectorNearestNeighbors" | "hash" | "raceInputs" | "toJson" | "compare" | "evaluate" | "randomNumber" | "shuffle" | "comment" | "imagetoMD" | "httpCall" | "delay" | "appendToDataset" | "createDataset" | "loadDataset" | "getAllDatasets" | "datasetNearestNeighbors" | "getDatasetRow" | "extractMarkdownCodeBlocks" | "assembleMessage" | "urlReference" | "destructure" | "replaceDataset" | "listGraphs" | "graphReference" | "callGraph" | "delegateFunctionCall" | "playAudio" | "chatLoop" | "readAllFiles" | "toMarkdownTable" | "cron" | "toTree" | "loopUntil" | "mcpDiscovery" | "mcpToolCall" | "mcpGetPrompt" | "referencedGraphAlias", import("./nodes/UserInputNode.js").UserInputNode | import("./nodes/TextNode.js").TextNode | import("./nodes/ChatNode.js").ChatNode | import("./nodes/PromptNode.js").PromptNode | import("./nodes/ExtractRegexNode.js").ExtractRegexNode | import("./nodes/CodeNode.js").CodeNode | import("./nodes/MatchNode.js").MatchNode | import("./nodes/IfNode.js").IfNode | import("./nodes/ReadDirectoryNode.js").ReadDirectoryNode | import("./nodes/ReadFileNode.js").ReadFileNode | import("./nodes/WriteFileNode.js").WriteFileNode | import("./nodes/IfElseNode.js").IfElseNode | import("./nodes/ChunkNode.js").ChunkNode | import("./nodes/GraphInputNode.js").GraphInputNode | import("./nodes/GraphOutputNode.js").GraphOutputNode | import("./nodes/SubGraphNode.js").SubGraphNode | import("./nodes/ArrayNode.js").ArrayNode | import("./nodes/ExtractJsonNode.js").ExtractJsonNode | import("./nodes/AssemblePromptNode.js").AssemblePromptNode | import("./nodes/ExtractYamlNode.js").ExtractYamlNode | import("./nodes/LoopControllerNode.js").LoopControllerNode | import("./nodes/TrimChatMessagesNode.js").TrimChatMessagesNode | import("./nodes/ExternalCallNode.js").ExternalCallNode | import("./nodes/ExtractObjectPathNode.js").ExtractObjectPathNode | import("./nodes/RaiseEventNode.js").RaiseEventNode | import("./nodes/ContextNode.js").ContextNode | import("./nodes/CoalesceNode.js").CoalesceNode | import("./nodes/PassthroughNode.js").PassthroughNode | import("./nodes/PopNode.js").PopNode | import("./nodes/SetGlobalNode.js").SetGlobalNode | import("./nodes/GetGlobalNode.js").GetGlobalNode | import("./nodes/WaitForEventNode.js").WaitForEventNode | import("./nodes/ToolNode.js").GptFunctionNode | import("./nodes/ToYamlNode.js").ToYamlNode | import("./nodes/GetEmbeddingNode.js").GetEmbeddingNode | import("./nodes/VectorStoreNode.js").VectorStoreNode | import("./nodes/VectorNearestNeighborsNode.js").VectorNearestNeighborsNode | import("./nodes/HashNode.js").HashNode | import("./nodes/AbortGraphNode.js").AbortGraphNode | import("./nodes/RaceInputsNode.js").RaceInputsNode | import("./nodes/ToJsonNode.js").ToJsonNode | import("./nodes/JoinNode.js").JoinNode | import("./nodes/FilterNode.js").FilterNode | import("./nodes/ObjectNode.js").ObjectNode | import("./nodes/BooleanNode.js").BooleanNode | import("./nodes/CompareNode.js").CompareNode | import("./nodes/EvaluateNode.js").EvaluateNode | import("./nodes/NumberNode.js").NumberNode | import("./nodes/RandomNumberNode.js").RandomNumberNode | import("./nodes/ShuffleNode.js").ShuffleNode | import("./nodes/CommentNode.js").CommentNode | import("./nodes/ImageToMDNode.js").ImageToMDNode | import("./nodes/ImageNode.js").ImageNode | import("./nodes/AudioNode.js").AudioNode | import("./nodes/HttpCallNode.js").HttpCallNode | import("./nodes/DelayNode.js").DelayNode | import("./nodes/AppendToDatasetNode.js").AppendToDatasetNode | import("./nodes/CreateDatasetNode.js").CreateDatasetNode | import("./nodes/LoadDatasetNode.js").LoadDatasetNode | import("./nodes/GetAllDatasetsNode.js").GetAllDatasetsNode | import("./nodes/SplitNode.js").SplitNode | import("./nodes/DatasetNearestNeigborsNode.js").DatasetNearestNeighborsNode | import("./nodes/GetDatasetRowNode.js").GetDatasetRowNode | import("./nodes/SliceNode.js").SliceNode | import("./nodes/ExtractMarkdownCodeBlocksNode.js").ExtractMarkdownCodeBlocksNode | import("./nodes/AssembleMessageNode.js").AssembleMessageNode | import("./nodes/URLReferenceNode.js").UrlReferenceNode | import("./nodes/DestructureNode.js").DestructureNode | import("./nodes/ReplaceDatasetNode.js").ReplaceDatasetNode | import("./nodes/ListGraphsNode.js").ListGraphsNode | import("./nodes/GraphReferenceNode.js").GraphReferenceNode | import("./nodes/CallGraphNode.js").CallGraphNode | import("./nodes/DelegateFunctionCallNode.js").DelegateFunctionCallNode | import("./nodes/PlayAudioNode.js").PlayAudioNode | import("./nodes/DocumentNode.js").DocumentNode | import("./nodes/ChatLoopNode.js").ChatLoopNode | import("./nodes/ReadAllFilesNode.js").ReadAllFilesNode | import("./nodes/ToMarkdownTableNode.js").ToMarkdownTableNode | import("./nodes/CronNode.js").CronNode | import("./nodes/ToTreeNode.js").ToTreeNode | import("./nodes/LoopUntilNode.js").LoopUntilNode | (import("./NodeBase.js").NodeBase & {
+    type: "mcpDiscovery";
+    data: import("./nodes/MCPDiscoveryNode.js").MCPDiscoveryNodeData;
+    variants?: import("./NodeBase.js").ChartNodeVariant<import("./nodes/MCPDiscoveryNode.js").MCPDiscoveryNodeData>[] | undefined;
+}) | import("./nodes/MCPToolCallNode.js").MCPToolCallNode | import("./nodes/MCPGetPromptNode.js").MCPGetPromptNode | import("./nodes/ReferencedGraphAliasNode.js").ReferencedGraphAliasNode>;
 export { globalRivetNodeRegistry };
 export type BuiltInNodes = typeof globalRivetNodeRegistry.NodesType;
 export type BuiltInNodeType = typeof globalRivetNodeRegistry.NodeTypesType;

package/dist/types/model/ProcessContext.d.ts

@@ -1,5 +1,5 @@
 import type { Opaque } from 'type-fest';
-import { type Settings, type NativeApi, type Project, type DataValue, type ExternalFunction, type Outputs, type GraphId, type GraphProcessor, type ScalarOrArrayDataValue, type DatasetProvider, type ChartNode, type AttachedNodeData, type AudioProvider, type StringArrayDataValue, type ProjectId } from '../index.js';
+import { type Settings, type NativeApi, type Project, type DataValue, type ExternalFunction, type Outputs, type GraphId, type GraphProcessor, type ScalarOrArrayDataValue, type DatasetProvider, type ChartNode, type AttachedNodeData, type AudioProvider, type StringArrayDataValue, type ProjectId, type MCPProvider } from '../index.js';
 import type { Tokenizer } from '../integrations/Tokenizer.js';
 import type { CodeRunner } from '../integrations/CodeRunner.js';
 import type { ProjectReferenceLoader } from './ProjectReferenceLoader.js';
@@ -8,6 +8,8 @@ export type ProcessContext = {
     nativeApi?: NativeApi;
     /** Sets the dataset provider to be used for all dataset node calls. */
     datasetProvider?: DatasetProvider;
+    /** Provider for all MCP node functionality */
+    mcpProvider?: MCPProvider;
     /** The provider responsible for being able to play audio. Undefined if unsupported in this context. */
     audioProvider?: AudioProvider;
     /** Sets the tokenizer that will be used for all nodes. If unset, the default GptTokenizerTokenizer will be used. */
@@ -46,6 +48,8 @@ export type InternalProcessContext<T extends ChartNode = ChartNode> = ProcessCon
     graphInputs: Record<string, DataValue>;
     /** Outputs from the graph. A GraphOutputNode will set these. */
     graphOutputs: Record<string, DataValue>;
+    /** Stores the resolved output values of GraphInput nodes during execution, keyed by the node's data.id. */
+    graphInputNodeValues: Record<string, DataValue>;
     /** The tokenizer to use to tokenize all strings.s */
     tokenizer: Tokenizer;
     /** The current node being executed. */

package/dist/types/model/Project.d.ts

@@ -1,6 +1,7 @@
 import type { Opaque } from 'type-fest';
 import { type GraphId, type NodeGraph } from './NodeGraph.js';
 import { type PluginLoadSpec } from './PluginLoadSpec.js';
+import type { MCP } from '../integrations/mcp/MCPProvider.js';
 export type ProjectId = Opaque<string, 'ProjectId'>;
 export type DataId = Opaque<string, 'DataId'>;
 export type Project = {
@@ -17,6 +18,7 @@ export type ProjectMetadata = {
     description: string;
     mainGraphId?: GraphId;
     path?: string;
+    mcpServer?: MCP.Config;
 };
 /** A reference to another project file. Project references cannot be cyclic. */
 export type ProjectReference = {

package/dist/types/model/nodes/GetAllDatasetsNode.d.ts

@@ -1,4 +1,4 @@
-import type { ChartNode, EditorDefinition, InternalProcessContext, NodeInputDefinition, NodeOutputDefinition, NodeUIData, Outputs } from '../../index.js';
+import type { ChartNode, EditorDefinition, InternalProcessContext, NodeInputDefinition, NodeOutputDefinition, NodeUIData, Outputs, Inputs } from '../../index.js';
 import { NodeImpl } from '../NodeImpl.js';
 export type GetAllDatasetsNode = ChartNode<'getAllDatasets', GetAllDatasetsNodeData>;
 type GetAllDatasetsNodeData = {};
@@ -8,7 +8,7 @@ export declare class GetAllDatasetsNodeImpl extends NodeImpl<GetAllDatasetsNode>
     getOutputDefinitions(): NodeOutputDefinition[];
     static getUIData(): NodeUIData;
     getEditors(): EditorDefinition<GetAllDatasetsNode>[] | Promise<EditorDefinition<GetAllDatasetsNode>[]>;
-    process(context: InternalProcessContext): Promise<Outputs>;
+    process(_inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
 }
 export declare const getAllDatasetsNode: import("../NodeDefinition.js").NodeDefinition<GetAllDatasetsNode>;
 export {};

package/dist/types/model/nodes/MCPDiscoveryNode.d.ts

@@ -0,0 +1,9 @@
+import type { ChartNode } from '../NodeBase.js';
+import { type MCPBaseNodeData } from '../../integrations/mcp/MCPBase.js';
+type MCPDiscoveryNode = ChartNode<'mcpDiscovery', MCPDiscoveryNodeData>;
+export type MCPDiscoveryNodeData = MCPBaseNodeData & {
+    useToolsOutput?: boolean;
+    usePromptsOutput?: boolean;
+};
+export declare const mcpDiscoveryNode: import("../NodeDefinition.js").NodeDefinition<MCPDiscoveryNode>;
+export {};

package/dist/types/model/nodes/MCPGetPromptNode.d.ts

@@ -0,0 +1,23 @@
+import { type ChartNode, type NodeInputDefinition, type NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
+import { type Inputs, type Outputs } from '../GraphProcessor.js';
+import { type EditorDefinition, type InternalProcessContext } from '../../index.js';
+import { type MCPBaseNodeData } from '../../integrations/mcp/MCPBase.js';
+import type { RivetUIContext } from '../RivetUIContext.js';
+export type MCPGetPromptNode = ChartNode<'mcpGetPrompt', MCPGetPromptNodeData>;
+export type MCPGetPromptNodeData = MCPBaseNodeData & {
+    promptName: string;
+    promptArguments?: string;
+    usePromptNameInput?: boolean;
+    usePromptArgumentsInput?: boolean;
+};
+export declare class MCPGetPromptNodeImpl extends NodeImpl<MCPGetPromptNode> {
+    static create(): MCPGetPromptNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(context: RivetUIContext): Promise<EditorDefinition<MCPGetPromptNode>[]>;
+    getBody(context: RivetUIContext): string;
+    static getUIData(): NodeUIData;
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const mcpGetPromptNode: import("../NodeDefinition.js").NodeDefinition<MCPGetPromptNode>;

package/dist/types/model/nodes/MCPToolCallNode.d.ts

@@ -0,0 +1,26 @@
+import { type ChartNode, type NodeInputDefinition, type NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
+import { type Inputs, type Outputs } from '../GraphProcessor.js';
+import { type InternalProcessContext } from '../../index.js';
+import { type MCPBaseNodeData } from '../../integrations/mcp/MCPBase.js';
+import type { RivetUIContext } from '../RivetUIContext.js';
+import { type EditorDefinition } from '../EditorDefinition.js';
+export type MCPToolCallNode = ChartNode<'mcpToolCall', MCPToolCallNodeData>;
+export type MCPToolCallNodeData = MCPBaseNodeData & {
+    toolName: string;
+    toolArguments?: string;
+    toolCallId?: string;
+    useToolNameInput?: boolean;
+    useToolArgumentsInput?: boolean;
+    useToolCallIdInput?: boolean;
+};
+export declare class MCPToolCallNodeImpl extends NodeImpl<MCPToolCallNode> {
+    static create(): MCPToolCallNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(context: RivetUIContext): Promise<EditorDefinition<MCPToolCallNode>[]>;
+    getBody(context: RivetUIContext): string;
+    static getUIData(): NodeUIData;
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const mcpToolCallNode: import("../NodeDefinition.js").NodeDefinition<MCPToolCallNode>;

package/dist/types/model/nodes/ObjectNode.d.ts

@@ -2,6 +2,7 @@ import { type ChartNode, type NodeInputDefinition, type NodeOutputDefinition } f
 import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
 import { type DataValue } from '../DataValue.js';
 import { type EditorDefinition } from '../EditorDefinition.js';
+import type { InternalProcessContext } from '../ProcessContext.js';
 export type ObjectNode = ChartNode<'object', ObjectNodeData>;
 export type ObjectNodeData = {
     jsonTemplate: string;
@@ -12,7 +13,7 @@ export declare class ObjectNodeImpl extends NodeImpl<ObjectNode> {
     getOutputDefinitions(): NodeOutputDefinition[];
     getEditors(): EditorDefinition<ObjectNode>[];
     static getUIData(): NodeUIData;
-    interpolate(baseString: string, values: Record<string, any>): string;
-    process(inputs: Record<string, DataValue>): Promise<Record<string, DataValue>>;
+    interpolate(baseString: string, values: Record<string, any>, graphInputNodeValues?: Record<string, DataValue>, contextValues?: Record<string, DataValue>): string;
+    process(inputs: Record<string, DataValue>, context: InternalProcessContext): Promise<Record<string, DataValue>>;
 }
 export declare const objectNode: import("../NodeDefinition.js").NodeDefinition<ObjectNode>;

package/dist/types/model/nodes/TextNode.d.ts

@@ -5,6 +5,7 @@ import { type EditorDefinition, type NodeBodySpec } from '../../index.js';
 export type TextNode = ChartNode<'text', TextNodeData>;
 export type TextNodeData = {
     text: string;
+    normalizeLineEndings?: boolean;
 };
 export declare class TextNodeImpl extends NodeImpl<TextNode> {
     static create(): TextNode;
@@ -12,7 +13,7 @@ export declare class TextNodeImpl extends NodeImpl<TextNode> {
     getOutputDefinitions(): NodeOutputDefinition[];
     getEditors(): EditorDefinition<TextNode>[];
     getBody(): string | NodeBodySpec | undefined;
-    process(inputs: Record<string, DataValue>): Promise<Record<string, DataValue>>;
+    process(inputs: Record<string, DataValue>, context: any): Promise<Record<string, DataValue>>;
     static getUIData(): NodeUIData;
 }
 export declare const textNode: import("../NodeDefinition.js").NodeDefinition<TextNode>;

package/dist/types/plugins/anthropic/anthropic.d.ts

@@ -87,6 +87,22 @@ export declare const anthropicModels: {
         };
         displayName: string;
     };
+    'claude-sonnet-4-20250514': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'claude-opus-4-20250514': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
 };
 export type AnthropicModels = keyof typeof anthropicModels;
 export declare const anthropicModelOptions: {
@@ -161,6 +177,7 @@ export type ChatMessageOptions = {
         input_schema: object;
     }[];
     beta?: string;
+    additionalHeaders?: Record<string, string>;
 };
 export type ChatCompletionOptions = {
     apiEndpoint: string;
@@ -174,6 +191,7 @@ export type ChatCompletionOptions = {
     top_k?: number;
     signal?: AbortSignal;
     stream?: boolean;
+    additionalHeaders?: Record<string, string>;
 };
 export type ChatCompletionChunk = {
     completion: string;
@@ -287,9 +305,9 @@ export type ChatMessageCitation = {
     page_number: number;
     end_page_number: number;
 };
-export declare function streamChatCompletions({ apiEndpoint, apiKey, signal, ...rest }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;
-export declare function callMessageApi({ apiEndpoint, apiKey, signal, tools, beta, ...rest }: ChatMessageOptions): Promise<ChatMessageResponse>;
-export declare function streamMessageApi({ apiEndpoint, apiKey, signal, beta, ...rest }: ChatMessageOptions): AsyncGenerator<ChatMessageChunk>;
+export declare function streamChatCompletions({ apiEndpoint, apiKey, signal, additionalHeaders, ...rest }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;
+export declare function callMessageApi({ apiEndpoint, apiKey, signal, tools, beta, additionalHeaders, ...rest }: ChatMessageOptions): Promise<ChatMessageResponse>;
+export declare function streamMessageApi({ apiEndpoint, apiKey, signal, beta, additionalHeaders, ...rest }: ChatMessageOptions): AsyncGenerator<ChatMessageChunk>;
 export declare class AnthropicError extends Error {
     readonly response: Response;
     readonly responseJson: unknown;

package/dist/types/plugins/anthropic/nodes/ChatAnthropicNode.d.ts

@@ -13,6 +13,10 @@ export type ChatAnthropicNodeConfigData = {
     endpoint?: string;
     overrideModel?: string;
     enableCitations?: boolean;
+    headers?: {
+        key: string;
+        value: string;
+    }[];
 };
 export type ChatAnthropicNodeData = ChatAnthropicNodeConfigData & {
     useModelInput: boolean;
@@ -25,6 +29,7 @@ export type ChatAnthropicNodeData = ChatAnthropicNodeConfigData & {
     useStopInput: boolean;
     useEndpointInput: boolean;
     useOverrideModelInput: boolean;
+    useHeadersInput?: boolean;
     /** Given the same set of inputs, return the same output without hitting GPT */
     cache: boolean;
     useAsGraphPartialOutput?: boolean;

package/dist/types/plugins/google/google.d.ts

@@ -1,4 +1,4 @@
-import { type Content, type FunctionCall, type Tool } from '@google/generative-ai';
+import { type Content, type FunctionCall, type Tool } from '@google/genai';
 export type GoogleModelDeprecated = {
     maxTokens: number;
     cost: {
@@ -27,6 +27,14 @@ export declare const googleModelsDeprecated: {
 };
 export type GoogleModelsDeprecated = keyof typeof googleModelsDeprecated;
 export declare const generativeAiGoogleModels: {
+    'gemini-2.5-flash-preview-04-17': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
     'gemini-2.0-flash-001': {
         maxTokens: number;
         cost: {
@@ -138,6 +146,8 @@ export type StreamGenerativeAiOptions = {
     topK: number | undefined;
     signal?: AbortSignal;
     tools: Tool[] | undefined;
+    thinkingBudget?: number;
+    additionalHeaders?: Record<string, string>;
 };
-export declare function streamGenerativeAi({ apiKey, model, systemPrompt, prompt, maxOutputTokens, temperature, topP, topK, signal, tools, }: StreamGenerativeAiOptions): AsyncGenerator<ChatCompletionChunk>;
+export declare function streamGenerativeAi({ apiKey, model, systemPrompt, prompt, maxOutputTokens, temperature, topP, topK, signal, tools, thinkingBudget, additionalHeaders, }: StreamGenerativeAiOptions): AsyncGenerator<ChatCompletionChunk>;
 export declare function streamChatCompletions({ project, location, applicationCredentials, model, signal, max_output_tokens, temperature, top_p, top_k, prompt, }: ChatCompletionOptions): AsyncGenerator<ChatCompletionChunk>;

package/dist/types/plugins/google/nodes/ChatGoogleNode.d.ts

@@ -8,6 +8,11 @@ export type ChatGoogleNodeConfigData = {
     top_p?: number;
     top_k?: number;
     maxTokens: number;
+    thinkingBudget: number | undefined;
+    headers?: {
+        key: string;
+        value: string;
+    }[];
 };
 export type ChatGoogleNodeData = ChatGoogleNodeConfigData & {
     useModelInput: boolean;
@@ -17,6 +22,8 @@ export type ChatGoogleNodeData = ChatGoogleNodeConfigData & {
     useUseTopPInput: boolean;
     useMaxTokensInput: boolean;
     useToolCalling: boolean;
+    useThinkingBudgetInput: boolean;
+    useHeadersInput?: boolean;
     /** Given the same set of inputs, return the same output without hitting GPT */
     cache: boolean;
     useAsGraphPartialOutput?: boolean;

package/dist/types/utils/interpolation.d.ts

@@ -1,4 +1,9 @@
+import type { DataValue } from '../model/DataValue.js';
 export declare const TOKEN_MATCH_REGEX: RegExp;
 export declare const ESCAPED_TOKEN_REGEX: RegExp;
-export declare function interpolate(baseString: string, values: Record<string, string>): string;
+export declare const ESCAPED_ESCAPED_TOKEN_REGEX: RegExp;
+export declare function unwrapPotentialDataValue(value: any): any;
+export declare function resolveExpressionRawValue(source: Record<string, any> | undefined, expression: string, sourceType: 'graphInputs' | 'context'): any | undefined;
+export declare function resolveExpressionToString(source: Record<string, any> | undefined, expression: string, sourceType: 'graphInputs' | 'context'): string | undefined;
+export declare function interpolate(template: string, variables: Record<string, DataValue | string | undefined>, graphInputValues?: Record<string, DataValue>, contextValues?: Record<string, DataValue>): string;
 export declare function extractInterpolationVariables(template: string): string[];

package/dist/types/utils/openai.d.ts

@@ -196,6 +196,30 @@ export declare const openaiModels: {
         };
         displayName: string;
     };
+    'gpt-4.1': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    o3: {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
+    'o4-mini': {
+        maxTokens: number;
+        cost: {
+            prompt: number;
+            completion: number;
+        };
+        displayName: string;
+    };
     'local-model': {
         maxTokens: number;
         cost: {

package/package.json

@@ -2,7 +2,7 @@
   "name": "@alpic80/rivet-core",
   "license": "MIT",
   "repository": "https://github.com/castortech/rivet",
-  "version": "1.24.0-aidon.5",
+  "version": "1.24.2-aidon.10",
   "packageManager": "yarn@3.5.0",
   "main": "dist/cjs/bundle.cjs",
   "module": "dist/esm/index.js",
@@ -46,9 +46,9 @@
   "dependencies": {
     "@gentrace/core": "^2.2.5",
     "@google-cloud/vertexai": "^0.1.3",
-    "@google/generative-ai": "^0.24.0",
-    "@huggingface/inference": "^2.6.4",
-    "@ironclad/rivet-core": "npm:@alpic80/rivet-core@1.24.0-aidon.5",
+    "@google/genai": "^0.12.0",
+    "@huggingface/inference": "^4.13.0",
+    "@ironclad/rivet-core": "npm:@alpic80/rivet-core@1.24.2-aidon.10",
     "assemblyai": "^4.6.0",
     "autoevals": "^0.0.26",
     "cron-parser": "^4.9.0",
@@ -56,12 +56,12 @@
     "emittery": "^1.0.1",
     "emittery-0-13": "npm:emittery@^0.13.1",
     "gpt-tokenizer": "^2.1.2",
-    "jsonpath-plus": "^10.2.0",
+    "jsonpath-plus": "^10.3.0",
     "lodash-es": "^4.17.21",
     "mdast-util-gfm-table": "^2.0.0",
     "mdast-util-to-markdown": "^2.1.2",
     "minimatch": "^9.0.3",
-    "nanoid": "^3.3.6",
+    "nanoid": "^3.3.8",
     "openai": "^4.28.4",
     "p-queue": "^7.4.1",
     "p-queue-6": "npm:p-queue@^6.0.0",
@@ -83,7 +83,7 @@
     "@types/yaml": "^1.9.7",
     "@typescript-eslint/eslint-plugin": "^8.24.0",
     "@typescript-eslint/parser": "^8.24.0",
-    "esbuild": "^0.19.5",
+    "esbuild": "^0.25.12",
     "eslint": "^9.20.1",
     "eslint-import-resolver-typescript": "^3.6.1",
     "eslint-plugin-import": "^2.31.0",