@alpic80/rivet-core 1.19.1-aidon.1

package/dist/esm/model/NodeBase.js

@@ -0,0 +1 @@
+export {};

package/dist/esm/model/NodeBodySpec.js

@@ -0,0 +1 @@
+export {};

package/dist/esm/model/NodeDefinition.js

@@ -0,0 +1,12 @@
+export function nodeDefinition(impl, displayName) {
+    return {
+        impl,
+        displayName,
+    };
+}
+export function pluginNodeDefinition(impl, displayName) {
+    return {
+        impl,
+        displayName,
+    };
+}

package/dist/esm/model/NodeGraph.js

@@ -0,0 +1,14 @@
+import { nanoid } from 'nanoid/non-secure';
+import {} from './NodeBase.js';
+import {} from '../utils/serialization/serializationUtils.js';
+export function emptyNodeGraph() {
+    return {
+        nodes: [],
+        connections: [],
+        metadata: {
+            id: nanoid(),
+            name: 'Untitled Graph',
+            description: '',
+        },
+    };
+}

package/dist/esm/model/NodeImpl.js

@@ -0,0 +1,49 @@
+export class NodeImpl {
+    chartNode;
+    constructor(chartNode) {
+        this.chartNode = chartNode;
+    }
+    get id() {
+        return this.chartNode.id;
+    }
+    get type() {
+        return this.chartNode.type;
+    }
+    get title() {
+        return this.chartNode.title;
+    }
+    get visualData() {
+        return this.chartNode.visualData;
+    }
+    get data() {
+        return this.chartNode.data;
+    }
+    getEditors(_context) {
+        return [];
+    }
+    getBody(_context) {
+        return undefined;
+    }
+}
+export class PluginNodeImplClass extends NodeImpl {
+    impl;
+    constructor(chartNode, impl) {
+        super(chartNode);
+        this.impl = impl;
+    }
+    getInputDefinitions(connections, nodes, project) {
+        return this.impl.getInputDefinitions(this.data, connections, nodes, project);
+    }
+    getOutputDefinitions(connections, nodes, project) {
+        return this.impl.getOutputDefinitions(this.data, connections, nodes, project);
+    }
+    process(inputData, context) {
+        return this.impl.process(this.data, inputData, context);
+    }
+    getEditors(context) {
+        return this.impl.getEditors(this.data, context);
+    }
+    getBody(context) {
+        return this.impl.getBody(this.data, context);
+    }
+}

package/dist/esm/model/NodeRegistration.js

@@ -0,0 +1,144 @@
+import {} from '../index.js';
+import { mapValues, values } from '../utils/typeSafety.js';
+import {} from './RivetPlugin.js';
+import {} from './RivetUIContext.js';
+import { PluginNodeImplClass } from './NodeImpl.js';
+export class NodeRegistration {
+    NodesType = undefined;
+    NodeTypesType = undefined;
+    #infos = {};
+    #plugins = [];
+    #implsMap = {};
+    #nodeTypes = [];
+    register(definition, plugin) {
+        const newRegistration = this;
+        const typeStr = definition.impl.create(undefined).type;
+        if (newRegistration.#infos[typeStr]) {
+            throw new Error(`Duplicate node type: ${typeStr}`);
+        }
+        newRegistration.#infos[typeStr] = {
+            displayName: definition.displayName,
+            impl: definition.impl,
+            plugin,
+        };
+        newRegistration.#implsMap[typeStr] = {
+            impl: definition.impl,
+            pluginImpl: undefined,
+        };
+        newRegistration.#nodeTypes.push(typeStr);
+        return newRegistration;
+    }
+    registerPluginNode(definition, plugin) {
+        const newRegistration = this;
+        const typeStr = definition.impl.create().type;
+        if (newRegistration.#infos[typeStr]) {
+            throw new Error(`Duplicate node type: ${typeStr}`);
+        }
+        const pluginClass = class extends PluginNodeImplClass {
+            static create() {
+                return definition.impl.create();
+            }
+            static getUIData(context) {
+                return definition.impl.getUIData(context);
+            }
+        };
+        newRegistration.#infos[typeStr] = {
+            displayName: definition.displayName,
+            impl: pluginClass,
+            plugin,
+            pluginImpl: definition.impl,
+        };
+        newRegistration.#implsMap[typeStr] = {
+            impl: pluginClass,
+            pluginImpl: definition.impl,
+        };
+        newRegistration.#nodeTypes.push(typeStr);
+        return newRegistration;
+    }
+    get #dynamicImpls() {
+        return this.#implsMap;
+    }
+    get #dynamicDisplayNames() {
+        const displayNameMap = mapValues(this.#infos, (info) => info.displayName);
+        return displayNameMap;
+    }
+    registerPlugin(plugin) {
+        if (plugin.register) {
+            plugin.register((definition) => this.registerPluginNode(definition, plugin));
+        }
+        this.#plugins.push(plugin);
+    }
+    create(type) {
+        const info = this.#infos[type];
+        if (!info) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return info.impl.create(info.pluginImpl);
+    }
+    createDynamic(type) {
+        const implClass = this.#dynamicImpls[type];
+        if (!implClass) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return implClass.impl.create(implClass.pluginImpl);
+    }
+    createImpl(node) {
+        const type = node.type;
+        const info = this.#infos[type];
+        if (!info) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        const { impl: ImplClass, pluginImpl } = info;
+        const impl = new ImplClass(node, pluginImpl);
+        if (!impl) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return impl;
+    }
+    createDynamicImpl(node) {
+        const { type } = node;
+        const ImplClass = this.#dynamicImpls[type];
+        if (!ImplClass) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        // eslint-disable-next-line new-cap
+        const impl = new ImplClass.impl(node, ImplClass.pluginImpl);
+        if (!impl) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return impl;
+    }
+    getDisplayName(type) {
+        const info = this.#infos[type];
+        if (!info) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return info.displayName;
+    }
+    getDynamicDisplayName(type) {
+        const displayName = this.#dynamicDisplayNames[type];
+        if (!displayName) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return displayName;
+    }
+    isRegistered(type) {
+        return this.#infos[type] !== undefined;
+    }
+    getNodeTypes() {
+        return this.#nodeTypes;
+    }
+    getNodeConstructors() {
+        return values(this.#dynamicImpls).map((info) => info.impl);
+    }
+    getPluginFor(type) {
+        const info = this.#infos[type];
+        if (!info) {
+            throw new Error(`Unknown node type: ${type}`);
+        }
+        return info.plugin;
+    }
+    getPlugins() {
+        return this.#plugins;
+    }
+}

package/dist/esm/model/Nodes.js

@@ -0,0 +1,227 @@
+import { NodeRegistration } from './NodeRegistration.js';
+import { userInputNode } from './nodes/UserInputNode.js';
+export * from './nodes/UserInputNode.js';
+import { textNode } from './nodes/TextNode.js';
+export * from './nodes/TextNode.js';
+import { chatNode } from './nodes/ChatNode.js';
+export * from './nodes/ChatNode.js';
+import { promptNode } from './nodes/PromptNode.js';
+export * from './nodes/PromptNode.js';
+import { extractRegexNode } from './nodes/ExtractRegexNode.js';
+export * from './nodes/ExtractRegexNode.js';
+import { codeNode } from './nodes/CodeNode.js';
+export * from './nodes/CodeNode.js';
+import { matchNode } from './nodes/MatchNode.js';
+export * from './nodes/MatchNode.js';
+import { ifNode } from './nodes/IfNode.js';
+export * from './nodes/IfNode.js';
+import { readDirectoryNode } from './nodes/ReadDirectoryNode.js';
+export * from './nodes/ReadDirectoryNode.js';
+import { readFileNode } from './nodes/ReadFileNode.js';
+export * from './nodes/ReadFileNode.js';
+import { ifElseNode } from './nodes/IfElseNode.js';
+export * from './nodes/IfElseNode.js';
+import { chunkNode } from './nodes/ChunkNode.js';
+export * from './nodes/ChunkNode.js';
+import { graphInputNode } from './nodes/GraphInputNode.js';
+export * from './nodes/GraphInputNode.js';
+import { graphOutputNode } from './nodes/GraphOutputNode.js';
+export * from './nodes/GraphOutputNode.js';
+import { subGraphNode } from './nodes/SubGraphNode.js';
+export * from './nodes/SubGraphNode.js';
+import { arrayNode } from './nodes/ArrayNode.js';
+export * from './nodes/ArrayNode.js';
+import { extractJsonNode } from './nodes/ExtractJsonNode.js';
+export * from './nodes/ExtractJsonNode.js';
+import { assemblePromptNode } from './nodes/AssemblePromptNode.js';
+export * from './nodes/ExtractYamlNode.js';
+import { loopControllerNode } from './nodes/LoopControllerNode.js';
+export * from './nodes/AssemblePromptNode.js';
+import { trimChatMessagesNode } from './nodes/TrimChatMessagesNode.js';
+export * from './nodes/LoopControllerNode.js';
+import { extractYamlNode } from './nodes/ExtractYamlNode.js';
+export * from './nodes/TrimChatMessagesNode.js';
+import { externalCallNode } from './nodes/ExternalCallNode.js';
+export * from './nodes/ExternalCallNode.js';
+import { extractObjectPathNode } from './nodes/ExtractObjectPathNode.js';
+export * from './nodes/ExtractObjectPathNode.js';
+import { raiseEventNode } from './nodes/RaiseEventNode.js';
+export * from './nodes/RaiseEventNode.js';
+import { contextNode } from './nodes/ContextNode.js';
+export * from './nodes/ContextNode.js';
+import { coalesceNode } from './nodes/CoalesceNode.js';
+export * from './nodes/CoalesceNode.js';
+import { passthroughNode } from './nodes/PassthroughNode.js';
+export * from './nodes/PassthroughNode.js';
+import { popNode } from './nodes/PopNode.js';
+export * from './nodes/PopNode.js';
+import { setGlobalNode } from './nodes/SetGlobalNode.js';
+export * from './nodes/SetGlobalNode.js';
+import { getGlobalNode } from './nodes/GetGlobalNode.js';
+export * from './nodes/GetGlobalNode.js';
+import { waitForEventNode } from './nodes/WaitForEventNode.js';
+export * from './nodes/WaitForEventNode.js';
+import { gptFunctionNode } from './nodes/GptFunctionNode.js';
+export * from './nodes/GptFunctionNode.js';
+import { toYamlNode } from './nodes/ToYamlNode.js';
+export * from './nodes/ToYamlNode.js';
+import { getEmbeddingNode } from './nodes/GetEmbeddingNode.js';
+export * from './nodes/GetEmbeddingNode.js';
+import { vectorStoreNode } from './nodes/VectorStoreNode.js';
+export * from './nodes/VectorStoreNode.js';
+import { vectorNearestNeighborsNode } from './nodes/VectorNearestNeighborsNode.js';
+export * from './nodes/VectorNearestNeighborsNode.js';
+import { hashNode } from './nodes/HashNode.js';
+export * from './nodes/HashNode.js';
+import { abortGraphNode } from './nodes/AbortGraphNode.js';
+export * from './nodes/AbortGraphNode.js';
+import { raceInputsNode } from './nodes/RaceInputsNode.js';
+export * from './nodes/RaceInputsNode.js';
+import { toJsonNode } from './nodes/ToJsonNode.js';
+export * from './nodes/ToJsonNode.js';
+import { joinNode } from './nodes/JoinNode.js';
+export * from './nodes/JoinNode.js';
+import { filterNode } from './nodes/FilterNode.js';
+export * from './nodes/FilterNode.js';
+import { objectNode } from './nodes/ObjectNode.js';
+export * from './nodes/ObjectNode.js';
+import { booleanNode } from './nodes/BooleanNode.js';
+export * from './nodes/BooleanNode.js';
+import { compareNode } from './nodes/CompareNode.js';
+export * from './nodes/CompareNode.js';
+import { evaluateNode } from './nodes/EvaluateNode.js';
+export * from './nodes/EvaluateNode.js';
+import { numberNode } from './nodes/NumberNode.js';
+export * from './nodes/NumberNode.js';
+import { randomNumberNode } from './nodes/RandomNumberNode.js';
+export * from './nodes/RandomNumberNode.js';
+import { shuffleNode } from './nodes/ShuffleNode.js';
+export * from './nodes/ShuffleNode.js';
+import { commentNode } from './nodes/CommentNode.js';
+export * from './nodes/CommentNode.js';
+import { imageNode } from './nodes/ImageNode.js';
+export * from './nodes/ImageNode.js';
+import { audioNode } from './nodes/AudioNode.js';
+export * from './nodes/AudioNode.js';
+import { httpCallNode } from './nodes/HttpCallNode.js';
+export * from './nodes/HttpCallNode.js';
+import { delayNode } from './nodes/DelayNode.js';
+export * from './nodes/DelayNode.js';
+import { appendToDatasetNode } from './nodes/AppendToDatasetNode.js';
+export * from './nodes/AppendToDatasetNode.js';
+import { createDatasetNode } from './nodes/CreateDatasetNode.js';
+export * from './nodes/CreateDatasetNode.js';
+import { loadDatasetNode } from './nodes/LoadDatasetNode.js';
+export * from './nodes/LoadDatasetNode.js';
+import { getAllDatasetsNode } from './nodes/GetAllDatasetsNode.js';
+export * from './nodes/GetAllDatasetsNode.js';
+import { splitNode } from './nodes/SplitNode.js';
+export * from './nodes/SplitNode.js';
+import { datasetNearestNeighborsNode } from './nodes/DatasetNearestNeigborsNode.js';
+export * from './nodes/DatasetNearestNeigborsNode.js';
+import { getDatasetRowNode } from './nodes/GetDatasetRowNode.js';
+export * from './nodes/GetDatasetRowNode.js';
+import { sliceNode } from './nodes/SliceNode.js';
+export * from './nodes/SliceNode.js';
+import { extractMarkdownCodeBlocksNode } from './nodes/ExtractMarkdownCodeBlocksNode.js';
+export * from './nodes/ExtractMarkdownCodeBlocksNode.js';
+import { assembleMessageNode } from './nodes/AssembleMessageNode.js';
+export * from './nodes/AssembleMessageNode.js';
+import { urlReferenceNode } from './nodes/URLReferenceNode.js';
+export * from './nodes/URLReferenceNode.js';
+import { destructureNode } from './nodes/DestructureNode.js';
+export * from './nodes/DestructureNode.js';
+import { replaceDatasetNode } from './nodes/ReplaceDatasetNode.js';
+export * from './nodes/ReplaceDatasetNode.js';
+import { listGraphsNode } from './nodes/ListGraphsNode.js';
+export * from './nodes/ListGraphsNode.js';
+import { graphReferenceNode } from './nodes/GraphReferenceNode.js';
+export * from './nodes/GraphReferenceNode.js';
+import { callGraphNode } from './nodes/CallGraphNode.js';
+export * from './nodes/CallGraphNode.js';
+import { delegateFunctionCallNode } from './nodes/DelegateFunctionCallNode.js';
+export * from './nodes/DelegateFunctionCallNode.js';
+import { playAudioNode } from './nodes/PlayAudioNode.js';
+export * from './nodes/PlayAudioNode.js';
+export * from './nodes/CallGraphNode.js';
+export const registerBuiltInNodes = (registry) => {
+    return registry
+        .register(toYamlNode)
+        .register(userInputNode)
+        .register(textNode)
+        .register(chatNode)
+        .register(promptNode)
+        .register(extractRegexNode)
+        .register(codeNode)
+        .register(matchNode)
+        .register(ifNode)
+        .register(readDirectoryNode)
+        .register(readFileNode)
+        .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(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);
+};
+let globalRivetNodeRegistry = registerBuiltInNodes(new NodeRegistration());
+export { globalRivetNodeRegistry };
+/** Resets the global node registry to a fresh one with only built-in nodes registered. */
+export function resetGlobalRivetNodeRegistry() {
+    globalRivetNodeRegistry = registerBuiltInNodes(new NodeRegistration());
+}

package/dist/esm/model/PluginLoadSpec.js

@@ -0,0 +1 @@
+export {};

package/dist/esm/model/ProcessContext.js

@@ -0,0 +1 @@
+import {} from '../index.js';

package/dist/esm/model/Project.js

@@ -0,0 +1,2 @@
+import {} from './NodeGraph.js';
+import {} from './PluginLoadSpec.js';

package/dist/esm/model/RivetPlugin.js

@@ -0,0 +1 @@
+import {} from './NodeBase.js';

package/dist/esm/model/RivetUIContext.js

@@ -0,0 +1 @@
+import {} from '../index.js';

package/dist/esm/model/Settings.js

@@ -0,0 +1 @@
+export {};

package/dist/esm/model/nodes/AbortGraphNode.js

@@ -0,0 +1,100 @@
+import { nanoid } from 'nanoid/non-secure';
+import { NodeImpl } from '../NodeImpl.js';
+import { dedent } from 'ts-dedent';
+import { nodeDefinition } from '../NodeDefinition.js';
+import { coerceTypeOptional } from '../../utils/coerceType.js';
+export class AbortGraphNodeImpl extends NodeImpl {
+    static create() {
+        const chartNode = {
+            type: 'abortGraph',
+            title: 'Abort Graph',
+            id: nanoid(),
+            visualData: {
+                x: 0,
+                y: 0,
+                width: 200,
+            },
+            data: {
+                successfully: true,
+                errorMessage: '',
+            },
+        };
+        return chartNode;
+    }
+    getInputDefinitions() {
+        const inputs = [
+            {
+                id: 'data',
+                title: 'Data or Error',
+                dataType: 'any',
+                description: 'The message to abort the graph with.',
+            },
+        ];
+        if (this.data.useSuccessfullyInput) {
+            inputs.push({
+                id: 'successfully',
+                title: 'Successfully',
+                dataType: 'boolean',
+                description: 'Whether to successfully abort the graph (early-exit), or error abort the graph.',
+            });
+        }
+        return inputs;
+    }
+    getOutputDefinitions() {
+        return [];
+    }
+    getEditors() {
+        return [
+            {
+                type: 'toggle',
+                label: 'Successfully Abort',
+                dataKey: 'successfully',
+                useInputToggleDataKey: 'useSuccessfullyInput',
+            },
+            {
+                type: 'string',
+                label: 'Error Message (if not successfully aborting)',
+                dataKey: 'errorMessage',
+            },
+        ];
+    }
+    getBody() {
+        return dedent `
+      ${this.data.useSuccessfullyInput
+            ? 'Success depends on input'
+            : this.data.successfully
+                ? 'Successfully Abort'
+                : this.data.errorMessage
+                    ? `Error Abort: ${this.data.errorMessage}`
+                    : 'Error Abort'}
+    `;
+    }
+    static getUIData() {
+        return {
+            infoBoxBody: dedent `
+        Aborts the execution of the entire graph immediately.
+
+        Can either "successfully" abort the graph (early-exit), or "error" abort the graph.
+      `,
+            infoBoxTitle: 'Abort Graph Node',
+            contextMenuTitle: 'Abort Graph',
+            group: ['Logic'],
+        };
+    }
+    async process(inputs, context) {
+        const successfully = this.data.useSuccessfullyInput
+            ? coerceTypeOptional(inputs['successfully'], 'boolean') ?? this.data.successfully
+            : this.data.successfully;
+        if (successfully) {
+            context.abortGraph();
+        }
+        else {
+            const errorMessage = coerceTypeOptional(inputs['data'], 'string')?.trim() ||
+                this.data.errorMessage ||
+                'Graph aborted with error';
+            context.abortGraph(errorMessage);
+        }
+        return {};
+    }
+}
+export const abortGraphNode = nodeDefinition(AbortGraphNodeImpl, 'Abort Graph');