graphai 0.4.8 → 0.5.0

package/lib/node.d.ts

@@ -29,6 +29,7 @@ export declare class ComputedNode extends Node {
     transactionId: undefined | number;
     readonly anyInput: boolean;
     dataSources: Array<DataSource>;
+    inputNames?: Array<string>;
     pendings: Set<string>;
     private ifSource?;
     private unlessSource?;

package/lib/node.js

@@ -40,23 +40,41 @@ class ComputedNode extends Node {
         this.params = data.params ?? {};
         this.console = data.console ?? {};
         this.filterParams = data.filterParams ?? {};
+        this.retryLimit = data.retry ?? graph.retryLimit ?? 0;
+        this.timeout = data.timeout;
+        this.isResult = data.isResult ?? false;
+        this.priority = data.priority ?? 0;
+        this.anyInput = data.anyInput ?? false;
+        if (!data.inputs) {
+            this.dataSources = [];
+        }
+        else if (Array.isArray(data.inputs)) {
+            this.dataSources = (data.inputs ?? []).map((input) => (0, utils_2.parseNodeName)(input, graph.version));
+        }
+        else {
+            const inputs = data.inputs;
+            const keys = Object.keys(inputs);
+            this.inputNames = keys;
+            this.dataSources = keys.map((key) => (0, utils_2.parseNodeName)(inputs[key], graph.version));
+        }
+        this.pendings = new Set(this.dataSources.filter((source) => source.nodeId).map((source) => source.nodeId));
         if (typeof data.agent === "string") {
             this.agentId = data.agent;
         }
         else {
             (0, utils_2.assert)(typeof data.agent === "function", "agent must be either string or function");
             const agent = data.agent;
-            this.agentFunction = async ({ inputs }) => {
-                return agent(...inputs);
-            };
+            if (this.inputNames) {
+                this.agentFunction = async ({ namedInputs }) => {
+                    return agent(namedInputs);
+                };
+            }
+            else {
+                this.agentFunction = async ({ inputs }) => {
+                    return agent(...inputs);
+                };
+            }
         }
-        this.retryLimit = data.retry ?? graph.retryLimit ?? 0;
-        this.timeout = data.timeout;
-        this.isResult = data.isResult ?? false;
-        this.priority = data.priority ?? 0;
-        this.anyInput = data.anyInput ?? false;
-        this.dataSources = (data.inputs ?? []).map((input) => (0, utils_2.parseNodeName)(input, graph.version));
-        this.pendings = new Set(this.dataSources.filter((source) => source.nodeId).map((source) => source.nodeId));
         if (typeof data.graph === "string") {
             const source = (0, utils_2.parseNodeName)(data.graph, graph.version);
             (0, utils_2.assert)(!!source.nodeId, `Invalid data source ${data.graph}`);
@@ -232,6 +250,7 @@ class ComputedNode extends Node {
             const context = {
                 params: params,
                 inputs: previousResults,
+                namedInputs: {},
                 debugInfo: {
                     nodeId: this.nodeId,
                     agentId: this.agentId,
@@ -242,6 +261,13 @@ class ComputedNode extends Node {
                 agentFilters: this.graph.agentFilters,
                 log: localLog,
             };
+            if (this.inputNames) {
+                context.namedInputs = this.inputNames.reduce((tmp, name, index) => {
+                    tmp[name] = previousResults[index];
+                    return tmp;
+                }, {});
+                context.inputs = [];
+            }
             // NOTE: We use the existence of graph object in the agent-specific params to determine
             // if this is a nested agent or not.
             if (this.nestedGraph) {

package/lib/type.d.ts

@@ -30,7 +30,7 @@ export type AgentAnonymousFunction = (...params: any[]) => unknown;
 export type AgentFilterParams = Record<string, any>;
 export type ComputedNodeData = {
     agent: string | AgentAnonymousFunction;
-    inputs?: Array<any>;
+    inputs?: Array<any> | Record<string, any>;
     anyInput?: boolean;
     params?: NodeDataParams;
     filterParams?: AgentFilterParams;
@@ -64,6 +64,7 @@ export type GraphOptions = {
 export type AgentFunctionContext<ParamsType = DefaultParamsType, InputDataType = DefaultInputData> = {
     params: NodeDataParams<ParamsType>;
     inputs: Array<InputDataType>;
+    namedInputs: Record<string, any>;
     debugInfo: {
         verbose: boolean;
         nodeId: string;
@@ -90,6 +91,8 @@ export type AgentFunctionInfo = {
     name: string;
     agent: AgentFunction<any, any, any>;
     mock: AgentFunction<any, any, any>;
+    inputs?: any;
+    output?: any;
     samples: {
         inputs: any;
         params: DefaultParamsType;

package/lib/validators/relation_validator.js

@@ -12,17 +12,34 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
         const nodeData = data.nodes[computedNodeId];
         pendings[computedNodeId] = new Set();
         if ("inputs" in nodeData && nodeData && nodeData.inputs) {
-            nodeData.inputs.forEach((inputNodeId) => {
-                const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId, data.version ?? 0.02).nodeId;
-                if (sourceNodeId) {
-                    if (!nodeIds.has(sourceNodeId)) {
-                        throw new common_1.ValidationError(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
+            if (Array.isArray(nodeData.inputs)) {
+                nodeData.inputs.forEach((inputNodeId) => {
+                    const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId, data.version ?? 0.02).nodeId;
+                    if (sourceNodeId) {
+                        if (!nodeIds.has(sourceNodeId)) {
+                            throw new common_1.ValidationError(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
+                        }
+                        waitlist[sourceNodeId] === undefined && (waitlist[sourceNodeId] = new Set());
+                        pendings[computedNodeId].add(sourceNodeId);
+                        waitlist[sourceNodeId].add(computedNodeId);
                     }
-                    waitlist[sourceNodeId] === undefined && (waitlist[sourceNodeId] = new Set());
-                    pendings[computedNodeId].add(sourceNodeId);
-                    waitlist[sourceNodeId].add(computedNodeId);
-                }
-            });
+                });
+            }
+            else {
+                const keys = Object.keys(nodeData.inputs);
+                keys.forEach((key) => {
+                    const inputNodeId = nodeData.inputs[key];
+                    const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId, data.version ?? 0.3).nodeId;
+                    if (sourceNodeId) {
+                        if (!nodeIds.has(sourceNodeId)) {
+                            throw new common_1.ValidationError(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
+                        }
+                        waitlist[sourceNodeId] === undefined && (waitlist[sourceNodeId] = new Set());
+                        pendings[computedNodeId].add(sourceNodeId);
+                        waitlist[sourceNodeId].add(computedNodeId);
+                    }
+                });
+            }
         }
     });
     // TODO. validate update

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.4.8",
+  "version": "0.5.0",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [