korekt-cli 0.11.0 → 0.12.0

package/package.json

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

package/src/formatter.js

@@ -76,7 +76,7 @@ function toAbsolutePath(filePath) {
  * @param {Object} data - The API response data
  */
 export function formatReviewOutput(data) {
-  const { review, summary, change_classification: changeClassification } = data.data;
+  const { review, summary, change_classification: changeClassification, model } = data.data;
 
   console.log(chalk.bold.blue('🤖 Automated Code Review Results\n'));
 
@@ -162,4 +162,9 @@ export function formatReviewOutput(data) {
       console.log(); // Add a blank line for spacing
     });
   }
+
+  // Footer with model info
+  if (model) {
+    console.log(chalk.gray(`Analyzed with ${model}`));
+  }
 }

package/src/git-logic.js

@@ -215,6 +215,14 @@ export async function runUncommittedReview(mode = 'unstaged', fileRulesConfig =
       }
     }
 
+    // Check if file count exceeds max_files_for_content limit
+    if (config.max_files_for_content && fileList.length > config.max_files_for_content) {
+      throw new Error(
+        `Too many files: ${fileList.length} files exceeds the limit of ${config.max_files_for_content}. ` +
+          `Please reduce the number of changed files before reviewing.`
+      );
+    }
+
     // Check if this is a large change set (only if config has large_pr_threshold)
     const isLargePr = config.large_pr_threshold && fileList.length > config.large_pr_threshold;
     if (isLargePr) {
@@ -548,6 +556,14 @@ export async function runLocalReview(
       }
     }
 
+    // Check if file count exceeds max_files_for_content limit
+    if (config.max_files_for_content && filteredFileList.length > config.max_files_for_content) {
+      throw new Error(
+        `Too many files: ${filteredFileList.length} files exceeds the limit of ${config.max_files_for_content}. ` +
+          `Please split this PR into smaller reviews or use --ignore to exclude some files.`
+      );
+    }
+
     // Check if this is a large PR (only if config has large_pr_threshold)
     const isLargePr =
       config.large_pr_threshold && filteredFileList.length > config.large_pr_threshold;

package/src/index.js

@@ -22,6 +22,28 @@ export { detectCIProvider, truncateFileData, formatErrorOutput, getPrUrl } from
 const require = createRequire(import.meta.url);
 const { version } = require('../package.json');
 
+/**
+ * Fetch file processing rules from the API.
+ * These rules control which files to skip, which to show diff-only, etc.
+ * @param {string} apiEndpoint - The API endpoint URL
+ * @param {string} apiKey - The API key for authentication
+ * @returns {Promise<Object|null>} - Config object or null if fetch fails
+ */
+async function fetchFileRulesConfig(apiEndpoint, apiKey) {
+  try {
+    const configEndpoint = apiEndpoint.replace(/\/review\/?$/, '/config/file-rules');
+    const response = await axios.get(configEndpoint, {
+      headers: { Authorization: `Bearer ${apiKey}` },
+      timeout: 3000,
+    });
+    return response.data;
+  } catch {
+    // Non-fatal: continue without filtering if config unavailable
+    log(chalk.yellow('Warning: Could not fetch file rules config.'));
+    return null;
+  }
+}
+
 /**
  * Helper functions for clean output separation:
  * - log() writes to stderr (progress, info, errors)
@@ -50,6 +72,57 @@ async function confirmAction(message) {
   });
 }
 
+/**
+ * Available Gemini models for code review
+ */
+const GEMINI_MODELS = [
+  { value: 'gemini-2.5-pro', label: 'gemini-2.5-pro (high quality)' },
+  { value: 'gemini-2.5-flash', label: 'gemini-2.5-flash (avoid, the worst quality)' },
+  { value: 'gemini-3-pro-preview', label: 'gemini-3-pro-preview (experimental - the best model)' },
+  {
+    value: 'gemini-3-flash-preview',
+    label: 'gemini-3-flash-preview (experimental - the most efficient model)',
+  },
+];
+
+/**
+ * Prompt user to select a Gemini model
+ * @returns {Promise<string>} - Selected model name
+ * @throws {Error} - If not running in interactive terminal
+ */
+async function selectModel() {
+  // Check if running in interactive terminal
+  if (!process.stdin.isTTY) {
+    log(chalk.red('Error: --model flag requires a value in non-interactive environments.'));
+    log(chalk.gray('Example: kk review --model=gemini-2.5-pro'));
+    process.exit(1);
+  }
+
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stderr,
+  });
+
+  log(chalk.yellow('\nSelect Gemini model:\n'));
+  GEMINI_MODELS.forEach((model, index) => {
+    log(`  ${index + 1}. ${model.label}`);
+  });
+  log('');
+
+  return new Promise((resolvePromise) => {
+    rl.question(chalk.bold('Enter number (1-4): '), (answer) => {
+      rl.close();
+      const num = parseInt(answer, 10);
+      if (num >= 1 && num <= GEMINI_MODELS.length) {
+        resolvePromise(GEMINI_MODELS[num - 1].value);
+      } else {
+        log(chalk.red('Invalid selection. Please run the command again and enter 1-4.'));
+        process.exit(1);
+      }
+    });
+  });
+}
+
 /**
  * Run the CI integration script to post comments
  * @param {string} provider - CI provider (github, azure, bitbucket)
@@ -106,6 +179,8 @@ program
 Examples:
   $ kk review                      Review committed changes (auto-detect base)
   $ kk review main                 Review changes against main branch
+  $ kk review --model              Select Gemini model interactively
+  $ kk review -m gemini-2.5-pro    Use specific model
   $ kk stg --dry-run               Preview staged changes review
   $ kk diff                        Review unstaged changes
   $ kk review main --json          Output raw JSON (for CI/CD integration)
@@ -115,6 +190,7 @@ Common Options:
   --dry-run                        Show payload without sending to API
   --json                           Output raw API response as JSON
   --comment                        Post review results as PR comments
+  -m, --model [name]               Gemini model (picker if no value)
 
 Configuration:
   $ kk config --key YOUR_KEY
@@ -137,6 +213,7 @@ program
   .option('--json', 'Output raw API response as JSON')
   .option('--comment', 'Post review results as PR comments (auto-detects CI provider)')
   .option('--post-ticket', 'Post review results to linked ticket (e.g., JIRA)')
+  .option('-m, --model [name]', 'Gemini model to use (interactive picker if no value provided)')
   .action(async (targetBranch, options) => {
     const reviewTarget = targetBranch ? `against '${targetBranch}'` : '(auto-detecting fork point)';
 
@@ -157,8 +234,20 @@ program
       process.exit(1);
     }
 
+    // Handle model selection: if --model without value, show picker
+    let selectedModel = null;
+    if (options.model === true) {
+      selectedModel = await selectModel();
+      log(chalk.green(`Using model: ${selectedModel}\n`));
+    } else if (typeof options.model === 'string') {
+      selectedModel = options.model;
+    }
+
+    // Fetch file rules config from API
+    const fileRulesConfig = await fetchFileRulesConfig(apiEndpoint, apiKey);
+
     // Gather all data using our git logic module
-    const payload = await runLocalReview(targetBranch, options.ignore);
+    const payload = await runLocalReview(targetBranch, options.ignore, fileRulesConfig);
 
     if (!payload) {
       log(chalk.red('Could not proceed with review due to errors during analysis.'));
@@ -181,6 +270,11 @@ program
         changed_files: payload.changed_files.map((file) => truncateFileData(file)),
       };
 
+      // Add model if specified
+      if (selectedModel) {
+        displayPayload.model = selectedModel;
+      }
+
       log(JSON.stringify(displayPayload, null, 2));
       log(chalk.gray('\n💡 Run without --dry-run to send to API'));
       log(chalk.gray('💡 Diffs and content are truncated in dry-run for readability'));
@@ -230,6 +324,11 @@ program
       payload.post_to_ticket = true;
     }
 
+    // Add model if specified
+    if (selectedModel) {
+      payload.model = selectedModel;
+    }
+
     try {
       const response = await axios.post(apiEndpoint, payload, {
         headers: {
@@ -303,6 +402,7 @@ program
   .description('Review staged changes (git diff --cached)')
   .option('--dry-run', 'Show payload without sending to API')
   .option('--json', 'Output raw API response as JSON')
+  .option('-m, --model [name]', 'Gemini model to use (interactive picker if no value provided)')
   .action(async (options) => {
     log(chalk.blue.bold('🚀 Reviewing staged changes...'));
     await reviewUncommitted('staged', options);
@@ -314,6 +414,7 @@ program
   .description('Review unstaged changes (git diff)')
   .option('--dry-run', 'Show payload without sending to API')
   .option('--json', 'Output raw API response as JSON')
+  .option('-m, --model [name]', 'Gemini model to use (interactive picker if no value provided)')
   .action(async (options) => {
     log(chalk.blue.bold('🚀 Reviewing unstaged changes...'));
     await reviewUncommitted('unstaged', options);
@@ -334,8 +435,20 @@ async function reviewUncommitted(mode, options) {
     process.exit(1);
   }
 
+  // Handle model selection: if --model without value, show picker
+  let selectedModel = null;
+  if (options.model === true) {
+    selectedModel = await selectModel();
+    log(chalk.green(`Using model: ${selectedModel}\n`));
+  } else if (typeof options.model === 'string') {
+    selectedModel = options.model;
+  }
+
+  // Fetch file rules config from API
+  const fileRulesConfig = await fetchFileRulesConfig(apiEndpoint, apiKey);
+
   const { runUncommittedReview } = await import('./git-logic.js');
-  const payload = await runUncommittedReview(mode);
+  const payload = await runUncommittedReview(mode, fileRulesConfig);
 
   if (!payload) {
     log(chalk.red('No changes found or error occurred during analysis.'));
@@ -350,6 +463,11 @@ async function reviewUncommitted(mode, options) {
       changed_files: payload.changed_files.map((file) => truncateFileData(file)),
     };
 
+    // Add model if specified
+    if (selectedModel) {
+      displayPayload.model = selectedModel;
+    }
+
     log(JSON.stringify(displayPayload, null, 2));
     log(chalk.gray('\n💡 Run without --dry-run to send to API'));
     log(chalk.gray('💡 Diffs and content are truncated in dry-run for readability'));
@@ -382,6 +500,11 @@ async function reviewUncommitted(mode, options) {
     }
   }
 
+  // Add model if specified
+  if (selectedModel) {
+    payload.model = selectedModel;
+  }
+
   const spinner = ora('Submitting review to the AI...').start();
   const startTime = Date.now();
 

package/src/index.test.js

@@ -153,6 +153,36 @@ describe('CLI JSON output mode', () => {
       expect(displayFile.diff).not.toContain('truncated');
       expect(displayFile.content).not.toContain('truncated');
     });
+
+    it('should handle diff-only files without content', () => {
+      const file = {
+        path: 'test.js',
+        status: 'M',
+        diff: 'some diff here',
+        // no content field - this is a diff-only file
+      };
+
+      const displayFile = truncateFileData(file);
+
+      expect(displayFile.path).toBe('test.js');
+      expect(displayFile.status).toBe('M');
+      expect(displayFile.diff).toBe('some diff here');
+      expect(displayFile.content).toBeUndefined();
+    });
+
+    it('should handle files with neither diff nor content', () => {
+      const file = {
+        path: 'deleted.js',
+        status: 'D',
+      };
+
+      const displayFile = truncateFileData(file);
+
+      expect(displayFile.path).toBe('deleted.js');
+      expect(displayFile.status).toBe('D');
+      expect(displayFile.diff).toBeUndefined();
+      expect(displayFile.content).toBeUndefined();
+    });
   });
 
   describe('error formatting for JSON mode', () => {
@@ -439,6 +469,86 @@ describe('--comment flag behavior', () => {
   });
 });
 
+describe('--model flag behavior', () => {
+  describe('model selection logic', () => {
+    it('should treat --model without value as true (picker mode)', () => {
+      const options = { model: true };
+
+      // When --model is used without a value, Commander sets it to true
+      const shouldShowPicker = options.model === true;
+      const hasDirectValue = typeof options.model === 'string';
+
+      expect(shouldShowPicker).toBe(true);
+      expect(hasDirectValue).toBe(false);
+    });
+
+    it('should treat --model=value as string (direct mode)', () => {
+      const options = { model: 'gemini-2.5-flash' };
+
+      const shouldShowPicker = options.model === true;
+      const hasDirectValue = typeof options.model === 'string';
+
+      expect(shouldShowPicker).toBe(false);
+      expect(hasDirectValue).toBe(true);
+      expect(options.model).toBe('gemini-2.5-flash');
+    });
+
+    it('should not include model when flag is not used', () => {
+      const options = {};
+
+      const shouldIncludeModel = options.model !== undefined;
+
+      expect(shouldIncludeModel).toBe(false);
+    });
+  });
+
+  describe('payload model field', () => {
+    it('should add model to payload when specified', () => {
+      const payload = {
+        repo_url: 'https://github.com/user/repo',
+        changed_files: [],
+      };
+
+      const selectedModel = 'gemini-2.5-flash';
+      if (selectedModel) {
+        payload.model = selectedModel;
+      }
+
+      expect(payload.model).toBe('gemini-2.5-flash');
+    });
+
+    it('should not add model to payload when not specified', () => {
+      const payload = {
+        repo_url: 'https://github.com/user/repo',
+        changed_files: [],
+      };
+
+      const selectedModel = null;
+      if (selectedModel) {
+        payload.model = selectedModel;
+      }
+
+      expect(payload.model).toBeUndefined();
+    });
+  });
+
+  describe('available models', () => {
+    it('should have valid model values', () => {
+      const validModels = [
+        'gemini-2.5-pro',
+        'gemini-2.5-flash',
+        'gemini-3-pro-preview',
+        'gemini-3-flash-preview',
+      ];
+
+      // Test that all expected models are valid Gemini model names
+      validModels.forEach((model) => {
+        expect(model).toMatch(/^gemini-/);
+      });
+    });
+  });
+});
+
 describe('getPrUrl', () => {
   const originalEnv = process.env;
 

package/src/utils.js

@@ -73,19 +73,27 @@ export function getPrUrl() {
  * @returns {Object} File object with truncated diff and content
  */
 export function truncateFileData(file, maxLength = 500) {
-  return {
+  const result = {
     path: file.path,
     status: file.status,
     ...(file.old_path && { old_path: file.old_path }),
-    diff:
+  };
+
+  if (file.diff) {
+    result.diff =
       file.diff.length > maxLength
         ? `${file.diff.substring(0, maxLength)}... [truncated ${file.diff.length - maxLength} chars]`
-        : file.diff,
-    content:
+        : file.diff;
+  }
+
+  if (file.content) {
+    result.content =
       file.content.length > maxLength
         ? `${file.content.substring(0, maxLength)}... [truncated ${file.content.length - maxLength} chars]`
-        : file.content,
-  };
+        : file.content;
+  }
+
+  return result;
 }
 
 /**