@krotovm/gitlab-ai-review 1.0.21 → 1.0.22

package/dist/cli/ci-review.js

@@ -1,6 +1,6 @@
 /** @format */
 import OpenAI from "openai";
-import { AI_MAX_OUTPUT_TOKENS, buildAnswer, buildConsolidatePrompt, buildFileReviewPrompt, buildPrompt, buildTriagePrompt, buildVerificationPrompt, extractCompletionText, parseTriageResponse, } from "../prompt/index.js";
+import { buildAnswer, buildConsolidatePrompt, buildFileReviewPrompt, buildPrompt, buildTriagePrompt, buildVerificationPrompt, extractCompletionText, parseTriageResponse, } from "../prompt/index.js";
 import { fetchFileAtRef, searchRepository, } from "../gitlab/services.js";
 import { logToolUsageMinimal, MAX_FILE_TOOL_ROUNDS, MAX_TOOL_ROUNDS, TOOL_NAME_GET_FILE, TOOL_NAME_GREP, } from "./tooling.js";
 function buildReviewMetadata(changes, refs) {
@@ -219,7 +219,6 @@ export async function reviewMergeRequestWithTools(params) {
     const finalCompletion = await openaiInstance.chat.completions.create({
         model: aiModel,
         temperature: 0.2,
-        max_tokens: AI_MAX_OUTPUT_TOKENS,
         stream: false,
         messages,
     });
@@ -330,7 +329,6 @@ async function runFileReviewWithTools(params) {
     const final = await openaiInstance.chat.completions.create({
         model: aiModel,
         temperature: 0.2,
-        max_tokens: AI_MAX_OUTPUT_TOKENS,
         stream: false,
         messages,
     });
@@ -421,7 +419,6 @@ export async function reviewMergeRequestMultiPass(params) {
         const consolidateCompletion = await openaiInstance.chat.completions.create({
             model: aiModel,
             temperature: 0.1,
-            max_tokens: AI_MAX_OUTPUT_TOKENS,
             stream: false,
             messages: consolidateMessages,
         });
@@ -440,7 +437,6 @@ export async function reviewMergeRequestMultiPass(params) {
             const verificationCompletion = await openaiInstance.chat.completions.create({
                 model: aiModel,
                 temperature: 0.0,
-                max_tokens: AI_MAX_OUTPUT_TOKENS,
                 stream: false,
                 messages: verificationMessages,
             });

package/dist/gitlab/services.js

@@ -1,6 +1,6 @@
 /** @format */
 import OpenAI from "openai";
-import { AI_MAX_OUTPUT_TOKENS, AI_MODEL_TEMPERATURE } from "../prompt/index.js";
+import { AI_MODEL_TEMPERATURE } from "../prompt/index.js";
 import { GitLabError, OpenAIError, } from "./types.js";
 export const fetchPreEditFiles = async ({ gitLabBaseUrl, headers, changesOldPaths, ref }) => {
     const oldFilesRequestUrls = changesOldPaths.map((filePath) => {
@@ -87,7 +87,6 @@ export async function generateAICompletion(messages, openaiInstance, aiModel) {
         completion = await openaiInstance.chat.completions.create({
             model: aiModel,
             temperature: AI_MODEL_TEMPERATURE,
-            max_tokens: AI_MAX_OUTPUT_TOKENS,
             stream: false,
             messages,
         });
@@ -164,10 +163,18 @@ export const searchRepository = async ({ gitLabBaseUrl, headers, query, ref, pro
     if (res instanceof Error || !res.ok) {
         const responseDetails = await (async () => {
             if (res instanceof Error) {
-                return { url: url.toString(), error: { name: res.name, message: res.message } };
+                return {
+                    url: url.toString(),
+                    error: { name: res.name, message: res.message },
+                };
             }
             const bodyText = await res.text().catch(() => "");
-            return { url: url.toString(), status: res.status, statusText: res.statusText, body: bodyText.slice(0, 1000) };
+            return {
+                url: url.toString(),
+                status: res.status,
+                statusText: res.statusText,
+                body: bodyText.slice(0, 1000),
+            };
         })();
         return new GitLabError({
             name: "SEARCH_FAILED",

package/dist/prompt/index.js

@@ -1,6 +1,6 @@
 /** @format */
 import { buildMainSystemMessages, FILE_REVIEW_SYSTEM, TRIAGE_SYSTEM, } from "./messages.js";
-import { sanitizeGitLabMarkdown, truncateWithMarker } from "./utils.js";
+import { normalizeReviewFindingsMarkdown, sanitizeGitLabMarkdown, truncateWithMarker, } from "./utils.js";
 export const DEFAULT_PROMPT_LIMITS = {
     maxDiffs: 50,
     maxDiffChars: 16000,
@@ -273,6 +273,7 @@ export const buildAnswer = (completion) => {
     if (content === "") {
         return `${ERROR_ANSWER}\n\nError: Model returned an empty response body. Try another model (for example, gpt-4o-mini) or a different provider endpoint.\n\n---\n_${DISCLAIMER}_`;
     }
-    const safe = sanitizeGitLabMarkdown(content);
+    const normalizedFindings = normalizeReviewFindingsMarkdown(content);
+    const safe = sanitizeGitLabMarkdown(normalizedFindings);
     return `${safe}\n\n---\n_${DISCLAIMER}_`;
 };

package/dist/prompt/utils.js

@@ -12,3 +12,65 @@ export function sanitizeGitLabMarkdown(input) {
     const withClosedFence = fenceCount % 2 === 1 ? `${normalized}\n\`\`\`` : normalized;
     return withClosedFence;
 }
+const NO_ISSUES_SENTENCE = "No confirmed bugs or high-value optimizations found.";
+export function normalizeReviewFindingsMarkdown(input) {
+    const normalized = input.replace(/\r\n/g, "\n").trim();
+    if (normalized === "" || normalized === NO_ISSUES_SENTENCE)
+        return normalized;
+    const lines = normalized.split("\n");
+    const findings = [];
+    const headerRe = /^\s*-?\s*\[(high|medium)\]\s+(.+?)\s*$/i;
+    const fileRe = /^\s*[-*]?\s*File:\s*(.+?)\s*$/i;
+    const lineRe = /^\s*[-*]?\s*Line:\s*(.+?)\s*$/i;
+    const whyRe = /^\s*[-*]?\s*Why:\s*(.+?)\s*$/i;
+    for (let i = 0; i < lines.length; i += 1) {
+        const headerMatch = lines[i].match(headerRe);
+        if (headerMatch == null)
+            continue;
+        const severity = headerMatch[1].toLowerCase();
+        const title = headerMatch[2].trim();
+        let file = null;
+        let line = null;
+        let why = null;
+        let j = i + 1;
+        while (j < lines.length) {
+            const nextHeader = lines[j].match(headerRe);
+            if (nextHeader != null)
+                break;
+            if (file == null) {
+                const m = lines[j].match(fileRe);
+                if (m != null) {
+                    file = m[1].trim();
+                    j += 1;
+                    continue;
+                }
+            }
+            if (line == null) {
+                const m = lines[j].match(lineRe);
+                if (m != null) {
+                    line = m[1].trim();
+                    j += 1;
+                    continue;
+                }
+            }
+            if (why == null) {
+                const m = lines[j].match(whyRe);
+                if (m != null) {
+                    why = m[1].trim();
+                    j += 1;
+                    continue;
+                }
+            }
+            j += 1;
+        }
+        if (file != null && line != null && why != null) {
+            findings.push({ severity, title, file, line, why });
+            i = j - 1;
+        }
+    }
+    if (findings.length === 0)
+        return normalized;
+    return findings
+        .map((f) => `- [${f.severity}] ${f.title}\n  File: ${f.file}\n  Line: ${f.line}\n  Why: ${f.why}`)
+        .join("\n\n");
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@krotovm/gitlab-ai-review",
-  "version": "1.0.21",
+  "version": "1.0.22",
   "description": "CLI tool to generate AI code reviews for GitLab merge requests.",
   "main": "dist/cli.js",
   "bin": {