gitlab-ai-review 1.0.3 → 1.1.0

package/index.js

@@ -5,6 +5,7 @@
 
 import { getConfig, validateConfig } from './lib/config.js';
 import { GitLabClient } from './lib/gitlab-client.js';
+import { AIClient } from './lib/ai-client.js';
 
 /**
  * GitLab AI Review SDK 主类
@@ -12,7 +13,7 @@ import { GitLabClient } from './lib/gitlab-client.js';
 export class GitLabAIReview {
   constructor(options = {}) {
     this.name = 'GitLab AI Review SDK';
-    this.version = '1.0.0';
+    this.version = '1.0.3';
 
     // 如果传入了配置，使用手动配置；否则使用自动检测
     if (options.token || options.gitlab) {
@@ -26,6 +27,10 @@ export class GitLabAIReview {
           projectId: options.projectId || options.project?.projectId,
           mergeRequestIid: options.mergeRequestIid || options.project?.mergeRequestIid,
         },
+        ai: {
+          arkApiKey: options.arkApiKey || options.ai?.arkApiKey,
+          guardConfig: options.guardConfig || options.ai?.guardConfig,
+        },
       };
     } else {
       // 自动检测模式（CI/CD）
@@ -37,6 +42,14 @@ export class GitLabAIReview {
       token: this.config.gitlab.token,
       host: this.config.gitlab.url,
     });
+
+    // 创建 AI 客户端（如果配置了 API Key）
+    if (this.config.ai?.arkApiKey) {
+      this.aiClient = new AIClient({
+        apiKey: this.config.ai.arkApiKey,
+        ...options.aiConfig,
+      });
+    }
   }
 
   /**
@@ -112,6 +125,71 @@ export class GitLabAIReview {
     );
   }
 
+  /**
+   * 获取 AI 客户端
+   */
+  getAIClient() {
+    if (!this.aiClient) {
+      throw new Error('AI 客户端未初始化，请确保配置了 ARK_API_KEY');
+    }
+    return this.aiClient;
+  }
+
+  /**
+   * AI 审查单个文件的代码变更
+   */
+  async reviewCodeChange(change) {
+    const aiClient = this.getAIClient();
+    const guardConfig = this.config.ai?.guardConfig?.content || '';
+
+    return aiClient.reviewCode({
+      diff: change.diff,
+      fileName: change.new_path || change.old_path,
+      guardConfig,
+    });
+  }
+
+  /**
+   * AI 审查 MR 的所有代码变更
+   */
+  async reviewMergeRequest() {
+    const changes = await this.getMergeRequestChanges();
+    const aiClient = this.getAIClient();
+    const guardConfig = this.config.ai?.guardConfig?.content || '';
+
+    return aiClient.reviewMultipleFiles(changes, guardConfig);
+  }
+
+  /**
+   * AI 审查 MR 并自动添加评论
+   */
+  async reviewAndComment() {
+    const reviews = await this.reviewMergeRequest();
+    
+    // 构建评论内容
+    let comment = '## 🤖 AI 代码审查报告\n\n';
+    
+    for (const review of reviews) {
+      if (review.status === 'error') {
+        comment += `### ❌ ${review.fileName}\n`;
+        comment += `审查失败: ${review.error}\n\n`;
+        continue;
+      }
+
+      comment += `### 📄 ${review.fileName}\n\n`;
+      
+      if (review.reasoning) {
+        comment += `**思考过程**:\n${review.reasoning}\n\n`;
+      }
+      
+      comment += `**审查意见**:\n${review.content}\n\n`;
+      comment += `---\n\n`;
+    }
+
+    // 添加评论到 MR
+    return this.addComment(comment);
+  }
+
   /**
    * 测试方法
    */
@@ -123,6 +201,7 @@ export class GitLabAIReview {
 // 导出工具函数
 export { getConfig, validateConfig } from './lib/config.js';
 export { GitLabClient } from './lib/gitlab-client.js';
+export { AIClient } from './lib/ai-client.js';
 
 // 默认导出
 export default GitLabAIReview;

package/lib/ai-client.js

@@ -0,0 +1,207 @@
+/**
+ * AI 客户端 - 基于 ARK API (豆包大模型)
+ */
+
+import OpenAI from 'openai';
+
+/**
+ * AI 客户端类
+ */
+export class AIClient {
+  constructor(config = {}) {
+    const apiKey = config.apiKey || process.env.ARK_API_KEY;
+    const baseURL = config.baseURL || 'https://ark.cn-beijing.volces.com/api/v3';
+    const model = config.model || 'doubao-seed-1-6-251015';
+
+    if (!apiKey) {
+      throw new Error('ARK_API_KEY 未配置');
+    }
+
+    this.openai = new OpenAI({
+      apiKey,
+      baseURL,
+    });
+
+    this.model = model;
+    this.config = config;
+  }
+
+  /**
+   * 代码审查 - 非流式响应
+   * @param {Object} params - 审查参数
+   * @param {string} params.diff - 代码差异
+   * @param {string} params.fileName - 文件名
+   * @param {string} params.guardConfig - AI Review Guard 配置内容
+   * @returns {Promise<string>} 审查结果
+   */
+  async reviewCode({ diff, fileName, guardConfig = '' }) {
+    const systemPrompt = this._buildSystemPrompt(guardConfig);
+    const userPrompt = this._buildReviewPrompt(diff, fileName);
+
+    const completion = await this.openai.chat.completions.create({
+      messages: [
+        { role: 'system', content: systemPrompt },
+        { role: 'user', content: userPrompt },
+      ],
+      model: this.model,
+      reasoning_effort: this.config.reasoningEffort || 'medium',
+    });
+
+    return {
+      reasoning: completion.choices[0]?.message?.reasoning_content || '',
+      content: completion.choices[0]?.message?.content || '',
+      usage: completion.usage,
+    };
+  }
+
+  /**
+   * 代码审查 - 流式响应
+   * @param {Object} params - 审查参数
+   * @param {string} params.diff - 代码差异
+   * @param {string} params.fileName - 文件名
+   * @param {string} params.guardConfig - AI Review Guard 配置内容
+   * @param {Function} onChunk - 流式回调函数
+   * @returns {Promise<string>} 完整审查结果
+   */
+  async reviewCodeStream({ diff, fileName, guardConfig = '' }, onChunk) {
+    const systemPrompt = this._buildSystemPrompt(guardConfig);
+    const userPrompt = this._buildReviewPrompt(diff, fileName);
+
+    const stream = await this.openai.chat.completions.create({
+      messages: [
+        { role: 'system', content: systemPrompt },
+        { role: 'user', content: userPrompt },
+      ],
+      model: this.model,
+      reasoning_effort: this.config.reasoningEffort || 'medium',
+      stream: true,
+    });
+
+    let fullReasoning = '';
+    let fullContent = '';
+
+    for await (const part of stream) {
+      const reasoning = part.choices[0]?.delta?.reasoning_content || '';
+      const content = part.choices[0]?.delta?.content || '';
+
+      fullReasoning += reasoning;
+      fullContent += content;
+
+      if (onChunk) {
+        onChunk({ reasoning, content });
+      }
+    }
+
+    return {
+      reasoning: fullReasoning,
+      content: fullContent,
+    };
+  }
+
+  /**
+   * 通用聊天接口
+   * @param {Array} messages - 消息数组
+   * @param {Object} options - 可选参数
+   * @returns {Promise<Object>} AI 响应
+   */
+  async chat(messages, options = {}) {
+    const completion = await this.openai.chat.completions.create({
+      messages,
+      model: options.model || this.model,
+      reasoning_effort: options.reasoningEffort || 'medium',
+      stream: options.stream || false,
+      ...options,
+    });
+
+    if (options.stream) {
+      return completion;
+    }
+
+    return {
+      reasoning: completion.choices[0]?.message?.reasoning_content || '',
+      content: completion.choices[0]?.message?.content || '',
+      usage: completion.usage,
+    };
+  }
+
+  /**
+   * 构建系统提示词
+   * @private
+   */
+  _buildSystemPrompt(guardConfig) {
+    let prompt = `你是一个专业的代码审查助手，负责审查 GitLab Merge Request 的代码变更。
+
+你的职责是：
+1. 识别代码中的潜在问题（安全漏洞、性能问题、逻辑错误等）
+2. 提供具体的改进建议
+3. 指出不符合最佳实践的代码
+4. 评估代码的可维护性和可读性
+
+请以专业、建设性的语气提供审查意见。`;
+
+    if (guardConfig) {
+      prompt += `\n\n项目特定的审查规则：\n${guardConfig}`;
+    }
+
+    return prompt;
+  }
+
+  /**
+   * 构建代码审查提示词
+   * @private
+   */
+  _buildReviewPrompt(diff, fileName) {
+    return `请审查以下代码变更：
+
+**文件名**: ${fileName}
+
+**代码差异**:
+\`\`\`diff
+${diff}
+\`\`\`
+
+请提供：
+1. 主要问题（如果有）
+2. 改进建议
+3. 优点（如果有）
+
+如果代码没有问题，请简要说明代码质量良好。`;
+  }
+
+  /**
+   * 批量审查多个文件的变更
+   * @param {Array} changes - 变更数组
+   * @param {string} guardConfig - AI Review Guard 配置
+   * @returns {Promise<Array>} 审查结果数组
+   */
+  async reviewMultipleFiles(changes, guardConfig = '') {
+    const reviews = [];
+
+    for (const change of changes) {
+      try {
+        const result = await this.reviewCode({
+          diff: change.diff,
+          fileName: change.new_path || change.old_path,
+          guardConfig,
+        });
+
+        reviews.push({
+          fileName: change.new_path || change.old_path,
+          status: 'success',
+          ...result,
+        });
+      } catch (error) {
+        reviews.push({
+          fileName: change.new_path || change.old_path,
+          status: 'error',
+          error: error.message,
+        });
+      }
+    }
+
+    return reviews;
+  }
+}
+
+export default AIClient;
+

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gitlab-ai-review",
-  "version": "1.0.3",
+  "version": "1.1.0",
   "description": "GitLab AI Review SDK - 支持 CI/CD 自动配置和 ARK API",
   "main": "index.js",
   "type": "module",
@@ -25,5 +25,8 @@
   "license": "MIT",
   "engines": {
     "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "openai": "^4.73.0"
   }
 }