jbai-cli 1.9.6 → 2.1.0

package/bin/jbai-claude-opus.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'claude',
+  model: 'claude-opus-4-6',
+  label: 'Claude Code + Opus 4.6',
+});

package/bin/jbai-claude-sonnet.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'claude',
+  model: 'claude-sonnet-4-6',
+  label: 'Claude Code + Sonnet 4.6',
+});

package/bin/jbai-codex-5.2.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'codex',
+  model: 'gpt-5.2-codex',
+  label: 'Codex + GPT-5.2',
+});

package/bin/jbai-codex-5.3.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'codex',
+  model: 'gpt-5.3-codex',
+  label: 'Codex + GPT-5.3',
+});

package/bin/jbai-codex-rockhopper.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'codex',
+  model: 'rockhopper-alpha',
+  label: 'Codex + Rockhopper Alpha (OpenAI EAP)',
+});

package/bin/jbai-gemini-3.1.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'gemini',
+  model: 'gemini-3.1-pro-preview',
+  label: 'Gemini + 3.1 Pro Preview',
+});

package/bin/jbai-gemini-supernova.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'gemini',
+  model: 'supernova',
+  label: 'Gemini + Supernova (Google EAP)',
+});

package/bin/jbai-opencode-deepseek.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'opencode',
+  model: 'deepseek-r1',
+  label: 'OpenCode + DeepSeek R1',
+});

package/bin/jbai-opencode-grok.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'opencode',
+  model: 'xai-grok-4',
+  label: 'OpenCode + Grok 4 (xAI)',
+});

package/bin/jbai-opencode-rockhopper.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+require('../lib/shortcut').run({
+  tool: 'opencode',
+  model: 'rockhopper-alpha',
+  label: 'OpenCode + Rockhopper Alpha (OpenAI EAP)',
+});

package/bin/jbai.js

@@ -9,6 +9,7 @@ const os = require('os');
 const config = require('../lib/config');
 const { createHandoff } = require('../lib/handoff');
 const { getGroupsForTool, showModelsForTool } = require('../lib/model-list');
+const completions = require('../lib/completions');
 
 const TOOLS = {
   claude: {
@@ -66,6 +67,9 @@ COMMANDS:
   jbai install         Install all AI tools (claude, codex, gemini, opencode, goose, continue)
   jbai install claude  Install specific tool
   jbai doctor          Check which tools are installed
+  jbai completions     Print zsh completions to stdout
+  jbai completions --install   Add completions to ~/.zshrc
+  jbai completions --bash      Print bash completions
   jbai help            Show this help
 
 PROXY (for Codex Desktop, Cursor, etc.):
@@ -81,17 +85,31 @@ TOOL WRAPPERS:
   jbai-opencode Launch OpenCode with JetBrains AI
   jbai-council  Launch Claude + Codex + OpenCode in tmux council mode
 
+MODEL SHORTCUTS (super mode by default):
+  jbai-claude-opus              Claude Code + Opus 4.6
+  jbai-claude-sonnet            Claude Code + Sonnet 4.6
+  jbai-codex-5.3                Codex + GPT-5.3
+  jbai-codex-5.2                Codex + GPT-5.2
+  jbai-codex-rockhopper         Codex + Rockhopper Alpha (OpenAI EAP)
+  jbai-gemini-supernova         Gemini + Supernova (Google EAP)
+  jbai-gemini-3.1               Gemini + 3.1 Pro Preview
+  jbai-opencode-rockhopper      OpenCode + Rockhopper Alpha
+  jbai-opencode-grok            OpenCode + Grok 4 (xAI)
+  jbai-opencode-deepseek        OpenCode + DeepSeek R1
+
 SUPER MODE:
   Add --super (or --yolo or -s) to skip confirmations:
   jbai-claude --super               # Skip permission prompts
   jbai-codex --super                # Full auto mode
   jbai-gemini --super               # Auto-confirm changes
+  (All model shortcuts above run in super mode by default)
 
 EXAMPLES:
   jbai token set                    # Set your token
   jbai-claude                       # Start Claude Code
   jbai-codex exec "explain code"    # Run Codex task
-  jbai-gemini                       # Start Gemini CLI
+  jbai-codex-rockhopper             # Codex with Rockhopper (super mode)
+  jbai-gemini-supernova             # Gemini with Supernova (super mode)
   jbai-council                      # Launch all 3 agents in tmux
   jbai-council --super              # All agents in super mode
   jbai handoff --task "fix lint"    # Handoff task to Orca Lab
@@ -688,6 +706,9 @@ switch (command) {
     proxyMod.main();
     break;
   }
+  case 'completions':
+    completions.run(args);
+    break;
   case 'help':
   case '--help':
   case '-h':

package/lib/completions.js

@@ -0,0 +1,258 @@
+/**
+ * Shell completions for jbai-cli.
+ *
+ * `jbai completions`           — print zsh completion script to stdout
+ * `jbai completions --install` — append source line to ~/.zshrc
+ * `jbai completions --bash`    — print bash completion script to stdout
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const config = require('./config');
+
+// Read bin entries from package.json for the canonical list of commands.
+const pkg = require('../package.json');
+const ALL_BINS = Object.keys(pkg.bin).sort();
+
+// Shortcuts = bins that have a dash after the tool name (jbai-codex-5.3, etc.)
+const TOOLS = ['claude', 'codex', 'gemini', 'opencode', 'goose', 'continue', 'council', 'proxy'];
+
+// jbai subcommands
+const JBAI_SUBCOMMANDS = [
+  'token', 'test', 'handoff', 'env', 'models', 'install', 'doctor',
+  'status', 'proxy', 'help', 'version', 'completions',
+];
+
+const JBAI_TOKEN_SUBCOMMANDS = ['set', 'refresh'];
+const JBAI_ENV_VALUES = ['staging', 'production'];
+const JBAI_MODELS_VALUES = ['all', 'claude', 'codex', 'gemini', 'opencode', 'goose', 'continue'];
+const JBAI_INSTALL_VALUES = ['all', 'claude', 'codex', 'gemini', 'opencode', 'goose', 'continue'];
+const SUPER_FLAGS = ['--super', '--yolo', '-s'];
+const COMMON_FLAGS = ['--model', '--models', '--help'];
+
+// Collect all model names for --model completion
+const ALL_MODELS = [
+  ...config.MODELS.claude.available,
+  ...config.MODELS.openai.available,
+  ...config.MODELS.codex.available,
+  ...config.MODELS.gemini.available,
+  ...config.MODELS.grazie.available,
+];
+// Dedupe
+const UNIQUE_MODELS = [...new Set(ALL_MODELS)].sort();
+
+function generateZsh() {
+  // Group shortcuts by tool for descriptive completions
+  const shortcutDescriptions = {};
+  for (const bin of ALL_BINS) {
+    if (bin === 'jbai' || bin === 'jbai-proxy') continue;
+    // e.g. jbai-codex-rockhopper -> "Codex + Rockhopper Alpha"
+    shortcutDescriptions[bin] = bin;
+  }
+
+  return `#compdef jbai jbai-claude jbai-codex jbai-gemini jbai-opencode jbai-goose jbai-continue jbai-council jbai-proxy jbai-claude-opus jbai-claude-sonnet jbai-codex-5.2 jbai-codex-5.3 jbai-codex-rockhopper jbai-gemini-3.1 jbai-gemini-supernova jbai-opencode-rockhopper jbai-opencode-grok jbai-opencode-deepseek
+# ─── jbai-cli zsh completions (auto-generated) ───
+
+# Complete model names after --model flag
+_jbai_models() {
+  local models=(
+${UNIQUE_MODELS.map(m => `    '${m}'`).join('\n')}
+  )
+  _describe 'model' models
+}
+
+# Main jbai command completions
+_jbai() {
+  local -a subcommands=(
+    'token:Show or manage authentication token'
+    'test:Test API endpoints'
+    'handoff:Continue task in Orca Lab'
+    'env:Switch environment (staging/production)'
+    'models:List available models'
+    'install:Install AI tools'
+    'doctor:Check installed tools'
+    'status:Check installed tools'
+    'proxy:Manage local proxy'
+    'help:Show help'
+    'version:Show version'
+    'completions:Shell completions'
+  )
+
+  if (( CURRENT == 2 )); then
+    _describe 'command' subcommands
+    return
+  fi
+
+  case "\${words[2]}" in
+    token)
+      if (( CURRENT == 3 )); then
+        local -a token_cmds=('set:Set token interactively' 'refresh:Refresh token via API')
+        _describe 'token command' token_cmds
+      fi
+      ;;
+    env)
+      if (( CURRENT == 3 )); then
+        local -a envs=('staging' 'production')
+        _describe 'environment' envs
+      fi
+      ;;
+    models)
+      if (( CURRENT == 3 )); then
+        local -a tools=(${JBAI_MODELS_VALUES.map(v => `'${v}'`).join(' ')})
+        _describe 'tool filter' tools
+      fi
+      ;;
+    install)
+      if (( CURRENT == 3 )); then
+        local -a tools=(${JBAI_INSTALL_VALUES.map(v => `'${v}'`).join(' ')})
+        _describe 'tool' tools
+      fi
+      ;;
+    completions)
+      if (( CURRENT == 3 )); then
+        local -a opts=('--install:Add to ~/.zshrc' '--bash:Print bash completions')
+        _describe 'option' opts
+      fi
+      ;;
+  esac
+}
+
+# Tool wrapper completions (jbai-claude, jbai-codex, etc.)
+_jbai_tool_wrapper() {
+  _arguments -s \\
+    '--model[Model to use]:model:_jbai_models' \\
+    '-m[Model to use]:model:_jbai_models' \\
+    '--super[Enable super/auto mode]' \\
+    '--yolo[Enable super/auto mode]' \\
+    '-s[Enable super/auto mode]' \\
+    '--models[List available models]' \\
+    '--list-models[List available models]' \\
+    '--help[Show help]' \\
+    'models:List available models' \\
+    '*:pass-through:'
+}
+
+# Register completions
+compdef _jbai jbai
+compdef _jbai_tool_wrapper jbai-claude jbai-codex jbai-gemini jbai-opencode jbai-goose jbai-continue
+compdef _jbai_tool_wrapper jbai-claude-opus jbai-claude-sonnet
+compdef _jbai_tool_wrapper jbai-codex-5.2 jbai-codex-5.3 jbai-codex-rockhopper
+compdef _jbai_tool_wrapper jbai-gemini-3.1 jbai-gemini-supernova
+compdef _jbai_tool_wrapper jbai-opencode-rockhopper jbai-opencode-grok jbai-opencode-deepseek
+`;
+}
+
+function generateBash() {
+  return `# ─── jbai-cli bash completions (auto-generated) ───
+
+_jbai_completions() {
+  local cur="\${COMP_WORDS[COMP_CWORD]}"
+  local prev="\${COMP_WORDS[COMP_CWORD-1]}"
+
+  # jbai subcommands
+  if [[ "\${COMP_WORDS[0]}" == "jbai" && \${COMP_CWORD} -eq 1 ]]; then
+    COMPREPLY=( $(compgen -W "${JBAI_SUBCOMMANDS.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  # jbai token <sub>
+  if [[ "\${COMP_WORDS[1]}" == "token" && \${COMP_CWORD} -eq 2 ]]; then
+    COMPREPLY=( $(compgen -W "${JBAI_TOKEN_SUBCOMMANDS.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  # jbai env <value>
+  if [[ "\${COMP_WORDS[1]}" == "env" && \${COMP_CWORD} -eq 2 ]]; then
+    COMPREPLY=( $(compgen -W "${JBAI_ENV_VALUES.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  # jbai models <value>
+  if [[ "\${COMP_WORDS[1]}" == "models" && \${COMP_CWORD} -eq 2 ]]; then
+    COMPREPLY=( $(compgen -W "${JBAI_MODELS_VALUES.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  # jbai install <value>
+  if [[ "\${COMP_WORDS[1]}" == "install" && \${COMP_CWORD} -eq 2 ]]; then
+    COMPREPLY=( $(compgen -W "${JBAI_INSTALL_VALUES.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  # --model completion
+  if [[ "$prev" == "--model" || "$prev" == "-m" ]]; then
+    COMPREPLY=( $(compgen -W "${UNIQUE_MODELS.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  # Generic flags for tool wrappers
+  COMPREPLY=( $(compgen -W "--model --super --yolo --models --help models" -- "$cur") )
+}
+
+_jbai_tool_completions() {
+  local cur="\${COMP_WORDS[COMP_CWORD]}"
+  local prev="\${COMP_WORDS[COMP_CWORD-1]}"
+
+  if [[ "$prev" == "--model" || "$prev" == "-m" ]]; then
+    COMPREPLY=( $(compgen -W "${UNIQUE_MODELS.join(' ')}" -- "$cur") )
+    return
+  fi
+
+  COMPREPLY=( $(compgen -W "--model --super --yolo --models --help models" -- "$cur") )
+}
+
+complete -F _jbai_completions jbai
+complete -F _jbai_tool_completions jbai-claude jbai-codex jbai-gemini jbai-opencode jbai-goose jbai-continue
+complete -F _jbai_tool_completions jbai-claude-opus jbai-claude-sonnet
+complete -F _jbai_tool_completions jbai-codex-5.2 jbai-codex-5.3 jbai-codex-rockhopper
+complete -F _jbai_tool_completions jbai-gemini-3.1 jbai-gemini-supernova
+complete -F _jbai_tool_completions jbai-opencode-rockhopper jbai-opencode-grok jbai-opencode-deepseek
+`;
+}
+
+function getCompletionFilePath() {
+  return path.join(config.CONFIG_DIR, 'completions.zsh');
+}
+
+function installZsh() {
+  const completionFile = getCompletionFilePath();
+  const zshrc = path.join(os.homedir(), '.zshrc');
+
+  // Write completion file
+  config.ensureConfigDir();
+  fs.writeFileSync(completionFile, generateZsh(), { mode: 0o644 });
+  console.log(`Written: ${completionFile}`);
+
+  // Check if already sourced in .zshrc
+  const sourceLine = `[ -f "${completionFile}" ] && source "${completionFile}"`;
+
+  if (fs.existsSync(zshrc)) {
+    const content = fs.readFileSync(zshrc, 'utf-8');
+    if (content.includes('jbai') && content.includes('completions.zsh')) {
+      console.log('Already in ~/.zshrc — updating completion file only');
+      return;
+    }
+  }
+
+  fs.appendFileSync(zshrc, `\n# jbai-cli completions\n${sourceLine}\n`);
+  console.log('Added to ~/.zshrc');
+  console.log('\nRun: source ~/.zshrc   (or open a new terminal)');
+}
+
+function run(args) {
+  if (args.includes('--install')) {
+    installZsh();
+    return;
+  }
+
+  if (args.includes('--bash')) {
+    process.stdout.write(generateBash());
+    return;
+  }
+
+  // Default: print zsh completions to stdout
+  process.stdout.write(generateZsh());
+}
+
+module.exports = { run, generateZsh, generateBash, installZsh, getCompletionFilePath };

package/lib/config.js

@@ -35,6 +35,7 @@ const MODELS = {
     available: [
       // Claude 4.6 series (latest)
       'claude-opus-4-6',
+      'claude-sonnet-4-6',
       // Claude 4.5 series
       'claude-opus-4-5-20251101',
       'claude-sonnet-4-5-20250929',
@@ -83,9 +84,11 @@ const MODELS = {
   codex: {
     default: 'gpt-5.3-codex',
     available: [
+      // Hidden/EAP models (JB internal NDA)
+      'rockhopper-alpha',
+      'gpt-5.3-codex-api-preview',
       // Codex-specific models (responses API only, NOT available via chat/completions)
       'gpt-5.3-codex',
-      'gpt-5.3-codex-api-preview',
       // GPT-5.x chat models (also work via responses API)
       'gpt-5.2-2025-12-11',
       'gpt-5.2',
@@ -105,6 +108,9 @@ const MODELS = {
   gemini: {
     default: 'gemini-2.5-flash',
     available: [
+      // Hidden/EAP models (JB internal NDA)
+      'supernova',
+      'gemini-3.1-pro-preview',
       // Gemini 3.x (preview)
       'gemini-3-pro-preview',
       'gemini-3-flash-preview',
@@ -125,24 +131,31 @@ const MODELS = {
       // Google
       'google-gemini-3-1-pro',
       'google-gemini-3-0-flash',
+      'google-gemini-3-0-pro-snowball',
       'google-chat-gemini-pro-2.5',
       'google-gemini-2.5-flash',
       // DeepSeek
       'deepseek-r1',
-      'deepseek-chat-v3',
+      'openrouter-deepseek-v3-2',
       // Mistral
       'mistral-large',
       'mistral-small',
+      'openrouter-mistral-large-2512',
       // xAI / Grok
       'xai-grok-4',
       'xai-grok-4-1-fast',
-      'xai-grok-3',
+      'xai-grok-code-fast-1',
       // Qwen
       'qwen-max',
       'qwen-plus',
-      // Meta (via OpenRouter)
-      'openrouter-meta-llama-4-maverick',
-      'openrouter-meta-llama-4-scout',
+      // Kimi
+      'openrouter-kimi-k2-5',
+      'openrouter-kimi-k2-thinking',
+      // MiniMax
+      'openrouter-minimax-m2-1',
+      // ZhipuAI
+      'openrouter-zhipuai-glm-4-7',
+      'openrouter-zhipuai-glm-4-7-flash',
     ]
   }
 };

package/lib/postinstall.js

@@ -46,5 +46,8 @@ Next steps:
    $ jbai-goose      # Goose (Block)
    $ jbai-continue   # Continue CLI
 
+4. Enable tab completions:
+   $ jbai completions --install
+
 Run 'jbai help' for more options.
 `);

package/lib/shortcut.js

@@ -0,0 +1,47 @@
+/**
+ * Shared logic for per-model shortcut scripts.
+ *
+ * Each shortcut (e.g. jbai-codex-rockhopper) is a thin wrapper that calls
+ * the base tool wrapper (jbai-codex, jbai-claude, …) with a fixed model
+ * and super mode enabled by default.
+ *
+ * Usage in a shortcut script:
+ *   require('../lib/shortcut').run({
+ *     tool: 'codex',          // base wrapper to delegate to
+ *     model: 'rockhopper-alpha',
+ *     label: 'Rockhopper Alpha',
+ *   });
+ */
+
+const { spawn } = require('child_process');
+const path = require('path');
+
+function run({ tool, model, label }) {
+  const args = process.argv.slice(2);
+
+  // Already force super mode + the pinned model
+  const extraArgs = ['--super', '--model', model];
+
+  // If user already passed --model, skip our default
+  if (args.includes('--model') || args.includes('-m')) {
+    extraArgs.splice(extraArgs.indexOf('--model'), 2);
+  }
+
+  const wrapperScript = path.join(__dirname, '..', 'bin', `jbai-${tool}.js`);
+  const finalArgs = [wrapperScript, ...extraArgs, ...args];
+
+  console.log(`⚡ ${label || model} (super mode)`);
+
+  const child = spawn(process.execPath, finalArgs, {
+    stdio: 'inherit',
+    env: process.env,
+  });
+
+  child.on('exit', (code) => process.exit(code || 0));
+  child.on('error', (err) => {
+    console.error(`Error: ${err.message}`);
+    process.exit(1);
+  });
+}
+
+module.exports = { run };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jbai-cli",
-  "version": "1.9.6",
+  "version": "2.1.0",
   "description": "CLI wrappers to use AI coding tools (Claude Code, Codex, Gemini CLI, OpenCode, Goose, Continue) with JetBrains AI Platform",
   "keywords": [
     "jetbrains",
@@ -30,9 +30,19 @@
     "jbai": "bin/jbai.js",
     "jbai-proxy": "bin/jbai-proxy.js",
     "jbai-claude": "bin/jbai-claude.js",
+    "jbai-claude-opus": "bin/jbai-claude-opus.js",
+    "jbai-claude-sonnet": "bin/jbai-claude-sonnet.js",
     "jbai-codex": "bin/jbai-codex.js",
+    "jbai-codex-5.2": "bin/jbai-codex-5.2.js",
+    "jbai-codex-5.3": "bin/jbai-codex-5.3.js",
+    "jbai-codex-rockhopper": "bin/jbai-codex-rockhopper.js",
     "jbai-gemini": "bin/jbai-gemini.js",
+    "jbai-gemini-3.1": "bin/jbai-gemini-3.1.js",
+    "jbai-gemini-supernova": "bin/jbai-gemini-supernova.js",
     "jbai-opencode": "bin/jbai-opencode.js",
+    "jbai-opencode-rockhopper": "bin/jbai-opencode-rockhopper.js",
+    "jbai-opencode-grok": "bin/jbai-opencode-grok.js",
+    "jbai-opencode-deepseek": "bin/jbai-opencode-deepseek.js",
     "jbai-goose": "bin/jbai-goose.js",
     "jbai-continue": "bin/jbai-continue.js",
     "jbai-council": "bin/jbai-council.js"