graphai 0.2.4 → 0.2.5

package/lib/experimental_agents/llm_agents/groq_agent.js

@@ -3,11 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.gloqAgent = void 0;
 const groq_sdk_1 = require("groq-sdk");
 const utils_1 = require("../../utils/utils");
-const groq = process.env.GROQ_API_KEY
-    ? new groq_sdk_1.Groq({
-        apiKey: process.env.GROQ_API_KEY,
-    })
-    : undefined;
+const groq = process.env.GROQ_API_KEY ? new groq_sdk_1.Groq({ apiKey: process.env.GROQ_API_KEY }) : undefined;
 const gloqAgent = async ({ params, inputs }) => {
     (0, utils_1.assert)(groq !== undefined, "The GROQ_API_KEY environment variable is missing.");
     const query = params?.query ? [params.query] : [];

package/lib/node.d.ts

@@ -21,6 +21,7 @@ export declare class ComputedNode extends Node {
     retryCount: number;
     readonly agentId?: string;
     readonly timeout?: number;
+    readonly priority: number;
     error?: Error;
     transactionId: undefined | number;
     readonly anyInput: boolean;

package/lib/node.js

@@ -43,6 +43,7 @@ class ComputedNode extends Node {
         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));
         this.pendings = new Set(this.dataSources.filter((source) => source.nodeId).map((source) => source.nodeId));

package/lib/task_manager.d.ts

@@ -11,8 +11,14 @@ export declare class TaskManager {
     prepareForNesting(): void;
     restoreAfterNesting(): void;
     getStatus(verbose?: boolean): {
-        runningNodes?: string[] | undefined;
-        queuedNodes?: string[] | undefined;
+        runningNodes: string[];
+        queuedNodes: string[];
+        concurrency: number;
+        queue: number;
+        running: number;
+    } | {
+        runningNodes?: undefined;
+        queuedNodes?: undefined;
         concurrency: number;
         queue: number;
         running: number;

package/lib/task_manager.js

@@ -27,7 +27,12 @@ class TaskManager {
     // Node will call this method to put itself in the execution queue.
     // We call the associated callback function when it is dequeued.
     addTask(node, graphId, callback) {
-        this.taskQueue.push({ node, graphId, callback });
+        // Finder tasks in the queue, which has either the same or higher priority.
+        const count = this.taskQueue.filter((task) => {
+            return task.node.priority >= node.priority;
+        }).length;
+        (0, utils_1.assert)(count <= this.taskQueue.length, "TaskManager.addTask: Something is really wrong.");
+        this.taskQueue.splice(count, 0, { node, graphId, callback });
         this.dequeueTaskIfPossible();
     }
     isRunning(graphId) {
@@ -53,16 +58,17 @@ class TaskManager {
         this.concurrency--;
     }
     getStatus(verbose = false) {
+        const nodes = verbose
+            ? {
+                runningNodes: Array.from(this.runningNodes).map((node) => node.nodeId),
+                queuedNodes: this.taskQueue.map((task) => task.node.nodeId),
+            }
+            : {};
         return {
             concurrency: this.concurrency,
             queue: this.taskQueue.length,
             running: this.runningNodes.size,
-            ...(verbose
-                ? {
-                    runningNodes: Array.from(this.runningNodes).map((node) => node.nodeId),
-                    queuedNodes: this.taskQueue.map((task) => task.node.nodeId),
-                }
-                : {}),
+            ...nodes,
         };
     }
 }

package/lib/type.d.ts

@@ -34,6 +34,7 @@ export type ComputedNodeData = {
     timeout?: number;
     graph?: GraphData;
     isResult?: boolean;
+    priority?: number;
 };
 export type NodeData = StaticNodeData | ComputedNodeData;
 export type LoopData = {

package/lib/validators/common.js

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

package/package.json

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