ai-code-review-toolkit 0.0.2

package/README.md

@@ -0,0 +1,275 @@
+# AI Code Review [中文文档](README.zh-CN.md)
+
+AI-powered git pre-commit hook for automated code review with customizable rules.
+
+## Features
+
+- 🤖 **Multi-Model Support**: Works with OpenAI、Deepseek and Ollama、LM Studio
+- 🔍 **Smart Diff Analysis**: Focuses on meaningful changes, ignores deletions
+- ⚙️ **Customizable Rules**: Security, performance, style checks
+- ✏️ **Custom Prompts**: Fully customize review criteria and prompts
+- 📊 **Graded Feedback**: High/Medium/Low severity classification
+- 🛠 **Easy Integration**: Simple npm install and config
+
+## Installation
+
+```bash
+npm install --save-dev ai-code-review-kit
+# or
+yarn add -D ai-code-review-kit
+```
+
+Add to git pre-commit hook:
+
+```bash
+npx ai-review install
+```
+
+Check / uninstall / reinstall:
+
+```bash
+npx ai-review check
+npx ai-review uninstall
+npx ai-review install --force
+```
+
+Or if you have husky installed:
+
+```bash
+npx ai-review install
+```
+
+Yarn PnP is supported. If `.pnp.cjs` exists, the installed hook will prefer running via `yarn ai-review run` (fallback to `corepack yarn` if available).
+
+Legacy (run by husky command):
+
+```JSON
+.package.json
+{
+  "husky": {
+    "hooks": {
+      "pre-commit": "npx ai-review run"
+    }
+  }
+}
+```
+
+## Configuration
+
+### Option 1: package.json
+
+```json
+"aiCheckConfig": {
+  "providerType": "openai",
+  "apiKey": "your-api-key",
+  "model": "gpt-4",
+  "baseURL": "https://api.openai.com",
+  "timeoutMs": 120000,
+  "concurrency": 2,
+  "maxRetries": 1,
+  "retryDelayMs": 500,
+  "diffContextLines": 3,
+  "maxChunkSize": 4000,
+  "language": "english",
+  "ignoreDeletions": true,
+  "stripUnchangedCommentLines": true,
+  "checkSecurity": true,
+  "checkPerformance": true,
+  "checkStyle": false,
+  "enabledFileExtensions": ".html, .js, .jsx, .ts, .tsx, .vue"
+}
+```
+
+### Option 2: .env file
+
+```env
+providerType=openai
+baseURL=http://localhost:11434
+model=gpt-4
+maxChunkSize=4000
+language=chinese
+```
+
+If both `package.json` (`aiCheckConfig`) and `.env` exist, configurations will be merged and `.env` will take precedence.
+
+Environment variables (process.env) override both `.env` and `package.json`.
+
+To avoid conflicts with common system env vars (e.g. `LANGUAGE`), this tool only reads:
+
+- Exact config keys (case-sensitive), e.g. `providerType`, `baseURL`, `apiKey`
+- Or prefixed keys (recommended): `AI_REVIEW_*` / `AI_CODE_REVIEW_KIT_*`, e.g. `AI_REVIEW_PROVIDER_TYPE`, `AI_REVIEW_BASE_URL`, `AI_REVIEW_API_KEY`
+
+Note on `baseURL` / `embeddingsBaseURL`:
+
+- Use an API root like `https://example.com` or `https://example.com/v1`.
+- Avoid setting a full endpoint like `.../v1/responses` / `.../v1/chat/completions`. (The tool will try to normalize these, but root URLs are more predictable.)
+
+## Knowledge Base (Local RAG)
+
+This project supports a **local, project-level RAG** based on **LanceDB**.
+
+Workflow:
+
+1. Build index locally (one-time / incremental per your needs)
+2. During review, the tool embeds the current diff chunk, retrieves top-K relevant snippets from the index, and injects them into the prompt.
+3. If index is missing, RAG will be skipped automatically (fallback to normal review).
+
+Build index:
+
+```bash
+npx ai-review index
+```
+
+Minimal config (add into `aiCheckConfig`):
+
+```json
+{
+  "enableRag": true,
+  "knowledgeBasePaths": ["."],
+  "knowledgeBaseIndexDir": ".ai-reviewer-cache/lancedb",
+  "knowledgeBaseTable": "project_kb",
+  "ragTopK": 6,
+  "ragMaxChars": 8000,
+  "embeddingsBaseURL": "https://api.openai.com",
+  "embeddingsModel": "text-embedding-3-small"
+}
+```
+
+Notes:
+
+- Cache directory: `.ai-reviewer-cache/` (recommended to keep ignored by git)
+- LanceDB keeps historical versions by default. This tool prunes old versions after each `ai-review index` run to avoid unbounded cache growth. You can also safely delete `.ai-reviewer-cache/` anytime and rebuild.
+- Embeddings config precedence:
+  - `embeddingsBaseURL/embeddingsApiKey/embeddingsModel` if provided
+  - otherwise fallback to `baseURL/apiKey` (if compatible)
+
+### Full Configuration Options
+
+| Parameter | Type | Default | Description |
+|-----------|------|---------|-------------|
+| providerType | string | "openai" | AI provider type (openai、deepseek、zhipu, ollama or LMStudio) |
+| apiKey | string | - | Provider API key (not required for Ollama or LMStudio) |
+| model | string | "gpt-3.5-turbo" | Model name |
+| temperature | number | 0.2 | Controls randomness of AI output (higher = more random) |
+| baseURL | string | `"https://api.openai.com"` (OpenAI)<br>`"https://api.deepseek.com"` (Deepseek)<br>`"https://open.bigmodel.cn/api/paas/v4"` (Zhipu)<br>`"http://localhost:11434"` (Ollama)<br>`"http://127.0.0.1:1234"` (LM Studio)| API base URL (can include or omit the `/v1` suffix) |
+| useResponsesApi | boolean | false | Force using the Responses API (`/responses`) for OpenAI-compatible providers |
+| timeoutMs | number | 120000 | Request timeout (ms) |
+| concurrency | number | 2 | Max parallel AI requests |
+| maxRetries | number | 1 | Retry count on transient API/format errors |
+| retryDelayMs | number | 500 | Base retry delay (ms), exponential backoff |
+| diffContextLines | number | 3 | `git diff --unified` context lines |
+| maxChunkSize | number | 12000 | Max diff chunk size (characters) |
+| customPrompts | string | '' |Custom prompt templates. When provided, these will completely replace the default security (checkSecurity), performance (checkPerformance) and style (checkStyle) checks. |
+| language | string | "chinese" | Output language |
+| strict | boolean | true | Fail on API errors |
+| skipReview | boolean | false | Skip AI review (always pass) |
+| correctedResult | boolean | true | When the result field in the AI's returned result does not match the specific detection item result in the list, the system will automatically correct the final determination result based on the actual detection problem.  |
+| showNormal | boolean | false | Show low/medium severity issues |
+| ignoreDeletions | boolean | true | Remove deleted lines (`-...`) before sending diff to AI |
+| stripUnchangedCommentLines | boolean | true | Remove unchanged comment-only context lines (e.g. ` // ...`) |
+| checkSecurity | boolean | true | Enable security checks |
+| checkPerformance | boolean | true | Enable performance checks |
+| checkStyle | boolean | false | Enable style checks |
+| enabledFileExtensions | string | '.html, .js, .jsx, .ts, .tsx, .vue' | File types to review |
+| enableRag | boolean | false | Enable local RAG knowledge retrieval |
+| knowledgeBasePaths | string\|string[] | '' | Paths to index (e.g. ["."] or "src,docs") |
+| knowledgeBaseIndexDir | string | '.ai-reviewer-cache/lancedb' | LanceDB index directory (relative to repo root) |
+| knowledgeBaseTable | string | 'project_kb' | LanceDB table name |
+| ragTopK | number | 6 | Retrieve top K snippets |
+| ragMaxChars | number | 8000 | Max chars injected into prompt |
+| embeddingsProviderType | string | '' | Embeddings provider type (defaults to `providerType`) |
+| embeddingsBaseURL | string | '' | Embeddings baseURL (provider default if omitted) |
+| embeddingsApiKey | string | '' | API key for embeddings endpoint |
+| embeddingsModel | string | provider default | Embeddings model (OpenAI: `text-embedding-3-small`, Zhipu: `embedding-3`) |
+| embeddingsDimensions | number | null | Optional embeddings output dimensions (if supported by the model/provider) |
+
+## Review Process
+
+1. **Diff Extraction**: Gets staged changes via `git diff --cached`
+2. **File Filtering**: Only processes specified file extensions
+3. **Chunk Splitting**: Splits large diffs into manageable chunks
+4. **AI Analysis**: Sends chunks to configured AI provider
+5. **Result Aggregation**: Combines results from all chunks
+6. **Output**: Displays issues grouped by severity
+
+## Example Output
+
+```bash
+Find 1 changed files...
+Running code review with AI: The content will be reviewed in 1 sessions for better accuracy.
+
+X Code review was not passed.Please fix the following high-level issues and try again.
+- src/auth.js: [高] - 安全问题 - 硬编码的API密钥
+  Suggested fix: 使用环境变量存储敏感信息
+- src/db.js: [中] - 性能问题 - 缺少数据库连接池
+  Suggested fix: 实现连接池减少连接开销
+
+√ Code review passed.
+
+```
+
+## Supported Providers
+
+### OpenAI
+
+- Required: `apiKey`
+- Optional: `model` (default: gpt-3.5-turbo), `baseURL` (default: https://api.openai.com)
+- Models: gpt-4, gpt-3.5-turbo
+  
+### Deepseek
+
+- Required: `apiKey`
+- Optional: `model` (default: deepseek-chat), `baseURL` (default: https://api.deepseek.com)
+- Models: deepseek-chat、deepseek-reasoner
+
+### Zhipu (BigModel)
+
+- Required: `apiKey`
+- Optional: `model` (default: glm-4), `baseURL` (default: https://open.bigmodel.cn/api/paas/v4)
+- Embeddings: set `embeddingsProviderType: "zhipu"` and `embeddingsModel: "embedding-3"` (optional `embeddingsDimensions`)
+
+### Ollama (Local AI Models)
+
+- Required: None (runs locally)
+- Optional: `model` (default: gpt-3.5-turbo), `baseURL` (default: http://localhost:11434)
+- Setup:
+  1. Install Ollama: https://ollama.ai/
+  2. Download models: `ollama pull <model-name>`
+  3. Common models: llama2, codellama, mistral
+- Example .env configuration:
+  ```env
+  providerType=ollama
+  model=codellama
+  baseURL=http://localhost:11434
+  ```
+
+### LMStudio (Local AI Models)
+
+- Required: None (runs locally)
+- Optional: `model` (default: qwen/qwq-32b), `baseURL` (default: http://127.0.0.1:1234)
+
+## Troubleshooting
+
+### Hook not running
+
+- Verify `.git/hooks/pre-commit` exists and is executable
+- Check file contains `node path/to/ai-review.js`
+
+### API Errors
+
+- Verify API key and base URL (not required for Ollama)
+- Check network connectivity
+- Set `strict: false` to allow commit on API errors
+
+### No changes found
+
+- Check `enabledFileExtensions` matches your file types
+- Verify changes are staged (`git add`)
+
+### The returned data format does not conform to the specification
+
+The model fails to produce valid JSON output, indicating potential limitations in its instruction-following capability. Ensure the model is capable of generating structured data.
+
+## License
+
+ISC

package/README.zh-CN.md

@@ -0,0 +1,268 @@
+# AI 代码审查 [English Md](README.md)
+
+基于人工智能的git pre-commit钩子，提供自动化代码审查功能，支持自定义规则。
+
+## 功能特性
+
+- 🤖 **多模型支持**: 支持OpenAI、Deepseek、Ollama和LM Studio
+- 🔍 **智能差异分析**: 聚焦有意义的变更，忽略删除内容
+- ⚙️ **可定制规则**: 安全检查、性能优化、代码风格
+- ✏️ **自定义提示**: 完全自定义审查标准和提示词
+- 📊 **分级反馈**: 高/中/低严重性分类
+- 🛠 **简单集成**: 通过npm安装和配置
+
+## 安装
+
+```bash
+npm install --save-dev ai-code-review-kit
+# 或
+yarn add -D ai-code-review-kit
+```
+
+添加到git pre-commit钩子:
+
+```bash
+npx ai-review install
+```
+
+检查 / 卸载 / 强制重装：
+
+```bash
+npx ai-review check
+npx ai-review uninstall
+npx ai-review install --force
+```
+
+如果已安装husky:
+
+```bash
+npx ai-review install
+```
+
+支持 Yarn PnP：如果仓库根目录存在`.pnp.cjs`，已安装的 hook 会优先通过 `yarn ai-review run` 执行（若可用则回退到 `corepack yarn`）。
+
+旧方式（由 husky 命令执行）：
+
+```JSON
+.package.json
+{
+  "husky": {
+    "hooks": {
+      "pre-commit": "npx ai-review run"
+    }
+  }
+}
+```
+
+## 配置
+
+### 方式1: package.json
+
+```json
+"aiCheckConfig": {
+  "providerType": "openai",
+  "apiKey": "your-api-key",
+  "model": "gpt-4",
+  "baseURL": "https://api.openai.com",
+  "timeoutMs": 120000,
+  "concurrency": 2,
+  "maxRetries": 1,
+  "retryDelayMs": 500,
+  "diffContextLines": 3,
+  "maxChunkSize": 4000,
+  "language": "english",
+  "ignoreDeletions": true,
+  "stripUnchangedCommentLines": true,
+  "checkSecurity": true,
+  "checkPerformance": true,
+  "checkStyle": false,
+  "enabledFileExtensions": ".html, .js, .jsx, .ts, .tsx, .vue"
+}
+```
+
+### 方式2: .env文件
+
+```env
+providerType=openai
+baseURL=http://localhost:11434
+model=gpt-4
+maxChunkSize=4000
+language=chinese
+```
+
+如果同时存在`package.json`（`aiCheckConfig`）和`.env`，将合并配置，并以`.env`优先。
+
+环境变量（process.env）优先级最高，会覆盖 `.env` 和 `package.json`。
+
+为避免与系统常见环境变量冲突（例如 `LANGUAGE`），本工具只会读取：
+
+- 与配置项完全同名（区分大小写）的 env，例如 `providerType`、`baseURL`、`apiKey`
+- 或带前缀（推荐）的 env：`AI_REVIEW_*` / `AI_CODE_REVIEW_KIT_*`，例如 `AI_REVIEW_PROVIDER_TYPE`、`AI_REVIEW_BASE_URL`、`AI_REVIEW_API_KEY`
+
+关于 `baseURL` / `embeddingsBaseURL`：
+
+- 建议填写 API 根路径，例如 `https://example.com` 或 `https://example.com/v1`
+- 避免填写完整 endpoint，例如 `.../v1/responses` / `.../v1/chat/completions`（工具会尝试自动归一化，但根路径更稳定）
+
+## 项目知识库（本地 RAG）
+
+支持基于 **LanceDB** 的本地项目级 RAG：先离线构建索引，review 时针对当前 diff chunk 做向量检索，把相关片段注入到 prompt 中增强审查。
+
+构建索引（手动触发）：
+
+```bash
+npx ai-review index
+```
+
+最小配置示例（写入 `aiCheckConfig`）：
+
+```json
+{
+  "enableRag": true,
+  "knowledgeBasePaths": ["."],
+  "knowledgeBaseIndexDir": ".ai-reviewer-cache/lancedb",
+  "knowledgeBaseTable": "project_kb",
+  "ragTopK": 6,
+  "ragMaxChars": 8000,
+  "embeddingsBaseURL": "https://api.openai.com",
+  "embeddingsModel": "text-embedding-3-small"
+}
+```
+
+说明：
+
+- 缓存目录：`.ai-reviewer-cache/`（建议加入 gitignore，不提交）
+- LanceDB 默认会保留历史版本。本工具会在每次 `ai-review index` 后自动清理旧版本，避免缓存无限增长；你也可以随时安全删除 `.ai-reviewer-cache/` 并重新构建。
+- 若索引不存在，会提示你执行 `ai-review index`，并自动降级为普通 review（不会阻断）
+- embeddings 配置优先级：优先使用 `embeddingsBaseURL/embeddingsApiKey/embeddingsModel`，否则回退到 `baseURL/apiKey`（需兼容 OpenAI embeddings 接口）
+
+### 完整配置选项
+
+| 参数 | 类型 | 默认值 | 描述 |
+|-----------|------|---------|-------------|
+| providerType | string | "openai" | AI提供商类型(openai、deepseek、zhipu、ollama或LMStudio) |
+| apiKey | string | - | 提供商API密钥(Ollama和LM Studio不需要) |
+| model | string | "gpt-3.5-turbo" | 模型名称 |
+| temperature | number | 0.2 | 控制AI输出的随机性(值越高越随机) |
+| baseURL | string | `"https://api.openai.com"` (OpenAI)<br>`"https://api.deepseek.com"` (Deepseek)<br>`"https://open.bigmodel.cn/api/paas/v4"` (智谱)<br>`"http://localhost:11434"` (Ollama)<br>`"http://127.0.0.1:1234"` (LM Studio)| API基础URL（可包含或省略`/v1`后缀） |
+| useResponsesApi | boolean | false | 强制对 OpenAI 兼容提供商使用 Responses API（`/responses`） |
+| timeoutMs | number | 120000 | 请求超时时间(毫秒) |
+| concurrency | number | 2 | 最大并行AI请求数 |
+| maxRetries | number | 1 | 调用失败/格式错误时的重试次数 |
+| retryDelayMs | number | 500 | 重试基础延迟(毫秒)，指数退避 |
+| diffContextLines | number | 3 | `git diff --unified`上下文行数 |
+| maxChunkSize | number | 12000 | 最大差异块大小(字符数) |
+| customPrompts | string | '' |自定义提示模板。当提供时，将完全替换默认的安全(checkSecurity)、性能(checkPerformance)和风格(checkStyle)检查。 |
+| language | string | "chinese" | 输出语言 |
+| strict | boolean | true | 调用API时如果发生错误导致pre-commit结果不通过 |
+| skipReview | boolean | false | 跳过 AI 审查（直接通过） |
+| correctedResult | boolean | true | 当AI返回结果中的result字段与列表中特定检测项结果不匹配时，系统会根据实际检测问题自动修正最终判定结果。 |
+| showNormal | boolean | false | 显示低/中严重性问题 |
+| ignoreDeletions | boolean | true | 发送给AI前移除删除行（`-...`） |
+| stripUnchangedCommentLines | boolean | true | 移除未变更的纯注释上下文行（例如` // ...`） |
+| checkSecurity | boolean | true | 启用安全检查 |
+| checkPerformance | boolean | true | 启用性能检查 |
+| checkStyle | boolean | false | 启用风格检查 |
+| enabledFileExtensions | string | '.html, .js, .jsx, .ts, .tsx, .vue' | 需要审查的文件类型 |
+| enableRag | boolean | false | 开启本地 RAG 知识检索增强 |
+| knowledgeBasePaths | string\|string[] | '' | 需要索引的路径（例如 ["."] 或 "src,docs"） |
+| knowledgeBaseIndexDir | string | '.ai-reviewer-cache/lancedb' | LanceDB 索引目录（相对仓库根目录） |
+| knowledgeBaseTable | string | 'project_kb' | LanceDB 表名 |
+| ragTopK | number | 6 | 检索 Top K 片段 |
+| ragMaxChars | number | 8000 | 注入 prompt 的最大字符数 |
+| embeddingsProviderType | string | '' | embeddings 提供商类型（默认跟随 `providerType`） |
+| embeddingsBaseURL | string | '' | embeddings baseURL（不填则使用提供商默认值） |
+| embeddingsApiKey | string | '' | embeddings 接口的 API key |
+| embeddingsModel | string | 提供商默认 | embeddings 模型（OpenAI: `text-embedding-3-small`，智谱: `embedding-3`） |
+| embeddingsDimensions | number | null | embeddings 输出维度（可选，需模型/服务端支持） |
+
+## 审查流程
+
+1. **差异提取**: 通过`git diff --cached`获取暂存变更
+2. **文件过滤**: 仅处理指定扩展名的文件
+3. **分块处理**: 将大差异分割为可管理的块
+4. **AI分析**: 将块发送到配置的AI提供商
+5. **结果聚合**: 合并所有块的结果
+6. **输出**: 按严重性分组显示问题
+
+## 示例输出
+
+```bash
+Find 1 changed files...
+Running code review with AI: The content will be reviewed in 1 sessions for better accuracy.
+
+X Code review was not passed.Please fix the following high-level issues and try again.
+- src/auth.js: [高] - 安全问题 - 硬编码的API密钥
+  Suggested fix: 使用环境变量存储敏感信息
+- src/db.js: [中] - 性能问题 - 缺少数据库连接池
+  Suggested fix: 实现连接池减少连接开销
+
+√ Code review passed.
+
+```
+
+## 支持的提供商
+
+### OpenAI
+
+- 必填: `apiKey`
+- 可选: `model` (默认: gpt-3.5-turbo), `baseURL` (默认: https://api.openai.com)
+- 模型: gpt-4, gpt-3.5-turbo
+  
+### Deepseek
+
+- 必填: `apiKey`
+- 可选: `model` (默认: deepseek-chat), `baseURL` (默认: https://api.deepseek.com)
+- 模型: deepseek-chat、deepseek-reasoner
+
+### 智谱（BigModel）
+
+- 必填: `apiKey`
+- 可选: `model`（默认: glm-4）, `baseURL`（默认: https://open.bigmodel.cn/api/paas/v4）
+- embeddings：建议设置 `embeddingsProviderType: "zhipu"` 且 `embeddingsModel: "embedding-3"`（可选 `embeddingsDimensions`）
+
+### Ollama (本地AI模型)
+
+- 必填: 无(本地运行)
+- 可选: `model` (默认: gpt-3.5-turbo), `baseURL` (默认: http://localhost:11434)
+- 设置:
+  1. 安装Ollama: https://ollama.ai/
+  2. 下载模型: `ollama pull <模型名称>`
+  3. 常用模型: llama2, codellama, mistral
+- 示例.env配置:
+  ```env
+  providerType=ollama
+  model=codellama
+  baseURL=http://localhost:11434
+  ```
+
+### LMStudio (本地AI模型)
+
+- 必填: 无(本地运行)
+- 可选: `model` (默认: qwen/qwq-32b), `baseURL` (默认: http://127.0.0.1:1234)
+
+## 故障排除
+
+### 钩子未运行
+
+- 检查`.git/hooks/pre-commit`是否存在且可执行
+- 确认文件包含`node path/to/ai-review.js`
+
+### API错误
+
+- 验证API密钥和基础URL(Ollama不需要)
+- 检查网络连接
+- 设置`strict: false`允许API错误时提交
+
+### 未发现变更
+
+- 检查`enabledFileExtensions`是否匹配您的文件类型
+- 确认变更已暂存(`git add`)
+
+### The returned data format does not conform to the specification
+
+模型未能生成有效的JSON输出，表明其遵循指令的能力可能有限。请确保模型能够生成结构化数据。
+
+## 许可证
+
+ISC

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "ai-code-review-toolkit",
+  "version": "0.0.2",
+  "description": "AI-powered git code review toolkit with optional local RAG knowledge base",
+  "main": "./src/index.js",
+  "type": "module",
+  "exports": {
+    ".": "./src/index.js"
+  },
+  "files": [
+    "src",
+    "README.md",
+    "README.zh-CN.md"
+  ],
+  "scripts": {
+    "pre-commit": "node src/ai-review.js",
+    "test": "node --test"
+  },
+  "bin": {
+    "ai-review": "./src/ai-review-cli.js"
+  },
+  "keywords": [
+    "ai",
+    "code-review",
+    "git",
+    "git-hooks",
+    "rag",
+    "lancedb"
+  ],
+  "author": "puppy,liweihao",
+  "license": "ISC",
+  "dependencies": {
+    "@lancedb/lancedb": "^0.26.2",
+    "axios": "^1.13.6",
+    "chalk": "^5.4.1"
+  }
+}

package/src/AIError.js

@@ -0,0 +1,7 @@
+export default class AIError extends Error {
+  constructor(message, options = {}) {
+    super(message);
+    this.name = 'AIError';
+    this.options = options;
+  }
+}

package/src/ai-review-cli.js

@@ -0,0 +1,79 @@
+#!/usr/bin/env node
+
+import chalk from 'chalk'
+import { main as hookMain } from './cli.js'
+import { main as runMain } from './ai-review.js'
+import { main as indexMain } from './kb-index.js'
+
+function printHelp() {
+  console.log(`Usage: ai-review <command> [options]
+
+Commands:
+  install     Install git pre-commit hook (auto-detect husky)
+  check       Check whether hook is installed
+  uninstall   Remove installed hook block
+  run         Run code review for staged changes
+  index       Build local RAG knowledge index (LanceDB)
+`)
+}
+
+function hasTargetArg(args) {
+  return args.some((a, i) => {
+    if (a === '--target' || a === '-t') return true
+    if (typeof a === 'string' && a.startsWith('--target=')) return true
+    // -t <value> handled above
+    return false
+  })
+}
+
+export async function main(argv = process.argv.slice(2)) {
+  const [command, ...rest] = argv
+  if (!command || command === '-h' || command === '--help') {
+    printHelp()
+    process.exit(command ? 0 : 1)
+  }
+
+  if (command === 'install') {
+    const args = [...rest]
+    if (!hasTargetArg(args)) {
+      args.unshift('auto')
+      args.unshift('--target')
+    }
+    return hookMain(args)
+  }
+
+  if (command === 'check') {
+    const args = [...rest]
+    if (!hasTargetArg(args)) {
+      args.unshift('auto')
+      args.unshift('--target')
+    }
+    return hookMain(['--check', ...args])
+  }
+
+  if (command === 'uninstall' || command === 'remove') {
+    const args = [...rest]
+    if (!hasTargetArg(args)) {
+      args.unshift('auto')
+      args.unshift('--target')
+    }
+    return hookMain(['--uninstall', ...args])
+  }
+
+  if (command === 'run') {
+    return await runMain(rest)
+  }
+
+  if (command === 'index') {
+    return await indexMain(rest)
+  }
+
+  console.error(chalk.red(`Unknown command: ${command}`))
+  printHelp()
+  process.exit(1)
+}
+
+main().catch((err) => {
+  console.error(chalk.red(err?.message || err))
+  process.exit(1)
+})