@receptron/graphai_cli 0.4.7 → 0.5.0

package/lib/agentdoc.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/lib/agentdoc.js

@@ -0,0 +1,32 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_1 = __importDefault(require("fs"));
+const docs_1 = require("./docs");
+const main = async () => {
+    const path = process.cwd();
+    const packageJson = JSON.parse(fs_1.default.readFileSync(path + "/package.json", "utf8"));
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const agents = require(path + "/lib/index");
+    const agentAttribute = (key) => {
+        if (key === "packageName") {
+            return packageJson.name;
+        }
+        if (key === "description") {
+            return packageJson.description;
+        }
+        if (key === "agents") {
+            return Object.keys(agents).join(", ");
+        }
+    };
+    const temp = (0, docs_1.readTemplate)("readme.md");
+    const md = ["packageName", "description", "agents"].reduce((tmp, key) => {
+        tmp = tmp.replaceAll("{" + key + "}", agentAttribute(key));
+        return tmp;
+    }, temp);
+    fs_1.default.writeFileSync(path + "/README.md", md);
+};
+main();

package/lib/docs.d.ts

@@ -1 +1 @@
-export {};
+export declare const readTemplate: (file: string) => string;

package/lib/docs.js

@@ -26,6 +26,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.readTemplate = void 0;
 const utils_1 = require("graphai/lib/utils/utils");
 const json_schema_generator_1 = __importDefault(require("json-schema-generator"));
 const packages = __importStar(require("@graphai/agents"));
@@ -33,9 +34,11 @@ const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const agentAttribute = (agentInfo, key) => {
     if (key === "samples") {
-        return agentInfo.samples
-            .map((sample) => {
+        return Array.from(agentInfo.samples.keys())
+            .map((key) => {
+            const sample = agentInfo.samples[key];
             return [
+                `### Sample${key}`,
                 "#### inputs",
                 "```json",
                 JSON.stringify(sample.inputs, null, 2),
@@ -54,12 +57,23 @@ const agentAttribute = (agentInfo, key) => {
             .join("\n");
     }
     if (key === "schemas") {
-        return agentInfo.samples
-            .map((sample) => {
+        if (agentInfo.inputs && agentInfo.output) {
+            return [
+                "#### inputs",
+                "```json",
+                JSON.stringify(agentInfo.inputs, null, 2),
+                "````",
+                "#### output",
+                "```json",
+                JSON.stringify(agentInfo.output, null, 2),
+                "````",
+            ].join("\n\n");
+        }
+        if (agentInfo.samples && agentInfo.samples[0]) {
+            const sample = agentInfo.samples[0];
             return ["#### inputs", "```json", JSON.stringify((0, json_schema_generator_1.default)(sample.inputs), null, 2), "````"].join("\n\n");
-            // return JSON.stringify(agentInfo.samples, null, 2);
-        })
-            .join("\n");
+        }
+        return "";
     }
     if (key === "resultKey") {
         return agentInfo.samples
@@ -73,8 +87,9 @@ const agentAttribute = (agentInfo, key) => {
 const readTemplate = (file) => {
     return fs_1.default.readFileSync(path_1.default.resolve(__dirname) + "/../templates/" + file, "utf8");
 };
+exports.readTemplate = readTemplate;
 const agentMd = (agentInfo) => {
-    const template = readTemplate("agent.md");
+    const template = (0, exports.readTemplate)("agent.md");
     const md = ["name", "description", "author", "repository", "license", "samples", "schemas", "resultKey"].reduce((tmp, key) => {
         tmp = tmp.replace("{" + key + "}", agentAttribute(agentInfo, key));
         return tmp;

package/lib/graphai_cli.js

@@ -34,7 +34,7 @@ const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const yaml_1 = __importDefault(require("yaml"));
 const args_1 = require("./args");
-const test_utils_1 = require("@graphai/test_utils");
+const test_utils_1 = require("@receptron/test_utils");
 const options_1 = require("./options");
 const mermaid_1 = require("./mermaid");
 const fileFullPath = (file) => {

package/lib/mermaid.js

@@ -9,15 +9,31 @@ const mermaid = (graphData) => {
         // label / name
         if ("agent" in node) {
             if (node.inputs) {
-                node.inputs.forEach((input) => {
-                    const source = (0, utils_1.parseNodeName)(input, graphData.version ?? 0.2);
-                    if (source.propIds) {
-                        console.log(` ${source.nodeId}(${source.nodeId}) -- ${source.propIds.join(".")} --> ${nodeId}`);
-                    }
-                    else {
-                        console.log(` ${source.nodeId}(${source.nodeId}) --> ${nodeId}`);
-                    }
-                });
+                if (Array.isArray(node.inputs)) {
+                    node.inputs.forEach((input) => {
+                        const source = (0, utils_1.parseNodeName)(input, graphData.version ?? 0.2);
+                        if (source.propIds) {
+                            console.log(` ${source.nodeId}(${source.nodeId}) -- ${source.propIds.join(".")} --> ${nodeId}`);
+                        }
+                        else {
+                            console.log(` ${source.nodeId}(${source.nodeId}) --> ${nodeId}`);
+                        }
+                    });
+                }
+                else {
+                    // LATER: Display the inputName as well.
+                    const inputNames = Object.keys(node.inputs);
+                    inputNames.forEach((inputName) => {
+                        const input = node.inputs[inputName];
+                        const source = (0, utils_1.parseNodeName)(input, graphData.version ?? 0.2);
+                        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) {

package/package.json

@@ -1,10 +1,11 @@
 {
   "name": "@receptron/graphai_cli",
-  "version": "0.4.7",
+  "version": "0.5.0",
   "description": "GraphAI command line tools.",
   "main": "lib/graphai_cli.js",
   "bin": {
-    "graphai": "lib/graphai_cli.js"
+    "graphai": "lib/graphai_cli.js",
+    "agentdoc": "lib/agentdoc.js"
   },
   "files": [
     "./lib"
@@ -35,15 +36,16 @@
     "@types/yargs": "^17.0.32"
   },
   "dependencies": {
-    "@graphai/agents": "^0.0.3",
-    "@graphai/test_utils": "^0.0.1",
+    "@graphai/agents": "^0.0.6",
+    "@receptron/test_utils": "^0.0.2",
     "dotenv": "^16.4.5",
-    "graphai": "^0.4.6",
+    "graphai": "^0.5.1",
     "json-schema-generator": "^2.0.6",
     "yargs": "^17.7.2"
   },
   "types": "./lib/graphai_cli.d.ts",
   "directories": {
-    "lib": "lib"
+    "lib": "lib",
+    "template": "template"
   }
 }