jbai-cli 1.2.4 → 1.3.0

package/README.md

@@ -2,7 +2,7 @@
 
 **Use AI coding tools with your JetBrains AI subscription** — no separate API keys needed.
 
-One token, all tools: Claude Code, Codex, Aider, OpenCode.
+One token, all tools: Claude Code, Codex, Aider, Gemini CLI, OpenCode.
 
 ## Install
 
@@ -89,6 +89,7 @@ jbai-aider --super
 | Claude Code | `--dangerously-skip-permissions` |
 | Codex | `--full-auto` |
 | Aider | `--yes` |
+| Gemini CLI | `--yes` |
 | OpenCode | `--yes` |
 
 ⚠️ **Use with caution** - super mode allows the AI to make changes without confirmation.

package/bin/jbai-claude.js

@@ -33,11 +33,14 @@ if (superMode) {
 }
 
 // Set environment for Claude Code
-// Only set API_KEY (not AUTH_TOKEN) to avoid conflicts
+// Use ANTHROPIC_CUSTOM_HEADERS for the Grazie auth header
+// Remove /v1 from endpoint - Claude Code adds it automatically
+const baseUrl = endpoints.anthropic.replace(/\/v1$/, '');
 const env = {
   ...process.env,
-  ANTHROPIC_BASE_URL: endpoints.anthropic,
-  ANTHROPIC_API_KEY: token
+  ANTHROPIC_BASE_URL: baseUrl,
+  ANTHROPIC_API_KEY: 'placeholder',
+  ANTHROPIC_CUSTOM_HEADERS: `Grazie-Authenticate-JWT: ${token}`
 };
 
 // Remove any existing auth token that might conflict

package/bin/jbai-gemini.js

@@ -0,0 +1,66 @@
+#!/usr/bin/env node
+
+/**
+ * jbai-gemini - Gemini CLI wrapper for JetBrains AI Platform
+ *
+ * Uses GEMINI_CLI_CUSTOM_HEADERS and GEMINI_BASE_URL for authentication
+ */
+
+const { spawn } = require('child_process');
+const config = require('../lib/config');
+
+const token = config.getToken();
+if (!token) {
+  console.error('❌ No token found. Run: jbai token set');
+  process.exit(1);
+}
+
+if (config.isTokenExpired(token)) {
+  console.error('⚠️  Token expired. Run: jbai token refresh');
+  process.exit(1);
+}
+
+const endpoints = config.getEndpoints();
+let args = process.argv.slice(2);
+
+// Check for super mode (--super, --yolo, -s)
+const superFlags = ['--super', '--yolo', '-s'];
+const superMode = args.some(a => superFlags.includes(a));
+args = args.filter(a => !superFlags.includes(a));
+
+// Check if model specified
+const hasModel = args.includes('--model') || args.includes('-m');
+let finalArgs = hasModel ? args : ['--model', config.MODELS.gemini.default, ...args];
+
+// Add super mode flags (auto-confirm)
+if (superMode) {
+  finalArgs = ['--yes', ...finalArgs];
+  console.log('🚀 Super mode: --yes (auto-confirm) enabled');
+}
+
+// Set environment for Gemini CLI
+// Uses GEMINI_CLI_CUSTOM_HEADERS for auth (supported since Nov 2025)
+const env = {
+  ...process.env,
+  GEMINI_BASE_URL: endpoints.google,
+  GEMINI_API_KEY: 'placeholder',
+  GEMINI_CLI_CUSTOM_HEADERS: `Grazie-Authenticate-JWT: ${token}`
+};
+
+const child = spawn('gemini', finalArgs, {
+  stdio: 'inherit',
+  env
+});
+
+child.on('error', (err) => {
+  if (err.code === 'ENOENT') {
+    console.error(`❌ Gemini CLI not found.\n`);
+    console.error(`Install with: npm install -g @google/gemini-cli`);
+    console.error(`Or run: jbai install gemini`);
+  } else {
+    console.error(`Error: ${err.message}`);
+  }
+  process.exit(1);
+});
+
+child.on('exit', (code) => process.exit(code || 0));

package/bin/jbai-opencode.js

@@ -55,11 +55,13 @@ if (!opencodeConfig.provider) {
 const envVarName = environment === 'staging' ? 'GRAZIE_STAGING_TOKEN' : 'GRAZIE_API_TOKEN';
 
 // Add/update JetBrains provider with custom header (using env var reference)
+// Use openai-compatible SDK which properly sends custom headers
+// Point to OpenAI endpoint for compatibility
 opencodeConfig.provider[providerName] = {
-  npm: '@ai-sdk/anthropic',
+  npm: '@ai-sdk/openai-compatible',
   name: `JetBrains AI (${environment})`,
   options: {
-    baseURL: endpoints.anthropic,
+    baseURL: endpoints.openai,
     apiKey: `{env:${envVarName}}`,
     headers: {
       'Grazie-Authenticate-JWT': `{env:${envVarName}}`
@@ -68,10 +70,15 @@ opencodeConfig.provider[providerName] = {
   models: {}
 };
 
-// Add Claude models
-config.MODELS.claude.available.forEach(model => {
+// Add OpenAI models (since we use openai-compatible SDK)
+// Set context and output limits to avoid "max_tokens too large" errors
+config.MODELS.openai.available.forEach(model => {
   opencodeConfig.provider[providerName].models[model] = {
-    name: model
+    name: model,
+    limit: {
+      context: 128000,
+      output: 8192
+    }
   };
 });
 
@@ -84,7 +91,7 @@ let finalArgs = [];
 
 if (!hasModel) {
   // Use provider/model format for OpenCode
-  finalArgs.push('--model', `${providerName}/${config.MODELS.claude.default}`);
+  finalArgs.push('--model', `${providerName}/${config.MODELS.openai.default}`);
 }
 
 // Add super mode flags

package/bin/jbai.js

@@ -24,10 +24,16 @@ const TOOLS = {
     install: 'pipx install aider-chat',
     check: 'aider --version'
   },
+  gemini: {
+    name: 'Gemini CLI',
+    command: 'gemini',
+    install: 'npm install -g @google/gemini-cli',
+    check: 'gemini --version'
+  },
   opencode: {
     name: 'OpenCode',
     command: 'opencode',
-    install: 'go install github.com/opencode-ai/opencode@latest',
+    install: 'npm install -g opencode-ai',
     check: 'opencode --version'
   }
 };
@@ -53,6 +59,7 @@ TOOL WRAPPERS:
   jbai-claude   Launch Claude Code with JetBrains AI
   jbai-codex    Launch Codex CLI with JetBrains AI
   jbai-aider    Launch Aider with JetBrains AI
+  jbai-gemini   Launch Gemini CLI with JetBrains AI
   jbai-opencode Launch OpenCode with JetBrains AI
 
 SUPER MODE:
@@ -277,8 +284,6 @@ function doctor() {
       console.log(`   Install: ${tool.install}`);
     }
   }
-
-  console.log('\nTip: For Gemini models, use Aider: jbai-aider --model gemini/gemini-2.5-pro');
 }
 
 async function installTools(toolKey) {

package/lib/config.js

@@ -138,10 +138,15 @@ const TOOLS = {
     command: 'aider',
     install: 'pipx install aider-chat'
   },
+  gemini: {
+    name: 'Gemini CLI',
+    command: 'gemini',
+    install: 'npm install -g @google/gemini-cli'
+  },
   opencode: {
     name: 'OpenCode',
     command: 'opencode',
-    install: 'go install github.com/opencode-ai/opencode@latest'
+    install: 'npm install -g opencode-ai'
   }
 };
 

package/package.json

@@ -1,17 +1,19 @@
 {
   "name": "jbai-cli",
-  "version": "1.2.4",
-  "description": "CLI wrappers to use AI coding tools (Claude Code, Codex, Aider, OpenCode) with JetBrains AI Platform",
+  "version": "1.3.0",
+  "description": "CLI wrappers to use AI coding tools (Claude Code, Codex, Aider, Gemini CLI, OpenCode) with JetBrains AI Platform",
   "keywords": [
     "jetbrains",
     "ai",
     "claude",
     "codex",
     "aider",
+    "gemini",
     "opencode",
     "cli",
     "openai",
-    "anthropic"
+    "anthropic",
+    "google"
   ],
   "author": "Andrii Shchupliak",
   "license": "MIT",
@@ -28,6 +30,7 @@
     "jbai-claude": "./bin/jbai-claude.js",
     "jbai-codex": "./bin/jbai-codex.js",
     "jbai-aider": "./bin/jbai-aider.js",
+    "jbai-gemini": "./bin/jbai-gemini.js",
     "jbai-opencode": "./bin/jbai-opencode.js"
   },
   "files": [