draco-cli 3.0.3 → 3.1.0

package/README.md

@@ -43,11 +43,20 @@ draco --model=gpt-4 "refactor this function"
 
 **Environment Variables:**
 ```bash
-export OPENCODE_ZEN_KEY="your-api-key"
-export DRACO_MODEL="nemotron-3-ultra-free"
+export GEMINI_API_KEY="your-google-ai-studio-key"   # https://aistudio.google.com/apikey
+export DRACO_MODEL="gemini-3.8-flash"                # default
 export DRACO_APPROVALS="suggest"  # suggest | auto-edit | full-auto
 ```
 
+**Models (Gemini free tier):**
+
+| Model | Best for |
+|-------|----------|
+| `gemini-3.8-flash` | **Default.** Fast, balanced, multimodal |
+| `gemini-3.1-pro-preview` | Complex reasoning & coding |
+| `gemini-3.5-flash-lite` | High-frequency lightweight tasks |
+| `gemini-2.5-flash` / `gemini-2.5-pro` | Previous-generation free tier |
+
 ## Project Structure
 
 ```

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "draco-cli",
-  "version": "3.0.3",
-  "description": "Draco — a zero-login, terminal-first AI coding agent for the OpenCode Zen API.",
+  "version": "3.1.0",
+  "description": "Draco — a terminal-first AI coding agent powered by the Gemini API (Google AI Studio).",
   "license": "MIT",
   "author": "Anuj",
   "bin": {
@@ -15,8 +15,8 @@
     "ai",
     "cli",
     "coding-agent",
-    "opencode",
-    "zen",
+    "gemini",
+    "google-ai-studio",
     "terminal",
     "agent"
   ],

package/src/cli/slash-commands.js

@@ -72,7 +72,7 @@ async function handle(line, rt) {
           process.stdout.write(`${marker}${id}\n`);
         }
         process.stdout.write(
-          `${ui.c.gray}(active marked with *; some accounts use an "opencode/" prefix — switch with /model opencode/<id>)${ui.c.reset}\n`
+          `${ui.c.gray}(active marked with *; switch with /model <id>)${ui.c.reset}\n`
         );
       } catch (e) {
         ui.error(
@@ -233,6 +233,7 @@ async function handle(line, rt) {
       const mcpCount = rt.mcp ? rt.mcp.listTools().length : 0;
       const lines = [
         `version:       ${cfg.version}`,
+        `provider:      ${cfg.provider || 'Google AI Studio'}`,
         `endpoint:      ${cfg.apiBase}`,
         `model:         ${rt.agent.activeModel}`,
         `approvals:     ${cfg.approvals}`,

package/src/config.js

@@ -12,15 +12,23 @@ try {
   // keep fallback version
 }
 
-const DEFAULT_API_BASE = 'https://opencode.ai';
-const DEFAULT_MODEL = 'nemotron-3-ultra-free';
-const DEFAULT_FALLBACK_CHAIN = [
-  'nemotron-3-ultra-free',
-  'deepseek-v4-flash-free',
-  'big-pickle',
-  'mimo-v2.5-free',
+const PROVIDER_NAME = 'Google AI Studio';
+const DEFAULT_API_BASE = 'https://generativelanguage.googleapis.com/v1beta/openai';
+const DEFAULT_API_PATH = '';
+const DEFAULT_MODEL = 'gemini-3.8-flash';
+const DEFAULT_FALLBACK_CHAIN = ['gemini-3.8-flash', 'gemini-3.1-pro-preview'];
+const KNOWN_FREE_MODELS = [
+  'gemini-3.8-flash',
+  'gemini-3.7-flash',
+  'gemini-3.6-flash',
+  'gemini-3.5-flash',
+  'gemini-3.5-flash-lite',
+  'gemini-3.1-pro-preview',
+  'gemini-3.1-flash-lite',
+  'gemini-2.5-flash',
+  'gemini-2.5-flash-lite',
+  'gemini-2.5-pro',
 ];
-const KNOWN_FREE_MODELS = DEFAULT_FALLBACK_CHAIN.slice();
 
 const CHAT_TIMEOUT_MS = 60000;
 const MODELS_TIMEOUT_MS = 15000;
@@ -117,11 +125,15 @@ function validateConfig(config) {
   const warnings = [];
   
   if (!config.apiKey) {
-    warnings.push('No API key configured. Set OPENCODE_ZEN_KEY or enter when prompted.');
+    warnings.push(
+      'No API key configured. Get one at https://aistudio.google.com/apikey and set GEMINI_API_KEY.'
+    );
   }
-  
-  if (config.model && !KNOWN_FREE_MODELS.includes(config.model) && !config.model.includes('/')) {
-    warnings.push(`Model "${config.model}" is not in the known free models list.`);
+
+  if (config.model && !KNOWN_FREE_MODELS.includes(config.model)) {
+    warnings.push(
+      `Model "${config.model}" is not on the Gemini free-tier list. It may be paid-only or invalid.`
+    );
   }
   
   if (config.fallbackChain && config.fallbackChain.length === 0) {
@@ -160,10 +172,10 @@ function resolveConfig(cliFlags) {
       ? persisted.fallbackChain
       : DEFAULT_FALLBACK_CHAIN.slice();
 
-  let apiKey = env.OPENCODE_ZEN_KEY || '';
+  let apiKey = env.GEMINI_API_KEY || env.GOOGLE_API_KEY || '';
   let keySource = null;
   if (apiKey) {
-    keySource = 'environment';
+    keySource = env.GEMINI_API_KEY ? 'environment' : 'environment (GOOGLE_API_KEY)';
   } else if (typeof persisted.apiKey === 'string' && persisted.apiKey.trim()) {
     apiKey = persisted.apiKey.trim();
     keySource = 'saved';
@@ -171,9 +183,11 @@ function resolveConfig(cliFlags) {
 
   const config = {
     version: VERSION,
+    provider: PROVIDER_NAME,
     apiKey,
     keySource,
-    apiBase: (env.DRACO_API_BASE || DEFAULT_API_BASE).replace(/\/+$/, ''),
+    apiBase: DEFAULT_API_BASE,
+    apiPath: DEFAULT_API_PATH,
     model,
     yolo,
     approvals,
@@ -194,6 +208,8 @@ module.exports = {
   APPROVAL_MODES,
   DEFAULT_APPROVALS,
   DEFAULT_API_BASE,
+  DEFAULT_API_PATH,
+  PROVIDER_NAME,
   DEFAULT_MODEL,
   DEFAULT_FALLBACK_CHAIN,
   KNOWN_FREE_MODELS,

package/src/index.js

@@ -33,11 +33,11 @@ function printUsage() {
       'One-shot mode:   draco "your prompt here" — answer once, then exit',
       '',
       'Environment:',
-      '  OPENCODE_ZEN_KEY        Optional. Skips the first-run key prompt.',
-      '  DRACO_MODEL             Default model override (default: nemotron-3-ultra-free)',
+      '  GEMINI_API_KEY          Google AI Studio key (skips the first-run prompt)',
+      '  GOOGLE_API_KEY          Alias of GEMINI_API_KEY',
+      '  DRACO_MODEL             Default model override (default: gemini-3.8-flash)',
       '  DRACO_APPROVALS         suggest | auto-edit | full-auto',
       '  DRACO_YOLO              "true" to auto-approve shell commands',
-      '  DRACO_API_BASE          Override the API endpoint',
       '  DRACO_SEARCH_PROVIDER   Search backend when a key is set (default: brave)',
       '  DRACO_SEARCH_API_KEY    Optional. Use Brave instead of built-in DuckDuckGo',
     ].join('\n') + '\n'
@@ -55,9 +55,11 @@ function installGlobalHandlers() {
 
 async function ensureApiKey(cfg, nextLine) {
   if (cfg.apiKey) return true;
-  ui.info('No API key found. Get a free one at https://opencode.ai');
+  ui.info('No API key found. Create a free one at https://aistudio.google.com/apikey');
   for (let attempt = 0; attempt < 5; attempt++) {
-    process.stdout.write(`${ui.c.yellow}?${ui.c.reset} Enter your OpenCode Zen API key: `);
+    process.stdout.write(
+      `${ui.c.yellow}?${ui.c.reset} Paste your Google AI Studio API key: `
+    );
     const line = await nextLine();
     if (line === null) return false;
     const key = line.trim();
@@ -205,7 +207,7 @@ async function main(argv) {
 
   const hasKey = await ensureApiKey(cfg, nextLine);
   if (!hasKey) {
-    ui.error('No API key provided. Set OPENCODE_ZEN_KEY or enter it when prompted.');
+    ui.error('No API key provided. Set GEMINI_API_KEY or enter it when prompted.');
     process.exit(1);
   }
 

package/src/providers/openai-compatible.js

@@ -5,7 +5,15 @@ const {
   MODELS_TIMEOUT_MS,
 } = require('../config.js');
 
-const DEFAULT_PROBE_MODEL = 'nemotron-3-ultra-free';
+const DEFAULT_PROBE_MODEL = 'gemini-3.8-flash';
+
+function endpoint(cfg, suffix) {
+  return `${cfg.apiBase}${cfg.apiPath || ''}/${suffix}`;
+}
+
+function probeModel(cfg) {
+  return (cfg.model && cfg.model.trim()) || DEFAULT_PROBE_MODEL;
+}
 
 class ApiError extends Error {
   constructor(status, message, bodyText) {
@@ -64,7 +72,7 @@ async function listModels(cfg) {
   try {
     let res;
     try {
-      res = await fetch(`${cfg.apiBase}/zen/v1/models`, {
+      res = await fetch(endpoint(cfg, 'models'), {
         signal: controller.signal,
         headers: { Authorization: `Bearer ${cfg.apiKey}` },
       });
@@ -132,7 +140,7 @@ async function streamChatCompletion(opts) {
 
   let res;
   try {
-    res = await fetch(`${cfg.apiBase}/zen/v1/chat/completions`, {
+    res = await fetch(endpoint(cfg, 'chat/completions'), {
       method: 'POST',
       signal: controller.signal,
       headers: {
@@ -253,7 +261,7 @@ async function probeApiKey(cfg, apiKey) {
   try {
     let res;
     try {
-      res = await fetch(`${cfg.apiBase}/zen/v1/chat/completions`, {
+      res = await fetch(endpoint(cfg, 'chat/completions'), {
         method: 'POST',
         signal: controller.signal,
         headers: {
@@ -261,7 +269,7 @@ async function probeApiKey(cfg, apiKey) {
           'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-          model: DEFAULT_PROBE_MODEL,
+          model: probeModel(cfg),
           messages: [{ role: 'user', content: 'ping' }],
           max_tokens: 1,
         }),
@@ -272,13 +280,22 @@ async function probeApiKey(cfg, apiKey) {
       }
       return { verdict: 'unreachable', detail: e.message };
     }
-    if (res.ok || res.status === 429 || res.status === 400) {
-      // 429 = rate-limited but authenticated; 400 ModelError = auth fine, model busy
+    if (res.ok || res.status === 429) {
+      // 429 = rate-limited but authenticated
       return { verdict: 'ok', status: res.status };
     }
     if (res.status === 401 || res.status === 403) {
       return { verdict: 'invalid', status: res.status };
     }
+    if (res.status === 400) {
+      // Gemini reports a bad key as 400 INVALID_ARGUMENT, so inspect the body.
+      // A real model/param error still means the key itself was accepted.
+      const body = await safeBodyText(res);
+      if (/api key|api_key|API_KEY|unauthenticated|permission/i.test(body)) {
+        return { verdict: 'invalid', status: res.status };
+      }
+      return { verdict: 'ok', status: res.status };
+    }
     return { verdict: 'unknown', status: res.status };
   } finally {
     clearTimeout(timer);

package/src/ui/terminal.js

@@ -9,8 +9,8 @@ const c = theme.c;
 function splash(cfg) {
   process.stdout.write(theme.welcomeBanner(cfg) + '\n');
   process.stdout.write('\n');
-  process.stdout.write(`Endpoint: OpenCode Zen API Server\n`);
-  process.stdout.write(`Model: Free Dynamic Routing Tier\n`);
+  process.stdout.write(`Endpoint: ${cfg.provider || 'Google AI Studio'} (Gemini API)\n`);
+  process.stdout.write(`${c.gray}Free-tier models via aistudio.google.com${c.reset}\n`);
   process.stdout.write('\n');
   process.stdout.write(`${c.bold}Workspace:${c.reset} ${cfg.workspaceRoot}\n`);
   process.stdout.write(`${c.gray}Active model: ${cfg.model}${c.reset}\n`);