graphai 0.2.2 → 0.2.4

package/lib/graphai.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GraphAI = void 0;
-const type_1 = require("./type");
 const node_1 = require("./node");
 const utils_1 = require("./utils/utils");
 const validator_1 = require("./validator");
@@ -144,7 +143,7 @@ class GraphAI {
     }
     // for computed
     pushQueue(node) {
-        node.state = type_1.NodeState.Queued;
+        node.beforeAddTask();
         this.taskManager.addTask(node, this.graphId, (_node) => {
             (0, utils_1.assert)(node.nodeId === _node.nodeId, "GraphAI.pushQueue node mismatch");
             node.execute();

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.d.ts

@@ -32,6 +32,7 @@ export declare class ComputedNode extends Node {
     isReadyNode(): boolean;
     private retry;
     private checkDataAvailability;
+    beforeAddTask(): void;
     removePending(nodeId: string): void;
     private isCurrentTransaction;
     private executeTimeout;

package/lib/node.js

@@ -87,6 +87,11 @@ class ComputedNode extends Node {
         });
         return results.length > 0;
     }
+    // This method is called right before the Graph add this node to the task manager.
+    beforeAddTask() {
+        this.state = type_1.NodeState.Queued;
+        this.log.beforeAddTask(this, this.graph);
+    }
     // This method is called when the data became available on one of nodes,
     // which this node needs data from.
     removePending(nodeId) {

package/lib/transaction_log.d.ts

@@ -23,5 +23,6 @@ export declare class TransactionLog {
     onInjected(node: StaticNode, graph: GraphAI, injectFrom?: string): void;
     onComplete(node: ComputedNode, graph: GraphAI, localLog: TransactionLog[]): void;
     beforeExecute(node: ComputedNode, graph: GraphAI, transactionId: number, inputs: ResultData[]): void;
+    beforeAddTask(node: ComputedNode, graph: GraphAI): void;
     onError(node: ComputedNode, graph: GraphAI, errorMessage: string): void;
 }

package/lib/transaction_log.js

@@ -45,6 +45,11 @@ class TransactionLog {
         graph.setLoopLog(this);
         graph.appendLog(this);
     }
+    beforeAddTask(node, graph) {
+        this.state = node.state;
+        graph.setLoopLog(this);
+        graph.appendLog(this);
+    }
     onError(node, graph, errorMessage) {
         this.state = node.state;
         this.errorMessage = errorMessage;

package/lib/type.d.ts

@@ -7,8 +7,7 @@ export declare enum NodeState {
     Failed = "failed",
     TimedOut = "timed-out",
     Completed = "completed",
-    Injected = "injected",
-    Dispatched = "dispatched"
+    Injected = "injected"
 }
 export type DefaultResultData = Record<string, any> | string | number | Array<DefaultResultData>;
 export type DefaultInputData = Record<string, any>;

package/lib/type.js

@@ -10,5 +10,4 @@ var NodeState;
     NodeState["TimedOut"] = "timed-out";
     NodeState["Completed"] = "completed";
     NodeState["Injected"] = "injected";
-    NodeState["Dispatched"] = "dispatched";
 })(NodeState || (exports.NodeState = NodeState = {}));

package/package.json

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