MarceloAdan73/ai-agent-toolkit
GitHub: MarceloAdan73/ai-agent-toolkit
一套包含五个 AI agent 的命令行工具集，用于自动化文档生成、测试编写、代码审查、重构建议和安全审计等软件开发生命周期任务。
Stars: 2 | Forks: 0
# 🤖 AI AGENT TOOLKIT
**🚀 包含 5 个 AI 驱动的 CLI agent 套件，用于自动化软件开发工作流。**
[](https://github.com/MarceloAdan73/ai-agent-toolkit/actions)
[](https://github.com/MarceloAdan73/ai-agent-toolkit)
[](LICENSE)
[](https://nodejs.org)
[](https://typescriptlang.org)
[](https://vitest.dev)
[](#-agents)
[](#-supported-ai-providers)
[](https://github.com/MarceloAdan73/ai-agent-toolkit/pulls)
[](https://www.npmjs.com/package/@aiagentkit/ai-agent-toolkit)
[](https://www.npmjs.com/package/@aiagentkit/agent-doc-generator)
[](https://www.npmjs.com/package/@aiagentkit/agent-test-generator)
[](https://www.npmjs.com/package/@aiagentkit/agent-code-review)
[](https://www.npmjs.com/package/@aiagentkit/agent-refactor)
[](https://www.npmjs.com/package/@aiagentkit/agent-security-audit)
[📦 安装](#-installation) · [🤖 Agents](#-agents) · [🧠 Providers](#-supported-ai-providers) · [⚙️ CLI 参数](#%EF%B8%8F-cli-flags) · [🤝 参与贡献](#-contributing)
## 📋 概述
AI Agent Toolkit 是一个 **monorepo**，汇集了 5 个独立的 AI agent，每个 agent 专门负责不同的开发任务。它们共享统一的 CLI、通用技术栈，并支持多个 AI provider。
```
npx ai-toolkit doc ./src # 📄 automatic documentation
npx ai-toolkit test ./src # 🧪 unit tests
npx ai-toolkit review ./src # 🔍 code review
npx ai-toolkit refactor ./src # ♻️ refactoring
npx ai-toolkit audit ./src # 🛡️ security audit
```
## 🤖 Agents
| 命令 | 包 | 版本 | 描述 |
|:-------:|---------|:-------:|-------------|
| `ai-toolkit doc` | [`@aiagentkit/agent-doc-generator`](https://www.npmjs.com/package/@aiagentkit/agent-doc-generator) | v1.0.0 | 📄 自动生成 JSDoc 文档 |
| `ai-toolkit test` | [`@aiagentkit/agent-test-generator`](https://www.npmjs.com/package/@aiagentkit/agent-test-generator) | v1.0.0 | 🧪 使用 Vitest 生成单元测试 |
| `ai-toolkit review` | [`@aiagentkit/agent-code-review`](https://www.npmjs.com/package/@aiagentkit/agent-code-review) | v1.0.0 | 🔍 分析代码质量和圈复杂度 |
| `ai-toolkit refactor` | [`@aiagentkit/agent-refactor`](https://www.npmjs.com/package/@aiagentkit/agent-refactor) | v1.0.0 | ♻️ 建议并应用带有 diff 的重构 |
| `ai-toolkit audit` | [`@aiagentkit/agent-security-audit`](https://www.npmjs.com/package/@aiagentkit/agent-security-audit) | v1.0.0 | 🛡️ OWASP Top 10 安全审计 |
## 🧠 支持的 AI Providers
支持 5 个 provider，具备**自动检测**功能。如果你不指定 `--provider`，该工具包会检测你已配置的选项。
| Provider | 默认模型 | 认证 | 备注 |
|----------|--------------|------|-------|
| 🔷 **Gemini** | `gemini-2.5-flash` | `GEMINI_API_KEY` | 提供免费额度 |
| 🟢 **OpenAI** | `gpt-4o` | `OPENAI_API_KEY` | GPT-4o, GPT-4o-mini |
| 🟠 **Anthropic** | `claude-sonnet-4-20250514` | `ANTHROPIC_API_KEY` | 最适合编程 |
| 🔵 **DeepSeek** | `deepseek-chat` | `DEEPSEEK_API_KEY` | 最具性价比 |
| 🦙 **Ollama** | 本地模型 | 无需密钥 | 本地、免费、完全隐私 |
**🔍 检测优先级：** OpenAI > Anthropic > DeepSeek > Gemini > Ollama
```
# 🎯 显式指定 provider
npx ai-toolkit doc --path ./src --provider openai
npx ai-toolkit audit --path ./src --provider ollama --model qwen2.5-coder:7b
```
## 📦 安装
### 🚀 从 npm 安装（已发布）
```
npm install -g @aiagentkit/ai-agent-toolkit
npx ai-toolkit doc --path ./src
```
或者单独安装 agent：
```
npx @aiagentkit/agent-doc-generator --path ./src
npx @aiagentkit/agent-test-generator --path ./src
```
### 📥 从仓库安装
```
git clone https://github.com/MarceloAdan73/ai-agent-toolkit.git
cd ai-agent-toolkit
npm install && npm run build
# ⚙️ 配置 provider
cp .env.example .env
# 使用你的 API key 编辑 .env
# ▶️ 运行
node toolkit/dist/cli.js doc --path ./src
```
## 🏗️ 架构
```
ai-agent-toolkit/
├── package.json # 📦 npm workspaces
├── toolkit/ # 🖥️ Unified CLI (ai-toolkit)
│ └── src/cli.ts
├── agent-doc-generator/ # 📄 documentation
├── agent-test-generator/ # 🧪 unit tests
├── agent-code-review/ # 🔍 code review
├── agent-refactor/ # ♻️ refactoring
└── agent-security-audit/ # 🛡️ OWASP security audit
```
每个 agent 包含：`src/`（逻辑）、`tests/`（Vitest）、`index.ts`（入口点）、`package.json`（bin、files、engines）。
## 🛠️ 技术栈
| 层级 | 技术 |
|-------|------------|
| 💻 语言 | TypeScript（严格模式） |
| ⚡ Runtime | Node.js >= 18（ES Modules） |
| 🧠 AI Providers | Gemini, OpenAI, Anthropic, DeepSeek, Ollama |
| 🖥️ CLI | Commander.js |
| ⏳ 进度 UI | ora |
| 🧪 测试 | Vitest（596 个测试） |
| 🔄 CI/CD | GitHub Actions |
| 📦 打包 | tsc（TypeScript 编译器） |
## ⚙️ CLI 参数
| 参数 | 描述 | 默认值 |
|------|-------------|---------|
| `--path` | 📁 要处理的文件或文件夹 | *(必填)* |
| `--provider` | 🧠 `gemini` \| `openai` \| `anthropic` \| `deepseek` \| `ollama` | 自动检测 |
| `--model` | 🤖 要使用的模型 | 取决于 provider |
| `--base-url` | 🌐 服务器 URL（Ollama） | `http://localhost:11434` |
| `--api-key` | 🔑 API 密钥（Ollama 可选） | `.env` |
| `--max-chars` | 📏 每个文件的字符限制 | `15000` |
| `--output` | 📂 输出文件夹/文件 | 位于源文件旁 |
| `--extensions` | 📎 要处理的扩展名（CSV） | 全部 |
| `--split` | 📑 每个源文件生成一个输出文件 | `false` |
| `--format` | 📄 `terminal` \| `markdown` \| `html` \| `pdf` | `terminal` |
| `--severity` | ⚠️ 最低严重级别（审计） | `info` |
| `--dry-run` | 👁️ 仅预览，不写入 | `false` |
| `--verbose` | 📝 详细日志 | `false` |
## ✨ 功能
| 功能 | 描述 |
|---------|-------------|
| 🧠 **多 provider** | 5 个支持自动检测的 AI provider |
| 💾 **智能缓存** | MD5 哈希，跳过未更改的文件 |
| 🔄 **弹性重试** | API 出错时进行 3 次带退避的重试 |
| ✂️ **安全截断** | 大文件在截断时会发出警告 |
| 📎 **扩展名过滤** | 使用 `--extensions .ts,.prisma` 限制范围 |
| 📄 **导出** | Terminal、Markdown、HTML 或 PDF |
| 👁️ **试运行 (Dry-run)** | 在写入更改前预览 |
| 📑 **拆分模式** | 为每个分析的源文件生成一个输出文件 |
## 📜 脚本
```
npm run build # 🏗️ build all packages
npm run test # 🧪 596 tests
npm run typecheck # 🔍 strict typecheck
npm run test:integration # 🔗 tests with real providers (requires API keys)
```
## 🗺️ 路线图
- [x] ✅ 5 个功能完备的 agent，包含 596 个测试
- [x] ✅ 统一的 CLI `ai-toolkit`
- [x] ✅ 多 provider + 自动检测
- [x] ✅ 响应缓存
- [x] ✅ HTML/PDF 导出
- [x] ✅ 使用 GitHub Actions 进行 CI/CD
- [x] ✅ 准备好进行 npm 发布
- [x] ✅ 发布到 npm
- [ ] ⏳ 与真实 provider 的集成测试
- [ ] ⏳ 通过 `.rc` 文件进行配置
## 📄 许可证
[MIT](./LICENSE) — 🧑💻 Marcelo Adan
**[🐙 GitHub](https://github.com/MarceloAdan73)** · **[🐛 报告 Bug](https://github.com/MarceloAdan73/ai-agent-toolkit/issues)** · **[💡 请求功能](https://github.com/MarceloAdan73/ai-agent-toolkit/issues)**
标签：CLI, MITM代理, SOC Prime, TypeScript, WiFi技术, 人工智能, 代码审查, 代码重构, 安全插件, 开发工具, 用户模式Hook绕过, 网络调试, 自动化, 自动化攻击