graphai 1.0.11 → 1.0.13

package/lib/bundle.esm.js

@@ -12,6 +12,55 @@ var NodeState;
     NodeState["Skipped"] = "skipped";
 })(NodeState || (NodeState = {}));
 
+const enabledLevels = {
+    debug: true,
+    info: true,
+    log: true,
+    warn: true,
+    error: true,
+};
+let customLogger = null;
+function setLevelEnabled(level, enabled) {
+    enabledLevels[level] = enabled;
+}
+function setLogger(logger) {
+    customLogger = logger;
+}
+function output(level, ...args) {
+    if (!enabledLevels[level])
+        return;
+    if (customLogger) {
+        customLogger(level, ...args);
+    }
+    else {
+        (console[level] || console.log)(...args);
+    }
+}
+function debug(...args) {
+    output("debug", ...args);
+}
+function info(...args) {
+    output("info", ...args);
+}
+function log(...args) {
+    output("log", ...args);
+}
+function warn(...args) {
+    output("warn", ...args);
+}
+function error(...args) {
+    output("error", ...args);
+}
+const GraphAILogger = {
+    setLevelEnabled,
+    setLogger,
+    debug,
+    info,
+    log,
+    warn,
+    error,
+};
+
 const sleep = async (milliseconds) => {
     return await new Promise((resolve) => setTimeout(resolve, milliseconds));
 };
@@ -42,7 +91,7 @@ function assert(condition, message, isWarn = false) {
         if (!isWarn) {
             throw new Error(message);
         }
-        console.warn("warn: " + message);
+        GraphAILogger.warn("warn: " + message);
     }
 }
 const isObject = (x) => {
@@ -158,6 +207,7 @@ const inputs2dataSources = (inputs) => {
     }
     return parseNodeName(inputs);
 };
+// TODO: Maybe it's a remnant of old array inputs. Check and delete.
 const dataSourceNodeIds = (sources) => {
     if (!Array.isArray(sources)) {
         throw new Error("sources must be array!! maybe inputs is invalid");
@@ -300,7 +350,7 @@ const propStringFunction = (result, propId) => {
             if (sliceMatch[1] !== undefined) {
                 return result.slice(Number(sliceMatch[1]));
             }
-            console.log(sliceMatch);
+            GraphAILogger.warn("slice is not valid format: " + sliceMatch);
         }
         const splitMatch = propId.match(/^split\(([-_:;.,\s\n]+)\)$/);
         if (splitMatch) {
@@ -342,7 +392,7 @@ const utilsFunctions = (input, nodes) => {
         return nodes[loopCounterKey].result;
     }
     // If a placeholder does not match any key, replace it with an empty string.
-    console.warn("not match template utility function: ${" + input + "}");
+    GraphAILogger.warn("not match template utility function: ${" + input + "}");
     return "";
 };
 
@@ -381,7 +431,7 @@ const innerGetDataFromSource = (result, propIds, propFunctions) => {
         const propId = propIds[0];
         const ret = getNestedData(result, propId, propFunctions);
         if (ret === undefined) {
-            console.error(`prop: ${propIds.join(".")} is not hit`);
+            GraphAILogger.error(`prop: ${propIds.join(".")} is not hit`);
         }
         if (propIds.length > 1) {
             return innerGetDataFromSource(ret, propIds.slice(1), propFunctions);
@@ -495,14 +545,14 @@ class Node {
             return;
         }
         else if (this.console === true || this.console.after === true) {
-            console.log(typeof result === "string" ? result : JSON.stringify(result, null, 2));
+            GraphAILogger.log(typeof result === "string" ? result : JSON.stringify(result, null, 2));
         }
         else if (this.console.after) {
             if (isObject(this.console.after)) {
-                console.log(JSON.stringify(resultsOf(this.console.after, { self: { result } }, this.graph.propFunctions, true), null, 2));
+                GraphAILogger.log(JSON.stringify(resultsOf(this.console.after, { self: { result } }, this.graph.propFunctions, true), null, 2));
             }
             else {
-                console.log(this.console.after);
+                GraphAILogger.log(this.console.after);
             }
         }
     }
@@ -660,7 +710,7 @@ class ComputedNode extends Node {
     // and attempt to retry (if specified).
     executeTimeout(transactionId) {
         if (this.state === NodeState.Executing && this.isCurrentTransaction(transactionId)) {
-            console.warn(`-- timeout ${this.timeout} with ${this.nodeId}`);
+            GraphAILogger.warn(`-- timeout ${this.timeout} with ${this.nodeId}`);
             this.retry(NodeState.TimedOut, Error("Timeout"));
         }
     }
@@ -753,7 +803,7 @@ class ComputedNode extends Node {
             if (!this.isCurrentTransaction(transactionId)) {
                 // This condition happens when the agent function returns
                 // after the timeout (either retried or not).
-                console.log(`-- transactionId mismatch with ${this.nodeId} (probably timeout)`);
+                GraphAILogger.log(`-- transactionId mismatch with ${this.nodeId} (probably timeout)`);
                 return;
             }
             // after process
@@ -791,20 +841,20 @@ class ComputedNode extends Node {
     // the retry if specified.
     errorProcess(error, transactionId, namedInputs) {
         if (error instanceof Error && error.message !== strIntentionalError) {
-            console.error(`<-- NodeId: ${this.nodeId}, Agent: ${this.agentId}`);
-            console.error({ namedInputs });
-            console.error(error);
-            console.error("-->");
+            GraphAILogger.error(`<-- NodeId: ${this.nodeId}, Agent: ${this.agentId}`);
+            GraphAILogger.error({ namedInputs });
+            GraphAILogger.error(error);
+            GraphAILogger.error("-->");
         }
         if (!this.isCurrentTransaction(transactionId)) {
-            console.warn(`-- transactionId mismatch with ${this.nodeId} (not timeout)`);
+            GraphAILogger.warn(`-- transactionId mismatch with ${this.nodeId} (not timeout)`);
             return;
         }
         if (error instanceof Error) {
             this.retry(NodeState.Failed, error);
         }
         else {
-            console.error(`-- NodeId: ${this.nodeId}: Unknown error was caught`);
+            GraphAILogger.error(`-- NodeId: ${this.nodeId}: Unknown error was caught`);
             this.retry(NodeState.Failed, Error("Unknown"));
         }
     }
@@ -852,10 +902,10 @@ class ComputedNode extends Node {
             return;
         }
         else if (this.console === true || this.console.before === true) {
-            console.log(JSON.stringify(context.namedInputs, null, 2));
+            GraphAILogger.log(JSON.stringify(context.namedInputs, null, 2));
         }
         else if (this.console.before) {
-            console.log(this.console.before);
+            GraphAILogger.log(this.console.before);
         }
     }
 }
@@ -1247,6 +1297,7 @@ class GraphAI {
         bypassAgentIds: [],
         config: {},
         graphLoader: undefined,
+        forceLoop: false,
     }) {
         this.logs = [];
         this.config = {};
@@ -1254,11 +1305,11 @@ class GraphAI {
         this.callbacks = [];
         this.repeatCount = 0;
         if (!graphData.version && !options.taskManager) {
-            console.warn("------------ missing version number");
+            GraphAILogger.warn("------------ missing version number");
         }
         this.version = graphData.version ?? graphDataLatestVersion;
         if (this.version < graphDataLatestVersion) {
-            console.warn(`------------ upgrade to ${graphDataLatestVersion}!`);
+            GraphAILogger.warn(`------------ upgrade to ${graphDataLatestVersion}!`);
         }
         this.retryLimit = graphData.retry; // optional
         this.graphId = `${Date.now().toString(36)}-${Math.random().toString(36).substr(2, 9)}`; // URL.createObjectURL(new Blob()).slice(-36);
@@ -1269,6 +1320,7 @@ class GraphAI {
         this.bypassAgentIds = options.bypassAgentIds ?? [];
         this.config = options.config;
         this.graphLoader = options.graphLoader;
+        this.forceLoop = options.forceLoop ?? false;
         this.loop = graphData.loop;
         this.verbose = graphData.verbose === true;
         this.onComplete = (__isAbort) => {
@@ -1281,7 +1333,7 @@ class GraphAI {
             nodes: {
                 ...graphData.nodes,
                 [loopCounterKey]: { value: 0, update: `:${loopCounterKey}.add(1)` },
-            }
+            },
         };
         this.nodes = this.createNodes(this.graphData);
         this.initializeStaticNodes(true);
@@ -1371,15 +1423,15 @@ class GraphAI {
         }
         this.pushReadyNodesIntoQueue();
         if (!this.isRunning()) {
-            console.warn("-- nothing to execute");
+            GraphAILogger.warn("-- nothing to execute");
             return {};
         }
         return new Promise((resolve, reject) => {
             this.onComplete = (isAbort = false) => {
                 const errors = this.errors();
-                const nodeIds = Object.keys(errors);
-                if (nodeIds.length > 0 || isAbort) {
-                    reject(errors[nodeIds[0]]);
+                const errorNodeIds = Object.keys(errors);
+                if (errorNodeIds.length > 0 || isAbort) {
+                    reject(errors[errorNodeIds[0]]);
                 }
                 else {
                     resolve(this.results(all));
@@ -1420,6 +1472,10 @@ class GraphAI {
     // Check if there is any running computed nodes.
     // In case of no running computed note, start the another iteration if ncessary (loop)
     processLoopIfNecessary() {
+        //
+        if (!this.forceLoop && Object.keys(this.errors()).length > 0) {
+            return false;
+        }
         this.repeatCount++;
         const loop = this.loop;
         if (!loop) {
@@ -1499,5 +1555,5 @@ class GraphAI {
     }
 }
 
-export { GraphAI, NodeState, ValidationError, agentInfoWrapper, assert, debugResultKey, defaultAgentInfo, defaultConcurrency, defaultTestContext, graphDataLatestVersion, inputs2dataSources, isComputedNodeData, isObject, isStaticNodeData, parseNodeName, sleep, strIntentionalError };
+export { GraphAI, GraphAILogger, NodeState, ValidationError, agentInfoWrapper, assert, debugResultKey, defaultAgentInfo, defaultConcurrency, defaultTestContext, graphDataLatestVersion, inputs2dataSources, isComputedNodeData, isObject, isStaticNodeData, parseNodeName, sleep, strIntentionalError };
 //# sourceMappingURL=bundle.esm.js.map