graphai 0.2.1 → 0.2.3

package/lib/graphai.d.ts

@@ -12,6 +12,7 @@ export declare class GraphAI {
     private readonly logs;
     readonly callbackDictonary: AgentFunctionDictonary;
     readonly taskManager: TaskManager;
+    readonly retryLimit?: number;
     nodes: GraphNodes;
     onLogCallback: (__log: TransactionLog, __isUpdate: boolean) => void;
     verbose: boolean;

package/lib/graphai.js

@@ -12,14 +12,16 @@ class GraphAI {
     // or we are about to start n-th iteration (n>2).
     createNodes(data) {
         const nodes = Object.keys(data.nodes).reduce((_nodes, nodeId) => {
-            const isStaticNode = "value" in data.nodes[nodeId];
-            if (isStaticNode) {
-                _nodes[nodeId] = new node_1.StaticNode(nodeId, data.nodes[nodeId], this);
+            const nodeData = data.nodes[nodeId];
+            if ("value" in nodeData) {
+                _nodes[nodeId] = new node_1.StaticNode(nodeId, nodeData, this);
             }
-            else {
-                const nodeData = data.nodes[nodeId];
+            else if ("agentId" in nodeData) {
                 _nodes[nodeId] = new node_1.ComputedNode(this.graphId, nodeId, nodeData, this);
             }
+            else {
+                throw new Error("Unknown node: " + nodeId);
+            }
             return _nodes;
         }, {});
         // Generate the waitlist for each node.
@@ -72,6 +74,7 @@ class GraphAI {
             console.log("------------ no version");
         }
         this.version = data.version ?? 0.2;
+        this.retryLimit = data.retry; // optional
         this.graphId = URL.createObjectURL(new Blob()).slice(-36);
         this.data = data;
         this.callbackDictonary = callbackDictonary;

package/lib/index.d.ts

@@ -1,4 +1,4 @@
 import { GraphAI } from "./graphai";
 export { GraphAI };
-export { AgentFunction, AgentFunctionDictonary, GraphData, ResultDataDictonary, ResultData } from "./type";
+export { AgentFunction, AgentFunctionDictonary, GraphData, ResultDataDictonary, ResultData, NodeState } from "./type";
 export type { TransactionLog } from "./transaction_log";

package/lib/index.js

@@ -1,5 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GraphAI = void 0;
+exports.NodeState = exports.GraphAI = void 0;
 const graphai_1 = require("./graphai");
 Object.defineProperty(exports, "GraphAI", { enumerable: true, get: function () { return graphai_1.GraphAI; } });
+var type_1 = require("./type");
+Object.defineProperty(exports, "NodeState", { enumerable: true, get: function () { return type_1.NodeState; } });

package/lib/node.js

@@ -40,7 +40,7 @@ class ComputedNode extends Node {
         this.params = data.params ?? {};
         this.nestedGraph = data.graph;
         this.agentId = data.agentId;
-        this.retryLimit = data.retry ?? 0;
+        this.retryLimit = data.retry ?? graph.retryLimit ?? 0;
         this.timeout = data.timeout;
         this.isResult = data.isResult ?? false;
         this.anyInput = data.anyInput ?? false;

package/lib/type.d.ts

@@ -47,6 +47,7 @@ export type GraphData = {
     concurrency?: number;
     loop?: LoopData;
     verbose?: boolean;
+    retry?: number;
 };
 export type AgentFunctionContext<ParamsType, InputDataType> = {
     params: NodeDataParams<ParamsType>;

package/lib/validators/common.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.staticNodeAttributeKeys = exports.computedNodeAttributeKeys = exports.graphDataAttributeKeys = void 0;
-exports.graphDataAttributeKeys = ["nodes", "concurrency", "agentId", "loop", "verbose"];
+exports.graphDataAttributeKeys = ["nodes", "concurrency", "agentId", "loop", "verbose", "version"];
 exports.computedNodeAttributeKeys = ["inputs", "anyInput", "params", "retry", "timeout", "agentId", "graph", "isResult"];
 exports.staticNodeAttributeKeys = ["value", "update", "isResult"];

package/lib/validators/relation_validator.js

@@ -10,7 +10,7 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
     computedNodeIds.forEach((computedNodeId) => {
         const nodeData = data.nodes[computedNodeId];
         pendings[computedNodeId] = new Set();
-        if (nodeData.inputs) {
+        if ("inputs" in nodeData && nodeData && nodeData.inputs) {
             nodeData.inputs.forEach((inputNodeId) => {
                 const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId).nodeId;
                 if (sourceNodeId) {
@@ -27,8 +27,8 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
     // TODO. validate update
     staticNodeIds.forEach((staticNodeId) => {
         const nodeData = data.nodes[staticNodeId];
-        const update = nodeData.update;
-        if (update) {
+        if ("value" in nodeData && nodeData.update) {
+            const update = nodeData.update;
             const updateNodeId = (0, utils_1.parseNodeName)(update).nodeId;
             if (!updateNodeId) {
                 throw new Error("Update it a literal");

package/package.json

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