graphai 2.0.4 → 2.0.6

package/lib/graphai.js

@@ -21,7 +21,8 @@ class GraphAI {
                 _nodes[nodeId] = new node_1.ComputedNode(this.graphId, nodeId, nodeData, this);
             }
             else {
-                _nodes[nodeId] = new node_1.StaticNode(nodeId, nodeData, this);
+                const updateValue = this.staticNodeInitData[nodeId];
+                _nodes[nodeId] = new node_1.StaticNode(nodeId, updateValue !== undefined ? { ...nodeData, value: updateValue } : nodeData, this);
             }
             return _nodes;
         }, {});
@@ -45,7 +46,7 @@ class GraphAI {
         return (0, data_source_1.getDataFromSource)(source.nodeId ? results[source.nodeId] : undefined, source, this.propFunctions);
     }
     // for static
-    initializeStaticNodes(enableConsoleLog = false) {
+    setStaticNodeResults(enableConsoleLog = false) {
         // If the result property is specified, inject it.
         // If the previousResults exists (indicating we are in a loop),
         // process the update property (nodeId or nodeId.propId).
@@ -54,7 +55,7 @@ class GraphAI {
             if (node?.isStaticNode) {
                 const value = node?.value;
                 if (value !== undefined) {
-                    this.injectValue(nodeId, value, nodeId);
+                    node.setResultValue(nodeId);
                 }
                 if (enableConsoleLog) {
                     node.consoleLog();
@@ -72,7 +73,7 @@ class GraphAI {
                 const update = node?.update;
                 if (update && previousResults) {
                     const result = this.getValueFromResults(update, previousResults);
-                    this.injectValue(nodeId, result, update.nodeId);
+                    this.updateStaticNodeValue(nodeId, result, update.nodeId);
                 }
                 if (enableConsoleLog) {
                     node.consoleLog();
@@ -88,6 +89,7 @@ class GraphAI {
         graphLoader: undefined,
         forceLoop: false,
     }) {
+        this.staticNodeInitData = {};
         this.logs = [];
         this.config = {};
         this.onLogCallback = (__log, __isUpdate) => { };
@@ -125,7 +127,6 @@ class GraphAI {
             },
         };
         this.nodes = this.createNodes(this.graphData);
-        this.initializeStaticNodes(true);
     }
     getAgentFunctionInfo(agentId) {
         if (agentId && this.agentFunctionInfoDictionary[agentId]) {
@@ -202,6 +203,7 @@ class GraphAI {
     }
     // Public API
     async run(all = false) {
+        this.setStaticNodeResults();
         if (Object.values(this.nodes)
             .filter((node) => node.isStaticNode)
             .some((node) => node.result === undefined && node.update === undefined)) {
@@ -228,14 +230,18 @@ class GraphAI {
             };
         });
     }
-    abort() {
+    abort(isChild = false) {
         if (this.isRunning()) {
             this.resetPending();
+            // Stop All Running node.
         }
         // For an agent like an event agent, where an external promise remains unresolved,
         // aborting and then retrying can cause nodes or the graph to execute again.
         // To prevent this, the transactionId is updated to ensure the retry fails.
         Object.values(this.nodes).forEach((node) => node.isComputedNode && (node.transactionId = undefined));
+        if (!isChild) {
+            this.taskManager.reset();
+        }
         this.onComplete(this.isRunning());
     }
     resetPending() {
@@ -273,6 +279,7 @@ class GraphAI {
         // We need to update static nodes, before checking the condition
         const previousResults = this.results(true, true); // results from previous loop
         this.updateStaticNodes(previousResults);
+        this.setStaticNodeResults();
         if (loop.count === undefined || this.repeatCount < loop.count) {
             if (loop.while) {
                 const source = (0, utils_1.parseNodeName)(loop.while);
@@ -282,8 +289,9 @@ class GraphAI {
                     return false; // while condition is not met
                 }
             }
-            this.initializeGraphAI();
+            this.nodes = this.createNodes(this.graphData);
             this.updateStaticNodes(previousResults, true);
+            this.setStaticNodeResults();
             this.pushReadyNodesIntoQueue();
             return true; // Indicating that we are going to continue.
         }
@@ -294,7 +302,7 @@ class GraphAI {
             throw new Error("This GraphAI instance is running");
         }
         this.nodes = this.createNodes(this.graphData);
-        this.initializeStaticNodes();
+        this.setStaticNodeResults();
     }
     setPreviousResults(previousResults) {
         this.updateStaticNodes(previousResults);
@@ -324,9 +332,13 @@ class GraphAI {
     }
     // Public API
     injectValue(nodeId, value, injectFrom) {
+        this.staticNodeInitData[nodeId] = value;
+        this.updateStaticNodeValue(nodeId, value, injectFrom);
+    }
+    updateStaticNodeValue(nodeId, value, injectFrom) {
         const node = this.nodes[nodeId];
         if (node && node.isStaticNode) {
-            node.injectValue(value, injectFrom);
+            node.updateValue(value, injectFrom);
         }
         else {
             throw new Error(`injectValue with Invalid nodeId, ${nodeId}`);

package/lib/node.d.ts

@@ -72,7 +72,8 @@ export declare class StaticNode extends Node {
     readonly isStaticNode = true;
     readonly isComputedNode = false;
     constructor(nodeId: string, data: StaticNodeData, graph: GraphAI);
-    injectValue(value: ResultData, injectFrom?: string): void;
+    updateValue(value: ResultData, injectFrom?: string): void;
+    setResultValue(injectFrom?: string): void;
     consoleLog(): void;
 }
 export type GraphNodes = Record<string, ComputedNode | StaticNode>;

package/lib/node.js

@@ -140,7 +140,7 @@ class ComputedNode extends Node {
             this.updateState(type_1.NodeState.Abort);
         }
         if (this.debugInfo && this.debugInfo.subGraphs) {
-            this.debugInfo.subGraphs.forEach((graph) => graph.abort());
+            this.debugInfo.subGraphs.forEach((graph) => graph.abort(true));
         }
     }
     isReadyNode() {
@@ -418,10 +418,13 @@ class StaticNode extends Node {
         this.isResult = data.isResult ?? false;
         this.console = data.console ?? {};
     }
-    injectValue(value, injectFrom) {
-        this.state = type_1.NodeState.Injected;
+    updateValue(value, injectFrom) {
         this.value = value;
-        this.result = value;
+        this.log.onInjected(this, this.graph, injectFrom);
+    }
+    setResultValue(injectFrom) {
+        this.state = type_1.NodeState.Injected;
+        this.result = this.value;
         this.log.onInjected(this, this.graph, injectFrom);
         this.onSetResult();
     }

package/lib/task_manager.d.ts

@@ -23,4 +23,5 @@ export declare class TaskManager {
         queue: number;
         running: number;
     };
+    reset(): void;
 }

package/lib/task_manager.js

@@ -68,5 +68,9 @@ class TaskManager {
             ...nodes,
         };
     }
+    reset() {
+        this.taskQueue.length = 0;
+        this.runningNodes.clear();
+    }
 }
 exports.TaskManager = TaskManager;

package/lib/type.d.ts

@@ -129,6 +129,7 @@ export type AgentFunctionInfoSample = {
     params: DefaultParamsType;
     result: any;
     graph?: GraphData;
+    description?: string;
 };
 export type AgentFunctionInfo = {
     name: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "2.0.4",
+  "version": "2.0.6",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/bundle.cjs.js",
   "module": "lib/bundle.esm.js",
@@ -29,8 +29,8 @@
   },
   "homepage": "https://github.com/receptron/graphai#readme",
   "devDependencies": {
-    "typedoc": "^0.28.4",
-    "typedoc-plugin-markdown": "^4.6.3"
+    "typedoc": "^0.28.5",
+    "typedoc-plugin-markdown": "^4.6.4"
   },
   "types": "./lib/index.d.ts",
   "directories": {