ampcode-connector 0.1.8 → 0.1.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ampcode-connector",
-  "version": "0.1.8",
+  "version": "0.1.9",
   "description": "Proxy AmpCode through local OAuth subscriptions (Claude Code, Codex, Gemini CLI, Antigravity)",
   "license": "MIT",
   "repository": {
@@ -40,16 +40,16 @@
     "format": "biome check --write src/ tests/"
   },
   "devDependencies": {
-    "@biomejs/biome": "^2.4.0",
+    "@biomejs/biome": "^2.4.2",
+    "@google/genai": "^1.42.0",
     "@types/bun": "^1.3.9"
   },
   "peerDependencies": {
-    "typescript": "^5"
+    "typescript": "^5.9.3"
   },
   "dependencies": {
-    "@google/genai": "^1.41.0",
-    "@kreuzberg/html-to-markdown": "^2.25.0",
-    "ampcode-connector": "^0.1.5",
+    "@kreuzberg/html-to-markdown": "^2.25.1",
+    "ampcode-connector": "^0.1.8",
     "exa-js": "^2.4.0"
   }
 }

package/src/cli/setup.ts

@@ -131,6 +131,7 @@ export async function setup(): Promise<void> {
 
   for (const p of providers) {
     const connected = p.accounts.filter((a) => a.status === "connected");
+    const disabled = p.accounts.filter((a) => a.status === "disabled");
     const total = p.accounts.filter((a) => a.status !== "disconnected");
 
     if (connected.length > 0) {
@@ -140,7 +141,10 @@ export async function setup(): Promise<void> {
         .filter(Boolean)
         .join(", ");
       const info = emails ? `  ${s.dim}${emails}${s.reset}` : "";
-      line(`  ${p.label.padEnd(16)} ${s.green}${connected.length} account(s)${s.reset}${info}`);
+      const disabledInfo = disabled.length > 0 ? `  ${s.red}${disabled.length} disabled${s.reset}` : "";
+      line(`  ${p.label.padEnd(16)} ${s.green}${connected.length} account(s)${s.reset}${info}${disabledInfo}`);
+    } else if (disabled.length > 0) {
+      line(`  ${p.label.padEnd(16)} ${s.red}${disabled.length} disabled${s.reset}`);
     } else if (total.length > 0) {
       line(`  ${p.label.padEnd(16)} ${s.yellow}${total.length} expired${s.reset}`);
     } else {

package/src/cli/status.ts

@@ -1,7 +1,8 @@
 import type { Credentials, ProviderName } from "../auth/store.ts";
 import * as store from "../auth/store.ts";
+import { cooldown, type QuotaPool } from "../routing/cooldown.ts";
 
-export type ConnectionStatus = "connected" | "expired" | "disconnected";
+export type ConnectionStatus = "connected" | "expired" | "disabled" | "disconnected";
 
 export interface AccountStatus {
   account: number;
@@ -23,8 +24,16 @@ const PROVIDERS: { name: ProviderName; label: string; sublabel?: string }[] = [
   { name: "google", label: "Google", sublabel: "Gemini CLI + Antigravity" },
 ];
 
-function connectionOf(creds: Credentials): ConnectionStatus {
+const POOL_MAP: Record<ProviderName, QuotaPool[]> = {
+  anthropic: ["anthropic"],
+  codex: ["codex"],
+  google: ["gemini", "antigravity"],
+};
+
+function connectionOf(name: ProviderName, account: number, creds: Credentials): ConnectionStatus {
   if (!creds.refreshToken) return "disconnected";
+  const pools = POOL_MAP[name];
+  if (pools.some((p) => cooldown.isExhausted(p, account))) return "disabled";
   return store.fresh(creds) ? "connected" : "expired";
 }
 
@@ -35,7 +44,7 @@ export function all(): ProviderStatus[] {
     sublabel,
     accounts: store.getAll(name).map((e) => ({
       account: e.account,
-      status: connectionOf(e.credentials),
+      status: connectionOf(name, e.account, e.credentials),
       email: e.credentials.email,
       expiresAt: e.credentials.expiresAt,
     })),

package/src/cli/tui.ts

@@ -16,6 +16,7 @@ const oauthConfigs: Record<ProviderName, OAuthConfig> = {
 const ICON: Record<ConnectionStatus, string> = {
   connected: `${s.green}●${s.reset}`,
   expired: `${s.yellow}●${s.reset}`,
+  disabled: `${s.red}●${s.reset}`,
   disconnected: `${s.dim}○${s.reset}`,
 };
 
@@ -109,7 +110,9 @@ function formatInfo(a: AccountStatus): string {
   if (a.status === "disconnected") return `${s.dim}—${s.reset}`;
 
   const parts: string[] = [];
-  parts.push(a.status === "connected" ? `${s.green}connected${s.reset}` : `${s.yellow}expired${s.reset}`);
+  if (a.status === "connected") parts.push(`${s.green}connected${s.reset}`);
+  else if (a.status === "disabled") parts.push(`${s.red}disabled${s.reset}`);
+  else parts.push(`${s.yellow}expired${s.reset}`);
   if (a.expiresAt && a.status === "connected") parts.push(`${s.dim}${status.remaining(a.expiresAt)}${s.reset}`);
   if (a.email) parts.push(`${s.dim}${a.email}${s.reset}`);
   return parts.join(`${s.dim} · ${s.reset}`);

package/src/providers/anthropic.ts

@@ -32,6 +32,7 @@ export const provider: Provider = {
       streaming: body.stream,
       providerName: "Anthropic",
       rewrite,
+      email: store.get("anthropic", account)?.email,
       headers: {
         ...stainlessHeaders,
         Accept: body.stream ? "text/event-stream" : "application/json",

package/src/providers/antigravity.ts

@@ -52,7 +52,7 @@ export const provider: Provider = {
 
     const gemini = path.gemini(sub);
     const action = gemini?.action ?? "generateContent";
-    const model = gemini?.model ?? "";
+    const model = gemini?.model === "gemini-3-flash-preview" ? "gemini-3-flash" : (gemini?.model ?? "");
     const requestBody = maybeWrap(body.parsed, body.forwardBody, projectId, model, {
       userAgent: "antigravity",
       requestIdPrefix: "agent",
@@ -60,7 +60,7 @@ export const provider: Provider = {
     });
     const unwrapThenRewrite = withUnwrap(rewrite);
 
-    return tryEndpoints(requestBody, body.stream, headers, action, unwrapThenRewrite);
+    return tryEndpoints(requestBody, body.stream, headers, action, unwrapThenRewrite, creds?.email);
   },
 };
 
@@ -70,13 +70,14 @@ async function tryEndpoints(
   headers: Record<string, string>,
   action: string,
   rewrite?: (data: string) => string,
+  email?: string,
 ): Promise<Response> {
   let lastError: Error | null = null;
 
   for (const endpoint of endpoints) {
     const url = buildUrl(endpoint, action);
     try {
-      const response = await forward({ url, body, streaming, headers, providerName: "Antigravity", rewrite });
+      const response = await forward({ url, body, streaming, headers, providerName: "Antigravity", rewrite, email });
       if (response.status < 500) return response;
       lastError = new Error(`${endpoint} returned ${response.status}`);
       logger.debug("Endpoint 5xx, trying next", { provider: "Antigravity" });

package/src/providers/base.ts

@@ -26,6 +26,7 @@ interface ForwardOptions {
   headers: Record<string, string>;
   providerName: string;
   rewrite?: (data: string) => string;
+  email?: string;
 }
 
 const RETRYABLE_STATUS = new Set([408, 500, 502, 503, 504]);
@@ -64,7 +65,8 @@ export async function forward(opts: ForwardOptions): Promise<Response> {
 
     if (!response.ok) {
       const text = await response.text();
-      logger.error(`${opts.providerName} API error (${response.status})`, { error: text.slice(0, 200) });
+      const ctx = opts.email ? ` account=${opts.email}` : "";
+      logger.error(`${opts.providerName} API error (${response.status})${ctx}`, { error: text.slice(0, 200) });
       return new Response(text, { status: response.status, headers: { "Content-Type": contentType } });
     }
 

package/src/providers/codex.ts

@@ -41,6 +41,7 @@ export const provider: Provider = {
       providerName: "OpenAI Codex",
       // Skip generic rewrite when we need full response transform
       rewrite: needsResponseTransform ? undefined : rewrite,
+      email: store.get("codex", account)?.email,
       headers: {
         "Content-Type": "application/json",
         Authorization: `Bearer ${accessToken}`,
@@ -149,6 +150,7 @@ function transformForCodex(
   // Remove fields the Codex backend doesn't accept
   delete parsed.max_tokens;
   delete parsed.max_completion_tokens;
+  delete parsed.max_output_tokens;
   // Chat Completions fields not in Responses API
   delete parsed.frequency_penalty;
   delete parsed.logprobs;

package/src/providers/gemini.ts

@@ -67,6 +67,7 @@ export const provider: Provider = {
       headers,
       providerName: "Gemini CLI",
       rewrite: unwrapThenRewrite,
+      email: store.get("google", account)?.email,
     });
   },
 };

package/src/routing/cooldown.ts

@@ -12,6 +12,8 @@ interface CooldownEntry {
 }
 
 /** When detected as exhausted, cooldown for this long. */
+/** 403 = account disabled/revoked — long cooldown. */
+const FORBIDDEN_COOLDOWN_MS = 24 * 3600_000;
 const EXHAUSTED_COOLDOWN_MS = 2 * 3600_000;
 /** Retry-After threshold (seconds) above which we consider quota exhausted. */
 const EXHAUSTED_THRESHOLD_S = 300;
@@ -65,6 +67,13 @@ export class CooldownTracker {
     this.entries.set(k, entry);
   }
 
+  /** 403 = account forbidden/revoked. Immediately disable for 24h. */
+  record403(pool: QuotaPool, account: number): void {
+    const k = this.key(pool, account);
+    this.entries.set(k, { until: Date.now() + FORBIDDEN_COOLDOWN_MS, exhausted: true, consecutive429: 0 });
+    logger.warn(`Account disabled (403): ${k}`, { cooldownHours: FORBIDDEN_COOLDOWN_MS / 3600_000 });
+  }
+
   recordSuccess(pool: QuotaPool, account: number): void {
     this.entries.delete(this.key(pool, account));
   }

package/src/server/server.ts

@@ -5,6 +5,7 @@ import type { ProxyConfig } from "../config/config.ts";
 import * as rewriter from "../proxy/rewriter.ts";
 import * as upstream from "../proxy/upstream.ts";
 import { affinity } from "../routing/affinity.ts";
+import { cooldown } from "../routing/cooldown.ts";
 import { tryReroute, tryWithCachePreserve } from "../routing/retry.ts";
 import { recordSuccess, routeRequest } from "../routing/router.ts";
 import { handleInternal, isLocalMethod } from "../tools/internal.ts";
@@ -31,7 +32,7 @@ export function startServer(config: ProxyConfig): ReturnType<typeof Bun.serve> {
         logger.error("Unhandled server error", { error: String(err) });
         return Response.json({ error: "Internal proxy error" }, { status });
       } finally {
-        logger.info(`${req.method} ${url.pathname} ${status}`, { duration: Date.now() - startTime });
+        logger.info(`${req.method} ${url.pathname}${url.search} ${status}`, { duration: Date.now() - startTime });
       }
     },
   });
@@ -120,6 +121,10 @@ async function handleProvider(
         );
         response = rerouted ?? (await fallbackUpstream(req, body, config));
       }
+    } else if (handlerResponse.status === 403 && route.pool) {
+      cooldown.record403(route.pool, route.account);
+      if (threadId) affinity.clear(threadId, providerName);
+      response = await fallbackUpstream(req, body, config);
     } else if (handlerResponse.status === 401) {
       logger.debug("Local provider denied, falling back to upstream");
       response = await fallbackUpstream(req, body, config);