@receptron/graphai_cli 2.0.1 → 2.0.4

package/lib/graphai_cli.js

@@ -42,6 +42,7 @@ const graphai_1 = require("graphai");
 const packages = __importStar(require("@graphai/agents"));
 const token_bound_string_agent_1 = require("@graphai/token_bound_string_agent");
 const vanilla_node_agents_1 = require("@graphai/vanilla_node_agents");
+const stream_agent_filter_1 = require("@graphai/stream_agent_filter");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const yaml_1 = __importDefault(require("yaml"));
@@ -87,7 +88,36 @@ const main = async () => {
             console.log(yaml_1.default.stringify(graph_data, null, 2));
             return;
         }
-        const graph = new graphai_1.GraphAI(graph_data, agents);
+        const fds = {};
+        let fdsIndex = 3;
+        const getFd = (nodeId) => {
+            if (fds[nodeId]) {
+                return fds[nodeId];
+            }
+            fds[nodeId] = fs_1.default.createWriteStream(null, { fd: fdsIndex });
+            fds[nodeId].on("error", () => {
+                // nothing
+            });
+            fdsIndex = fdsIndex + 1;
+            return fds[nodeId];
+        };
+        const consoleStreamAgentFilter = (0, stream_agent_filter_1.streamAgentFilterGenerator)((context, data) => {
+            const fd = getFd(context.debugInfo.nodeId);
+            try {
+                fd.write(data);
+            }
+            catch (e) {
+                // nothinfg
+            }
+            process.stdout.write(data);
+        });
+        const agentFilters = [
+            {
+                name: "consoleStreamAgentFilter",
+                agent: consoleStreamAgentFilter,
+            },
+        ];
+        const graph = new graphai_1.GraphAI(graph_data, agents, { agentFilters });
         if (args_1.args.verbose) {
             graph.onLogCallback = test_utils_1.callbackLog;
         }

package/lib/test.d.ts

@@ -0,0 +1,4 @@
+declare const fs: any;
+declare const fd3: any;
+declare let count: number;
+declare const interval: NodeJS.Timeout;

package/lib/test.js

@@ -0,0 +1,14 @@
+"use strict";
+const fs = require("fs");
+const fd3 = fs.createWriteStream(null, { fd: 3 });
+console.log(fd3.writable);
+let count = 0;
+const interval = setInterval(() => {
+    process.stdout.write(`stdout: ${count}\n`);
+    fd3.write(`fd3: ${count}\n`);
+    count++;
+    if (count >= 5) {
+        clearInterval(interval);
+        fd3.end();
+    }
+}, 100);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@receptron/graphai_cli",
-  "version": "2.0.1",
+  "version": "2.0.4",
   "description": "GraphAI command line tools.",
   "main": "lib/graphai_cli.js",
   "bin": {
@@ -17,7 +17,7 @@
     "test": "yarn run test_sh",
     "test_sh": "./scripts/test.sh",
     "test_node": "node --test  -r tsconfig-paths/register --require ts-node/register ./test_yaml/test_*.ts",
-    "cli": "npx ts-node  -r tsconfig-paths/register ./src/graphai_cli.ts",
+    "cli": "node --require ts-node/register  -r tsconfig-paths/register ./src/graphai_cli.ts",
     "b": "yarn run format && yarn run eslint && yarn run build"
   },
   "repository": {
@@ -34,10 +34,11 @@
     "@types/yargs": "^17.0.33"
   },
   "dependencies": {
-    "@graphai/agent_filters": "^2.0.0",
     "@graphai/agents": "^2.0.4",
+    "@graphai/stream_agent_filter": "^2.0.1",
     "@graphai/token_bound_string_agent": "^1.0.1",
     "@graphai/vanilla_node_agents": "^2.0.0",
+    "@receptron/graphai_cli": "./receptron-graphai_cli-2.0.3-alpha.tgz",
     "@receptron/test_utils": "^2.0.0",
     "dotenv": "^16.5.0",
     "graphai": "^2.0.5",