@agentforge/cli 0.3.9 → 0.4.1

package/dist/index.js

@@ -1,5 +1,5 @@
 import { Command } from 'commander';
-import chalk from 'chalk';
+import chalk7 from 'chalk';
 import path from 'path';
 import ora from 'ora';
 import inquirer from 'inquirer';
@@ -12,20 +12,20 @@ import { execa } from 'execa';
 var Logger = class {
   spinner = null;
   info(message) {
-    console.log(chalk.blue("\u2139"), message);
+    console.log(chalk7.blue("\u2139"), message);
   }
   success(message) {
-    console.log(chalk.green("\u2714"), message);
+    console.log(chalk7.green("\u2714"), message);
   }
   warn(message) {
-    console.log(chalk.yellow("\u26A0"), message);
+    console.log(chalk7.yellow("\u26A0"), message);
   }
   error(message) {
-    console.log(chalk.red("\u2716"), message);
+    console.log(chalk7.red("\u2716"), message);
   }
   debug(message) {
     if (process.env.DEBUG) {
-      console.log(chalk.gray("\u{1F41B}"), message);
+      console.log(chalk7.gray("\u{1F41B}"), message);
     }
   }
   startSpinner(message) {
@@ -58,19 +58,19 @@ var Logger = class {
     console.log();
   }
   divider() {
-    console.log(chalk.gray("\u2500".repeat(50)));
+    console.log(chalk7.gray("\u2500".repeat(50)));
   }
   header(message) {
     this.newLine();
-    console.log(chalk.bold.cyan(message));
+    console.log(chalk7.bold.cyan(message));
     this.divider();
   }
   code(code) {
-    console.log(chalk.gray("  " + code));
+    console.log(chalk7.gray("  " + code));
   }
   list(items) {
     items.forEach((item) => {
-      console.log(chalk.gray("  \u2022"), item);
+      console.log(chalk7.gray("  \u2022"), item);
     });
   }
 };
@@ -177,12 +177,13 @@ async function promptAgentSetup(defaults = {}) {
   ]);
 }
 async function promptToolSetup(defaults = {}) {
-  return inquirer.prompt([
+  const answers = await inquirer.prompt([
     {
       type: "input",
       name: "name",
       message: "Tool name:",
       default: defaults.name,
+      when: () => !defaults.name,
       validate: (input) => {
         if (!input) return "Tool name is required";
         if (!/^[a-zA-Z][a-zA-Z0-9]*$/.test(input)) {
@@ -201,25 +202,52 @@ async function promptToolSetup(defaults = {}) {
         { name: "File - File operations", value: "file" },
         { name: "Utility - General utilities", value: "utility" }
       ],
-      default: defaults.category || "utility"
+      default: defaults.category || "utility",
+      when: () => !defaults.category
     },
     {
       type: "input",
       name: "description",
       message: "Tool description:",
       default: defaults.description,
+      when: () => !defaults.description,
       validate: (input) => {
         if (!input) return "Tool description is required";
         return true;
       }
     },
+    {
+      type: "list",
+      name: "structure",
+      message: "Tool structure:",
+      choices: [
+        {
+          name: "Single file - Simple tools (<150 lines, single responsibility)",
+          value: "single"
+        },
+        {
+          name: "Multi-file - Complex tools (multiple providers, >150 lines)",
+          value: "multi"
+        }
+      ],
+      default: defaults.structure || "single",
+      when: () => !defaults.structure
+    },
     {
       type: "confirm",
       name: "generateTests",
       message: "Generate tests?",
-      default: defaults.generateTests !== false
+      default: defaults.generateTests !== false,
+      when: () => defaults.generateTests === void 0
     }
   ]);
+  return {
+    name: defaults.name || answers.name,
+    category: defaults.category || answers.category,
+    description: defaults.description || answers.description,
+    structure: defaults.structure || answers.structure,
+    generateTests: defaults.generateTests !== void 0 ? defaults.generateTests : answers.generateTests
+  };
 }
 var __filename$1 = fileURLToPath(import.meta.url);
 var __dirname$1 = path.dirname(__filename$1);
@@ -411,9 +439,9 @@ async function createCommand(projectName, options) {
       initGit: options.git
     });
     logger.newLine();
-    logger.info(`Creating project: ${chalk.cyan(answers.projectName)}`);
-    logger.info(`Template: ${chalk.cyan(answers.template)}`);
-    logger.info(`Package manager: ${chalk.cyan(answers.packageManager)}`);
+    logger.info(`Creating project: ${chalk7.cyan(answers.projectName)}`);
+    logger.info(`Template: ${chalk7.cyan(answers.template)}`);
+    logger.info(`Package manager: ${chalk7.cyan(answers.packageManager)}`);
     logger.newLine();
     logger.startSpinner("Creating project directory...");
     await ensureDir(targetPath);
@@ -461,7 +489,7 @@ async function createCommand(projectName, options) {
       }
     }
     logger.newLine();
-    logger.success(chalk.bold.green("\u2728 Project created successfully!"));
+    logger.success(chalk7.bold.green("\u2728 Project created successfully!"));
     logger.newLine();
     logger.header("\u{1F4DD} Next Steps");
     logger.list([
@@ -598,8 +626,8 @@ async function agentCreateCommand(name, options) {
       generateTests: options.test
     });
     logger.newLine();
-    logger.info(`Creating agent: ${chalk.cyan(answers.name)}`);
-    logger.info(`Pattern: ${chalk.cyan(answers.pattern)}`);
+    logger.info(`Creating agent: ${chalk7.cyan(answers.name)}`);
+    logger.info(`Pattern: ${chalk7.cyan(answers.pattern)}`);
     logger.newLine();
     const cwd = process.cwd();
     const agentDir = path.join(cwd, "src", "agents");
@@ -619,12 +647,12 @@ async function agentCreateCommand(name, options) {
       logger.succeedSpinner("Test file created");
     }
     logger.newLine();
-    logger.success(chalk.bold.green("\u2728 Agent created successfully!"));
+    logger.success(chalk7.bold.green("\u2728 Agent created successfully!"));
     logger.newLine();
     logger.header("\u{1F4DD} Next Steps");
     logger.list([
-      `Edit ${chalk.cyan(`src/agents/${answers.name}.ts`)} to customize your agent`,
-      answers.generateTests ? `Run ${chalk.cyan(`pnpm test tests/agents/${answers.name}.test.ts`)} to test your agent` : ""
+      `Edit ${chalk7.cyan(`src/agents/${answers.name}.ts`)} to customize your agent`,
+      answers.generateTests ? `Run ${chalk7.cyan(`pnpm test tests/agents/${answers.name}.test.ts`)} to test your agent` : ""
     ].filter(Boolean));
   } catch (error) {
     logger.error(`Failed to create agent: ${error.message}`);
@@ -744,10 +772,10 @@ async function agentListCommand(options) {
     const agentFiles = await findFiles("*.ts", agentDir);
     if (agentFiles.length === 0) {
       logger.warn("No agents found");
-      logger.info(`Create an agent with: ${chalk.cyan("agentforge agent:create <name>")}`);
+      logger.info(`Create an agent with: ${chalk7.cyan("agentforge agent:create <name>")}`);
       return;
     }
-    logger.info(`Found ${chalk.cyan(agentFiles.length)} agent(s):
+    logger.info(`Found ${chalk7.cyan(agentFiles.length)} agent(s):
 `);
     for (const file of agentFiles) {
       const agentName = path.basename(file, ".ts");
@@ -756,22 +784,22 @@ async function agentListCommand(options) {
         const content = await readFile(agentPath);
         const pattern = extractPattern(content);
         const description = extractDescription(content);
-        logger.info(chalk.bold.cyan(`  ${agentName}`));
+        logger.info(chalk7.bold.cyan(`  ${agentName}`));
         if (pattern) {
           logger.info(`    Pattern: ${pattern}`);
         }
         if (description) {
           logger.info(`    Description: ${description}`);
         }
-        logger.info(`    Path: ${chalk.gray(agentPath)}`);
+        logger.info(`    Path: ${chalk7.gray(agentPath)}`);
         logger.newLine();
       } else {
-        logger.info(`  \u2022 ${chalk.cyan(agentName)}`);
+        logger.info(`  \u2022 ${chalk7.cyan(agentName)}`);
       }
     }
     if (!options.verbose) {
       logger.newLine();
-      logger.info(`Use ${chalk.cyan("--verbose")} for more details`);
+      logger.info(`Use ${chalk7.cyan("--verbose")} for more details`);
     }
   } catch (error) {
     logger.error(`Failed to list agents: ${error.message}`);
@@ -798,10 +826,10 @@ async function agentTestCommand(name, options) {
     const testFile = path.join(cwd, "tests", "agents", `${name}.test.ts`);
     if (!await pathExists(testFile)) {
       logger.error(`Test file not found: ${testFile}`);
-      logger.info(`Create tests with: ${chalk.cyan(`agentforge agent:create ${name} --test`)}`);
+      logger.info(`Create tests with: ${chalk7.cyan(`agentforge agent:create ${name} --test`)}`);
       process.exit(1);
     }
-    logger.info(`Testing agent: ${chalk.cyan(name)}`);
+    logger.info(`Testing agent: ${chalk7.cyan(name)}`);
     logger.info(`Watch mode: ${options.watch ? "Yes" : "No"}`);
     logger.newLine();
     const packageManager = await detectPackageManager(cwd);
@@ -819,8 +847,8 @@ async function agentTestCommand(name, options) {
 async function agentDeployCommand(name, options) {
   try {
     logger.header("\u{1F680} Deploy Agent");
-    logger.info(`Agent: ${chalk.cyan(name)}`);
-    logger.info(`Environment: ${chalk.cyan(options.environment || "production")}`);
+    logger.info(`Agent: ${chalk7.cyan(name)}`);
+    logger.info(`Environment: ${chalk7.cyan(options.environment || "production")}`);
     logger.info(`Dry run: ${options.dryRun ? "Yes" : "No"}`);
     logger.newLine();
     if (options.dryRun) {
@@ -836,7 +864,7 @@ async function agentDeployCommand(name, options) {
       logger.succeedSpinner("Agent deployed successfully");
     }
     logger.newLine();
-    logger.success(chalk.bold.green("\u2728 Deployment completed!"));
+    logger.success(chalk7.bold.green("\u2728 Deployment completed!"));
     logger.newLine();
     logger.info("Note: Actual deployment implementation coming soon");
     logger.info("For now, please use the deployment templates in the templates/deployment directory");
@@ -852,38 +880,37 @@ async function toolCreateCommand(name, options) {
     const answers = await promptToolSetup({
       name,
       category: options.category,
+      structure: options.structure,
+      description: options.description,
       generateTests: options.test
     });
     logger.newLine();
-    logger.info(`Creating tool: ${chalk.cyan(answers.name)}`);
-    logger.info(`Category: ${chalk.cyan(answers.category)}`);
+    logger.info(`Creating tool: ${chalk7.cyan(answers.name)}`);
+    logger.info(`Category: ${chalk7.cyan(answers.category)}`);
+    logger.info(`Structure: ${chalk7.cyan(answers.structure)}`);
     logger.newLine();
     const cwd = process.cwd();
-    const toolDir = path.join(cwd, "src", "tools");
-    const toolFile = path.join(toolDir, `${answers.name}.ts`);
-    logger.startSpinner("Creating tool file...");
-    await ensureDir(toolDir);
-    const toolContent = generateToolContent(answers.name, answers.category, answers.description);
-    await writeFile(toolFile, toolContent);
-    logger.succeedSpinner("Tool file created");
-    if (answers.generateTests) {
-      logger.startSpinner("Creating test file...");
-      const testDir = path.join(cwd, "tests", "tools");
-      const testFile = path.join(testDir, `${answers.name}.test.ts`);
-      await ensureDir(testDir);
-      const testContent = generateTestContent2(answers.name);
-      await writeFile(testFile, testContent);
-      logger.succeedSpinner("Test file created");
+    if (answers.structure === "multi") {
+      await createMultiFileTool(cwd, answers);
+    } else {
+      await createSingleFileTool(cwd, answers);
     }
     logger.newLine();
-    logger.success(chalk.bold.green("\u2728 Tool created successfully!"));
+    logger.success(chalk7.bold.green("\u2728 Tool created successfully!"));
     logger.newLine();
     logger.header("\u{1F4DD} Next Steps");
-    logger.list([
-      `Edit ${chalk.cyan(`src/tools/${answers.name}.ts`)} to implement your tool`,
-      answers.generateTests ? `Run ${chalk.cyan(`pnpm test tests/tools/${answers.name}.test.ts`)} to test your tool` : "",
+    const nextSteps = answers.structure === "multi" ? [
+      `Edit ${chalk7.cyan(`src/tools/${answers.name}/index.ts`)} to implement your tool`,
+      `Add providers in ${chalk7.cyan(`src/tools/${answers.name}/providers/`)}`,
+      `Define types in ${chalk7.cyan(`src/tools/${answers.name}/types.ts`)}`,
+      answers.generateTests ? `Run ${chalk7.cyan(`pnpm test ${answers.name}`)} to test your tool` : "",
       `Register the tool in your agent's tool registry`
-    ].filter(Boolean));
+    ] : [
+      `Edit ${chalk7.cyan(`src/tools/${answers.name}.ts`)} to implement your tool`,
+      answers.generateTests ? `Run ${chalk7.cyan(`pnpm test tests/tools/${answers.name}.test.ts`)} to test your tool` : "",
+      `Register the tool in your agent's tool registry`
+    ];
+    logger.list(nextSteps.filter(Boolean));
   } catch (error) {
     logger.error(`Failed to create tool: ${error.message}`);
     process.exit(1);
@@ -939,6 +966,46 @@ describe('${capitalize2(name)} Tool', () => {
 function capitalize2(str) {
   return str.charAt(0).toUpperCase() + str.slice(1);
 }
+async function createSingleFileTool(cwd, answers) {
+  const toolDir = path.join(cwd, "src", "tools");
+  const toolFile = path.join(toolDir, `${answers.name}.ts`);
+  logger.startSpinner("Creating tool file...");
+  await ensureDir(toolDir);
+  const toolContent = generateToolContent(answers.name, answers.category, answers.description);
+  await writeFile(toolFile, toolContent);
+  logger.succeedSpinner("Tool file created");
+  if (answers.generateTests) {
+    logger.startSpinner("Creating test file...");
+    const testDir = path.join(cwd, "tests", "tools");
+    const testFile = path.join(testDir, `${answers.name}.test.ts`);
+    await ensureDir(testDir);
+    const testContent = generateTestContent2(answers.name);
+    await writeFile(testFile, testContent);
+    logger.succeedSpinner("Test file created");
+  }
+}
+async function createMultiFileTool(cwd, answers) {
+  const toolDir = path.join(cwd, "src", "tools", answers.name);
+  logger.startSpinner("Creating tool directory structure...");
+  await ensureDir(toolDir);
+  const templatePath = getTemplatePath("tool-multi");
+  const replacements = {
+    TOOL_NAME: answers.name,
+    TOOL_NAME_PASCAL: capitalize2(answers.name),
+    TOOL_NAME_CAMEL: answers.name.charAt(0).toLowerCase() + answers.name.slice(1),
+    TOOL_DESCRIPTION: answers.description,
+    TOOL_CATEGORY: answers.category
+  };
+  await copyTemplate(templatePath, toolDir, replacements);
+  logger.succeedSpinner("Tool directory structure created");
+  if (!answers.generateTests) {
+    logger.startSpinner("Cleaning up test files...");
+    const fs4 = await import('fs-extra');
+    const testDir = path.join(toolDir, "__tests__");
+    await fs4.remove(testDir);
+    logger.succeedSpinner("Test files removed");
+  }
+}
 async function toolListCommand(options) {
   try {
     logger.header("\u{1F4CB} List Tools");
@@ -947,7 +1014,7 @@ async function toolListCommand(options) {
     const toolFiles = await findFiles("*.ts", toolDir);
     if (toolFiles.length === 0) {
       logger.warn("No tools found");
-      logger.info(`Create a tool with: ${chalk.cyan("agentforge tool:create <name>")}`);
+      logger.info(`Create a tool with: ${chalk7.cyan("agentforge tool:create <name>")}`);
       return;
     }
     let filteredTools = toolFiles;
@@ -966,7 +1033,7 @@ async function toolListCommand(options) {
       logger.warn(`No tools found in category: ${options.category}`);
       return;
     }
-    logger.info(`Found ${chalk.cyan(filteredTools.length)} tool(s):
+    logger.info(`Found ${chalk7.cyan(filteredTools.length)} tool(s):
 `);
     for (const file of filteredTools) {
       const toolName = path.basename(file, ".ts");
@@ -975,22 +1042,22 @@ async function toolListCommand(options) {
         const content = await readFile(toolPath);
         const category = extractCategory(content);
         const description = extractDescription2(content);
-        logger.info(chalk.bold.cyan(`  ${toolName}`));
+        logger.info(chalk7.bold.cyan(`  ${toolName}`));
         if (category) {
           logger.info(`    Category: ${category}`);
         }
         if (description) {
           logger.info(`    Description: ${description}`);
         }
-        logger.info(`    Path: ${chalk.gray(toolPath)}`);
+        logger.info(`    Path: ${chalk7.gray(toolPath)}`);
         logger.newLine();
       } else {
-        logger.info(`  \u2022 ${chalk.cyan(toolName)}`);
+        logger.info(`  \u2022 ${chalk7.cyan(toolName)}`);
       }
     }
     if (!options.verbose) {
       logger.newLine();
-      logger.info(`Use ${chalk.cyan("--verbose")} for more details`);
+      logger.info(`Use ${chalk7.cyan("--verbose")} for more details`);
     }
   } catch (error) {
     logger.error(`Failed to list tools: ${error.message}`);
@@ -1012,10 +1079,10 @@ async function toolTestCommand(name, options) {
     const testFile = path.join(cwd, "tests", "tools", `${name}.test.ts`);
     if (!await pathExists(testFile)) {
       logger.error(`Test file not found: ${testFile}`);
-      logger.info(`Create tests with: ${chalk.cyan(`agentforge tool:create ${name} --test`)}`);
+      logger.info(`Create tests with: ${chalk7.cyan(`agentforge tool:create ${name} --test`)}`);
       process.exit(1);
     }
-    logger.info(`Testing tool: ${chalk.cyan(name)}`);
+    logger.info(`Testing tool: ${chalk7.cyan(name)}`);
     logger.info(`Watch mode: ${options.watch ? "Yes" : "No"}`);
     logger.newLine();
     const packageManager = await detectPackageManager(cwd);
@@ -1033,8 +1100,8 @@ async function toolTestCommand(name, options) {
 async function toolPublishCommand(name, options) {
   try {
     logger.header("\u{1F4E6} Publish Tool");
-    logger.info(`Tool: ${chalk.cyan(name)}`);
-    logger.info(`Tag: ${chalk.cyan(options.tag || "latest")}`);
+    logger.info(`Tool: ${chalk7.cyan(name)}`);
+    logger.info(`Tag: ${chalk7.cyan(options.tag || "latest")}`);
     logger.info(`Dry run: ${options.dryRun ? "Yes" : "No"}`);
     logger.newLine();
     if (options.dryRun) {
@@ -1066,7 +1133,7 @@ async function toolPublishCommand(name, options) {
       logger.succeedSpinner("Published to npm");
     }
     logger.newLine();
-    logger.success(chalk.bold.green("\u2728 Tool published successfully!"));
+    logger.success(chalk7.bold.green("\u2728 Tool published successfully!"));
     logger.newLine();
     logger.info("Note: Actual npm publishing implementation coming soon");
     logger.info("For now, please use npm publish manually");
@@ -1091,7 +1158,7 @@ agent.command("list").description("List all agents").option("-v, --verbose", "Sh
 agent.command("test <name>").description("Test a specific agent").option("-w, --watch", "Watch mode").action(agentTestCommand);
 agent.command("deploy <name>").description("Deploy an agent").option("-e, --environment <env>", "Deployment environment", "production").option("--dry-run", "Dry run without actual deployment").action(agentDeployCommand);
 var tool = program.command("tool").description("Manage tools");
-tool.command("create <name>").description("Create a new tool").option("-c, --category <category>", "Tool category (web, data, file, utility)", "utility").option("--no-test", "Skip test generation").action(toolCreateCommand);
+tool.command("create <name>").description("Create a new tool").option("-c, --category <category>", "Tool category (web, data, file, utility)", "utility").option("-s, --structure <structure>", "Tool structure (single, multi)", "single").option("-d, --description <description>", "Tool description").option("--no-test", "Skip test generation").action(toolCreateCommand);
 tool.command("list").description("List all tools").option("-c, --category <category>", "Filter by category").option("-v, --verbose", "Show detailed information").action(toolListCommand);
 tool.command("test <name>").description("Test a specific tool").option("-w, --watch", "Watch mode").action(toolTestCommand);
 tool.command("publish <name>").description("Publish a tool to npm").option("--tag <tag>", "npm tag", "latest").option("--dry-run", "Dry run without actual publishing").action(toolPublishCommand);
@@ -1101,7 +1168,7 @@ async function run() {
     await program.parseAsync(process.argv);
   } catch (error) {
     if (error.code !== "commander.help" && error.code !== "commander.version") {
-      console.error(chalk.red("Error:"), error.message);
+      console.error(chalk7.red("Error:"), error.message);
       process.exit(1);
     }
   }