korekt-cli 0.13.2 → 0.13.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "korekt-cli",
-  "version": "0.13.2",
+  "version": "0.13.4",
   "description": "AI-powered code review CLI - Keep your kode korekt",
   "main": "src/index.js",
   "bin": {

package/src/git-logic.js

@@ -18,7 +18,9 @@ export function truncateContent(content, maxLines = 2000) {
   const halfMax = Math.floor(maxLines / 2);
   const head = lines.slice(0, halfMax).join('\n');
   const tail = lines.slice(-halfMax).join('\n');
-  return `${head}\n\n... [truncated] ...\n\n${tail}`;
+  const linesKept = 2 * halfMax;
+  const omittedLines = lines.length - linesKept;
+  return `${head}\n\n... [${omittedLines} lines truncated] ...\n\n${tail}`;
 }
 
 /**

package/src/git-logic.test.js

@@ -450,7 +450,7 @@ describe('truncateContent', () => {
     // Should contain head and tail
     expect(result).toContain('line 0');
     expect(result).toContain('line 999'); // Last line of head (first 1000 lines)
-    expect(result).toContain('... [truncated] ...');
+    expect(result).toContain('... [1000 lines truncated] ...');
     expect(result).toContain('line 2000'); // First line of tail (last 1000 lines)
     expect(result).toContain('line 2999');
 
@@ -464,7 +464,7 @@ describe('truncateContent', () => {
 
     expect(result).toContain('line 0');
     expect(result).toContain('line 49'); // Last of first 50
-    expect(result).toContain('... [truncated] ...');
+    expect(result).toContain('... [100 lines truncated] ...');
     expect(result).toContain('line 150'); // First of last 50
     expect(result).toContain('line 199');
   });

package/src/index.js

@@ -19,6 +19,9 @@ import { detectCIProvider, truncateFileData, formatErrorOutput } from './utils.j
 // Re-export utilities for backward compatibility
 export { detectCIProvider, truncateFileData, formatErrorOutput, getPrUrl } from './utils.js';
 
+// Export for testing
+export { GEMINI_MODELS, resolveModel };
+
 const require = createRequire(import.meta.url);
 const { version } = require('../package.json');
 
@@ -103,6 +106,19 @@ const GEMINI_MODELS = [
   },
 ];
 
+/**
+ * Resolve model input - supports numeric shortcuts (1-4) or full model names
+ * @param {string} input - Model input (number or name)
+ * @returns {string} - Resolved model name
+ */
+function resolveModel(input) {
+  const num = parseInt(input, 10);
+  if (!isNaN(num) && num >= 1 && num <= GEMINI_MODELS.length) {
+    return GEMINI_MODELS[num - 1].value;
+  }
+  return input;
+}
+
 /**
  * Prompt user to select a Gemini model
  * @returns {Promise<string>} - Selected model name
@@ -258,7 +274,8 @@ program
       selectedModel = await selectModel();
       log(chalk.green(`Using model: ${selectedModel}\n`));
     } else if (typeof options.model === 'string') {
-      selectedModel = options.model;
+      selectedModel = resolveModel(options.model);
+      log(chalk.green(`Using model: ${selectedModel}\n`));
     }
 
     // Fetch file rules config from API
@@ -465,7 +482,8 @@ async function reviewUncommitted(mode, options) {
     selectedModel = await selectModel();
     log(chalk.green(`Using model: ${selectedModel}\n`));
   } else if (typeof options.model === 'string') {
-    selectedModel = options.model;
+    selectedModel = resolveModel(options.model);
+    log(chalk.green(`Using model: ${selectedModel}\n`));
   }
 
   // Fetch file rules config from API

package/src/index.test.js

@@ -5,6 +5,7 @@ import {
   detectCIProvider,
   getPrUrl,
   handleSkippedResponse,
+  resolveModel,
 } from './index.js';
 
 describe('CLI JSON output mode', () => {
@@ -553,6 +554,34 @@ describe('--model flag behavior', () => {
       });
     });
   });
+
+  describe('numeric model shortcuts', () => {
+    it('should resolve "1" to gemini-2.5-pro', () => {
+      expect(resolveModel('1')).toBe('gemini-2.5-pro');
+    });
+
+    it('should resolve "2" to gemini-2.5-flash', () => {
+      expect(resolveModel('2')).toBe('gemini-2.5-flash');
+    });
+
+    it('should resolve "3" to gemini-3-pro-preview', () => {
+      expect(resolveModel('3')).toBe('gemini-3-pro-preview');
+    });
+
+    it('should resolve "4" to gemini-3-flash-preview', () => {
+      expect(resolveModel('4')).toBe('gemini-3-flash-preview');
+    });
+
+    it('should pass through full model names unchanged', () => {
+      expect(resolveModel('gemini-3-pro-preview')).toBe('gemini-3-pro-preview');
+    });
+
+    it('should pass through invalid input unchanged', () => {
+      expect(resolveModel('invalid')).toBe('invalid');
+      expect(resolveModel('0')).toBe('0');
+      expect(resolveModel('5')).toBe('5');
+    });
+  });
 });
 
 describe('skipped response handling', () => {