@graphai/agent_generator 0.0.2

package/README.md

@@ -0,0 +1,32 @@
+# graphai agent generator
+
+This tool is a graphai application in which the user writes a simple specification, and based on that specification, AI fully automates the creation of a graphai agent.
+
+## update spec
+
+edit template/spec.md
+
+# set openai key
+
+edit .env and set OPENAI_API_KEY=sk-....
+
+## run
+
+```
+yarn run run
+```
+
+## then
+
+generate agent in tmp directory
+
+
+
+
+
+
+
+
+
+
+

package/lib/index.d.ts

@@ -0,0 +1,2 @@
+import templateAgent from "./template_agent";
+export { templateAgent };

package/lib/index.js

@@ -0,0 +1,8 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.templateAgent = void 0;
+const template_agent_1 = __importDefault(require("./template_agent"));
+exports.templateAgent = template_agent_1.default;

package/lib/run.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/run.js

@@ -0,0 +1,376 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+//仕様と必要な情報（npmやデータ）をテキストで用意する
+// generatorを呼び出してスケルトンを作る
+// agentの中身を実装する
+// unit testを動かす -> 失敗したら結果やエラーを元に3を再実装
+// できたら、documentもつくる
+const path = __importStar(require("node:path"));
+const graphai_1 = require("graphai");
+const openai_agent_1 = require("@graphai/openai_agent");
+const vanilla_1 = require("@graphai/vanilla");
+const vanilla_node_agents_1 = require("@graphai/vanilla_node_agents");
+const shell_utilty_agent_1 = require("@graphai/shell_utilty_agent");
+const dotenv_1 = __importDefault(require("dotenv"));
+dotenv_1.default.config({ quiet: true });
+const tools = [
+    {
+        type: "function",
+        function: {
+            name: "generate_package",
+            description: "generate agent skelton",
+            parameters: {
+                type: "object",
+                properties: {
+                    agentName: {
+                        type: "string",
+                        description: "The agent name. Separate words with spaces.",
+                    },
+                    description: {
+                        type: "string",
+                        description: "Explain what the agent does",
+                    },
+                    category: {
+                        type: "string",
+                        description: "category of the agent.",
+                    },
+                    npmPackages: {
+                        type: "string",
+                        description: "list of npm package if you need. Separate packages with spaces. ",
+                    },
+                },
+                required: ["agentName", "description", "category"],
+            },
+        },
+    },
+];
+/*
+const tools_npminstall = [
+  {
+    type: "function",
+    function: {
+      name: "installNpm",
+      description: "install npm package",
+      parameters: {
+        type: "object",
+        properties: {
+          npmPackages: {
+            type: "string",
+            description: "list of npm package if you need. Separate packages with spaces. ",
+          },
+        },
+        required: ["npmPackages"],
+      },
+    },
+  },
+];
+*/
+const main = async () => {
+    const graphData = {
+        version: 0.5,
+        nodes: {
+            packageBaseDir: {
+                value: "",
+            },
+            templateBaseDir: {
+                value: "",
+            },
+            specPrompt: {
+                value: "",
+            },
+            implementPrompt: {
+                value: "",
+            },
+            errorPrompt: {
+                value: "",
+            },
+            specFileReader: {
+                agent: "fileReadAgent",
+                inputs: {
+                    array: ["template/spec.md", "template/spec_base.md"],
+                },
+                params: {
+                    baseDir: ":templateBaseDir",
+                    outputType: "text",
+                },
+                console: { after: true },
+                isResult: true,
+            },
+            specFile: {
+                agent: "copyAgent",
+                inputs: { data: "${:specFileReader.array.$0}\n\n${:specFileReader.array.$1}" },
+                isResult: true,
+            },
+            specLLM: {
+                agent: "openAIAgent",
+                inputs: {
+                    prompt: "${:specPrompt}\n\n ${:specFile.data}",
+                    tools,
+                },
+                console: { after: true },
+            },
+            packageInfo: {
+                agent: "stringCaseVariantsAgent",
+                params: {
+                    suffix: "agent",
+                },
+                inputs: {
+                    text: ":specLLM.tool.arguments.agentName",
+                },
+                isResult: true,
+            },
+            createSkeleton: {
+                agent: "runShellAgent",
+                inputs: {
+                    commands: [
+                        "npm",
+                        "create",
+                        "graphai-agent@latest",
+                        "--",
+                        "-c",
+                        "--agentName",
+                        ":packageInfo.kebabCase",
+                        "--description",
+                        ":specLLM.tool.arguments.description",
+                        "--author",
+                        "me",
+                        "--license",
+                        "MIT",
+                        "--category",
+                        ":specLLM.tool.arguments.category",
+                        "--outdir",
+                        ":packageBaseDir",
+                    ],
+                    baseDir: ":packageBaseDir",
+                },
+            },
+            yarnInstall: {
+                agent: "runShellAgent",
+                params: {},
+                inputs: {
+                    commands: ["yarn", "install"],
+                    dirs: [":packageBaseDir", ":packageInfo.kebabCase"],
+                    waiting: ":createSkeleton",
+                },
+            },
+            packageDir: {
+                agent: "copyAgent",
+                inputs: {
+                    text: "${:packageBaseDir}/${:packageInfo.kebabCase}",
+                },
+            },
+            yarnAdd: {
+                agent: "runShellAgent",
+                if: ":specLLM.tool.arguments.npmPackages",
+                defaultValue: {},
+                inputs: {
+                    commands: ["yarn", "add", "${:specLLM.tool.arguments.npmPackages}"],
+                    baseDir: ":packageDir.text",
+                    wait: ":yarnInstall",
+                },
+            },
+            srcFile: {
+                agent: "pathUtilsAgent",
+                params: { method: "join" },
+                inputs: { dirs: [":packageInfo.kebabCase", "src", "${:packageInfo.snakeCase}.ts"] },
+            },
+            programmer: {
+                agent: "nestedAgent",
+                isResult: true,
+                inputs: {
+                    waiting: ":yarnAdd",
+                    packageInfo: ":packageInfo",
+                    srcFile: ":srcFile",
+                    specFile: ":specFile",
+                    packageBaseDir: ":packageBaseDir",
+                    implementPrompt: ":implementPrompt",
+                    errorPrompt: ":errorPrompt",
+                    npmPackages: [":specLLM.tool.arguments.npmPackages"],
+                },
+                graph: {
+                    loop: {
+                        while: ":yarnTest.error",
+                    },
+                    nodes: {
+                        npmPackages: {
+                            update: ":npmPackagesStack.array",
+                        },
+                        error: {
+                            value: "",
+                            update: ":yarnTest.stdout",
+                        },
+                        sourceFile: {
+                            agent: "fileReadAgent",
+                            inputs: {
+                                file: ":srcFile.path",
+                            },
+                            params: {
+                                baseDir: ":packageBaseDir",
+                                outputType: "text",
+                            },
+                        },
+                        llm: {
+                            agent: "openAIAgent",
+                            inputs: {
+                                system: ":specFile.data",
+                                prompt: "${:implementPrompt}\n\n###ソース###\n\n${:sourceFile.data}\n\n\n###npmは以下が追加さています###\n${:npmPackages.join(,)}\n\n###${:errorPrompt}###\n\n${:error}",
+                                // model: "o1-mini",
+                                // tools: tools_npminstall,
+                            },
+                            console: { before: true, after: true },
+                        },
+                        yarnAdd: {
+                            agent: "runShellAgent",
+                            if: ":llm.tool.arguments.npmPackages",
+                            inputs: {
+                                command: ["yarn", "add", "${:llm.tool.arguments.npmPackages}"],
+                                dirs: [":packageBaseDir", ":packageInfo.kebabCase"],
+                            },
+                            defaultValue: {},
+                        },
+                        res: {
+                            agent: "copyAgent",
+                            inputs: {
+                                text: ":llm.text.codeBlock()",
+                            },
+                            isResult: true,
+                        },
+                        writeFile: {
+                            if: ":res.text",
+                            defaultValue: {},
+                            agent: "fileWriteAgent",
+                            inputs: {
+                                file: ":srcFile.path",
+                                text: ":llm.text.codeBlock()",
+                            },
+                            params: {
+                                baseDir: ":packageBaseDir",
+                            },
+                            console: {
+                                before: true,
+                            },
+                        },
+                        yarnTest: {
+                            agent: "runShellAgent",
+                            params: {},
+                            inputs: {
+                                commands: ["yarn", "run", "test"],
+                                waiting: ":writeFile",
+                                dirs: [":packageBaseDir", ":packageInfo.kebabCase"],
+                            },
+                            console: {
+                                after: true,
+                            },
+                        },
+                        npmPackagesStack: {
+                            agent: "pushAgent",
+                            inputs: {
+                                array: ":npmPackages",
+                                items: [":llm.tool.arguments.npmPackages"],
+                            },
+                            console: {
+                                after: true,
+                            },
+                        },
+                        /*
+                        isLoop: {
+                          agent: "copyAgent",
+                          inputs: { array: [":yarnTest.error", ":yarnAdd"]}
+                          },
+                        */
+                    },
+                },
+            },
+            final: {
+                agent: "runShellAgent",
+                params: {},
+                inputs: {
+                    commands: ["yarn", "run", "build"], // , "&&", "yarn", "run", "doc"
+                    dirs: [":packageBaseDir", ":packageInfo.kebabCase"],
+                    waiting: ":programmer",
+                },
+            },
+            final2: {
+                agent: "runShellAgent",
+                params: {},
+                inputs: {
+                    commands: ["yarn", "run", "doc"],
+                    dirs: [":packageBaseDir", ":packageInfo.kebabCase"],
+                    waiting: ":programmer",
+                },
+            },
+            /*
+            writeSpec: {
+              agent: "fileWriteAgent",
+              inputs: {
+                file: "spec.txt",
+                waiting: ":programmer",
+                text: ":specFileReader.array.$0",
+              },
+              params: {
+                baseDir: "${:packageBaseDir}/${:packageInfo.kebabCase}",
+                outputType: "text",
+              },
+            },
+            */
+        },
+    };
+    const graph = new graphai_1.GraphAI(graphData, {
+        openAIAgent: openai_agent_1.openAIAgent,
+        copyAgent: vanilla_1.copyAgent,
+        pushAgent: vanilla_1.pushAgent,
+        fileReadAgent: vanilla_node_agents_1.fileReadAgent,
+        fileWriteAgent: vanilla_node_agents_1.fileWriteAgent,
+        nestedAgent: vanilla_1.nestedAgent,
+        runShellAgent: shell_utilty_agent_1.runShellAgent,
+        stringCaseVariantsAgent: vanilla_1.stringCaseVariantsAgent,
+        pathUtilsAgent: vanilla_node_agents_1.pathUtilsAgent,
+    });
+    graph.injectValue("templateBaseDir", path.resolve(__dirname, ".."));
+    // graph.injectValue("packageBaseDir", "/Users/isamu/ss/llm/ai-generated-graphai-agents");
+    graph.injectValue("packageBaseDir", path.resolve(__dirname, "..", "tmp"));
+    graph.injectValue("specPrompt", "以下の仕様を元に必要な情報を教えて下さい。結果はgenerate_packageで返してください。npmパッケージが必要な場合はそれも一覧で返してください。");
+    graph.injectValue("packageBaseDir", path.resolve(__dirname, "..", "tmp"));
+    graph.injectValue("specPrompt", "以下の仕様を元に必要な情報を教えて下さい。結果はgenerate_packageで返してください。npmパッケージが必要な場合はそれも一覧で返してください。");
+    graph.injectValue("implementPrompt", "以下のソースを仕様に従って変更して");
+    graph.injectValue("errorPrompt", "エラー情報");
+    const result = (await graph.run());
+    console.log(result);
+};
+main();

package/lib/run2.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/run2.js

@@ -0,0 +1,100 @@
+"use strict";
+// import * as path from "node:path";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const graphai_1 = require("graphai");
+const openai_agent_1 = require("@graphai/openai_agent");
+const vanilla_1 = require("@graphai/vanilla");
+const vanilla_node_agents_1 = require("@graphai/vanilla_node_agents");
+const shell_utilty_agent_1 = require("@graphai/shell_utilty_agent");
+const dotenv_1 = __importDefault(require("dotenv"));
+dotenv_1.default.config({ quiet: true });
+const tools = [
+    {
+        type: "function",
+        function: {
+            name: "searchNpm",
+            description: "searchNpm",
+            parameters: {
+                type: "object",
+                properties: {
+                    keywords: {
+                        type: "string",
+                        description: "Consists of letters and numbers, with a space if more than one",
+                    },
+                },
+                required: ["keywords"],
+            },
+        },
+    },
+];
+const tools2 = [
+    {
+        type: "function",
+        function: {
+            name: "choiseNpmPackageName",
+            description: "choice npm package",
+            parameters: {
+                type: "object",
+                properties: {
+                    packageName: {
+                        type: "string",
+                        description: "npm package name",
+                    },
+                },
+                required: ["keywords"],
+            },
+        },
+    },
+];
+const main = async () => {
+    const graphData = {
+        version: 0.5,
+        nodes: {
+            prompt: {
+                // value: "s3のクライアントを作りたいです。",
+                value: "税務計算のソフトを作りたいです。",
+            },
+            npmKeywords: {
+                agent: "openAIAgent",
+                inputs: {
+                    prompt: "${:prompt} 必要なnpmを探すので検索を",
+                    tools,
+                },
+                console: { after: true },
+            },
+            search: {
+                agent: "runShellAgent",
+                params: {},
+                inputs: {
+                    command: "npm search ${:npmKeywords.tool.arguments.keywords}",
+                },
+                console: { after: true },
+            },
+            choiceNpm: {
+                agent: "openAIAgent",
+                inputs: {
+                    system: "${:prompt}推奨するnpmのパッケージをえらんでください。選んだらchoiseNpmPackageNameの関数をtoolsで返して",
+                    prompt: "${:search.text}",
+                    tools: tools2,
+                },
+                console: { after: true },
+            },
+        },
+    };
+    const graph = new graphai_1.GraphAI(graphData, {
+        openAIAgent: openai_agent_1.openAIAgent,
+        copyAgent: vanilla_1.copyAgent,
+        fileReadAgent: vanilla_node_agents_1.fileReadAgent,
+        fileWriteAgent: vanilla_node_agents_1.fileWriteAgent,
+        nestedAgent: vanilla_1.nestedAgent,
+        runShellAgent: shell_utilty_agent_1.runShellAgent,
+        stringCaseVariantsAgent: vanilla_1.stringCaseVariantsAgent,
+        pathUtilsAgent: vanilla_node_agents_1.pathUtilsAgent,
+    });
+    const result = (await graph.run());
+    console.log(result);
+};
+main();

package/lib/template_agent.d.ts

@@ -0,0 +1,4 @@
+import { AgentFunction, AgentFunctionInfo } from "graphai";
+export declare const templateAgent: AgentFunction;
+declare const templateAgentInfo: AgentFunctionInfo;
+export default templateAgentInfo;

package/lib/template_agent.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.templateAgent = void 0;
+const templateAgent = async ({ params, namedInputs }) => {
+    return { params, namedInputs };
+};
+exports.templateAgent = templateAgent;
+const templateAgentInfo = {
+    name: "templateAgent",
+    agent: exports.templateAgent,
+    mock: exports.templateAgent,
+    samples: [],
+    description: "Template Agent",
+    category: ["template"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = templateAgentInfo;

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@graphai/agent_generator",
+  "version": "0.0.2",
+  "description": "",
+  "main": "lib/index.js",
+  "files": [
+    "./lib"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "eslint": "eslint src --fix",
+    "format": "prettier --write '{src,tests}/**/*.ts'",
+    "doc": "echo hello",
+    "run": "npx ts-node src/run.ts",
+    "run2": "npx ts-node src/run2.ts",
+    "test_run": "node --test --require ts-node/register ./tests/run_*.ts",
+    "test": "echo hello",
+    "ci": "yarn run format && yarn run eslint && yarn run test && yarn run build"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/receptron/graphai-agents.git"
+  },
+  "author": "",
+  "license": "MIT",
+  "bugs": {
+    "url": "https://github.com/receptron/graphai-agents/issues"
+  },
+  "homepage": "https://github.com/receptron/graphai-agents#readme",
+  "devDependencies": {},
+  "dependencies": {
+    "@graphai/openai_agent": "^2.0.9",
+    "@graphai/shell_utilty_agent": "^0.1.0",
+    "@graphai/vanilla": "^2.0.12",
+    "@graphai/vanilla_node_agents": "^2.0.4",
+    "graphai": "^2.0.16"
+  }
+}