@aibridge/cli 0.3.0 → 0.5.0

package/src/quotaPreflight.test.ts

@@ -1,8 +1,14 @@
 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,
+  renderPreflightRefusal,
+} from './quotaPreflight.ts';
 
 test('evaluateAgyPreflight: exhausted model returns ok:false with resetTime', () => {
   const snapshot: AgyQuotaSnapshot = {
@@ -23,6 +29,7 @@ test('evaluateAgyPreflight: exhausted model returns ok:false with resetTime', ()
 
   assert.deepEqual(result, {
     ok: false,
+    kind: 'quota',
     message: 'agy model "Gemini 3.5 Flash (High)" is quota-exhausted',
     resetAt: '2024-01-02T12:00:00Z',
   });
@@ -122,6 +129,7 @@ test('evaluateCodexPreflight: limitReached returns ok:false', () => {
 
   assert.deepEqual(result, {
     ok: false,
+    kind: 'quota',
     message: 'codex quota limit reached',
     resetAt: '2024-01-01T17:00:00Z',
   });
@@ -176,3 +184,68 @@ 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,
+    kind: 'quota',
+    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 });
+});
+
+test('renderPreflightRefusal: auth kind uses unauthenticated wording', () => {
+  const msg = renderPreflightRefusal('plan', {
+    kind: 'auth',
+    message: 'grok session expired (401) — run `grok login`, then retry',
+    resetAt: undefined,
+  });
+  assert.strictEqual(
+    msg,
+    'aibridge plan: refusing — grok session expired (401) — run `grok login`, then retry. Running with --no-preflight would only send the delegate in unauthenticated. Or use a different --model.',
+  );
+});
+
+test('renderPreflightRefusal: quota kind keeps override wording', () => {
+  const msg = renderPreflightRefusal('subagent', {
+    kind: 'quota',
+    message: 'grok credit quota exhausted',
+    resetAt: undefined,
+  });
+  assert.strictEqual(
+    msg,
+    'aibridge subagent: refusing — grok credit quota exhausted. Use --no-preflight to override, or a claude-backend fallback (subagent --model sonnet|opus — bills the Claude subscription).',
+  );
+});

package/src/quotaPreflight.ts

@@ -1,10 +1,17 @@
 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 { isAuthExpired } from '@aibridge/proc';
 import { backendModelId, type ResolvedModel } from './models.ts';
 
 export type PreflightVerdict =
   | { readonly ok: true; readonly warning?: string }
-  | { readonly ok: false; readonly message: string; readonly resetAt: string | undefined };
+  | {
+      readonly ok: false;
+      readonly kind: 'auth' | 'quota';
+      readonly message: string;
+      readonly resetAt: string | undefined;
+    };
 
 export function evaluateAgyPreflight(
   snapshot: AgyQuotaSnapshot,
@@ -31,7 +38,12 @@ export function evaluateAgyPreflight(
         }
       }
     }
-    return { ok: false, message: `agy model "${backendModel}" is quota-exhausted`, resetAt };
+    return {
+      ok: false,
+      kind: 'quota',
+      message: `agy model "${backendModel}" is quota-exhausted`,
+      resetAt,
+    };
   }
 
   return { ok: true };
@@ -40,12 +52,30 @@ export function evaluateAgyPreflight(
 export function evaluateCodexPreflight(snapshot: CodexQuotaSnapshot): PreflightVerdict {
   if (snapshot.limitReached) {
     const resetAt = snapshot.windows.find(w => w.resetAt)?.resetAt;
-    return { ok: false, message: 'codex quota limit reached', resetAt };
+    return { ok: false, kind: 'quota', message: 'codex quota limit reached', resetAt };
   }
 
   const exhaustedWindow = snapshot.windows.find(w => w.usedPercent >= 100);
   if (exhaustedWindow) {
-    return { ok: false, message: 'codex quota limit reached', resetAt: exhaustedWindow.resetAt };
+    return {
+      ok: false,
+      kind: 'quota',
+      message: 'codex quota limit reached',
+      resetAt: exhaustedWindow.resetAt,
+    };
+  }
+
+  return { ok: true };
+}
+
+export function evaluateGrokPreflight(snapshot: GrokQuotaSnapshot): PreflightVerdict {
+  if (snapshot.usedPercent !== undefined && snapshot.usedPercent >= 100) {
+    return {
+      ok: false,
+      kind: 'quota',
+      message: 'grok credit quota exhausted',
+      resetAt: snapshot.periodEnd,
+    };
   }
 
   return { ok: true };
@@ -56,6 +86,10 @@ export async function preflightModel(resolved: ResolvedModel): Promise<Preflight
     return preflightCodex();
   }
 
+  if (resolved.spec.backend === 'grok') {
+    return preflightGrok();
+  }
+
   if (resolved.spec.backend !== 'agy') {
     return { ok: true };
   }
@@ -64,10 +98,9 @@ export async function preflightModel(resolved: ResolvedModel): Promise<Preflight
     const snapshot = await fetchAgyQuota();
     return evaluateAgyPreflight(snapshot, backendModelId(resolved));
   } catch (err) {
-    return {
-      ok: true,
-      warning: `quota preflight failed (${(err as Error).message}); proceeding`,
-    };
+    if (isAuthExpired(err))
+      return { ok: false, kind: 'auth', message: err.message, resetAt: undefined };
+    return { ok: true, warning: `quota preflight failed (${(err as Error).message}); proceeding` };
   }
 }
 
@@ -76,10 +109,20 @@ export async function preflightCodex(): Promise<PreflightVerdict> {
     const snapshot = await fetchCodexQuota();
     return evaluateCodexPreflight(snapshot);
   } catch (err) {
-    return {
-      ok: true,
-      warning: `quota preflight failed (${(err as Error).message}); proceeding`,
-    };
+    if (isAuthExpired(err))
+      return { ok: false, kind: 'auth', message: err.message, resetAt: undefined };
+    return { ok: true, warning: `quota preflight failed (${(err as Error).message}); proceeding` };
+  }
+}
+
+export async function preflightGrok(): Promise<PreflightVerdict> {
+  try {
+    const snapshot = await fetchGrokQuota();
+    return evaluateGrokPreflight(snapshot);
+  } catch (err) {
+    if (isAuthExpired(err))
+      return { ok: false, kind: 'auth', message: err.message, resetAt: undefined };
+    return { ok: true, warning: `quota preflight failed (${(err as Error).message}); proceeding` };
   }
 }
 
@@ -95,8 +138,11 @@ function formatReset(resetTime: string | undefined): string {
 
 export function renderPreflightRefusal(
   cmd: string,
-  verdict: { message: string; resetAt: string | undefined },
+  verdict: { kind: 'auth' | 'quota'; message: string; resetAt: string | undefined },
 ): string {
+  if (verdict.kind === 'auth') {
+    return `aibridge ${cmd}: refusing — ${verdict.message}. Running with --no-preflight would only send the delegate in unauthenticated. Or use a different --model.`;
+  }
   const resetClause = verdict.resetAt ? ` Resets ${formatReset(verdict.resetAt)}.` : '';
   return `aibridge ${cmd}: refusing — ${verdict.message}.${resetClause} Use --no-preflight to override, or a claude-backend fallback (subagent --model sonnet|opus — bills the Claude subscription).`;
 }