ai-test-cli 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Chijindu Azubuike
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,115 @@
+<h1 align="center">AI Test CLI 🧪🤖</h1>
+
+<p align="center">
+  <strong>An autonomous AI software testing engineer that dynamically generates, evaluates, and auto-fixes test suites for complex codebases.</strong>
+</p>
+
+<p align="center">
+  <a href="https://badge.fury.io/js/ai-test-cli"><img src="https://badge.fury.io/js/ai-test-cli.svg" alt="npm version" /></a>
+  <a href="https://opensource.org/licenses/MIT"><img src="https://img.shields.io/badge/License-MIT-yellow.svg" alt="License: MIT" /></a>
+</p>
+
+---
+
+## 🌟 Why AI Test CLI?
+
+Most AI testing tools just write boilerplate code and give up when tests fail. **AI Test CLI is different.** It operates like a real Senior QA engineer using a fully autonomous "Agentic Loop":
+
+1. **Intelligent Generation**: It reads your file, understands its imports, scans your workspace structure, and writes comprehensive tests using your existing framework (Jest, Vitest, Mocha, etc.).
+2. **Massive File Support**: Got an 11,000+ line controller? No problem. The **Agentic Chunking Generator** explores the file function-by-function, incrementally building out coverage without blowing up LLM token limits.
+3. **Smart Context for Local Models**: Running Ollama on a MacBook? The CLI automatically detects Local LLMs and engages "Extreme Chunking Mode" to guarantee your computer never crashes from context-window overflow.
+4. **Autonomous Auto-Fixing**: It actually *runs* the tests it writes. If they fail, it reads the error stack trace, uses search tools to inspect the broken dependencies, and dynamically applies patches until the test passes.
+5. **Bring Your Own Key (BYOK)**: Supports DeepSeek, OpenAI, Anthropic, Gemini, and Local Models (Ollama/LMStudio) via the Vercel AI SDK. 
+
+---
+
+## 🚀 Quickstart
+
+Install globally via npm:
+
+```bash
+npm install -g ai-test-cli
+```
+
+Initialize the configuration in your project root. This creates an `.aitestrc.json` file.
+
+```bash
+aitest init
+```
+
+Set your API key in your `.env` file, or set it in your environment:
+
+```bash
+# Depending on your chosen provider:
+export DEEPSEEK_API_KEY="sk-..."
+export OPENAI_API_KEY="sk-..."
+export ANTHROPIC_API_KEY="sk-..."
+export GEMINI_API_KEY="AIza..."
+export CUSTOM_API_KEY="sk-..."
+```
+
+---
+
+## 🛠️ Usage
+
+### Generate Tests
+
+Generate a test suite for a single file:
+```bash
+aitest generate --file path/to/your/file.ts
+```
+
+Generate tests for your entire project (this will skip files that already have tests):
+```bash
+aitest generate --all
+```
+
+**Iterative Coverage**: If a test file already exists, running `aitest generate` again will automatically evaluate the existing file for missing coverage and append tests for untested functions!
+
+### Auto-Fix Broken Tests
+
+Got a test suite that's failing because of broken mocks or outdated logic? Unleash the agentic fixer:
+```bash
+aitest fix
+```
+The AI will run your entire test suite, isolate the failures, analyze the error stack traces, and incrementally patch your test files until they turn green.
+
+### Explain Failures
+Don't want the AI to fix it for you? Just ask it to explain why a test is failing:
+```bash
+aitest explain
+```
+
+---
+
+## 🧠 Supported Providers
+
+The CLI supports the following providers out of the box via your `.aitestrc.json`:
+- `deepseek` (Highly Recommended: 128k context, extremely cheap)
+- `openai` 
+- `anthropic`
+- `gemini`
+- `ollama` (Local models, automatically triggers Smart Context chunking)
+
+---
+
+## 🛡️ Safety Mechanisms
+
+AI Test CLI is built with strict API billing safeguards:
+- **Stagnation Protection**: If the AI gets confused and fails to write a test chunk after 10 attempts, it safely aborts.
+- **Duplicate Action Protection**: If the AI falls into a hallucinatory repetition loop, the CLI detects it and breaks the loop instantly to save your API credits.
+- **Smart Context**: Drastically reduces token consumption for massive files by chunking out irrelevant lines during the repair loop.
+
+---
+
+## ☕ Support the Project
+
+Did AI Test CLI just save you 20 hours of grueling unit testing? Support the creator and buy them a coffee!
+
+<a href="https://buymeacoffee.com/cijaytechnh" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/v2/default-yellow.png" alt="Buy Me A Coffee" style="height: 60px !important;width: 217px !important;" ></a>
+
+---
+
+## 📄 License
+
+MIT License. See [LICENSE](LICENSE) for more information.

package/bin/aitest.js

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+import '../dist/index.js';

package/dist/commands/config.d.ts

@@ -0,0 +1,5 @@
+import { Command } from 'commander';
+
+declare const configCommand: Command;
+
+export { configCommand };

package/dist/commands/config.js

@@ -0,0 +1,80 @@
+// src/commands/config.ts
+import { Command } from "commander";
+
+// src/core/config.ts
+import { cosmiconfig } from "cosmiconfig";
+import { writeFile } from "fs/promises";
+import { resolve } from "path";
+
+// src/core/logger.ts
+import chalk from "chalk";
+import ora from "ora";
+var logger = {
+  info: (msg) => console.log(chalk.blue("\u2139"), msg),
+  success: (msg) => console.log(chalk.green("\u2714"), msg),
+  error: (msg) => console.error(chalk.red("\u2716"), msg),
+  warn: (msg) => console.warn(chalk.yellow("\u26A0"), msg),
+  log: (msg) => console.log(msg),
+  spinner: (text) => ora(text)
+};
+
+// src/core/config.ts
+var moduleName = "aitest";
+var explorer = cosmiconfig(moduleName);
+async function loadConfig() {
+  try {
+    const result = await explorer.search();
+    if (result) {
+      return result.config;
+    }
+    return null;
+  } catch (error) {
+    logger.error(`Failed to load config: ${error}`);
+    return null;
+  }
+}
+async function saveConfig(config, targetDir = process.cwd()) {
+  const configPath = resolve(targetDir, ".aitestrc.json");
+  try {
+    await writeFile(configPath, JSON.stringify(config, null, 2), "utf-8");
+    logger.success(`Saved config to ${configPath}`);
+  } catch (error) {
+    logger.error(`Failed to save config: ${error}`);
+    throw error;
+  }
+}
+
+// src/commands/config.ts
+var configCommand = new Command("config").description("Manage settings (e.g., provider, model, apiKey)");
+configCommand.command("set <key> <value>").description("Set a configuration value").action(async (key, value) => {
+  let config = await loadConfig();
+  if (!config) {
+    config = { provider: "openai", model: "gpt-4o" };
+  }
+  let parsedValue = value;
+  if (value === "true") parsedValue = true;
+  if (value === "false") parsedValue = false;
+  if (!isNaN(Number(value))) parsedValue = Number(value);
+  config[key] = parsedValue;
+  await saveConfig(config);
+});
+configCommand.command("list").description("List current configuration").action(async () => {
+  const config = await loadConfig();
+  if (!config) {
+    logger.warn("No configuration found. Run `aitest init`.");
+    return;
+  }
+  console.log("\n--- Current Configuration ---");
+  for (const [key, value] of Object.entries(config)) {
+    if (key === "apiKey") {
+      console.log(`${key}: ********* (Hidden)`);
+    } else {
+      console.log(`${key}: ${value}`);
+    }
+  }
+  console.log("---------------------------\n");
+});
+export {
+  configCommand
+};
+//# sourceMappingURL=config.js.map

package/dist/commands/config.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/commands/config.ts","../../src/core/config.ts","../../src/core/logger.ts"],"sourcesContent":["import { Command } from 'commander';\nimport { loadConfig, saveConfig } from '../core/config.js';\nimport { logger } from '../core/logger.js';\n\nexport const configCommand = new Command('config')\n  .description('Manage settings (e.g., provider, model, apiKey)');\n\nconfigCommand.command('set <key> <value>')\n  .description('Set a configuration value')\n  .action(async (key, value) => {\n    let config = await loadConfig();\n    if (!config) {\n      config = { provider: 'openai', model: 'gpt-4o' }; // fallback default\n    }\n    \n    // basic type coercion\n    let parsedValue: any = value;\n    if (value === 'true') parsedValue = true;\n    if (value === 'false') parsedValue = false;\n    if (!isNaN(Number(value))) parsedValue = Number(value);\n\n    (config as any)[key] = parsedValue;\n    \n    await saveConfig(config);\n  });\n\nconfigCommand.command('list')\n  .description('List current configuration')\n  .action(async () => {\n    const config = await loadConfig();\n    if (!config) {\n      logger.warn('No configuration found. Run `aitest init`.');\n      return;\n    }\n    \n    console.log('\\n--- Current Configuration ---');\n    for (const [key, value] of Object.entries(config)) {\n      if (key === 'apiKey') {\n         console.log(`${key}: ********* (Hidden)`);\n      } else {\n         console.log(`${key}: ${value}`);\n      }\n    }\n    console.log('---------------------------\\n');\n  });\n","import { cosmiconfig } from 'cosmiconfig';\nimport { writeFile } from 'fs/promises';\nimport { resolve } from 'path';\nimport { logger } from './logger.js';\n\nexport interface AITestConfig {\n  provider: string;\n  model: string;\n  apiKey?: string;\n  baseURL?: string;\n  customHeaders?: Record<string, string>;\n  temperature?: number;\n  autoFix?: boolean;\n  maxRetries?: number;\n}\n\nconst moduleName = 'aitest';\nconst explorer = cosmiconfig(moduleName);\n\nexport async function loadConfig(): Promise<AITestConfig | null> {\n  try {\n    const result = await explorer.search();\n    if (result) {\n      return result.config as AITestConfig;\n    }\n    return null;\n  } catch (error) {\n    logger.error(`Failed to load config: ${error}`);\n    return null;\n  }\n}\n\nexport async function saveConfig(config: AITestConfig, targetDir: string = process.cwd()): Promise<void> {\n  const configPath = resolve(targetDir, '.aitestrc.json');\n  try {\n    await writeFile(configPath, JSON.stringify(config, null, 2), 'utf-8');\n    logger.success(`Saved config to ${configPath}`);\n  } catch (error) {\n    logger.error(`Failed to save config: ${error}`);\n    throw error;\n  }\n}\n","import chalk from 'chalk';\nimport ora from 'ora';\n\nexport const logger = {\n  info: (msg: string) => console.log(chalk.blue('ℹ'), msg),\n  success: (msg: string) => console.log(chalk.green('✔'), msg),\n  error: (msg: string) => console.error(chalk.red('✖'), msg),\n  warn: (msg: string) => console.warn(chalk.yellow('⚠'), msg),\n  log: (msg: string) => console.log(msg),\n  spinner: (text: string) => ora(text),\n};\n"],"mappings":";AAAA,SAAS,eAAe;;;ACAxB,SAAS,mBAAmB;AAC5B,SAAS,iBAAiB;AAC1B,SAAS,eAAe;;;ACFxB,OAAO,WAAW;AAClB,OAAO,SAAS;AAET,IAAM,SAAS;AAAA,EACpB,MAAM,CAAC,QAAgB,QAAQ,IAAI,MAAM,KAAK,QAAG,GAAG,GAAG;AAAA,EACvD,SAAS,CAAC,QAAgB,QAAQ,IAAI,MAAM,MAAM,QAAG,GAAG,GAAG;AAAA,EAC3D,OAAO,CAAC,QAAgB,QAAQ,MAAM,MAAM,IAAI,QAAG,GAAG,GAAG;AAAA,EACzD,MAAM,CAAC,QAAgB,QAAQ,KAAK,MAAM,OAAO,QAAG,GAAG,GAAG;AAAA,EAC1D,KAAK,CAAC,QAAgB,QAAQ,IAAI,GAAG;AAAA,EACrC,SAAS,CAAC,SAAiB,IAAI,IAAI;AACrC;;;ADMA,IAAM,aAAa;AACnB,IAAM,WAAW,YAAY,UAAU;AAEvC,eAAsB,aAA2C;AAC/D,MAAI;AACF,UAAM,SAAS,MAAM,SAAS,OAAO;AACrC,QAAI,QAAQ;AACV,aAAO,OAAO;AAAA,IAChB;AACA,WAAO;AAAA,EACT,SAAS,OAAO;AACd,WAAO,MAAM,0BAA0B,KAAK,EAAE;AAC9C,WAAO;AAAA,EACT;AACF;AAEA,eAAsB,WAAW,QAAsB,YAAoB,QAAQ,IAAI,GAAkB;AACvG,QAAM,aAAa,QAAQ,WAAW,gBAAgB;AACtD,MAAI;AACF,UAAM,UAAU,YAAY,KAAK,UAAU,QAAQ,MAAM,CAAC,GAAG,OAAO;AACpE,WAAO,QAAQ,mBAAmB,UAAU,EAAE;AAAA,EAChD,SAAS,OAAO;AACd,WAAO,MAAM,0BAA0B,KAAK,EAAE;AAC9C,UAAM;AAAA,EACR;AACF;;;ADrCO,IAAM,gBAAgB,IAAI,QAAQ,QAAQ,EAC9C,YAAY,iDAAiD;AAEhE,cAAc,QAAQ,mBAAmB,EACtC,YAAY,2BAA2B,EACvC,OAAO,OAAO,KAAK,UAAU;AAC5B,MAAI,SAAS,MAAM,WAAW;AAC9B,MAAI,CAAC,QAAQ;AACX,aAAS,EAAE,UAAU,UAAU,OAAO,SAAS;AAAA,EACjD;AAGA,MAAI,cAAmB;AACvB,MAAI,UAAU,OAAQ,eAAc;AACpC,MAAI,UAAU,QAAS,eAAc;AACrC,MAAI,CAAC,MAAM,OAAO,KAAK,CAAC,EAAG,eAAc,OAAO,KAAK;AAErD,EAAC,OAAe,GAAG,IAAI;AAEvB,QAAM,WAAW,MAAM;AACzB,CAAC;AAEH,cAAc,QAAQ,MAAM,EACzB,YAAY,4BAA4B,EACxC,OAAO,YAAY;AAClB,QAAM,SAAS,MAAM,WAAW;AAChC,MAAI,CAAC,QAAQ;AACX,WAAO,KAAK,4CAA4C;AACxD;AAAA,EACF;AAEA,UAAQ,IAAI,iCAAiC;AAC7C,aAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,MAAM,GAAG;AACjD,QAAI,QAAQ,UAAU;AACnB,cAAQ,IAAI,GAAG,GAAG,sBAAsB;AAAA,IAC3C,OAAO;AACJ,cAAQ,IAAI,GAAG,GAAG,KAAK,KAAK,EAAE;AAAA,IACjC;AAAA,EACF;AACA,UAAQ,IAAI,+BAA+B;AAC7C,CAAC;","names":[]}

package/dist/commands/doctor.d.ts

@@ -0,0 +1,5 @@
+import { Command } from 'commander';
+
+declare const doctorCommand: Command;
+
+export { doctorCommand };

package/dist/commands/doctor.js

@@ -0,0 +1,185 @@
+// src/commands/doctor.ts
+import { Command } from "commander";
+
+// src/core/config.ts
+import { cosmiconfig } from "cosmiconfig";
+
+// src/core/logger.ts
+import chalk from "chalk";
+import ora from "ora";
+var logger = {
+  info: (msg) => console.log(chalk.blue("\u2139"), msg),
+  success: (msg) => console.log(chalk.green("\u2714"), msg),
+  error: (msg) => console.error(chalk.red("\u2716"), msg),
+  warn: (msg) => console.warn(chalk.yellow("\u26A0"), msg),
+  log: (msg) => console.log(msg),
+  spinner: (text) => ora(text)
+};
+
+// src/core/config.ts
+var moduleName = "aitest";
+var explorer = cosmiconfig(moduleName);
+async function loadConfig() {
+  try {
+    const result = await explorer.search();
+    if (result) {
+      return result.config;
+    }
+    return null;
+  } catch (error) {
+    logger.error(`Failed to load config: ${error}`);
+    return null;
+  }
+}
+
+// src/core/detector.ts
+import { readFileSync, existsSync } from "fs";
+import { resolve } from "path";
+function detectProjectInfo(targetDir = process.cwd()) {
+  const packageJsonPath = resolve(targetDir, "package.json");
+  const tsconfigPath = resolve(targetDir, "tsconfig.json");
+  let packageJson = {};
+  if (existsSync(packageJsonPath)) {
+    try {
+      packageJson = JSON.parse(readFileSync(packageJsonPath, "utf-8"));
+    } catch (e) {
+    }
+  }
+  const allDeps = {
+    ...packageJson.dependencies || {},
+    ...packageJson.devDependencies || {}
+  };
+  const language = existsSync(tsconfigPath) || allDeps["typescript"] ? "typescript" : "javascript";
+  let testRunner = "unknown";
+  if (allDeps["jest"]) testRunner = "jest";
+  else if (allDeps["vitest"]) testRunner = "vitest";
+  else if (allDeps["mocha"]) testRunner = "mocha";
+  let framework = "unknown";
+  if (allDeps["react"]) framework = "react";
+  else if (allDeps["vue"]) framework = "vue";
+  else if (allDeps["@angular/core"]) framework = "angular";
+  else if (allDeps["express"] || allDeps["@nestjs/core"] || allDeps["fastify"]) framework = "node";
+  let packageManager = "npm";
+  if (existsSync(resolve(targetDir, "pnpm-lock.yaml"))) packageManager = "pnpm";
+  else if (existsSync(resolve(targetDir, "yarn.lock"))) packageManager = "yarn";
+  else if (existsSync(resolve(targetDir, "bun.lockb"))) packageManager = "bun";
+  return { language, testRunner, framework, packageManager };
+}
+
+// src/core/ai.ts
+import { generateText } from "ai";
+import { createOpenAI } from "@ai-sdk/openai";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createGoogleGenerativeAI } from "@ai-sdk/google";
+import { createDeepSeek } from "@ai-sdk/deepseek";
+import * as dotenv from "dotenv";
+dotenv.config();
+function getAIModel(config2) {
+  const providerName = config2.provider.toLowerCase();
+  if (providerName === "openai") {
+    const openai = createOpenAI({
+      apiKey: config2.apiKey || process.env.OPENAI_API_KEY
+    });
+    return openai(config2.model || "gpt-4o");
+  }
+  if (providerName === "anthropic") {
+    const anthropic = createAnthropic({
+      apiKey: config2.apiKey || process.env.ANTHROPIC_API_KEY
+    });
+    return anthropic(config2.model || "claude-3-5-sonnet-20240620");
+  }
+  if (providerName === "gemini" || providerName === "google") {
+    const google = createGoogleGenerativeAI({
+      apiKey: config2.apiKey || process.env.GEMINI_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY
+    });
+    return google(config2.model || "gemini-2.5-flash");
+  }
+  if (providerName === "deepseek") {
+    const deepseek = createDeepSeek({
+      apiKey: config2.apiKey || process.env.DEEPSEEK_API_KEY
+    });
+    return deepseek(config2.model || "deepseek-coder");
+  }
+  if (providerName === "ollama") {
+    const ollama = createOpenAI({
+      baseURL: "http://localhost:11434/v1",
+      apiKey: "ollama"
+      // API key isn't strictly needed for local Ollama
+    });
+    return ollama(config2.model || "llama3.1");
+  }
+  if (providerName === "custom") {
+    const custom = createOpenAI({
+      baseURL: config2.baseURL,
+      apiKey: config2.apiKey || process.env.CUSTOM_API_KEY || "custom",
+      headers: config2.customHeaders
+    });
+    return custom(config2.model || "local-model");
+  }
+  throw new Error(`Unsupported AI provider: ${config2.provider}`);
+}
+async function askAI(config2, system, prompt) {
+  const model = getAIModel(config2);
+  const { text } = await generateText({
+    model,
+    system,
+    prompt,
+    maxRetries: 5,
+    temperature: config2.temperature || 0.1
+  });
+  return text;
+}
+
+// src/commands/doctor.ts
+var doctorCommand = new Command("doctor").description("Verify installation, configuration, and environment").action(async () => {
+  logger.info("Running AI Test CLI Diagnostics...\n");
+  let allChecksPassed = true;
+  const configSpinner = logger.spinner("Checking configuration...").start();
+  const config2 = await loadConfig();
+  if (config2) {
+    configSpinner.succeed("Configuration found");
+  } else {
+    configSpinner.fail("Configuration missing. Run `aitest init`");
+    allChecksPassed = false;
+  }
+  const apiSpinner = logger.spinner("Checking API key...").start();
+  if (config2?.apiKey || process.env.OPENAI_API_KEY || process.env.ANTHROPIC_API_KEY || process.env.GEMINI_API_KEY) {
+    apiSpinner.succeed("API key configured");
+  } else {
+    apiSpinner.warn("No explicit API key found in config or environment variables (might be okay for local models)");
+  }
+  if (config2) {
+    const modelSpinner = logger.spinner(`Testing access to ${config2.provider} (${config2.model})...`).start();
+    try {
+      const response = await askAI(config2, 'Reply with exactly "OK"', "Test connection");
+      if (response.includes("OK") || response.trim() !== "") {
+        modelSpinner.succeed("Model connection successful");
+      } else {
+        modelSpinner.warn("Model responded, but unexpectedly");
+      }
+    } catch (error) {
+      modelSpinner.fail(`Model connection failed: ${error.message}`);
+      allChecksPassed = false;
+    }
+  }
+  const envSpinner = logger.spinner("Checking project environment...").start();
+  const projectInfo = detectProjectInfo();
+  if (projectInfo.packageManager !== "npm" && projectInfo.packageManager !== "yarn" && projectInfo.packageManager !== "pnpm" && projectInfo.packageManager !== "bun") {
+    envSpinner.fail("Supported package manager not found");
+    allChecksPassed = false;
+  } else if (projectInfo.testRunner === "unknown") {
+    envSpinner.warn(`Test runner not explicitly detected. Fallback may be used.`);
+  } else {
+    envSpinner.succeed(`Project environment valid (${projectInfo.language}, ${projectInfo.packageManager}, ${projectInfo.testRunner})`);
+  }
+  console.log("\n--- Diagnostic Summary ---");
+  if (allChecksPassed) {
+    logger.success("\u2705 All critical checks passed. You are ready to generate tests!");
+  } else {
+    logger.warn("\u26A0\uFE0F Some checks failed. Please resolve the issues above before proceeding.");
+  }
+});
+export {
+  doctorCommand
+};
+//# sourceMappingURL=doctor.js.map

package/dist/commands/doctor.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/commands/doctor.ts","../../src/core/config.ts","../../src/core/logger.ts","../../src/core/detector.ts","../../src/core/ai.ts"],"sourcesContent":["import { Command } from 'commander';\nimport { loadConfig } from '../core/config.js';\nimport { logger } from '../core/logger.js';\nimport { detectProjectInfo } from '../core/detector.js';\nimport { askAI } from '../core/ai.js';\n\nexport const doctorCommand = new Command('doctor')\n  .description('Verify installation, configuration, and environment')\n  .action(async () => {\n    logger.info('Running AI Test CLI Diagnostics...\\n');\n\n    let allChecksPassed = true;\n\n    // 1. Check Configuration\n    const configSpinner = logger.spinner('Checking configuration...').start();\n    const config = await loadConfig();\n    if (config) {\n      configSpinner.succeed('Configuration found');\n    } else {\n      configSpinner.fail('Configuration missing. Run `aitest init`');\n      allChecksPassed = false;\n    }\n\n    // 2. Check API Key\n    const apiSpinner = logger.spinner('Checking API key...').start();\n    if (config?.apiKey || process.env.OPENAI_API_KEY || process.env.ANTHROPIC_API_KEY || process.env.GEMINI_API_KEY) {\n      apiSpinner.succeed('API key configured');\n    } else {\n      apiSpinner.warn('No explicit API key found in config or environment variables (might be okay for local models)');\n    }\n\n    // 3. Test Model Access\n    if (config) {\n      const modelSpinner = logger.spinner(`Testing access to ${config.provider} (${config.model})...`).start();\n      try {\n        const response = await askAI(config, 'Reply with exactly \"OK\"', 'Test connection');\n        if (response.includes('OK') || response.trim() !== '') {\n          modelSpinner.succeed('Model connection successful');\n        } else {\n          modelSpinner.warn('Model responded, but unexpectedly');\n        }\n      } catch (error: any) {\n        modelSpinner.fail(`Model connection failed: ${error.message}`);\n        allChecksPassed = false;\n      }\n    }\n\n    // 4. Check Project Environment\n    const envSpinner = logger.spinner('Checking project environment...').start();\n    const projectInfo = detectProjectInfo();\n    \n    if (projectInfo.packageManager !== 'npm' && projectInfo.packageManager !== 'yarn' && projectInfo.packageManager !== 'pnpm' && projectInfo.packageManager !== 'bun') {\n      envSpinner.fail('Supported package manager not found');\n      allChecksPassed = false;\n    } else if (projectInfo.testRunner === 'unknown') {\n      envSpinner.warn(`Test runner not explicitly detected. Fallback may be used.`);\n    } else {\n      envSpinner.succeed(`Project environment valid (${projectInfo.language}, ${projectInfo.packageManager}, ${projectInfo.testRunner})`);\n    }\n\n    console.log('\\n--- Diagnostic Summary ---');\n    if (allChecksPassed) {\n      logger.success('✅ All critical checks passed. You are ready to generate tests!');\n    } else {\n      logger.warn('⚠️ Some checks failed. Please resolve the issues above before proceeding.');\n    }\n  });\n","import { cosmiconfig } from 'cosmiconfig';\nimport { writeFile } from 'fs/promises';\nimport { resolve } from 'path';\nimport { logger } from './logger.js';\n\nexport interface AITestConfig {\n  provider: string;\n  model: string;\n  apiKey?: string;\n  baseURL?: string;\n  customHeaders?: Record<string, string>;\n  temperature?: number;\n  autoFix?: boolean;\n  maxRetries?: number;\n}\n\nconst moduleName = 'aitest';\nconst explorer = cosmiconfig(moduleName);\n\nexport async function loadConfig(): Promise<AITestConfig | null> {\n  try {\n    const result = await explorer.search();\n    if (result) {\n      return result.config as AITestConfig;\n    }\n    return null;\n  } catch (error) {\n    logger.error(`Failed to load config: ${error}`);\n    return null;\n  }\n}\n\nexport async function saveConfig(config: AITestConfig, targetDir: string = process.cwd()): Promise<void> {\n  const configPath = resolve(targetDir, '.aitestrc.json');\n  try {\n    await writeFile(configPath, JSON.stringify(config, null, 2), 'utf-8');\n    logger.success(`Saved config to ${configPath}`);\n  } catch (error) {\n    logger.error(`Failed to save config: ${error}`);\n    throw error;\n  }\n}\n","import chalk from 'chalk';\nimport ora from 'ora';\n\nexport const logger = {\n  info: (msg: string) => console.log(chalk.blue('ℹ'), msg),\n  success: (msg: string) => console.log(chalk.green('✔'), msg),\n  error: (msg: string) => console.error(chalk.red('✖'), msg),\n  warn: (msg: string) => console.warn(chalk.yellow('⚠'), msg),\n  log: (msg: string) => console.log(msg),\n  spinner: (text: string) => ora(text),\n};\n","import { readFileSync, existsSync } from 'fs';\nimport { resolve } from 'path';\n\nexport interface ProjectInfo {\n  language: 'javascript' | 'typescript';\n  testRunner: 'jest' | 'vitest' | 'mocha' | 'unknown';\n  framework: 'react' | 'vue' | 'angular' | 'node' | 'unknown';\n  packageManager: 'npm' | 'yarn' | 'pnpm' | 'bun';\n}\n\nexport function detectProjectInfo(targetDir: string = process.cwd()): ProjectInfo {\n  const packageJsonPath = resolve(targetDir, 'package.json');\n  const tsconfigPath = resolve(targetDir, 'tsconfig.json');\n  \n  let packageJson: any = {};\n  if (existsSync(packageJsonPath)) {\n    try {\n      packageJson = JSON.parse(readFileSync(packageJsonPath, 'utf-8'));\n    } catch (e) {\n      // ignore\n    }\n  }\n\n  const allDeps = {\n    ...(packageJson.dependencies || {}),\n    ...(packageJson.devDependencies || {})\n  };\n\n  const language = existsSync(tsconfigPath) || allDeps['typescript'] ? 'typescript' : 'javascript';\n  \n  let testRunner: ProjectInfo['testRunner'] = 'unknown';\n  if (allDeps['jest']) testRunner = 'jest';\n  else if (allDeps['vitest']) testRunner = 'vitest';\n  else if (allDeps['mocha']) testRunner = 'mocha';\n\n  let framework: ProjectInfo['framework'] = 'unknown';\n  if (allDeps['react']) framework = 'react';\n  else if (allDeps['vue']) framework = 'vue';\n  else if (allDeps['@angular/core']) framework = 'angular';\n  else if (allDeps['express'] || allDeps['@nestjs/core'] || allDeps['fastify']) framework = 'node';\n\n  let packageManager: ProjectInfo['packageManager'] = 'npm';\n  if (existsSync(resolve(targetDir, 'pnpm-lock.yaml'))) packageManager = 'pnpm';\n  else if (existsSync(resolve(targetDir, 'yarn.lock'))) packageManager = 'yarn';\n  else if (existsSync(resolve(targetDir, 'bun.lockb'))) packageManager = 'bun';\n\n  return { language, testRunner, framework, packageManager };\n}\n","import { generateText } from 'ai';\nimport { createOpenAI } from '@ai-sdk/openai';\nimport { createAnthropic } from '@ai-sdk/anthropic';\nimport { createGoogleGenerativeAI } from '@ai-sdk/google';\nimport { createDeepSeek } from '@ai-sdk/deepseek';\nimport { AITestConfig } from './config.js';\nimport * as dotenv from 'dotenv';\n\ndotenv.config();\n\nexport function getAIModel(config: AITestConfig) {\n  const providerName = config.provider.toLowerCase();\n  \n  if (providerName === 'openai') {\n    const openai = createOpenAI({\n      apiKey: config.apiKey || process.env.OPENAI_API_KEY,\n    });\n    return openai(config.model || 'gpt-4o');\n  } \n  \n  if (providerName === 'anthropic') {\n    const anthropic = createAnthropic({\n      apiKey: config.apiKey || process.env.ANTHROPIC_API_KEY,\n    });\n    return anthropic(config.model || 'claude-3-5-sonnet-20240620');\n  }\n\n  if (providerName === 'gemini' || providerName === 'google') {\n    const google = createGoogleGenerativeAI({\n      apiKey: config.apiKey || process.env.GEMINI_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY,\n    });\n    return google(config.model || 'gemini-2.5-flash');\n  }\n\n  if (providerName === 'deepseek') {\n    const deepseek = createDeepSeek({\n      apiKey: config.apiKey || process.env.DEEPSEEK_API_KEY,\n    });\n    return deepseek(config.model || 'deepseek-coder');\n  }\n\n  if (providerName === 'ollama') {\n    const ollama = createOpenAI({\n      baseURL: 'http://localhost:11434/v1',\n      apiKey: 'ollama', // API key isn't strictly needed for local Ollama\n    });\n    return ollama(config.model || 'llama3.1');\n  }\n\n  if (providerName === 'custom') {\n    const custom = createOpenAI({\n      baseURL: config.baseURL,\n      apiKey: config.apiKey || process.env.CUSTOM_API_KEY || 'custom',\n      headers: config.customHeaders,\n    });\n    return custom(config.model || 'local-model');\n  }\n\n  throw new Error(`Unsupported AI provider: ${config.provider}`);\n}\n\nexport async function askAI(config: AITestConfig, system: string, prompt: string) {\n  const model = getAIModel(config);\n  \n  const { text } = await generateText({\n    model,\n    system,\n    prompt,\n    maxRetries: 5,\n    temperature: config.temperature || 0.1,\n  });\n\n  return text;\n}\n"],"mappings":";AAAA,SAAS,eAAe;;;ACAxB,SAAS,mBAAmB;;;ACA5B,OAAO,WAAW;AAClB,OAAO,SAAS;AAET,IAAM,SAAS;AAAA,EACpB,MAAM,CAAC,QAAgB,QAAQ,IAAI,MAAM,KAAK,QAAG,GAAG,GAAG;AAAA,EACvD,SAAS,CAAC,QAAgB,QAAQ,IAAI,MAAM,MAAM,QAAG,GAAG,GAAG;AAAA,EAC3D,OAAO,CAAC,QAAgB,QAAQ,MAAM,MAAM,IAAI,QAAG,GAAG,GAAG;AAAA,EACzD,MAAM,CAAC,QAAgB,QAAQ,KAAK,MAAM,OAAO,QAAG,GAAG,GAAG;AAAA,EAC1D,KAAK,CAAC,QAAgB,QAAQ,IAAI,GAAG;AAAA,EACrC,SAAS,CAAC,SAAiB,IAAI,IAAI;AACrC;;;ADMA,IAAM,aAAa;AACnB,IAAM,WAAW,YAAY,UAAU;AAEvC,eAAsB,aAA2C;AAC/D,MAAI;AACF,UAAM,SAAS,MAAM,SAAS,OAAO;AACrC,QAAI,QAAQ;AACV,aAAO,OAAO;AAAA,IAChB;AACA,WAAO;AAAA,EACT,SAAS,OAAO;AACd,WAAO,MAAM,0BAA0B,KAAK,EAAE;AAC9C,WAAO;AAAA,EACT;AACF;;;AE9BA,SAAS,cAAc,kBAAkB;AACzC,SAAS,eAAe;AASjB,SAAS,kBAAkB,YAAoB,QAAQ,IAAI,GAAgB;AAChF,QAAM,kBAAkB,QAAQ,WAAW,cAAc;AACzD,QAAM,eAAe,QAAQ,WAAW,eAAe;AAEvD,MAAI,cAAmB,CAAC;AACxB,MAAI,WAAW,eAAe,GAAG;AAC/B,QAAI;AACF,oBAAc,KAAK,MAAM,aAAa,iBAAiB,OAAO,CAAC;AAAA,IACjE,SAAS,GAAG;AAAA,IAEZ;AAAA,EACF;AAEA,QAAM,UAAU;AAAA,IACd,GAAI,YAAY,gBAAgB,CAAC;AAAA,IACjC,GAAI,YAAY,mBAAmB,CAAC;AAAA,EACtC;AAEA,QAAM,WAAW,WAAW,YAAY,KAAK,QAAQ,YAAY,IAAI,eAAe;AAEpF,MAAI,aAAwC;AAC5C,MAAI,QAAQ,MAAM,EAAG,cAAa;AAAA,WACzB,QAAQ,QAAQ,EAAG,cAAa;AAAA,WAChC,QAAQ,OAAO,EAAG,cAAa;AAExC,MAAI,YAAsC;AAC1C,MAAI,QAAQ,OAAO,EAAG,aAAY;AAAA,WACzB,QAAQ,KAAK,EAAG,aAAY;AAAA,WAC5B,QAAQ,eAAe,EAAG,aAAY;AAAA,WACtC,QAAQ,SAAS,KAAK,QAAQ,cAAc,KAAK,QAAQ,SAAS,EAAG,aAAY;AAE1F,MAAI,iBAAgD;AACpD,MAAI,WAAW,QAAQ,WAAW,gBAAgB,CAAC,EAAG,kBAAiB;AAAA,WAC9D,WAAW,QAAQ,WAAW,WAAW,CAAC,EAAG,kBAAiB;AAAA,WAC9D,WAAW,QAAQ,WAAW,WAAW,CAAC,EAAG,kBAAiB;AAEvE,SAAO,EAAE,UAAU,YAAY,WAAW,eAAe;AAC3D;;;AC/CA,SAAS,oBAAoB;AAC7B,SAAS,oBAAoB;AAC7B,SAAS,uBAAuB;AAChC,SAAS,gCAAgC;AACzC,SAAS,sBAAsB;AAE/B,YAAY,YAAY;AAEjB,cAAO;AAEP,SAAS,WAAWA,SAAsB;AAC/C,QAAM,eAAeA,QAAO,SAAS,YAAY;AAEjD,MAAI,iBAAiB,UAAU;AAC7B,UAAM,SAAS,aAAa;AAAA,MAC1B,QAAQA,QAAO,UAAU,QAAQ,IAAI;AAAA,IACvC,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,QAAQ;AAAA,EACxC;AAEA,MAAI,iBAAiB,aAAa;AAChC,UAAM,YAAY,gBAAgB;AAAA,MAChC,QAAQA,QAAO,UAAU,QAAQ,IAAI;AAAA,IACvC,CAAC;AACD,WAAO,UAAUA,QAAO,SAAS,4BAA4B;AAAA,EAC/D;AAEA,MAAI,iBAAiB,YAAY,iBAAiB,UAAU;AAC1D,UAAM,SAAS,yBAAyB;AAAA,MACtC,QAAQA,QAAO,UAAU,QAAQ,IAAI,kBAAkB,QAAQ,IAAI;AAAA,IACrE,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,kBAAkB;AAAA,EAClD;AAEA,MAAI,iBAAiB,YAAY;AAC/B,UAAM,WAAW,eAAe;AAAA,MAC9B,QAAQA,QAAO,UAAU,QAAQ,IAAI;AAAA,IACvC,CAAC;AACD,WAAO,SAASA,QAAO,SAAS,gBAAgB;AAAA,EAClD;AAEA,MAAI,iBAAiB,UAAU;AAC7B,UAAM,SAAS,aAAa;AAAA,MAC1B,SAAS;AAAA,MACT,QAAQ;AAAA;AAAA,IACV,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,UAAU;AAAA,EAC1C;AAEA,MAAI,iBAAiB,UAAU;AAC7B,UAAM,SAAS,aAAa;AAAA,MAC1B,SAASA,QAAO;AAAA,MAChB,QAAQA,QAAO,UAAU,QAAQ,IAAI,kBAAkB;AAAA,MACvD,SAASA,QAAO;AAAA,IAClB,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,aAAa;AAAA,EAC7C;AAEA,QAAM,IAAI,MAAM,4BAA4BA,QAAO,QAAQ,EAAE;AAC/D;AAEA,eAAsB,MAAMA,SAAsB,QAAgB,QAAgB;AAChF,QAAM,QAAQ,WAAWA,OAAM;AAE/B,QAAM,EAAE,KAAK,IAAI,MAAM,aAAa;AAAA,IAClC;AAAA,IACA;AAAA,IACA;AAAA,IACA,YAAY;AAAA,IACZ,aAAaA,QAAO,eAAe;AAAA,EACrC,CAAC;AAED,SAAO;AACT;;;AJnEO,IAAM,gBAAgB,IAAI,QAAQ,QAAQ,EAC9C,YAAY,qDAAqD,EACjE,OAAO,YAAY;AAClB,SAAO,KAAK,sCAAsC;AAElD,MAAI,kBAAkB;AAGtB,QAAM,gBAAgB,OAAO,QAAQ,2BAA2B,EAAE,MAAM;AACxE,QAAMC,UAAS,MAAM,WAAW;AAChC,MAAIA,SAAQ;AACV,kBAAc,QAAQ,qBAAqB;AAAA,EAC7C,OAAO;AACL,kBAAc,KAAK,0CAA0C;AAC7D,sBAAkB;AAAA,EACpB;AAGA,QAAM,aAAa,OAAO,QAAQ,qBAAqB,EAAE,MAAM;AAC/D,MAAIA,SAAQ,UAAU,QAAQ,IAAI,kBAAkB,QAAQ,IAAI,qBAAqB,QAAQ,IAAI,gBAAgB;AAC/G,eAAW,QAAQ,oBAAoB;AAAA,EACzC,OAAO;AACL,eAAW,KAAK,+FAA+F;AAAA,EACjH;AAGA,MAAIA,SAAQ;AACV,UAAM,eAAe,OAAO,QAAQ,qBAAqBA,QAAO,QAAQ,KAAKA,QAAO,KAAK,MAAM,EAAE,MAAM;AACvG,QAAI;AACF,YAAM,WAAW,MAAM,MAAMA,SAAQ,2BAA2B,iBAAiB;AACjF,UAAI,SAAS,SAAS,IAAI,KAAK,SAAS,KAAK,MAAM,IAAI;AACrD,qBAAa,QAAQ,6BAA6B;AAAA,MACpD,OAAO;AACL,qBAAa,KAAK,mCAAmC;AAAA,MACvD;AAAA,IACF,SAAS,OAAY;AACnB,mBAAa,KAAK,4BAA4B,MAAM,OAAO,EAAE;AAC7D,wBAAkB;AAAA,IACpB;AAAA,EACF;AAGA,QAAM,aAAa,OAAO,QAAQ,iCAAiC,EAAE,MAAM;AAC3E,QAAM,cAAc,kBAAkB;AAEtC,MAAI,YAAY,mBAAmB,SAAS,YAAY,mBAAmB,UAAU,YAAY,mBAAmB,UAAU,YAAY,mBAAmB,OAAO;AAClK,eAAW,KAAK,qCAAqC;AACrD,sBAAkB;AAAA,EACpB,WAAW,YAAY,eAAe,WAAW;AAC/C,eAAW,KAAK,4DAA4D;AAAA,EAC9E,OAAO;AACL,eAAW,QAAQ,8BAA8B,YAAY,QAAQ,KAAK,YAAY,cAAc,KAAK,YAAY,UAAU,GAAG;AAAA,EACpI;AAEA,UAAQ,IAAI,8BAA8B;AAC1C,MAAI,iBAAiB;AACnB,WAAO,QAAQ,qEAAgE;AAAA,EACjF,OAAO;AACL,WAAO,KAAK,qFAA2E;AAAA,EACzF;AACF,CAAC;","names":["config","config"]}

package/dist/commands/explain.d.ts

@@ -0,0 +1,5 @@
+import { Command } from 'commander';
+
+declare const explainCommand: Command;
+
+export { explainCommand };

package/dist/commands/explain.js

@@ -0,0 +1,146 @@
+// src/commands/explain.ts
+import { Command } from "commander";
+
+// src/core/config.ts
+import { cosmiconfig } from "cosmiconfig";
+
+// src/core/logger.ts
+import chalk from "chalk";
+import ora from "ora";
+var logger = {
+  info: (msg) => console.log(chalk.blue("\u2139"), msg),
+  success: (msg) => console.log(chalk.green("\u2714"), msg),
+  error: (msg) => console.error(chalk.red("\u2716"), msg),
+  warn: (msg) => console.warn(chalk.yellow("\u26A0"), msg),
+  log: (msg) => console.log(msg),
+  spinner: (text) => ora(text)
+};
+
+// src/core/config.ts
+var moduleName = "aitest";
+var explorer = cosmiconfig(moduleName);
+async function loadConfig() {
+  try {
+    const result = await explorer.search();
+    if (result) {
+      return result.config;
+    }
+    return null;
+  } catch (error) {
+    logger.error(`Failed to load config: ${error}`);
+    return null;
+  }
+}
+
+// src/core/ai.ts
+import { generateText } from "ai";
+import { createOpenAI } from "@ai-sdk/openai";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createGoogleGenerativeAI } from "@ai-sdk/google";
+import { createDeepSeek } from "@ai-sdk/deepseek";
+import * as dotenv from "dotenv";
+dotenv.config();
+function getAIModel(config2) {
+  const providerName = config2.provider.toLowerCase();
+  if (providerName === "openai") {
+    const openai = createOpenAI({
+      apiKey: config2.apiKey || process.env.OPENAI_API_KEY
+    });
+    return openai(config2.model || "gpt-4o");
+  }
+  if (providerName === "anthropic") {
+    const anthropic = createAnthropic({
+      apiKey: config2.apiKey || process.env.ANTHROPIC_API_KEY
+    });
+    return anthropic(config2.model || "claude-3-5-sonnet-20240620");
+  }
+  if (providerName === "gemini" || providerName === "google") {
+    const google = createGoogleGenerativeAI({
+      apiKey: config2.apiKey || process.env.GEMINI_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY
+    });
+    return google(config2.model || "gemini-2.5-flash");
+  }
+  if (providerName === "deepseek") {
+    const deepseek = createDeepSeek({
+      apiKey: config2.apiKey || process.env.DEEPSEEK_API_KEY
+    });
+    return deepseek(config2.model || "deepseek-coder");
+  }
+  if (providerName === "ollama") {
+    const ollama = createOpenAI({
+      baseURL: "http://localhost:11434/v1",
+      apiKey: "ollama"
+      // API key isn't strictly needed for local Ollama
+    });
+    return ollama(config2.model || "llama3.1");
+  }
+  if (providerName === "custom") {
+    const custom = createOpenAI({
+      baseURL: config2.baseURL,
+      apiKey: config2.apiKey || process.env.CUSTOM_API_KEY || "custom",
+      headers: config2.customHeaders
+    });
+    return custom(config2.model || "local-model");
+  }
+  throw new Error(`Unsupported AI provider: ${config2.provider}`);
+}
+async function askAI(config2, system, prompt) {
+  const model = getAIModel(config2);
+  const { text } = await generateText({
+    model,
+    system,
+    prompt,
+    maxRetries: 5,
+    temperature: config2.temperature || 0.1
+  });
+  return text;
+}
+
+// src/commands/explain.ts
+import { exec } from "child_process";
+import { promisify } from "util";
+var execAsync = promisify(exec);
+var explainCommand = new Command("explain").description("Explains test failures in plain English").action(async () => {
+  const config2 = await loadConfig();
+  if (!config2) {
+    logger.error("Configuration not found. Run `aitest init` first.");
+    return;
+  }
+  const spinner = logger.spinner("Running tests to gather failure data...").start();
+  let stdoutStr = "";
+  let stderrStr = "";
+  try {
+    await execAsync("npm test");
+    spinner.succeed("Tests passed! Nothing to explain.");
+    return;
+  } catch (error) {
+    stdoutStr = error.stdout || "";
+    stderrStr = error.stderr || error.message;
+    spinner.info("Test failure detected.");
+  }
+  const outputToAnalyze = `
+STDOUT:
+${stdoutStr.slice(-2e3)}
+
+STDERR:
+${stderrStr.slice(-2e3)}
+    `.trim();
+  logger.info("Analyzing root cause with AI...");
+  const analyzeSpinner = logger.spinner("Thinking...").start();
+  try {
+    const explanation = await askAI(
+      config2,
+      "You are an expert QA and software testing engineer. The user has failing tests. Explain the failure in plain English. Include:\n- Root cause\n- Stack analysis\n- Suggested fixes",
+      outputToAnalyze
+    );
+    analyzeSpinner.succeed("Explanation ready:");
+    console.log("\n" + explanation + "\n");
+  } catch (error) {
+    analyzeSpinner.fail("AI explanation failed.");
+    logger.error(error.message);
+  }
+});
+export {
+  explainCommand
+};
+//# sourceMappingURL=explain.js.map

package/dist/commands/explain.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/commands/explain.ts","../../src/core/config.ts","../../src/core/logger.ts","../../src/core/ai.ts"],"sourcesContent":["import { Command } from 'commander';\nimport { loadConfig } from '../core/config.js';\nimport { logger } from '../core/logger.js';\nimport { askAI } from '../core/ai.js';\nimport { exec } from 'child_process';\nimport { promisify } from 'util';\n\nconst execAsync = promisify(exec);\n\nexport const explainCommand = new Command('explain')\n  .description('Explains test failures in plain English')\n  .action(async () => {\n    const config = await loadConfig();\n    if (!config) {\n      logger.error('Configuration not found. Run `aitest init` first.');\n      return;\n    }\n\n    const spinner = logger.spinner('Running tests to gather failure data...').start();\n    \n    let stdoutStr = '';\n    let stderrStr = '';\n\n    try {\n      await execAsync('npm test');\n      spinner.succeed('Tests passed! Nothing to explain.');\n      return;\n    } catch (error: any) {\n      stdoutStr = error.stdout || '';\n      stderrStr = error.stderr || error.message;\n      spinner.info('Test failure detected.');\n    }\n\n    const outputToAnalyze = `\nSTDOUT:\n${stdoutStr.slice(-2000)}\n\nSTDERR:\n${stderrStr.slice(-2000)}\n    `.trim();\n\n    logger.info('Analyzing root cause with AI...');\n    const analyzeSpinner = logger.spinner('Thinking...').start();\n    \n    try {\n      const explanation = await askAI(\n        config,\n        'You are an expert QA and software testing engineer. The user has failing tests. Explain the failure in plain English. Include:\\n- Root cause\\n- Stack analysis\\n- Suggested fixes',\n        outputToAnalyze\n      );\n      \n      analyzeSpinner.succeed('Explanation ready:');\n      console.log('\\n' + explanation + '\\n');\n      \n    } catch (error: any) {\n      analyzeSpinner.fail('AI explanation failed.');\n      logger.error(error.message);\n    }\n  });\n","import { cosmiconfig } from 'cosmiconfig';\nimport { writeFile } from 'fs/promises';\nimport { resolve } from 'path';\nimport { logger } from './logger.js';\n\nexport interface AITestConfig {\n  provider: string;\n  model: string;\n  apiKey?: string;\n  baseURL?: string;\n  customHeaders?: Record<string, string>;\n  temperature?: number;\n  autoFix?: boolean;\n  maxRetries?: number;\n}\n\nconst moduleName = 'aitest';\nconst explorer = cosmiconfig(moduleName);\n\nexport async function loadConfig(): Promise<AITestConfig | null> {\n  try {\n    const result = await explorer.search();\n    if (result) {\n      return result.config as AITestConfig;\n    }\n    return null;\n  } catch (error) {\n    logger.error(`Failed to load config: ${error}`);\n    return null;\n  }\n}\n\nexport async function saveConfig(config: AITestConfig, targetDir: string = process.cwd()): Promise<void> {\n  const configPath = resolve(targetDir, '.aitestrc.json');\n  try {\n    await writeFile(configPath, JSON.stringify(config, null, 2), 'utf-8');\n    logger.success(`Saved config to ${configPath}`);\n  } catch (error) {\n    logger.error(`Failed to save config: ${error}`);\n    throw error;\n  }\n}\n","import chalk from 'chalk';\nimport ora from 'ora';\n\nexport const logger = {\n  info: (msg: string) => console.log(chalk.blue('ℹ'), msg),\n  success: (msg: string) => console.log(chalk.green('✔'), msg),\n  error: (msg: string) => console.error(chalk.red('✖'), msg),\n  warn: (msg: string) => console.warn(chalk.yellow('⚠'), msg),\n  log: (msg: string) => console.log(msg),\n  spinner: (text: string) => ora(text),\n};\n","import { generateText } from 'ai';\nimport { createOpenAI } from '@ai-sdk/openai';\nimport { createAnthropic } from '@ai-sdk/anthropic';\nimport { createGoogleGenerativeAI } from '@ai-sdk/google';\nimport { createDeepSeek } from '@ai-sdk/deepseek';\nimport { AITestConfig } from './config.js';\nimport * as dotenv from 'dotenv';\n\ndotenv.config();\n\nexport function getAIModel(config: AITestConfig) {\n  const providerName = config.provider.toLowerCase();\n  \n  if (providerName === 'openai') {\n    const openai = createOpenAI({\n      apiKey: config.apiKey || process.env.OPENAI_API_KEY,\n    });\n    return openai(config.model || 'gpt-4o');\n  } \n  \n  if (providerName === 'anthropic') {\n    const anthropic = createAnthropic({\n      apiKey: config.apiKey || process.env.ANTHROPIC_API_KEY,\n    });\n    return anthropic(config.model || 'claude-3-5-sonnet-20240620');\n  }\n\n  if (providerName === 'gemini' || providerName === 'google') {\n    const google = createGoogleGenerativeAI({\n      apiKey: config.apiKey || process.env.GEMINI_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY,\n    });\n    return google(config.model || 'gemini-2.5-flash');\n  }\n\n  if (providerName === 'deepseek') {\n    const deepseek = createDeepSeek({\n      apiKey: config.apiKey || process.env.DEEPSEEK_API_KEY,\n    });\n    return deepseek(config.model || 'deepseek-coder');\n  }\n\n  if (providerName === 'ollama') {\n    const ollama = createOpenAI({\n      baseURL: 'http://localhost:11434/v1',\n      apiKey: 'ollama', // API key isn't strictly needed for local Ollama\n    });\n    return ollama(config.model || 'llama3.1');\n  }\n\n  if (providerName === 'custom') {\n    const custom = createOpenAI({\n      baseURL: config.baseURL,\n      apiKey: config.apiKey || process.env.CUSTOM_API_KEY || 'custom',\n      headers: config.customHeaders,\n    });\n    return custom(config.model || 'local-model');\n  }\n\n  throw new Error(`Unsupported AI provider: ${config.provider}`);\n}\n\nexport async function askAI(config: AITestConfig, system: string, prompt: string) {\n  const model = getAIModel(config);\n  \n  const { text } = await generateText({\n    model,\n    system,\n    prompt,\n    maxRetries: 5,\n    temperature: config.temperature || 0.1,\n  });\n\n  return text;\n}\n"],"mappings":";AAAA,SAAS,eAAe;;;ACAxB,SAAS,mBAAmB;;;ACA5B,OAAO,WAAW;AAClB,OAAO,SAAS;AAET,IAAM,SAAS;AAAA,EACpB,MAAM,CAAC,QAAgB,QAAQ,IAAI,MAAM,KAAK,QAAG,GAAG,GAAG;AAAA,EACvD,SAAS,CAAC,QAAgB,QAAQ,IAAI,MAAM,MAAM,QAAG,GAAG,GAAG;AAAA,EAC3D,OAAO,CAAC,QAAgB,QAAQ,MAAM,MAAM,IAAI,QAAG,GAAG,GAAG;AAAA,EACzD,MAAM,CAAC,QAAgB,QAAQ,KAAK,MAAM,OAAO,QAAG,GAAG,GAAG;AAAA,EAC1D,KAAK,CAAC,QAAgB,QAAQ,IAAI,GAAG;AAAA,EACrC,SAAS,CAAC,SAAiB,IAAI,IAAI;AACrC;;;ADMA,IAAM,aAAa;AACnB,IAAM,WAAW,YAAY,UAAU;AAEvC,eAAsB,aAA2C;AAC/D,MAAI;AACF,UAAM,SAAS,MAAM,SAAS,OAAO;AACrC,QAAI,QAAQ;AACV,aAAO,OAAO;AAAA,IAChB;AACA,WAAO;AAAA,EACT,SAAS,OAAO;AACd,WAAO,MAAM,0BAA0B,KAAK,EAAE;AAC9C,WAAO;AAAA,EACT;AACF;;;AE9BA,SAAS,oBAAoB;AAC7B,SAAS,oBAAoB;AAC7B,SAAS,uBAAuB;AAChC,SAAS,gCAAgC;AACzC,SAAS,sBAAsB;AAE/B,YAAY,YAAY;AAEjB,cAAO;AAEP,SAAS,WAAWA,SAAsB;AAC/C,QAAM,eAAeA,QAAO,SAAS,YAAY;AAEjD,MAAI,iBAAiB,UAAU;AAC7B,UAAM,SAAS,aAAa;AAAA,MAC1B,QAAQA,QAAO,UAAU,QAAQ,IAAI;AAAA,IACvC,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,QAAQ;AAAA,EACxC;AAEA,MAAI,iBAAiB,aAAa;AAChC,UAAM,YAAY,gBAAgB;AAAA,MAChC,QAAQA,QAAO,UAAU,QAAQ,IAAI;AAAA,IACvC,CAAC;AACD,WAAO,UAAUA,QAAO,SAAS,4BAA4B;AAAA,EAC/D;AAEA,MAAI,iBAAiB,YAAY,iBAAiB,UAAU;AAC1D,UAAM,SAAS,yBAAyB;AAAA,MACtC,QAAQA,QAAO,UAAU,QAAQ,IAAI,kBAAkB,QAAQ,IAAI;AAAA,IACrE,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,kBAAkB;AAAA,EAClD;AAEA,MAAI,iBAAiB,YAAY;AAC/B,UAAM,WAAW,eAAe;AAAA,MAC9B,QAAQA,QAAO,UAAU,QAAQ,IAAI;AAAA,IACvC,CAAC;AACD,WAAO,SAASA,QAAO,SAAS,gBAAgB;AAAA,EAClD;AAEA,MAAI,iBAAiB,UAAU;AAC7B,UAAM,SAAS,aAAa;AAAA,MAC1B,SAAS;AAAA,MACT,QAAQ;AAAA;AAAA,IACV,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,UAAU;AAAA,EAC1C;AAEA,MAAI,iBAAiB,UAAU;AAC7B,UAAM,SAAS,aAAa;AAAA,MAC1B,SAASA,QAAO;AAAA,MAChB,QAAQA,QAAO,UAAU,QAAQ,IAAI,kBAAkB;AAAA,MACvD,SAASA,QAAO;AAAA,IAClB,CAAC;AACD,WAAO,OAAOA,QAAO,SAAS,aAAa;AAAA,EAC7C;AAEA,QAAM,IAAI,MAAM,4BAA4BA,QAAO,QAAQ,EAAE;AAC/D;AAEA,eAAsB,MAAMA,SAAsB,QAAgB,QAAgB;AAChF,QAAM,QAAQ,WAAWA,OAAM;AAE/B,QAAM,EAAE,KAAK,IAAI,MAAM,aAAa;AAAA,IAClC;AAAA,IACA;AAAA,IACA;AAAA,IACA,YAAY;AAAA,IACZ,aAAaA,QAAO,eAAe;AAAA,EACrC,CAAC;AAED,SAAO;AACT;;;AHrEA,SAAS,YAAY;AACrB,SAAS,iBAAiB;AAE1B,IAAM,YAAY,UAAU,IAAI;AAEzB,IAAM,iBAAiB,IAAI,QAAQ,SAAS,EAChD,YAAY,yCAAyC,EACrD,OAAO,YAAY;AAClB,QAAMC,UAAS,MAAM,WAAW;AAChC,MAAI,CAACA,SAAQ;AACX,WAAO,MAAM,mDAAmD;AAChE;AAAA,EACF;AAEA,QAAM,UAAU,OAAO,QAAQ,yCAAyC,EAAE,MAAM;AAEhF,MAAI,YAAY;AAChB,MAAI,YAAY;AAEhB,MAAI;AACF,UAAM,UAAU,UAAU;AAC1B,YAAQ,QAAQ,mCAAmC;AACnD;AAAA,EACF,SAAS,OAAY;AACnB,gBAAY,MAAM,UAAU;AAC5B,gBAAY,MAAM,UAAU,MAAM;AAClC,YAAQ,KAAK,wBAAwB;AAAA,EACvC;AAEA,QAAM,kBAAkB;AAAA;AAAA,EAE1B,UAAU,MAAM,IAAK,CAAC;AAAA;AAAA;AAAA,EAGtB,UAAU,MAAM,IAAK,CAAC;AAAA,MAClB,KAAK;AAEP,SAAO,KAAK,iCAAiC;AAC7C,QAAM,iBAAiB,OAAO,QAAQ,aAAa,EAAE,MAAM;AAE3D,MAAI;AACF,UAAM,cAAc,MAAM;AAAA,MACxBA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAEA,mBAAe,QAAQ,oBAAoB;AAC3C,YAAQ,IAAI,OAAO,cAAc,IAAI;AAAA,EAEvC,SAAS,OAAY;AACnB,mBAAe,KAAK,wBAAwB;AAC5C,WAAO,MAAM,MAAM,OAAO;AAAA,EAC5B;AACF,CAAC;","names":["config","config"]}

package/dist/commands/fix.d.ts

@@ -0,0 +1,5 @@
+import { Command } from 'commander';
+
+declare const fixCommand: Command;
+
+export { fixCommand };