jstar-reviewer 2.1.1 → 2.1.4

package/README.md

@@ -38,7 +38,7 @@ curl -fsSL https://raw.githubusercontent.com/JStaRFilms/jstar-code-review/v2.0.0
 ### After Install:
 
 1.  **Check Config**: The tool now **auto-creates** `.env.example` and `.jstar/` when you run it.
-2.  **Add Keys**: Copy `.env.example` → `.env.local` and add your `GOOGLE_API_KEY` and `GROQ_API_KEY`.
+2.  **Add Keys**: Copy `.env.example` → `.env.local` and add your `GEMINI_API_KEY` and `GROQ_API_KEY`.
 3.  **Index**: Run `jstar init` (or `pnpm run index:init`) to build the brain.
 4.  **Review**: Stage changes (`git add`) and run `jstar review` (or `pnpm run review`).
 
@@ -91,7 +91,7 @@ pnpm install
 Create `.env.local`:
 
 ```env
-GOOGLE_API_KEY=your_gemini_key
+GEMINI_API_KEY=your_gemini_key
 GROQ_API_KEY=your_groq_key
 ```
 

package/bin/jstar.js

@@ -50,7 +50,7 @@ ${COLORS.bold}EXAMPLES:${COLORS.reset}
   jstar review
 
 ${COLORS.bold}ENVIRONMENT:${COLORS.reset}
-  GOOGLE_API_KEY    Required for Gemini embeddings
+  GEMINI_API_KEY    Required for Gemini embeddings (or GOOGLE_API_KEY)
   GROQ_API_KEY      Required for Groq LLM reviews
 
 ${COLORS.dim}Report issues: https://github.com/JStaRFilms/jstar-code-review${COLORS.reset}
@@ -135,7 +135,7 @@ function runScript(scriptName) {
 }
 
 const REQUIRED_ENV_VARS = {
-    'GOOGLE_API_KEY': '# Required: Google API key for Gemini embeddings\nGOOGLE_API_KEY=your_google_api_key_here',
+    'GEMINI_API_KEY': '# Required: Gemini API key (or GOOGLE_API_KEY)\nGEMINI_API_KEY=your_gemini_api_key_here',
     'GROQ_API_KEY': '# Required: Groq API key for LLM reviews\nGROQ_API_KEY=your_groq_api_key_here',
     'REVIEW_MODEL_NAME': '# Optional: Override the default model\n# REVIEW_MODEL_NAME=moonshotai/kimi-k2-instruct-0905'
 };

package/dist/scripts/config.js

@@ -42,7 +42,7 @@ const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 // --- Auto-Setup Logic ---
 const REQUIRED_ENV_VARS = {
-    'GOOGLE_API_KEY': '# Required: Google API key for Gemini embeddings\nGOOGLE_API_KEY=your_google_api_key_here',
+    'GEMINI_API_KEY': '# Required: Gemini API key (or GOOGLE_API_KEY)\nGEMINI_API_KEY=your_gemini_api_key_here',
     'GROQ_API_KEY': '# Required: Groq API key for LLM reviews\nGROQ_API_KEY=your_groq_api_key_here',
     'REVIEW_MODEL_NAME': '# Optional: Override the default model\n# REVIEW_MODEL_NAME=moonshotai/kimi-k2-instruct-0905'
 };

package/dist/scripts/gemini-embedding.js

@@ -6,9 +6,9 @@ class GeminiEmbedding {
     constructor() {
         // Stubs for BaseEmbedding compliance
         this.embedBatchSize = 10;
-        const apiKey = process.env.GOOGLE_API_KEY;
+        const apiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
         if (!apiKey) {
-            throw new Error("GOOGLE_API_KEY is missing from environment variables.");
+            throw new Error("GEMINI_API_KEY is missing from environment variables.");
         }
         this.genAI = new generative_ai_1.GoogleGenerativeAI(apiKey);
         // User requested 'text-embedding-004', which has better rate limits

package/dist/scripts/indexer.js

@@ -42,19 +42,52 @@ const mock_llm_1 = require("./mock-llm");
 const path = __importStar(require("path"));
 const fs = __importStar(require("fs"));
 const chalk_1 = __importDefault(require("chalk"));
+// IMPORTANT: Import config for side effects (loads dotenv from cwd)
+require("./config");
 // Configuration
 const STORAGE_DIR = path.join(process.cwd(), ".jstar", "storage");
-const SOURCE_DIR = path.join(process.cwd(), "scripts"); // Changed from src/ to scripts/
+// Smart source directory detection
+function getSourceDir() {
+    const cwd = process.cwd();
+    // 1. Check for --path argument
+    const args = process.argv.slice(2);
+    const pathArgIndex = args.indexOf('--path');
+    if (pathArgIndex !== -1 && args[pathArgIndex + 1]) {
+        const customPath = path.resolve(cwd, args[pathArgIndex + 1]);
+        if (fs.existsSync(customPath)) {
+            return customPath;
+        }
+        console.error(chalk_1.default.red(`❌ Custom path not found: ${customPath}`));
+        process.exit(1);
+    }
+    // 2. Try common source directories
+    const candidates = ['src', 'lib', 'app', 'scripts', '.'];
+    for (const dir of candidates) {
+        const fullPath = path.join(cwd, dir);
+        if (fs.existsSync(fullPath) && fs.statSync(fullPath).isDirectory()) {
+            // Skip '.' if there's a more specific match
+            if (dir === '.' && candidates.slice(0, -1).some(d => fs.existsSync(path.join(cwd, d)))) {
+                continue;
+            }
+            return fullPath;
+        }
+    }
+    // Default to cwd
+    return cwd;
+}
 async function main() {
     // 0. Environment Validation
-    if (!process.env.GOOGLE_API_KEY) {
-        console.error(chalk_1.default.red("❌ Missing GOOGLE_API_KEY!"));
+    const geminiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    if (!geminiKey) {
+        console.error(chalk_1.default.red("❌ Missing GEMINI_API_KEY (or GOOGLE_API_KEY)!"));
         console.log(chalk_1.default.yellow("\nPlease ensure you have a .env.local file. Check .env.example for a template.\n"));
         process.exit(1);
     }
     const args = process.argv.slice(2);
     const isWatch = args.includes("--watch");
+    const SOURCE_DIR = getSourceDir();
     console.log(chalk_1.default.blue("🧠 J-Star Indexer: Scanning codebase..."));
+    console.log(chalk_1.default.dim(`   Source: ${SOURCE_DIR}`));
     // 1. Load documents (Your Code)
     if (!fs.existsSync(SOURCE_DIR)) {
         console.error(chalk_1.default.red(`❌ Source directory not found: ${SOURCE_DIR}`));
@@ -114,8 +147,8 @@ async function main() {
     }
     catch (e) {
         console.error(chalk_1.default.red("❌ Indexing Failed:"), e.message);
-        if (e.message.includes("OpenAI")) {
-            console.log(chalk_1.default.yellow("👉 Tip: Make sure you have OPENAI_API_KEY in your .env file (or configure a local embedding model)."));
+        if (e.message.includes("API") || e.message.includes("key")) {
+            console.log(chalk_1.default.yellow("👉 Tip: Make sure you have GEMINI_API_KEY in your .env.local file."));
         }
         process.exit(1);
     }

package/dist/scripts/reviewer.js

@@ -49,7 +49,8 @@ const gemini_embedding_1 = require("./gemini-embedding");
 const mock_llm_1 = require("./mock-llm");
 const dashboard_1 = require("./dashboard");
 const llamaindex_1 = require("llamaindex");
-const google = (0, google_1.createGoogleGenerativeAI)({ apiKey: process.env.GOOGLE_API_KEY });
+const geminiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+const google = (0, google_1.createGoogleGenerativeAI)({ apiKey: geminiKey });
 const groq = (0, groq_1.createGroq)({ apiKey: process.env.GROQ_API_KEY });
 const embedModel = new gemini_embedding_1.GeminiEmbedding();
 const llm = new mock_llm_1.MockLLM();
@@ -126,10 +127,10 @@ function parseReviewResponse(text) {
 async function main() {
     console.log(chalk_1.default.blue("🕵️  J-Star Reviewer: Analyzing your changes...\n"));
     // 0. Environment Validation
-    if (!process.env.GOOGLE_API_KEY || !process.env.GROQ_API_KEY) {
+    if (!geminiKey || !process.env.GROQ_API_KEY) {
         console.error(chalk_1.default.red("❌ Missing API Keys!"));
         console.log(chalk_1.default.yellow("\nPlease ensure you have a .env.local file with:"));
-        console.log(chalk_1.default.white("- GOOGLE_API_KEY"));
+        console.log(chalk_1.default.white("- GEMINI_API_KEY (or GOOGLE_API_KEY)"));
         console.log(chalk_1.default.white("- GROQ_API_KEY"));
         console.log(chalk_1.default.white("\nCheck .env.example for a template.\n"));
         return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jstar-reviewer",
-  "version": "2.1.1",
+  "version": "2.1.4",
   "description": "Local-First, Context-Aware AI Code Reviewer - Works with any language",
   "bin": {
     "jstar": "bin/jstar.js"

package/scripts/config.ts

@@ -5,7 +5,7 @@ import { Severity } from "./types";
 
 // --- Auto-Setup Logic ---
 const REQUIRED_ENV_VARS = {
-    'GOOGLE_API_KEY': '# Required: Google API key for Gemini embeddings\nGOOGLE_API_KEY=your_google_api_key_here',
+    'GEMINI_API_KEY': '# Required: Gemini API key (or GOOGLE_API_KEY)\nGEMINI_API_KEY=your_gemini_api_key_here',
     'GROQ_API_KEY': '# Required: Groq API key for LLM reviews\nGROQ_API_KEY=your_groq_api_key_here',
     'REVIEW_MODEL_NAME': '# Optional: Override the default model\n# REVIEW_MODEL_NAME=moonshotai/kimi-k2-instruct-0905'
 };

package/scripts/gemini-embedding.ts

@@ -5,9 +5,9 @@ export class GeminiEmbedding {
     private model: any;
 
     constructor() {
-        const apiKey = process.env.GOOGLE_API_KEY;
+        const apiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
         if (!apiKey) {
-            throw new Error("GOOGLE_API_KEY is missing from environment variables.");
+            throw new Error("GEMINI_API_KEY is missing from environment variables.");
         }
         this.genAI = new GoogleGenerativeAI(apiKey);
         // User requested 'text-embedding-004', which has better rate limits

package/scripts/indexer.ts

@@ -9,25 +9,60 @@ import { MockLLM } from "./mock-llm";
 import * as path from "path";
 import * as fs from "fs";
 import chalk from "chalk";
-import { Config } from "./config";
+// IMPORTANT: Import config for side effects (loads dotenv from cwd)
+import "./config";
 
 // Configuration
 const STORAGE_DIR = path.join(process.cwd(), ".jstar", "storage");
-const SOURCE_DIR = path.join(process.cwd(), "scripts"); // Changed from src/ to scripts/
+
+// Smart source directory detection
+function getSourceDir(): string {
+    const cwd = process.cwd();
+
+    // 1. Check for --path argument
+    const args = process.argv.slice(2);
+    const pathArgIndex = args.indexOf('--path');
+    if (pathArgIndex !== -1 && args[pathArgIndex + 1]) {
+        const customPath = path.resolve(cwd, args[pathArgIndex + 1]);
+        if (fs.existsSync(customPath)) {
+            return customPath;
+        }
+        console.error(chalk.red(`❌ Custom path not found: ${customPath}`));
+        process.exit(1);
+    }
+
+    // 2. Try common source directories
+    const candidates = ['src', 'lib', 'app', 'scripts', '.'];
+    for (const dir of candidates) {
+        const fullPath = path.join(cwd, dir);
+        if (fs.existsSync(fullPath) && fs.statSync(fullPath).isDirectory()) {
+            // Skip '.' if there's a more specific match
+            if (dir === '.' && candidates.slice(0, -1).some(d => fs.existsSync(path.join(cwd, d)))) {
+                continue;
+            }
+            return fullPath;
+        }
+    }
+
+    // Default to cwd
+    return cwd;
+}
 
 async function main() {
     // 0. Environment Validation
-    if (!process.env.GOOGLE_API_KEY) {
-        console.error(chalk.red("❌ Missing GOOGLE_API_KEY!"));
+    const geminiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    if (!geminiKey) {
+        console.error(chalk.red("❌ Missing GEMINI_API_KEY (or GOOGLE_API_KEY)!"));
         console.log(chalk.yellow("\nPlease ensure you have a .env.local file. Check .env.example for a template.\n"));
         process.exit(1);
     }
 
     const args = process.argv.slice(2);
     const isWatch = args.includes("--watch");
+    const SOURCE_DIR = getSourceDir();
 
     console.log(chalk.blue("🧠 J-Star Indexer: Scanning codebase..."));
-
+    console.log(chalk.dim(`   Source: ${SOURCE_DIR}`));
 
     // 1. Load documents (Your Code)
     if (!fs.existsSync(SOURCE_DIR)) {
@@ -91,8 +126,8 @@ async function main() {
 
     } catch (e: any) {
         console.error(chalk.red("❌ Indexing Failed:"), e.message);
-        if (e.message.includes("OpenAI")) {
-            console.log(chalk.yellow("👉 Tip: Make sure you have OPENAI_API_KEY in your .env file (or configure a local embedding model)."));
+        if (e.message.includes("API") || e.message.includes("key")) {
+            console.log(chalk.yellow("👉 Tip: Make sure you have GEMINI_API_KEY in your .env.local file."));
         }
         process.exit(1);
     }

package/scripts/reviewer.ts

@@ -18,7 +18,8 @@ import {
     serviceContextFromDefaults
 } from "llamaindex";
 
-const google = createGoogleGenerativeAI({ apiKey: process.env.GOOGLE_API_KEY });
+const geminiKey = process.env.GEMINI_API_KEY || process.env.GOOGLE_API_KEY || process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+const google = createGoogleGenerativeAI({ apiKey: geminiKey });
 const groq = createGroq({ apiKey: process.env.GROQ_API_KEY });
 
 const embedModel = new GeminiEmbedding();
@@ -110,10 +111,10 @@ async function main() {
     console.log(chalk.blue("🕵️  J-Star Reviewer: Analyzing your changes...\n"));
 
     // 0. Environment Validation
-    if (!process.env.GOOGLE_API_KEY || !process.env.GROQ_API_KEY) {
+    if (!geminiKey || !process.env.GROQ_API_KEY) {
         console.error(chalk.red("❌ Missing API Keys!"));
         console.log(chalk.yellow("\nPlease ensure you have a .env.local file with:"));
-        console.log(chalk.white("- GOOGLE_API_KEY"));
+        console.log(chalk.white("- GEMINI_API_KEY (or GOOGLE_API_KEY)"));
         console.log(chalk.white("- GROQ_API_KEY"));
         console.log(chalk.white("\nCheck .env.example for a template.\n"));
         return;

package/setup.js

@@ -67,8 +67,8 @@ const SCRIPTS = {
 const ENV_EXAMPLE = `# J-Star Code Reviewer Configuration
 # Copy this to .env.local and fill in your keys
 
-# Required: Google API key for Gemini embeddings
-GOOGLE_API_KEY=your_google_api_key_here
+# Required: Gemini API key (or GOOGLE_API_KEY)
+GEMINI_API_KEY=your_gemini_api_key_here
 
 # Required: Groq API key for LLM reviews
 GROQ_API_KEY=your_groq_api_key_here
@@ -115,7 +115,7 @@ async function main() {
 
     // HARDCODED: Tagged release URL - NOT configurable for security
     // To update, modify this constant and publish a new version of setup.js
-    const BASE_URL = 'https://raw.githubusercontent.com/JStaRFilms/jstar-code-review/v2.1.0';
+    const BASE_URL = 'https://raw.githubusercontent.com/JStaRFilms/jstar-code-review/v2.1.4';
 
     // Validate URL matches our exact expected pattern (defense in depth)
     function isValidUrl(url) {
@@ -272,7 +272,7 @@ async function main() {
     // 6. Update .env.example (intelligently merge, don't override)
     const envExamplePath = path.join(cwd, '.env.example');
     const REQUIRED_ENV_VARS = {
-        'GOOGLE_API_KEY': '# Required: Google API key for Gemini embeddings\nGOOGLE_API_KEY=your_google_api_key_here',
+        'GEMINI_API_KEY': '# Required: Gemini API key (or GOOGLE_API_KEY)\nGEMINI_API_KEY=your_gemini_api_key_here',
         'GROQ_API_KEY': '# Required: Groq API key for LLM reviews\nGROQ_API_KEY=your_groq_api_key_here',
         'REVIEW_MODEL_NAME': '# Optional: Override the default model\n# REVIEW_MODEL_NAME=moonshotai/kimi-k2-instruct-0905'
     };
@@ -355,7 +355,7 @@ async function main() {
     console.log('\n🎉 J-Star Code Reviewer installed!\n');
     console.log('Next steps:');
     console.log('  1. Copy .env.example to .env.local');
-    console.log('  2. Add your GOOGLE_API_KEY and GROQ_API_KEY');
+    console.log('  2. Add your GEMINI_API_KEY and GROQ_API_KEY');
     console.log('  3. Run: pnpm run index:init');
     console.log('  4. Stage changes and run: pnpm run review');
     console.log('\n' + '─'.repeat(50) + '\n');