graphai 2.0.3 → 2.0.5

package/lib/bundle.esm.js

@@ -930,9 +930,13 @@ class StaticNode extends Node {
         this.isResult = data.isResult ?? false;
         this.console = data.console ?? {};
     }
-    injectValue(value, injectFrom) {
+    updateValue(value, injectFrom) {
+        this.value = value;
+        this.log.onInjected(this, this.graph, injectFrom);
+    }
+    setResultValue(injectFrom) {
         this.state = NodeState.Injected;
-        this.result = value;
+        this.result = this.value;
         this.log.onInjected(this, this.graph, injectFrom);
         this.onSetResult();
     }
@@ -1245,7 +1249,8 @@ class GraphAI {
                 _nodes[nodeId] = new ComputedNode(this.graphId, nodeId, nodeData, this);
             }
             else {
-                _nodes[nodeId] = new StaticNode(nodeId, nodeData, this);
+                const updateValue = this.staticNodeInitData[nodeId];
+                _nodes[nodeId] = new StaticNode(nodeId, updateValue !== undefined ? { ...nodeData, value: updateValue } : nodeData, this);
             }
             return _nodes;
         }, {});
@@ -1269,7 +1274,7 @@ class GraphAI {
         return 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).
@@ -1278,7 +1283,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();
@@ -1296,7 +1301,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();
@@ -1312,6 +1317,7 @@ class GraphAI {
         graphLoader: undefined,
         forceLoop: false,
     }) {
+        this.staticNodeInitData = {};
         this.logs = [];
         this.config = {};
         this.onLogCallback = (__log, __isUpdate) => { };
@@ -1349,7 +1355,6 @@ class GraphAI {
             },
         };
         this.nodes = this.createNodes(this.graphData);
-        this.initializeStaticNodes(true);
     }
     getAgentFunctionInfo(agentId) {
         if (agentId && this.agentFunctionInfoDictionary[agentId]) {
@@ -1426,6 +1431,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)) {
@@ -1497,6 +1503,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 = parseNodeName(loop.while);
@@ -1506,8 +1513,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.
         }
@@ -1518,7 +1526,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);
@@ -1548,9 +1556,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}`);