@receptron/graphai_cli 0.2.0 → 0.2.1

package/README.md

@@ -1,7 +1,13 @@
-# graphai cli
+# graphai cli🤖
 
 graphai command line tool
 
+## Install 🚀
+
+```sh
+npm i -g  @receptron/graphai_cli
+```
+
 ## Usage
 
 ```
@@ -16,9 +22,12 @@ Options:
       --help     Show help                                             [boolean]
       --version  Show version number                                   [boolean]
   -l, --list     agents list
-  -s, --sample   agents list                                            [string]
+  -s, --sample   agent sample data                                      [string]
   -d, --detail   agent detail                                           [string]
   -v, --verbose  verbose log               [boolean] [required] [default: false]
+  -m, --mermaid  mermaid                   [boolean] [required] [default: false]
+      --yaml     dump yaml                 [boolean] [required] [default: false]
+      --json     dump json                 [boolean] [required] [default: false]
       --log      output log                                             [string]
 ```
 

package/lib/args.d.ts

@@ -5,6 +5,9 @@ export declare const args: {
     sample: string | undefined;
     d: string | undefined;
     v: boolean;
+    m: boolean;
+    yaml: boolean;
+    json: boolean;
     log: string | undefined;
     yaml_or_json_file: string | undefined;
     _: (string | number)[];

package/lib/args.js

@@ -27,6 +27,25 @@ exports.args = yargs_1.default
     demandOption: true,
     default: false,
     type: "boolean",
+})
+    .option("m", {
+    alias: "mermaid",
+    describe: "mermaid",
+    demandOption: true,
+    default: false,
+    type: "boolean",
+})
+    .option("yaml", {
+    describe: "dump yaml",
+    demandOption: true,
+    default: false,
+    type: "boolean",
+})
+    .option("json", {
+    describe: "dump json",
+    demandOption: true,
+    default: false,
+    type: "boolean",
 })
     .option("log", {
     description: "output log",

package/lib/graphai_cli.d.ts

@@ -1,2 +1,2 @@
 #!/usr/bin/env node
-export {};
+import "dotenv/config";

package/lib/graphai_cli.js

@@ -29,12 +29,15 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const graphai_1 = require("graphai");
 const packages = __importStar(require("graphai/lib/experimental_agents/packages"));
+require("dotenv/config");
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
+const yaml_1 = __importDefault(require("yaml"));
 const args_1 = require("./args");
 const utils_1 = require("./utils");
 const file_utils_1 = require("./file_utils");
 const options_1 = require("./options");
+const mermaid_1 = require("./mermaid");
 const fileFullPath = (file) => {
     return path_1.default.resolve(process.cwd() + "/" + file) || "";
 };
@@ -54,6 +57,18 @@ const main = async () => {
     }
     try {
         const graph_data = (0, file_utils_1.readGraphaiData)(file_path);
+        if (args_1.args.mermaid) {
+            (0, mermaid_1.mermaid)(graph_data);
+            return;
+        }
+        if (args_1.args.json) {
+            console.log(JSON.stringify(graph_data, null, 2));
+            return;
+        }
+        if (args_1.args.yaml) {
+            console.log(yaml_1.default.stringify(graph_data, null, 2));
+            return;
+        }
         const agents = Object.entries(packages).reduce((tmp, current) => {
             const [k, v] = current;
             tmp[v.name] = v.agent;
@@ -63,8 +78,13 @@ const main = async () => {
         if (args_1.args.verbose) {
             graph.onLogCallback = utils_1.callbackLog;
         }
-        const results = await graph.run();
-        console.log(results);
+        try {
+            const results = await graph.run();
+            console.log(results);
+        }
+        catch (e) {
+            console.log("error", e);
+        }
         if (args_1.args.log) {
             const logfile = fileFullPath(args_1.args.log);
             fs_1.default.writeFileSync(logfile, JSON.stringify(graph.transactionLogs(), null, 2));

package/lib/mermaid.d.ts

@@ -0,0 +1,2 @@
+import { GraphData } from "graphai";
+export declare const mermaid: (graphData: GraphData) => void;

package/lib/mermaid.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mermaid = void 0;
+const utils_1 = require("graphai/lib/utils/utils");
+const mermaid = (graphData) => {
+    console.log("flowchart TD");
+    Object.keys(graphData.nodes).forEach((nodeId) => {
+        const node = graphData.nodes[nodeId];
+        // label / name
+        if ("agentId" in node) {
+            if (node.inputs) {
+                node.inputs.forEach((input) => {
+                    const source = (0, utils_1.parseNodeName)(input);
+                    if (source.propIds) {
+                        console.log(` ${source.nodeId}(${source.nodeId}) -- ${source.propIds.join(".")} --> ${nodeId}`);
+                    }
+                    else {
+                        console.log(` ${source.nodeId}(${source.nodeId}) --> ${nodeId}`);
+                    }
+                });
+            }
+        }
+        if ("value" in node) {
+            // console.log(node.value);
+        }
+    });
+};
+exports.mermaid = mermaid;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@receptron/graphai_cli",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "GraphAI command line tools.",
   "main": "lib/graphai_cli.js",
   "bin": {
@@ -47,7 +47,8 @@
     "deepmerge": "^4.3.1",
     "dotenv": "^16.4.5",
     "express": "^4.19.2",
-    "graphai": "0.2.0",
+    "graphai": "^0.2.7",
+    "groq-sdk": "^0.3.3",
     "openai": "^4.12.4",
     "slashgpt": "^0.0.8",
     "tiktoken": "^1.0.14",