@aibridge/cli 0.2.0 → 0.4.0

package/src/index.ts

@@ -6,9 +6,6 @@ export { getDriver } from './drivers.ts';
 export {
   type Backend,
   backendModelId,
-  DEFAULT_IMAGE_GEN,
-  DEFAULT_IMPLEMENTER,
-  DEFAULT_MODEL,
   type Effort,
   formatImageGenModelError,
   formatUnknownModelError,

package/src/models.ts

@@ -4,7 +4,7 @@
  * Models are registered by canonical, provider-qualified slug —
  * `<vendor>-<cli>/<model>[-<effort>]`, e.g. `openai-codex/gpt-5.6-sol-high`.
  * Canonical slugs only — no short aliases, by design. That holds on both sides:
- * `backendModel` is a pinned model id (`claude-opus-5`), never a moving vendor
+ * `backendModel` is a pinned model id (`claude-sonnet-5`), never a moving vendor
  * alias (`opus`), so a seat never silently changes model under you.
  */
 
@@ -31,7 +31,7 @@ export const MODELS: Record<string, ModelSpec> = {
     backend: 'grok',
     backendModel: 'grok-4.5',
     efforts: ['low', 'medium', 'high'],
-    brief: 'xAI Grok 4.5 via grok CLI — default for plan & review; own xAI login',
+    brief: 'xAI Grok 4.5 via grok CLI — own xAI login; ~30 req/min, ~1k msgs/day, single-flight',
   },
   'google-antigravity/gemini-3.6-flash': {
     slug: 'google-antigravity/gemini-3.6-flash',
@@ -40,7 +40,17 @@ export const MODELS: Record<string, ModelSpec> = {
     efforts: ['low', 'medium', 'high'],
     defaultEffort: 'high',
     brief:
-      'Google Gemini 3.6 Flash via agy — default for implement, also image-gen; own Antigravity login',
+      'Google Gemini 3.6 Flash via agy — own Antigravity login; quota shared across all Gemini tiers',
+  },
+  'google-antigravity/gemini-3.1-pro': {
+    slug: 'google-antigravity/gemini-3.1-pro',
+    backend: 'agy',
+    // agy exposes only -high and -low for this class — there is no medium tier.
+    efforts: ['low', 'high'],
+    backendModel: 'gemini-3.1-pro',
+    defaultEffort: 'high',
+    brief:
+      'Google Gemini 3.1 Pro via agy — own Antigravity login; quota shared across all Gemini tiers',
   },
   'google-antigravity/claude-sonnet-4-6': {
     slug: 'google-antigravity/claude-sonnet-4-6',
@@ -68,37 +78,56 @@ export const MODELS: Record<string, ModelSpec> = {
     backend: 'codex',
     backendModel: 'gpt-5.6-sol',
     efforts: ['low', 'medium', 'high', 'xhigh'],
-    brief: 'OpenAI Codex gpt-5.6-sol via codex CLI',
+    brief: 'OpenAI gpt-5.6-sol via codex CLI — frontier agentic coding; own ChatGPT login',
   },
-  'anthropic-claude/sonnet-5': {
-    slug: 'anthropic-claude/sonnet-5',
+  'openai-codex/gpt-5.6-terra': {
+    slug: 'openai-codex/gpt-5.6-terra',
+    backend: 'codex',
+    backendModel: 'gpt-5.6-terra',
+    efforts: ['low', 'medium', 'high', 'xhigh'],
+    brief: 'OpenAI gpt-5.6-terra via codex CLI — balanced, everyday coding; own ChatGPT login',
+  },
+  'openai-codex/gpt-5.6-luna': {
+    slug: 'openai-codex/gpt-5.6-luna',
+    backend: 'codex',
+    backendModel: 'gpt-5.6-luna',
+    efforts: ['low', 'medium', 'high', 'xhigh'],
+    brief: 'OpenAI gpt-5.6-luna via codex CLI — fast and affordable coding; own ChatGPT login',
+  },
+  'anthropic-claude/fable-5': {
+    slug: 'anthropic-claude/fable-5',
     backend: 'claude',
-    backendModel: 'claude-sonnet-5',
+    backendModel: 'claude-fable-5',
     efforts: ['low', 'medium', 'high', 'xhigh', 'max'],
-    brief: 'Claude Sonnet 5 via claude CLI — bills your Claude subscription',
+    defaultEffort: 'high',
+    brief:
+      'Claude Fable 5 via claude CLI — hardest, longest-running work; bills the claude CLI subscription',
   },
   'anthropic-claude/opus-5': {
     slug: 'anthropic-claude/opus-5',
     backend: 'claude',
-    backendModel: 'claude-opus-5',
+    backendModel: 'claude-opus-5[1m]',
     efforts: ['low', 'medium', 'high', 'xhigh', 'max'],
     defaultEffort: 'high',
-    brief: 'Claude Opus 5 via claude CLI (default effort: high) — bills subscription',
+    brief:
+      'Claude Opus 5, 1M context via claude CLI — everyday complex work; bills the claude CLI subscription',
   },
-  'anthropic-claude/opus-5-1m': {
-    slug: 'anthropic-claude/opus-5-1m',
+  'anthropic-claude/sonnet-5': {
+    slug: 'anthropic-claude/sonnet-5',
     backend: 'claude',
-    backendModel: 'claude-opus-5[1m]',
+    backendModel: 'claude-sonnet-5',
     efforts: ['low', 'medium', 'high', 'xhigh', 'max'],
-    defaultEffort: 'high',
-    brief: 'Claude Opus 5, 1M context via claude CLI — long-context work; bills subscription',
+    brief: 'Claude Sonnet 5 via claude CLI — routine work; bills the claude CLI subscription',
+  },
+  'anthropic-claude/haiku-4-5': {
+    slug: 'anthropic-claude/haiku-4-5',
+    backend: 'claude',
+    backendModel: 'claude-haiku-4-5-20251001',
+    efforts: ['low', 'medium', 'high', 'xhigh', 'max'],
+    brief: 'Claude Haiku 4.5 via claude CLI — quick answers; bills the claude CLI subscription',
   },
 } as const satisfies Record<string, ModelSpec>;
 
-export const DEFAULT_MODEL = 'xai-grok/grok-4.5';
-export const DEFAULT_IMPLEMENTER = 'google-antigravity/gemini-3.6-flash';
-export const DEFAULT_IMAGE_GEN = 'openai-codex/gpt-5.6-sol';
-
 export type ImageFormat = 'jpg' | 'png';
 
 const IMAGE_GEN_FORMATS: ReadonlyMap<Backend, ImageFormat> = new Map([

package/src/quotaPreflight.test.ts

@@ -1,8 +1,13 @@
 import assert from 'node:assert/strict';
 import type { AgyQuotaSnapshot } from '@aibridge/driver-agy';
 import type { CodexQuotaSnapshot } from '@aibridge/driver-codex';
+import type { GrokQuotaSnapshot } from '@aibridge/driver-grok';
 import { test } from 'vitest';
-import { evaluateAgyPreflight, evaluateCodexPreflight } from './quotaPreflight.ts';
+import {
+  evaluateAgyPreflight,
+  evaluateCodexPreflight,
+  evaluateGrokPreflight,
+} from './quotaPreflight.ts';
 
 test('evaluateAgyPreflight: exhausted model returns ok:false with resetTime', () => {
   const snapshot: AgyQuotaSnapshot = {
@@ -176,3 +181,43 @@ test('evaluateCodexPreflight: healthy (77% used) returns ok:true', () => {
 
   assert.deepEqual(result, { ok: true });
 });
+
+test('evaluateGrokPreflight: usedPercent at 100 refuses with resetAt', () => {
+  const snapshot: GrokQuotaSnapshot = {
+    fetchedAt: '2024-01-01T12:00:00Z',
+    periodType: 'weekly',
+    periodStart: '2024-01-01T00:00:00Z',
+    periodEnd: '2024-01-08T00:00:00Z',
+    usedPercent: 100,
+    products: [],
+    onDemandUsedCents: undefined,
+    onDemandCapCents: undefined,
+    prepaidBalanceCents: undefined,
+  };
+
+  const result = evaluateGrokPreflight(snapshot);
+
+  assert.deepEqual(result, {
+    ok: false,
+    message: 'grok credit quota exhausted',
+    resetAt: '2024-01-08T00:00:00Z',
+  });
+});
+
+test('evaluateGrokPreflight: healthy (17% used) returns ok:true', () => {
+  const snapshot: GrokQuotaSnapshot = {
+    fetchedAt: '2024-01-01T12:00:00Z',
+    periodType: 'weekly',
+    periodStart: '2024-01-01T00:00:00Z',
+    periodEnd: '2024-01-08T00:00:00Z',
+    usedPercent: 17,
+    products: [],
+    onDemandUsedCents: undefined,
+    onDemandCapCents: undefined,
+    prepaidBalanceCents: undefined,
+  };
+
+  const result = evaluateGrokPreflight(snapshot);
+
+  assert.deepEqual(result, { ok: true });
+});

package/src/quotaPreflight.ts

@@ -1,5 +1,6 @@
 import { type AgyQuotaSnapshot, fetchAgyQuota, findModelQuota } from '@aibridge/driver-agy';
 import { type CodexQuotaSnapshot, fetchCodexQuota } from '@aibridge/driver-codex';
+import { fetchGrokQuota, type GrokQuotaSnapshot } from '@aibridge/driver-grok';
 import { backendModelId, type ResolvedModel } from './models.ts';
 
 export type PreflightVerdict =
@@ -51,11 +52,27 @@ export function evaluateCodexPreflight(snapshot: CodexQuotaSnapshot): PreflightV
   return { ok: true };
 }
 
+export function evaluateGrokPreflight(snapshot: GrokQuotaSnapshot): PreflightVerdict {
+  if (snapshot.usedPercent !== undefined && snapshot.usedPercent >= 100) {
+    return {
+      ok: false,
+      message: 'grok credit quota exhausted',
+      resetAt: snapshot.periodEnd,
+    };
+  }
+
+  return { ok: true };
+}
+
 export async function preflightModel(resolved: ResolvedModel): Promise<PreflightVerdict> {
   if (resolved.spec.backend === 'codex') {
     return preflightCodex();
   }
 
+  if (resolved.spec.backend === 'grok') {
+    return preflightGrok();
+  }
+
   if (resolved.spec.backend !== 'agy') {
     return { ok: true };
   }
@@ -83,6 +100,18 @@ export async function preflightCodex(): Promise<PreflightVerdict> {
   }
 }
 
+export async function preflightGrok(): Promise<PreflightVerdict> {
+  try {
+    const snapshot = await fetchGrokQuota();
+    return evaluateGrokPreflight(snapshot);
+  } catch (err) {
+    return {
+      ok: true,
+      warning: `quota preflight failed (${(err as Error).message}); proceeding`,
+    };
+  }
+}
+
 function formatReset(resetTime: string | undefined): string {
   if (!resetTime) return '-';
   const ms = new Date(resetTime).getTime() - Date.now();