@alpic80/rivet-core 1.19.1-aidon.1

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

@@ -0,0 +1,55 @@
+import type { Inputs, Outputs } from './GraphProcessor.js';
+import type { ChartNode, NodeConnection, NodeId, NodeInputDefinition, NodeOutputDefinition } from './NodeBase.js';
+import type { Project } from './Project.js';
+import type { InternalProcessContext } from './ProcessContext.js';
+import type { EditorDefinition } from './EditorDefinition.js';
+import type { NodeBodySpec } from './NodeBodySpec.js';
+import type { RivetUIContext } from './RivetUIContext.js';
+export interface PluginNodeImpl<T extends ChartNode> {
+    getInputDefinitions(data: T['data'], connections: NodeConnection[], nodes: Record<NodeId, ChartNode>, project: Project): NodeInputDefinition[];
+    getOutputDefinitions(data: T['data'], connections: NodeConnection[], nodes: Record<NodeId, ChartNode>, project: Project): NodeOutputDefinition[];
+    process(data: T['data'], inputData: Inputs, context: InternalProcessContext): Promise<Outputs>;
+    getEditors(data: T['data'], context: RivetUIContext): EditorDefinition<T>[] | Promise<EditorDefinition<T>[]>;
+    getBody(data: T['data'], context: RivetUIContext): NodeBody | Promise<NodeBody>;
+    create(): T;
+    getUIData(context: RivetUIContext): NodeUIData | Promise<NodeUIData>;
+}
+export declare abstract class NodeImpl<T extends ChartNode, Type extends T['type'] = T['type']> {
+    readonly chartNode: T;
+    constructor(chartNode: T);
+    get id(): string;
+    get type(): Type;
+    get title(): string;
+    get visualData(): {
+        x: number;
+        y: number;
+    };
+    get data(): T['data'];
+    abstract getInputDefinitions(connections: NodeConnection[], nodes: Record<NodeId, ChartNode>, project: Project): NodeInputDefinition[];
+    abstract getOutputDefinitions(connections: NodeConnection[], nodes: Record<NodeId, ChartNode>, project: Project): NodeOutputDefinition[];
+    abstract process(inputData: Inputs, context: InternalProcessContext): Promise<Outputs>;
+    getEditors(_context: RivetUIContext): EditorDefinition<T>[] | Promise<EditorDefinition<T>[]>;
+    getBody(_context: RivetUIContext): NodeBody | Promise<NodeBody>;
+}
+export type NodeBody = string | NodeBodySpec | NodeBodySpec[] | undefined;
+export declare class PluginNodeImplClass<T extends ChartNode, Type extends T['type'] = T['type']> extends NodeImpl<T, Type> {
+    readonly impl: PluginNodeImpl<T>;
+    constructor(chartNode: T, impl: PluginNodeImpl<T>);
+    getInputDefinitions(connections: NodeConnection[], nodes: Record<NodeId, ChartNode>, project: Project): NodeInputDefinition[];
+    getOutputDefinitions(connections: NodeConnection[], nodes: Record<NodeId, ChartNode>, project: Project): NodeOutputDefinition[];
+    process(inputData: Inputs, context: InternalProcessContext): Promise<Outputs>;
+    getEditors(context: RivetUIContext): EditorDefinition<T>[] | Promise<EditorDefinition<T>[]>;
+    getBody(context: RivetUIContext): NodeBody | Promise<NodeBody>;
+}
+export type NodeUIData = {
+    contextMenuTitle?: string;
+    infoBoxTitle?: string;
+    infoBoxBody?: string;
+    infoBoxImageUri?: string;
+    group?: string | string[];
+};
+export type NodeImplConstructor<T extends ChartNode> = {
+    new (chartNode: T, pluginImpl: PluginNodeImpl<T> | undefined): NodeImpl<T>;
+    create(pluginImpl?: PluginNodeImpl<T>): T;
+    getUIData(context: RivetUIContext): NodeUIData | Promise<NodeUIData>;
+};

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

@@ -0,0 +1,24 @@
+import { type ChartNode, type NodeImplConstructor, type NodeImpl } from '../index.js';
+import type { NodeDefinition, PluginNodeDefinition } from './NodeDefinition.js';
+import { type RivetPlugin } from './RivetPlugin.js';
+export declare class NodeRegistration<NodeTypes extends string = never, Nodes extends ChartNode = never> {
+    #private;
+    NodesType: Nodes;
+    NodeTypesType: NodeTypes;
+    register<T extends ChartNode>(definition: NodeDefinition<T>, plugin?: RivetPlugin): NodeRegistration<NodeTypes | T['type'], Nodes | T>;
+    registerPluginNode<T extends ChartNode>(definition: PluginNodeDefinition<T>, plugin: RivetPlugin): NodeRegistration<NodeTypes | T['type'], Nodes | T>;
+    registerPlugin(plugin: RivetPlugin): void;
+    create<T extends NodeTypes>(type: T): Extract<Nodes, {
+        type: T;
+    }>;
+    createDynamic(type: string): ChartNode;
+    createImpl<T extends Nodes>(node: T): NodeImpl<T>;
+    createDynamicImpl(node: ChartNode): NodeImpl<ChartNode>;
+    getDisplayName<T extends NodeTypes>(type: T): string;
+    getDynamicDisplayName(type: string): string;
+    isRegistered(type: NodeTypes): boolean;
+    getNodeTypes(): NodeTypes[];
+    getNodeConstructors(): NodeImplConstructor<ChartNode>[];
+    getPluginFor(type: string): RivetPlugin | undefined;
+    getPlugins(): RivetPlugin[];
+}

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

@@ -0,0 +1,84 @@
+import { NodeRegistration } from './NodeRegistration.js';
+export * from './nodes/UserInputNode.js';
+export * from './nodes/TextNode.js';
+export * from './nodes/ChatNode.js';
+export * from './nodes/PromptNode.js';
+export * from './nodes/ExtractRegexNode.js';
+export * from './nodes/CodeNode.js';
+export * from './nodes/MatchNode.js';
+export * from './nodes/IfNode.js';
+export * from './nodes/ReadDirectoryNode.js';
+export * from './nodes/ReadFileNode.js';
+export * from './nodes/IfElseNode.js';
+export * from './nodes/ChunkNode.js';
+export * from './nodes/GraphInputNode.js';
+export * from './nodes/GraphOutputNode.js';
+export * from './nodes/SubGraphNode.js';
+export * from './nodes/ArrayNode.js';
+export * from './nodes/ExtractJsonNode.js';
+export * from './nodes/ExtractYamlNode.js';
+export * from './nodes/AssemblePromptNode.js';
+export * from './nodes/LoopControllerNode.js';
+export * from './nodes/TrimChatMessagesNode.js';
+export * from './nodes/ExternalCallNode.js';
+export * from './nodes/ExtractObjectPathNode.js';
+export * from './nodes/RaiseEventNode.js';
+export * from './nodes/ContextNode.js';
+export * from './nodes/CoalesceNode.js';
+export * from './nodes/PassthroughNode.js';
+export * from './nodes/PopNode.js';
+export * from './nodes/SetGlobalNode.js';
+export * from './nodes/GetGlobalNode.js';
+export * from './nodes/WaitForEventNode.js';
+export * from './nodes/GptFunctionNode.js';
+export * from './nodes/ToYamlNode.js';
+export * from './nodes/GetEmbeddingNode.js';
+export * from './nodes/VectorStoreNode.js';
+export * from './nodes/VectorNearestNeighborsNode.js';
+export * from './nodes/HashNode.js';
+export * from './nodes/AbortGraphNode.js';
+export * from './nodes/RaceInputsNode.js';
+export * from './nodes/ToJsonNode.js';
+export * from './nodes/JoinNode.js';
+export * from './nodes/FilterNode.js';
+export * from './nodes/ObjectNode.js';
+export * from './nodes/BooleanNode.js';
+export * from './nodes/CompareNode.js';
+export * from './nodes/EvaluateNode.js';
+export * from './nodes/NumberNode.js';
+export * from './nodes/RandomNumberNode.js';
+export * from './nodes/ShuffleNode.js';
+export * from './nodes/CommentNode.js';
+export * from './nodes/ImageNode.js';
+export * from './nodes/AudioNode.js';
+export * from './nodes/HttpCallNode.js';
+export * from './nodes/DelayNode.js';
+export * from './nodes/AppendToDatasetNode.js';
+export * from './nodes/CreateDatasetNode.js';
+export * from './nodes/LoadDatasetNode.js';
+export * from './nodes/GetAllDatasetsNode.js';
+export * from './nodes/SplitNode.js';
+export * from './nodes/DatasetNearestNeigborsNode.js';
+export * from './nodes/GetDatasetRowNode.js';
+export * from './nodes/SliceNode.js';
+export * from './nodes/ExtractMarkdownCodeBlocksNode.js';
+export * from './nodes/AssembleMessageNode.js';
+export * from './nodes/URLReferenceNode.js';
+export * from './nodes/DestructureNode.js';
+export * from './nodes/ReplaceDatasetNode.js';
+export * from './nodes/ListGraphsNode.js';
+export * from './nodes/GraphReferenceNode.js';
+export * from './nodes/CallGraphNode.js';
+export * from './nodes/DelegateFunctionCallNode.js';
+export * from './nodes/PlayAudioNode.js';
+export * from './nodes/CallGraphNode.js';
+export declare const registerBuiltInNodes: (registry: NodeRegistration) => NodeRegistration<"number" | "boolean" | "object" | "pop" | "join" | "slice" | "filter" | "match" | "split" | "image" | "audio" | "code" | "array" | "userInput" | "prompt" | "text" | "chat" | "extractRegex" | "if" | "readDirectory" | "readFile" | "ifElse" | "chunk" | "graphInput" | "graphOutput" | "subGraph" | "extractJson" | "assemblePrompt" | "extractYaml" | "loopController" | "trimChatMessages" | "externalCall" | "setGlobal" | "raiseEvent" | "getGlobal" | "abortGraph" | "extractObjectPath" | "context" | "coalesce" | "passthrough" | "waitForEvent" | "gptFunction" | "toYaml" | "getEmbedding" | "vectorStore" | "vectorNearestNeighbors" | "hash" | "raceInputs" | "toJson" | "compare" | "evaluate" | "randomNumber" | "shuffle" | "comment" | "httpCall" | "delay" | "appendToDataset" | "createDataset" | "loadDataset" | "getAllDatasets" | "datasetNearestNeighbors" | "getDatasetRow" | "extractMarkdownCodeBlocks" | "assembleMessage" | "urlReference" | "destructure" | "replaceDataset" | "listGraphs" | "graphReference" | "callGraph" | "delegateFunctionCall" | "playAudio", 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/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/GptFunctionNode.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/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>;
+declare let globalRivetNodeRegistry: NodeRegistration<"number" | "boolean" | "object" | "pop" | "join" | "slice" | "filter" | "match" | "split" | "image" | "audio" | "code" | "array" | "userInput" | "prompt" | "text" | "chat" | "extractRegex" | "if" | "readDirectory" | "readFile" | "ifElse" | "chunk" | "graphInput" | "graphOutput" | "subGraph" | "extractJson" | "assemblePrompt" | "extractYaml" | "loopController" | "trimChatMessages" | "externalCall" | "setGlobal" | "raiseEvent" | "getGlobal" | "abortGraph" | "extractObjectPath" | "context" | "coalesce" | "passthrough" | "waitForEvent" | "gptFunction" | "toYaml" | "getEmbedding" | "vectorStore" | "vectorNearestNeighbors" | "hash" | "raceInputs" | "toJson" | "compare" | "evaluate" | "randomNumber" | "shuffle" | "comment" | "httpCall" | "delay" | "appendToDataset" | "createDataset" | "loadDataset" | "getAllDatasets" | "datasetNearestNeighbors" | "getDatasetRow" | "extractMarkdownCodeBlocks" | "assembleMessage" | "urlReference" | "destructure" | "replaceDataset" | "listGraphs" | "graphReference" | "callGraph" | "delegateFunctionCall" | "playAudio", 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/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/GptFunctionNode.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/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>;
+export { globalRivetNodeRegistry };
+export type BuiltInNodes = typeof globalRivetNodeRegistry.NodesType;
+export type BuiltInNodeType = typeof globalRivetNodeRegistry.NodeTypesType;
+export type NodeOfType<T extends BuiltInNodeType> = Extract<BuiltInNodes, {
+    type: T;
+}>;
+/** Resets the global node registry to a fresh one with only built-in nodes registered. */
+export declare function resetGlobalRivetNodeRegistry(): void;

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

@@ -0,0 +1,17 @@
+export type BuiltInPluginLoadSpec = {
+    type: 'built-in';
+    id: string;
+    name: string;
+};
+export type URIPluginLoadSpec = {
+    type: 'uri';
+    id: string;
+    uri: string;
+};
+export type PackagePluginLoadSpec = {
+    type: 'package';
+    id: string;
+    package: string;
+    tag: string;
+};
+export type PluginLoadSpec = URIPluginLoadSpec | BuiltInPluginLoadSpec | PackagePluginLoadSpec;

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

@@ -0,0 +1,69 @@
+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 } from '../index.js';
+import type { Tokenizer } from '../integrations/Tokenizer.js';
+export type ProcessContext = {
+    settings: Settings;
+    nativeApi?: NativeApi;
+    /** Sets the dataset provider to be used for all dataset node calls. */
+    datasetProvider?: DatasetProvider;
+    /** 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. */
+    tokenizer?: Tokenizer;
+    /**
+     * If implemented, chat nodes will first call this to resolve their configured endpoint to a final endpoint.
+     * You can use this for adding auth headers, or to load balance between multiple endpoints.
+     */
+    getChatNodeEndpoint?: (configuredEndpoint: string, configuredModel: string) => ChatNodeEndpointInfo | Promise<ChatNodeEndpointInfo>;
+};
+export type ChatNodeEndpointInfo = {
+    endpoint: string;
+    headers: Record<string, string>;
+};
+export type ProcessId = Opaque<string, 'ProcessId'>;
+export type InternalProcessContext<T extends ChartNode = ChartNode> = ProcessContext & {
+    /** The executor that is running the current processor. */
+    executor: 'nodejs' | 'browser';
+    /** The project being executed. */
+    project: Project;
+    /** A signal that can be used when abort() is called on the GraphProcessor to abort the node's execution. */
+    signal: AbortSignal;
+    /** A unique ID for this specific execution of the node. */
+    processId: ProcessId;
+    /** Context values that are accessible on graphs and all subgraphs. */
+    contextValues: Record<string, DataValue>;
+    /** Inputs that were passed to the curent graph. Used for GraphInputNode. */
+    graphInputs: Record<string, DataValue>;
+    /** Outputs from the graph. A GraphOutputNode will set these. */
+    graphOutputs: Record<string, DataValue>;
+    /** The tokenizer to use to tokenize all strings.s */
+    tokenizer: Tokenizer;
+    /** The current node being executed. */
+    node: T;
+    /** For internal and advanced cases, gets the arbitrary data attached to the node during graph execution. */
+    attachedData: AttachedNodeData;
+    /** Raises a user event that can be listened for on the GraphProcessor. */
+    raiseEvent: (eventName: string, data: DataValue | undefined) => void;
+    waitEvent: (eventName: string) => Promise<DataValue | undefined>;
+    /** External functions that have been defined on the GraphProcessor (or its parent). */
+    externalFunctions: Record<string, ExternalFunction>;
+    /** Global cache shared by all nodes, is present for the entire execution of a graph (and shared in subgraphs). */
+    executionCache: Map<string, unknown>;
+    /** Call when the node has partial data but has not finished execution yet. */
+    onPartialOutputs?: (outputs: Outputs) => void;
+    /** Creates a subprocessor, for executing subgraphs. */
+    createSubProcessor: (subGraphId: GraphId | undefined, options?: {
+        signal?: AbortSignal;
+    }) => GraphProcessor;
+    /** Like context, but variables that are set during the run of the graph and can be read during the graph. Shared among all graphs and subgraphs. */
+    getGlobal: (id: string) => ScalarOrArrayDataValue | undefined;
+    /** Like context, but variables that are set during the run of the graph and can be read during the graph. Shared among all graphs and subgraphs. */
+    setGlobal: (id: string, value: ScalarOrArrayDataValue) => void;
+    waitForGlobal: (id: string) => Promise<ScalarOrArrayDataValue>;
+    /** Logs to GraphProcessor's trace event. */
+    trace: (message: string) => void;
+    /** Aborts the current graph, if there is an error, the graph is error aborted, and if undefined, then it is simply early-exited. */
+    abortGraph: (error?: Error | string) => void;
+    /** Gets a string plugin config value from the settings, falling back to a specified environment variable if set. */
+    getPluginConfig(name: string): string | undefined;
+};

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

@@ -0,0 +1,17 @@
+import type { Opaque } from 'type-fest';
+import { type GraphId, type NodeGraph } from './NodeGraph.js';
+import { type PluginLoadSpec } from './PluginLoadSpec.js';
+export type ProjectId = Opaque<string, 'ProjectId'>;
+export type DataId = Opaque<string, 'DataId'>;
+export type Project = {
+    metadata: {
+        id: ProjectId;
+        title: string;
+        description: string;
+        mainGraphId?: GraphId;
+        path?: string;
+    };
+    plugins?: PluginLoadSpec[];
+    graphs: Record<GraphId, NodeGraph>;
+    data?: Record<DataId, string>;
+};

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

@@ -0,0 +1,45 @@
+import { type ChartNode } from './NodeBase.js';
+import type { PluginNodeDefinition } from './NodeDefinition.js';
+export type RivetPlugin = {
+    /** The unique identifier of the plugin. Should be unique across all plugins. */
+    id: string;
+    /** The display name of the plugin - what is shown in the UI fr the plugin. */
+    name?: string;
+    /** Registers new nodes for the plugin to add. */
+    register?: (register: <T extends ChartNode>(definition: PluginNodeDefinition<T>) => void) => void;
+    /** The available configuration items and their specification, for configuring a plugin in the UI. */
+    configSpec?: RivetPluginConfigSpecs;
+    /** Defines additional context menu groups that the plugin adds. */
+    contextMenuGroups?: Array<{
+        id: string;
+        label: string;
+    }>;
+};
+export type RivetPluginConfigSpecs = Record<string, PluginConfigurationSpec>;
+export type PluginConfigurationSpecBase<T> = {
+    /** The type of the config value, how it should show as an editor in the UI. */
+    type: string;
+    /** The default value of the config item if unset. */
+    default?: T;
+    /** A description to show in the UI for the config setting. */
+    description?: string;
+    /** The label of the setting in the UI. */
+    label: string;
+};
+export type StringPluginConfigurationSpec = {
+    type: 'string';
+    default?: string;
+    label: string;
+    description?: string;
+    pullEnvironmentVariable?: true | string;
+    helperText?: string;
+};
+export type SecretPluginConfigurationSpec = {
+    type: 'secret';
+    default?: string;
+    label: string;
+    description?: string;
+    pullEnvironmentVariable?: true | string;
+    helperText?: string;
+};
+export type PluginConfigurationSpec = StringPluginConfigurationSpec | SecretPluginConfigurationSpec | PluginConfigurationSpecBase<number> | PluginConfigurationSpecBase<boolean>;

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

@@ -0,0 +1,18 @@
+import { type ChartNode, type DatasetProvider, type NodeGraph, type Project, type Settings } from '../index.js';
+/** Context accessible to UI functions such as getEditors() and getBody(). */
+export type RivetUIContext = {
+    /** The settings configured in the UI. */
+    settings: Settings;
+    /** The selected executor. */
+    executor: 'nodejs' | 'browser';
+    /** The provider that can control datasets. */
+    datasetProvider: DatasetProvider;
+    /** The current project loaded in the UI. */
+    project: Project;
+    /** The current graph selected in the UI. */
+    graph?: NodeGraph;
+    /** The current node selected in the UI. */
+    node?: ChartNode;
+    /** Gets a string plugin config value from the settings, falling back to a specified environment variable if set. */
+    getPluginConfig(name: string): string | undefined;
+};

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

@@ -0,0 +1,15 @@
+export interface Settings<PluginSettings = Record<string, Record<string, unknown>>> {
+    recordingPlaybackLatency?: number;
+    /** Configurable settings that a plugin can get and set. Settings can be available in the settings modal and are stored  */
+    pluginSettings?: PluginSettings;
+    /** A plugin can request environment variables to configure itself. Those can be populated here. */
+    pluginEnv?: {
+        [key: string]: string | undefined;
+    };
+    openAiKey?: string;
+    openAiOrganization?: string;
+    openAiEndpoint?: string;
+    /** Timeout in milliseconds before retrying a chat node call. */
+    chatNodeTimeout?: number;
+    chatNodeHeaders?: Record<string, string>;
+}

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

@@ -0,0 +1,22 @@
+import type { ChartNode, NodeOutputDefinition, NodeInputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
+import type { Inputs, Outputs } from '../GraphProcessor.js';
+import type { InternalProcessContext } from '../ProcessContext.js';
+import type { EditorDefinition } from '../../index.js';
+export type AbortGraphNode = ChartNode<'abortGraph', AbortGraphNodeData>;
+export type AbortGraphNodeData = {
+    /** Did the graph abort, but it's a success? Use this for early-exit instead of "error abort". */
+    successfully: boolean;
+    useSuccessfullyInput?: boolean;
+    errorMessage?: string;
+};
+export declare class AbortGraphNodeImpl extends NodeImpl<AbortGraphNode> {
+    static create(): AbortGraphNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(): EditorDefinition<AbortGraphNode>[];
+    getBody(): string | undefined;
+    static getUIData(): NodeUIData;
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const abortGraphNode: import("../NodeDefinition.js").NodeDefinition<AbortGraphNode>;

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

@@ -0,0 +1,21 @@
+import type { ChartNode, NodeInputDefinition, NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
+import type { InternalProcessContext } from '../ProcessContext.js';
+import type { DatasetId } from '../Dataset.js';
+import type { EditorDefinition } from '../EditorDefinition.js';
+import type { Inputs, Outputs } from '../GraphProcessor.js';
+export type AppendToDatasetNode = ChartNode<'appendToDataset', AppendToDatasetNodeData>;
+type AppendToDatasetNodeData = {
+    datasetId: DatasetId;
+    useDatasetIdInput?: boolean;
+};
+export declare class AppendToDatasetNodeImpl extends NodeImpl<AppendToDatasetNode> {
+    static create(): AppendToDatasetNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    static getUIData(): NodeUIData;
+    getEditors(): EditorDefinition<AppendToDatasetNode>[] | Promise<EditorDefinition<AppendToDatasetNode>[]>;
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const appendToDatasetNode: import("../NodeDefinition.js").NodeDefinition<AppendToDatasetNode>;
+export {};

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

@@ -0,0 +1,20 @@
+import { type ChartNode, type NodeConnection, type NodeInputDefinition, type NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeBody, type NodeUIData } from '../NodeImpl.js';
+import { type Inputs, type Outputs } from '../GraphProcessor.js';
+import { type EditorDefinition } from '../EditorDefinition.js';
+export type ArrayNode = ChartNode<'array', ArrayNodeData>;
+export type ArrayNodeData = {
+    flatten?: boolean;
+    flattenDeep?: boolean;
+};
+export declare class ArrayNodeImpl extends NodeImpl<ArrayNode> {
+    #private;
+    static create(): ArrayNode;
+    getInputDefinitions(connections: NodeConnection[]): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(): EditorDefinition<ArrayNode>[];
+    static getUIData(): NodeUIData;
+    getBody(): NodeBody;
+    process(inputs: Inputs): Promise<Outputs>;
+}
+export declare const arrayNode: import("../NodeDefinition.js").NodeDefinition<ArrayNode>;

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

@@ -0,0 +1,23 @@
+import { type ChartNode, type NodeConnection, type NodeInputDefinition, type NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeBody, type NodeUIData } from '../NodeImpl.js';
+import { type Inputs, type Outputs } from '../GraphProcessor.js';
+import type { EditorDefinition } from '../EditorDefinition.js';
+import type { RivetUIContext } from '../RivetUIContext.js';
+export type AssembleMessageNode = ChartNode<'assembleMessage', AssembleMessageNodeData>;
+export type AssembleMessageNodeData = {
+    type: 'system' | 'user' | 'assistant' | 'function';
+    useTypeInput: boolean;
+    toolCallId: string;
+    useToolCallIdInput?: boolean;
+};
+export declare class AssembleMessageNodeImpl extends NodeImpl<AssembleMessageNode> {
+    #private;
+    static create(): AssembleMessageNode;
+    getInputDefinitions(connections: NodeConnection[]): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(): EditorDefinition<AssembleMessageNode>[];
+    static getUIData(): NodeUIData;
+    getBody(_context: RivetUIContext): NodeBody | Promise<NodeBody>;
+    process(inputs: Inputs): Promise<Outputs>;
+}
+export declare const assembleMessageNode: import("../NodeDefinition.js").NodeDefinition<AssembleMessageNode>;

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

@@ -0,0 +1,20 @@
+import { type ChartNode, type NodeConnection, 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 } from '../EditorDefinition.js';
+import type { RivetUIContext } from '../RivetUIContext.js';
+import type { InternalProcessContext } from '../ProcessContext.js';
+export type AssemblePromptNode = ChartNode<'assemblePrompt', AssemblePromptNodeData>;
+export type AssemblePromptNodeData = {
+    computeTokenCount?: boolean;
+};
+export declare class AssemblePromptNodeImpl extends NodeImpl<AssemblePromptNode> {
+    #private;
+    static create(): AssemblePromptNode;
+    getInputDefinitions(connections: NodeConnection[]): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    static getUIData(): NodeUIData;
+    getEditors(_context: RivetUIContext): EditorDefinition<AssemblePromptNode>[];
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const assemblePromptNode: import("../NodeDefinition.js").NodeDefinition<AssemblePromptNode>;

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

@@ -0,0 +1,20 @@
+import { type ChartNode, type NodeInputDefinition, type NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
+import { type DataRef, type EditorDefinition, type Inputs, type InternalProcessContext, type Outputs } from '../../index.js';
+export type AudioNode = ChartNode<'audio', AudioNodeData>;
+type AudioNodeData = {
+    data?: DataRef;
+    useDataInput: boolean;
+    mediaType?: 'audio/wav' | 'audio/mp3' | 'audio/ogg';
+    useMediaTypeInput: boolean;
+};
+export declare class AudioNodeImpl extends NodeImpl<AudioNode> {
+    static create(): AudioNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(): EditorDefinition<AudioNode>[];
+    static getUIData(): NodeUIData;
+    process(inputData: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const audioNode: import("../NodeDefinition.js").NodeDefinition<AudioNode>;
+export {};

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

@@ -0,0 +1,19 @@
+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 } from '../../index.js';
+export type BooleanNode = ChartNode<'boolean', BooleanNodeData>;
+export type BooleanNodeData = {
+    value?: boolean;
+    useValueInput?: boolean;
+};
+export declare class BooleanNodeImpl extends NodeImpl<BooleanNode> {
+    static create(): BooleanNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    getEditors(): EditorDefinition<BooleanNode>[];
+    getBody(): string | undefined;
+    static getUIData(): NodeUIData;
+    process(inputs: Inputs): Promise<Outputs>;
+}
+export declare const booleanNode: import("../NodeDefinition.js").NodeDefinition<BooleanNode>;

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

@@ -0,0 +1,16 @@
+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 '../ProcessContext.js';
+export type CallGraphNode = ChartNode<'callGraph', CallGraphNodeData>;
+export type CallGraphNodeData = {
+    useErrorOutput?: boolean;
+};
+export declare class CallGraphNodeImpl extends NodeImpl<CallGraphNode> {
+    static create(): CallGraphNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    static getUIData(): NodeUIData;
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const callGraphNode: import("../NodeDefinition.js").NodeDefinition<CallGraphNode>;

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

@@ -0,0 +1,77 @@
+import { type ChartNode, type NodeInputDefinition, type NodeOutputDefinition } from '../NodeBase.js';
+import { NodeImpl, type NodeUIData } from '../NodeImpl.js';
+import { type ChatMessage } from '../DataValue.js';
+import type { Inputs, Outputs } from '../GraphProcessor.js';
+import { type InternalProcessContext } from '../ProcessContext.js';
+import { type EditorDefinition } from '../../index.js';
+export type ChatNode = ChartNode<'chat', ChatNodeData>;
+export type ChatNodeConfigData = {
+    model: string;
+    temperature: number;
+    useTopP: boolean;
+    top_p?: number;
+    maxTokens: number;
+    stop?: string;
+    presencePenalty?: number;
+    frequencyPenalty?: number;
+    enableFunctionUse?: boolean;
+    user?: string;
+    numberOfChoices?: number;
+    endpoint?: string;
+    overrideModel?: string;
+    overrideMaxTokens?: number;
+    headers?: {
+        key: string;
+        value: string;
+    }[];
+    seed?: number;
+    toolChoice?: 'none' | 'auto' | 'function';
+    toolChoiceFunction?: string;
+    responseFormat?: '' | 'text' | 'json' | 'json_schema';
+    parallelFunctionCalling?: boolean;
+    additionalParameters?: {
+        key: string;
+        value: string;
+    }[];
+    responseSchemaName?: string;
+};
+export type ChatNodeData = ChatNodeConfigData & {
+    useModelInput: boolean;
+    useTemperatureInput: boolean;
+    useTopPInput: boolean;
+    useTopP: boolean;
+    useUseTopPInput: boolean;
+    useMaxTokensInput: boolean;
+    useStop: boolean;
+    useStopInput: boolean;
+    usePresencePenaltyInput: boolean;
+    useFrequencyPenaltyInput: boolean;
+    useUserInput?: boolean;
+    useNumberOfChoicesInput?: boolean;
+    useEndpointInput?: boolean;
+    useHeadersInput?: boolean;
+    useSeedInput?: boolean;
+    useToolChoiceInput?: boolean;
+    useToolChoiceFunctionInput?: boolean;
+    useResponseFormatInput?: boolean;
+    useAdditionalParametersInput?: boolean;
+    useResponseSchemaNameInput?: boolean;
+    /** Given the same set of inputs, return the same output without hitting GPT */
+    cache: boolean;
+    useAsGraphPartialOutput?: boolean;
+};
+export declare class ChatNodeImpl extends NodeImpl<ChatNode> {
+    static create(): ChatNode;
+    getInputDefinitions(): NodeInputDefinition[];
+    getOutputDefinitions(): NodeOutputDefinition[];
+    static getUIData(): NodeUIData;
+    getEditors(): EditorDefinition<ChatNode>[];
+    getBody(): string;
+    process(inputs: Inputs, context: InternalProcessContext): Promise<Outputs>;
+}
+export declare const chatNode: import("../NodeDefinition.js").NodeDefinition<ChatNode>;
+export declare function getChatNodeMessages(inputs: Inputs): {
+    messages: ChatMessage[];
+    systemPrompt: import("../DataValue.js").DataValue | undefined;
+};
+export declare function getCostForTokens(tokenCount: number, type: 'prompt' | 'completion', costPerThousand: number): number;