@aigne/cli 1.0.0-9 → 1.1.0

package/CHANGELOG.md

@@ -0,0 +1,9 @@
+# Changelog
+
+## [1.1.0](https://github.com/AIGNE-io/aigne-framework/compare/cli-v1.0.0...cli-v1.1.0) (2025-04-08)
+
+
+### Features
+
+* add `serve` command for @aigne/cli ([#54](https://github.com/AIGNE-io/aigne-framework/issues/54)) ([1cca843](https://github.com/AIGNE-io/aigne-framework/commit/1cca843f1760abe832b6651108fa858130f47355))
+* add agent library support ([#51](https://github.com/AIGNE-io/aigne-framework/issues/51)) ([1f0d34d](https://github.com/AIGNE-io/aigne-framework/commit/1f0d34ddda3154283a4bc958ddb9b68b4ac106b0))

package/dist/cli.d.ts

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

package/dist/commands/aigne.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function createAIGNECommand(): Command;

package/dist/commands/aigne.js

@@ -1,16 +1,20 @@
 import { Command } from "commander";
-import { version } from "../../package.json" assert { type: "json" };
+import pkg from "../../package.json" with { type: "json" };
+import { asciiLogo } from "../utils/ascii-logo.js";
 import { createCreateCommand } from "./create.js";
 import { createRunCommand } from "./run.js";
+import { createServeCommand } from "./serve.js";
 import { createTestCommand } from "./test.js";
 export function createAIGNECommand() {
+    console.log(asciiLogo);
     return new Command()
         .name("aigne")
         .description("CLI for AIGNE framework")
-        .version(version)
+        .version(pkg.version)
         .addCommand(createRunCommand())
         .addCommand(createTestCommand())
         .addCommand(createCreateCommand())
+        .addCommand(createServeCommand())
         .showHelpAfterError(true)
         .showSuggestionAfterError(true);
 }

package/dist/commands/create.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function createCreateCommand(): Command;

package/dist/commands/create.js

@@ -1,21 +1,21 @@
 import { existsSync, mkdirSync, readdirSync } from "node:fs";
 import { cp } from "node:fs/promises";
-import { isAbsolute, join, resolve } from "node:path";
+import { isAbsolute, join, relative, resolve } from "node:path";
 import { Command } from "commander";
 import inquirer from "inquirer";
 export function createCreateCommand() {
     return new Command("create")
         .description("Create a new aigne project with agent config files")
         .argument("[path]", "Path to create the project directory", ".")
-        .action(async (path) => {
-        let projectPath = path;
-        if (projectPath === ".") {
+        .action(async (_path) => {
+        let path = _path;
+        if (path === ".") {
             const answers = await inquirer.prompt([
                 {
                     type: "input",
                     name: "projectName",
                     message: "Project name:",
-                    default: path !== "." ? path : "my-aigne-project",
+                    default: _path !== "." ? _path : "my-aigne-project",
                     validate: (input) => {
                         if (input.trim() === "")
                             return "Project name cannot be empty.";
@@ -23,16 +23,16 @@ export function createCreateCommand() {
                     },
                 },
             ]);
-            projectPath = answers.projectName;
+            path = answers.projectName;
         }
-        const absolutePath = isAbsolute(path) ? path : resolve(process.cwd(), path);
-        const isPathNotEmpty = existsSync(absolutePath) && readdirSync(absolutePath).length > 0;
+        path = isAbsolute(path) ? path : resolve(process.cwd(), path);
+        const isPathNotEmpty = existsSync(path) && readdirSync(path).length > 0;
         if (isPathNotEmpty) {
             const answers = await inquirer.prompt([
                 {
                     type: "confirm",
                     name: "overwrite",
-                    message: `The directory "${absolutePath}" is not empty. Do you want to remove its contents?`,
+                    message: `The directory "${path}" is not empty. Do you want to remove its contents?`,
                     default: false,
                 },
             ]);
@@ -51,18 +51,18 @@ export function createCreateCommand() {
                 default: "default",
             },
         ]);
-        mkdirSync(absolutePath, { recursive: true });
+        mkdirSync(path, { recursive: true });
         const templatePath = join(import.meta.dirname, "../../templates", template);
         if (!existsSync(templatePath))
             throw new Error(`Template "${template}" not found.`);
         const files = readdirSync(templatePath);
         for (const file of files) {
             const source = join(templatePath, file);
-            const destination = join(absolutePath, file);
+            const destination = join(path, file);
             await cp(source, destination, { recursive: true, force: true });
         }
         console.log("\n✅ Aigne project created successfully!");
-        console.log(`\nTo use your new agent, run:\n  cd ${path} && npx -y aigne run`);
+        console.log(`\nTo use your new agent, run:\n  cd ${relative(process.cwd(), path)} && aigne run`);
     })
         .showHelpAfterError(true)
         .showSuggestionAfterError(true);

package/dist/commands/run.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function createRunCommand(): Command;

package/dist/commands/run.js

@@ -19,15 +19,13 @@ export function createRunCommand() {
                 for (const agent of engine.agents) {
                     console.log(`- ${agent.name}`);
                 }
-                process.exit(1);
+                throw new Error(`Agent "${options.agent}" not found`);
             }
         }
         else {
             agent = engine.agents[0];
-            if (!agent) {
-                console.error("No agents found in the specified path.");
-                process.exit(1);
-            }
+            if (!agent)
+                throw new Error("No agents found in the specified path");
         }
         const user = engine.call(agent);
         await runChatLoopInTerminal(user, {});

package/dist/commands/serve.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function createServeCommand(): Command;

package/dist/commands/serve.js

@@ -0,0 +1,32 @@
+import { isAbsolute, resolve } from "node:path";
+import { ExecutionEngine } from "@aigne/core";
+import { tryOrThrow } from "@aigne/core/utils/type-utils.js";
+import { Command } from "commander";
+import { serveMCPServer } from "../utils/serve-mcp.js";
+const DEFAULT_PORT = () => tryOrThrow(() => {
+    const { PORT } = process.env;
+    if (!PORT)
+        return 3000;
+    const port = Number.parseInt(PORT);
+    if (!port || !Number.isInteger(port))
+        throw new Error(`Invalid PORT: ${PORT}`);
+    return port;
+}, (error) => new Error(`parse PORT error ${error.message}`));
+export function createServeCommand() {
+    return new Command("serve")
+        .description("Serve the agents in the specified directory as a MCP server")
+        .argument("[path]", "Path to the agents directory", ".")
+        .option("--mcp", "Serve the agents as a MCP server")
+        .option("--port <port>", "Port to run the MCP server on", (s) => Number.parseInt(s))
+        .action(async (path, options) => {
+        const absolutePath = isAbsolute(path) ? path : resolve(process.cwd(), path);
+        const port = options.port || DEFAULT_PORT();
+        const engine = await ExecutionEngine.load({ path: absolutePath });
+        if (options.mcp)
+            await serveMCPServer({ engine, port });
+        else
+            throw new Error("Default server is not supported yet. Please use --mcp option");
+    })
+        .showHelpAfterError(true)
+        .showSuggestionAfterError(true);
+}

package/dist/commands/test.d.ts

@@ -0,0 +1,2 @@
+import { Command } from "commander";
+export declare function createTestCommand(): Command;

package/dist/utils/ascii-logo.d.ts

@@ -0,0 +1 @@
+export declare const asciiLogo: string;

package/dist/utils/ascii-logo.js

@@ -0,0 +1,18 @@
+import chalk from "chalk";
+import gradient from "gradient-string";
+import pkg from "../../package.json" with { type: "json" };
+const modernGradient = gradient(["#4facfe", "#7367f0", "#f86aad"]);
+const logo = `
+     _    ___ ____ _   _ _____
+    / \\  |_ _/ ___| \\ | | ____|
+   / _ \\  | | |  _|  \\| |  _|
+  / ___ \\ | | |_| | |\\  | |___
+ /_/   \\_\\___\\____|_| \\_|_____|
+`;
+const frameworkInfo = `v${pkg.version}`;
+const logoLines = logo.split("\n");
+const maxLength = Math.max(...logoLines.filter((line) => line.trim()).map((line) => line.length));
+const versionText = frameworkInfo;
+const padding = Math.floor((maxLength - versionText.length) / 2);
+const centeredVersion = " ".repeat(padding) + versionText;
+export const asciiLogo = `${modernGradient(logo)}\n${chalk.cyan(centeredVersion)}\n\n`;

package/dist/utils/serve-mcp.d.ts

@@ -0,0 +1,5 @@
+import { type ExecutionEngine } from "@aigne/core";
+export declare function serveMCPServer({ engine, port }: {
+    engine: ExecutionEngine;
+    port: number;
+}): Promise<import("http").Server<typeof import("http").IncomingMessage, typeof import("http").ServerResponse>>;

package/dist/utils/serve-mcp.js

@@ -0,0 +1,71 @@
+import { getMessage } from "@aigne/core";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import express from "express";
+import { ZodObject } from "zod";
+export async function serveMCPServer({ engine, port }) {
+    const server = new McpServer({
+        name: engine.name || "aigne-mcp-server",
+        version: "1.0.0",
+    }, {
+        capabilities: { tools: {} },
+        instructions: engine.description,
+    });
+    for (const agent of engine.agents) {
+        const schema = agent.inputSchema;
+        if (!(schema instanceof ZodObject))
+            throw new Error("Agent input schema must be a ZodObject");
+        server.tool(agent.name, agent.description || "", schema.shape, async (input) => {
+            const result = await engine.call(agent, input);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: getMessage(result) || JSON.stringify(result),
+                    },
+                ],
+            };
+        });
+    }
+    const app = express();
+    const transports = {};
+    app.get("/sse", async (req, res) => {
+        const transport = new SSEServerTransport("/messages", res);
+        transports[transport.sessionId] = transport;
+        req.on("close", () => {
+            delete transports[transport.sessionId];
+        });
+        await server.connect(transport);
+    });
+    app.post("/messages", async (req, res) => {
+        const sessionId = req.query.sessionId;
+        const transport = transports[sessionId];
+        if (transport) {
+            await transport.handlePostMessage(req, res);
+        }
+        else {
+            throw new HttpError(400, "No transport found for sessionId");
+        }
+    });
+    app.use(((error, _req, res, _next) => {
+        console.error("handle route error", { error });
+        res
+            .status(error instanceof HttpError ? error.status : 500)
+            .json({ error: { message: error.message } });
+    }));
+    const { promise, resolve, reject } = Promise.withResolvers();
+    const httpServer = app.listen(port, (error) => {
+        if (error)
+            reject(error);
+        resolve();
+    });
+    await promise;
+    return httpServer;
+}
+class HttpError extends Error {
+    status;
+    constructor(status, message) {
+        super(message);
+        this.status = status;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/cli",
-  "version": "1.0.0-9",
+  "version": "1.1.0",
   "description": "cli for AIGNE framework",
   "publishConfig": {
     "access": "public"
@@ -22,24 +22,49 @@
   "bin": {
     "aigne": "dist/cli.js"
   },
+  "type": "module",
+  "exports": {
+    "./*": "./dist/*"
+  },
+  "typesVersions": {
+    "*": {
+      "*": [
+        "./dist/*"
+      ]
+    }
+  },
   "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.8.0",
+    "chalk": "^5.4.1",
     "commander": "^13.1.0",
+    "express": "^5.1.0",
+    "gradient-string": "^3.0.0",
     "inquirer": "^12.5.0",
-    "openai": "^4.89.1",
-    "@aigne/core": "^1.5.0"
+    "openai": "^4.91.1",
+    "zod": "^3.24.2",
+    "@aigne/core": "^1.6.0"
   },
   "devDependencies": {
-    "@types/bun": "^1.2.6",
-    "@types/node": "^22.13.14",
+    "@types/bun": "^1.2.8",
+    "@types/express": "^5.0.1",
+    "@types/gradient-string": "^1.1.6",
+    "@types/node": "^22.14.0",
+    "detect-port": "^2.1.0",
     "npm-run-all": "^4.1.5",
     "rimraf": "^6.0.1",
-    "typescript": "^5.8.2"
+    "typescript": "^5.8.2",
+    "ufo": "^1.6.0",
+    "@aigne/test-utils": "^1.0.0"
   },
   "scripts": {
     "lint": "tsc --noEmit",
     "build": "tsc --build tsconfig.build.json",
-    "clean": "rimraf dist coverage",
-    "test": "bun test",
-    "test:coverage": "bun test --coverage --coverage-reporter=lcov --coverage-reporter=text"
+    "clean": "rimraf dist test/coverage templates/coverage",
+    "test": "run-s test:src test:templates",
+    "test:coverage": "run-s test:src:coverage test:templates:coverage",
+    "test:src": "bun --cwd test test",
+    "test:src:coverage": "bun --cwd test test --coverage --coverage-reporter=lcov --coverage-reporter=text",
+    "test:templates": "cd templates && node --test",
+    "test:templates:coverage": "cd templates && mkdir -p coverage && node --test --experimental-test-coverage --test-reporter=lcov --test-reporter-destination=coverage/lcov.info --test-reporter=spec --test-reporter-destination=stdout"
   }
 }

package/templates/default/aigne.yaml

@@ -1,5 +1,9 @@
 chat_model:
+  provider: openai
   name: gpt-4o-mini
   temperature: 0.8
 agents:
   - chat.yaml
+tools:
+  - sandbox.js
+  - filesystem.yaml

package/templates/default/filesystem.yaml

@@ -0,0 +1,3 @@
+type: mcp
+command: npx
+args: ["-y", "@modelcontextprotocol/server-filesystem", "."]