@prometheus-ai/ai 0.5.4 → 0.5.8

package/src/providers/vision-guard.ts

@@ -1,3 +1,6 @@
+import { isDashscopeCompatibleModeUrl } from "@prometheus-ai/catalog/hosts";
+import { isQwenModelId } from "@prometheus-ai/catalog/identity";
+
 import type { ImageContent, Model, TextContent } from "../types";
 
 export const NON_VISION_IMAGE_PLACEHOLDER = "[image omitted: model does not support vision]";
@@ -42,11 +45,10 @@ export function joinTextWithImagePlaceholder(text: string, omittedImages: boolea
  * provider (issue #1859) can't drive the request into an unrecoverable 400.
  */
 export function isDashscopeCompatibleModeTextOnlyQwen(model: Model<"openai-completions">): boolean {
-	const baseUrl = model.baseUrl.toLowerCase();
-	if (!baseUrl.includes("dashscope") || !baseUrl.includes("aliyuncs.com") || !baseUrl.includes("/compatible-mode")) {
+	if (!isDashscopeCompatibleModeUrl(model.baseUrl)) {
 		return false;
 	}
 	const id = model.id.toLowerCase();
-	if (!id.includes("qwen")) return false;
+	if (!isQwenModelId(model.id)) return false;
 	return /\bqwen(?:[\d.]+)?-max\b/.test(id) || /\bqwen(?:[\d.]+)?-coder\b/.test(id);
 }

package/src/rate-limit-utils.ts

@@ -21,14 +21,24 @@ const ACCOUNT_RATE_LIMIT_PATTERN =
 
 /**
  * Classify a rate-limit error message into a reason category.
- * Priority order: MODEL_CAPACITY > RATE_LIMIT > QUOTA > SERVER_ERROR > UNKNOWN.
+ * Priority order: QUOTA (Antigravity "quota will reset") > MODEL_CAPACITY > QUOTA (account) >
+ * RATE_LIMIT > QUOTA (generic) > SERVER_ERROR > UNKNOWN.
  *
  * "resource exhausted" maps to MODEL_CAPACITY (transient, short wait)
- * "quota exceeded" maps to QUOTA_EXHAUSTED (long wait, switch account)
+ * "quota exceeded" / "quota will reset" maps to QUOTA_EXHAUSTED (long wait, switch account)
  */
 export function parseRateLimitReason(errorMessage: string): RateLimitReason {
 	const lower = errorMessage.toLowerCase();
 
+	// Antigravity / Cloud Code Assist surface multi-hour daily-quota exhaustion as
+	// "You have exhausted your capacity on this model. Your quota will reset after …".
+	// The literal "capacity" used to pre-empt the QUOTA branch even though "quota
+	// will reset" is the long-wait signal — short-circuit here before the
+	// MODEL_CAPACITY fallthrough so credential rotation (not 60s backoff) kicks in.
+	if (lower.includes("quota will reset") || lower.includes("exhausted your capacity")) {
+		return "QUOTA_EXHAUSTED";
+	}
+
 	if (
 		lower.includes("capacity") ||
 		lower.includes("overloaded") ||
@@ -84,7 +94,7 @@ export function calculateRateLimitBackoffMs(reason: RateLimitReason): number {
 
 /** Detect usage/quota limit errors in error messages (persistent, requires credential switch). */
 const USAGE_LIMIT_PATTERN =
-	/usage.?limit|usage_limit_reached|usage_not_included|limit_reached|quota.?exceeded|resource.?exhausted/i;
+	/usage.?limit|usage_limit_reached|usage_not_included|limit_reached|quota.?exceeded|quota.?reached|resource.?exhausted|exhausted your capacity|quota will reset/i;
 
 export function isUsageLimitError(errorMessage: string): boolean {
 	return USAGE_LIMIT_PATTERN.test(errorMessage) || ACCOUNT_RATE_LIMIT_PATTERN.test(errorMessage);

package/src/registry/aimlapi.ts

@@ -0,0 +1,6 @@
+import type { ProviderDefinition } from "./types";
+
+export const aimlApiProvider = {
+	id: "aimlapi",
+	name: "AIML API",
+} as const satisfies ProviderDefinition;

package/src/{utils/oauth → registry}/alibaba-coding-plan.ts

@@ -1,27 +1,11 @@
-/**
- * Alibaba Coding Plan login flow.
- *
- * Alibaba Coding Plan provides OpenAI-compatible models via https://coding-intl.dashscope.aliyuncs.com/v1.
- *
- * This is not OAuth - it's a simple API key flow:
- * 1. Open browser to Alibaba Cloud DashScope API key settings
- * 2. User copies their API key
- * 3. User pastes the API key into the CLI
- */
-
 import { validateOpenAICompatibleApiKey } from "./api-key-validation";
-import type { OAuthController } from "./types";
+import type { OAuthController, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
 
 const AUTH_URL = "https://modelstudio.console.alibabacloud.com/";
 const API_BASE_URL = "https://coding-intl.dashscope.aliyuncs.com/v1";
 const VALIDATION_MODEL = "qwen3.5-plus";
 
-/**
- * Login to Alibaba Coding Plan.
- *
- * Opens browser to API keys page, prompts user to paste their API key.
- * Returns the API key directly (not OAuthCredentials - this isn't OAuth).
- */
 export async function loginAlibabaCodingPlan(options: OAuthController): Promise<string> {
 	if (!options.onPrompt) {
 		throw new Error("Alibaba Coding Plan login requires onPrompt callback");
@@ -57,3 +41,9 @@ export async function loginAlibabaCodingPlan(options: OAuthController): Promise<
 
 	return trimmed;
 }
+
+export const alibabaCodingPlanProvider = {
+	id: "alibaba-coding-plan",
+	name: "Alibaba Coding Plan",
+	login: (cb: OAuthLoginCallbacks) => loginAlibabaCodingPlan(cb),
+} as const satisfies ProviderDefinition;

package/src/registry/amazon-bedrock.ts

@@ -0,0 +1,22 @@
+import { $env } from "@prometheus-ai/utils";
+import type { ProviderDefinition } from "./types";
+
+export const amazonBedrockProvider = {
+	id: "amazon-bedrock",
+	name: "Amazon Bedrock",
+	// Amazon Bedrock accepts bearer tokens, IAM keys, profiles, ECS/IRSA credential chains.
+	envKeys: () => {
+		const hasEcsCredentials =
+			!!$env.AWS_CONTAINER_CREDENTIALS_RELATIVE_URI || !!$env.AWS_CONTAINER_CREDENTIALS_FULL_URI;
+		const hasWebIdentity = !!$env.AWS_WEB_IDENTITY_TOKEN_FILE && !!$env.AWS_ROLE_ARN;
+		if (
+			$env.AWS_PROFILE ||
+			($env.AWS_ACCESS_KEY_ID && $env.AWS_SECRET_ACCESS_KEY) ||
+			$env.AWS_BEARER_TOKEN_BEDROCK ||
+			hasEcsCredentials ||
+			hasWebIdentity
+		) {
+			return "<authenticated>";
+		}
+	},
+} as const satisfies ProviderDefinition;

package/src/registry/anthropic.ts

@@ -0,0 +1,26 @@
+import { $pickenv } from "@prometheus-ai/utils";
+import { isFoundryEnabled } from "../utils/foundry";
+import type { OAuthCredentials, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
+export const anthropicProvider = {
+	id: "anthropic",
+	name: "Anthropic (Claude Pro/Max)",
+	// Foundry mode optionally switches Anthropic auth to enterprise gateway credentials.
+	envKeys: () =>
+		isFoundryEnabled()
+			? $pickenv("ANTHROPIC_FOUNDRY_API_KEY", "ANTHROPIC_OAUTH_TOKEN", "ANTHROPIC_API_KEY")
+			: $pickenv("ANTHROPIC_OAUTH_TOKEN", "ANTHROPIC_API_KEY"),
+	login: async (cb: OAuthLoginCallbacks) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { loginAnthropic } = await import("./oauth/anthropic");
+		return loginAnthropic(cb);
+	},
+	refreshToken: async (credentials: OAuthCredentials) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { refreshAnthropicToken } = await import("./oauth/anthropic");
+		return refreshAnthropicToken(credentials.refresh);
+	},
+	callbackPort: 54545,
+	pasteCodeFlow: true,
+} as const satisfies ProviderDefinition;

package/src/{utils/oauth → registry}/api-key-login.ts

@@ -6,8 +6,12 @@
  * optionally validate it, and return the trimmed key.
  */
 
-import { validateApiKeyAgainstModelsEndpoint, validateOpenAICompatibleApiKey } from "./api-key-validation";
-import type { OAuthController } from "./types";
+import {
+	validateAnthropicCompatibleApiKey,
+	validateApiKeyAgainstModelsEndpoint,
+	validateOpenAICompatibleApiKey,
+} from "./api-key-validation";
+import type { OAuthController } from "./oauth/types";
 
 type ChatCompletionsValidation = {
 	kind: "chat-completions";
@@ -16,6 +20,13 @@ type ChatCompletionsValidation = {
 	model: string;
 };
 
+type AnthropicMessagesValidation = {
+	kind: "anthropic-messages";
+	provider: string;
+	baseUrl: string;
+	model: string;
+};
+
 type ModelsEndpointValidation = {
 	kind: "models-endpoint";
 	provider: string;
@@ -34,7 +45,7 @@ export type ApiKeyLoginConfig = {
 	/** Placeholder string for the prompt (e.g. "sk-...", "csk-..."). */
 	placeholder: string;
 	/** Validation strategy, or `null` to skip validation. */
-	validation: ChatCompletionsValidation | ModelsEndpointValidation | null;
+	validation: ChatCompletionsValidation | AnthropicMessagesValidation | ModelsEndpointValidation | null;
 };
 
 export function createApiKeyLogin(config: ApiKeyLoginConfig): (options: OAuthController) => Promise<string> {
@@ -71,6 +82,16 @@ export function createApiKeyLogin(config: ApiKeyLoginConfig): (options: OAuthCon
 					baseUrl: config.validation.baseUrl,
 					model: config.validation.model,
 					signal: options.signal,
+					fetch: options.fetch,
+				});
+			} else if (config.validation.kind === "anthropic-messages") {
+				await validateAnthropicCompatibleApiKey({
+					provider: config.validation.provider,
+					apiKey: trimmed,
+					baseUrl: config.validation.baseUrl,
+					model: config.validation.model,
+					signal: options.signal,
+					fetch: options.fetch,
 				});
 			} else {
 				await validateApiKeyAgainstModelsEndpoint({
@@ -78,6 +99,7 @@ export function createApiKeyLogin(config: ApiKeyLoginConfig): (options: OAuthCon
 					apiKey: trimmed,
 					modelsUrl: config.validation.modelsUrl,
 					signal: options.signal,
+					fetch: options.fetch,
 				});
 			}
 		}

package/src/{utils/oauth → registry}/api-key-validation.ts

@@ -1,9 +1,20 @@
+import type { FetchImpl } from "../types";
+
 type OpenAICompatibleValidationOptions = {
 	provider: string;
 	apiKey: string;
 	baseUrl: string;
 	model: string;
 	signal?: AbortSignal;
+	fetch?: FetchImpl;
+};
+type AnthropicCompatibleValidationOptions = {
+	provider: string;
+	apiKey: string;
+	baseUrl: string;
+	model: string;
+	signal?: AbortSignal;
+	fetch?: FetchImpl;
 };
 
 type ModelListValidationOptions = {
@@ -11,10 +22,16 @@ type ModelListValidationOptions = {
 	apiKey: string;
 	modelsUrl: string;
 	signal?: AbortSignal;
+	fetch?: FetchImpl;
 };
 
 const VALIDATION_TIMEOUT_MS = 15_000;
 
+function normalizeAnthropicCompatibleBaseUrl(baseUrl: string): string {
+	const trimmed = baseUrl.trim().replace(/\/+$/, "");
+	return trimmed.endsWith("/v1") ? trimmed.slice(0, -3) : trimmed;
+}
+
 /**
  * Validate an API key against an OpenAI-compatible chat completions endpoint.
  *
@@ -23,8 +40,9 @@ const VALIDATION_TIMEOUT_MS = 15_000;
 export async function validateOpenAICompatibleApiKey(options: OpenAICompatibleValidationOptions): Promise<void> {
 	const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
 	const signal = options.signal ? AbortSignal.any([options.signal, timeoutSignal]) : timeoutSignal;
+	const fetchImpl = options.fetch ?? fetch;
 
-	const response = await fetch(`${options.baseUrl}/chat/completions`, {
+	const response = await fetchImpl(`${options.baseUrl}/chat/completions`, {
 		method: "POST",
 		headers: {
 			"Content-Type": "application/json",
@@ -56,6 +74,47 @@ export async function validateOpenAICompatibleApiKey(options: OpenAICompatibleVa
 	throw new Error(message);
 }
 
+/**
+ * Validate an API key against an Anthropic-compatible messages endpoint.
+ */
+export async function validateAnthropicCompatibleApiKey(options: AnthropicCompatibleValidationOptions): Promise<void> {
+	const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
+	const signal = options.signal ? AbortSignal.any([options.signal, timeoutSignal]) : timeoutSignal;
+	const baseUrl = normalizeAnthropicCompatibleBaseUrl(options.baseUrl);
+	const fetchImpl = options.fetch ?? fetch;
+
+	const response = await fetchImpl(`${baseUrl}/v1/messages`, {
+		method: "POST",
+		headers: {
+			"Content-Type": "application/json",
+			"anthropic-version": "2023-06-01",
+			"x-api-key": options.apiKey,
+		},
+		body: JSON.stringify({
+			model: options.model,
+			messages: [{ role: "user", content: "ping" }],
+			max_tokens: 1,
+		}),
+		signal,
+	});
+
+	if (response.ok) {
+		return;
+	}
+
+	let details = "";
+	try {
+		details = (await response.text()).trim();
+	} catch {
+		// ignore body parse errors, status is enough
+	}
+
+	const message = details
+		? `${options.provider} API key validation failed (${response.status}): ${details}`
+		: `${options.provider} API key validation failed (${response.status})`;
+	throw new Error(message);
+}
+
 /**
  * Validate an API key against a provider models endpoint.
  *
@@ -65,8 +124,9 @@ export async function validateOpenAICompatibleApiKey(options: OpenAICompatibleVa
 export async function validateApiKeyAgainstModelsEndpoint(options: ModelListValidationOptions): Promise<void> {
 	const timeoutSignal = AbortSignal.timeout(VALIDATION_TIMEOUT_MS);
 	const signal = options.signal ? AbortSignal.any([options.signal, timeoutSignal]) : timeoutSignal;
+	const fetchImpl = options.fetch ?? fetch;
 
-	const response = await fetch(options.modelsUrl, {
+	const response = await fetchImpl(options.modelsUrl, {
 		method: "GET",
 		headers: {
 			Authorization: `Bearer ${options.apiKey}`,

package/src/{utils/oauth → registry}/cerebras.ts

@@ -1,5 +1,6 @@
-/** Cerebras login flow (API key paste against https://api.cerebras.ai/v1). */
 import { createApiKeyLogin } from "./api-key-login";
+import type { OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
 
 export const loginCerebras = createApiKeyLogin({
 	providerLabel: "Cerebras",
@@ -14,3 +15,9 @@ export const loginCerebras = createApiKeyLogin({
 		model: "gpt-oss-120b",
 	},
 });
+
+export const cerebrasProvider = {
+	id: "cerebras",
+	name: "Cerebras",
+	login: (cb: OAuthLoginCallbacks) => loginCerebras(cb),
+} as const satisfies ProviderDefinition;

package/src/{utils/oauth → registry}/cloudflare-ai-gateway.ts

@@ -1,15 +1,5 @@
-/**
- * Cloudflare AI Gateway login flow.
- *
- * Cloudflare AI Gateway proxies upstream model providers.
- *
- * This is not OAuth - it's a simple API key flow:
- * 1. Open Cloudflare AI Gateway docs/dashboard
- * 2. User copies their Cloudflare AI Gateway token/API key
- * 3. User pastes the API key into the CLI
- */
-
-import type { OAuthController } from "./types";
+import type { OAuthController, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
 
 const AUTH_URL = "https://developers.cloudflare.com/ai-gateway/configuration/authentication/";
 
@@ -46,3 +36,9 @@ export async function loginCloudflareAiGateway(options: OAuthController): Promis
 
 	return trimmed;
 }
+
+export const cloudflareAiGatewayProvider = {
+	id: "cloudflare-ai-gateway",
+	name: "Cloudflare AI Gateway",
+	login: (cb: OAuthLoginCallbacks) => loginCloudflareAiGateway(cb),
+} as const satisfies ProviderDefinition;

package/src/registry/cursor.ts

@@ -0,0 +1,20 @@
+import type { OAuthCredentials, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
+export const cursorProvider = {
+	id: "cursor",
+	name: "Cursor (Claude, GPT, etc.)",
+	login: async (cb: OAuthLoginCallbacks) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { loginCursor } = await import("./oauth/cursor");
+		return loginCursor(
+			url => cb.onAuth({ url }),
+			cb.onProgress ? () => cb.onProgress?.("Waiting for browser authentication...") : undefined,
+		);
+	},
+	refreshToken: async (credentials: OAuthCredentials) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { refreshCursorToken } = await import("./oauth/cursor");
+		return refreshCursorToken(credentials.refresh);
+	},
+} as const satisfies ProviderDefinition;

package/src/{utils/oauth → registry}/deepseek.ts

@@ -1,13 +1,6 @@
-/**
- * DeepSeek login flow (API key paste against https://api.deepseek.com).
- *
- * Validation hits `GET /v1/models` so it authenticates the key without
- * depending on a specific model being enabled on the account. The previous
- * implementation issued a chat-completion against `deepseek-v4-pro`, which
- * 404s for accounts without that preview model even when the key is valid.
- */
 import { createApiKeyLogin } from "./api-key-login";
-import type { OAuthController, OAuthPrompt } from "./types";
+import type { OAuthController, OAuthLoginCallbacks, OAuthPrompt } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
 
 const innerLogin = createApiKeyLogin({
 	providerLabel: "DeepSeek",
@@ -22,17 +15,10 @@ const innerLogin = createApiKeyLogin({
 	},
 });
 
-/**
- * Normalize a pasted DeepSeek API key.
- *
- * Users frequently copy keys out of `curl` snippets that include the
- * `Authorization: Bearer …` prefix. Strip it so validation does not fail
- * with a confusing 401, and reject obviously empty input early.
- */
 export function normalizeDeepSeekApiKey(raw: string): string {
 	const trimmed = raw.trim();
 	if (!trimmed) {
-		return trimmed; // let the shared factory throw the canonical "API key is required"
+		return trimmed;
 	}
 	const stripped = trimmed.replace(/^bearer\b\s*/i, "");
 	if (!stripped) {
@@ -51,3 +37,9 @@ export const loginDeepSeek = async (options: OAuthController): Promise<string> =
 		: options;
 	return innerLogin(wrapped);
 };
+
+export const deepseekProvider = {
+	id: "deepseek",
+	name: "DeepSeek",
+	login: (cb: OAuthLoginCallbacks) => loginDeepSeek(cb),
+} as const satisfies ProviderDefinition;

package/src/registry/derived.ts

@@ -0,0 +1,9 @@
+import { PROVIDER_REGISTRY } from "./registry";
+
+/**
+ * Providers whose OAuth flow needs a pasted code/redirect URL rather than a
+ * local callback server. Consumed by the coding-agent login UX.
+ */
+export const PASTE_CODE_LOGIN_PROVIDERS: ReadonlySet<string> = new Set(
+	PROVIDER_REGISTRY.filter(p => p.pasteCodeFlow).map(p => p.id),
+);

package/src/{utils/oauth → registry}/firepass.ts

@@ -1,3 +1,7 @@
+import { createApiKeyLogin } from "./api-key-login";
+import type { OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
 /**
  * Fire Pass login flow.
  *
@@ -7,8 +11,6 @@
  * pings the chat completions endpoint with the router id directly.
  * See https://docs.fireworks.ai/firepass.
  */
-import { createApiKeyLogin } from "./api-key-login";
-
 export const loginFirepass = createApiKeyLogin({
 	providerLabel: "Fire Pass",
 	authUrl: "https://app.fireworks.ai/settings/users/api-keys",
@@ -22,3 +24,9 @@ export const loginFirepass = createApiKeyLogin({
 		model: "accounts/fireworks/routers/kimi-k2p6-turbo",
 	},
 });
+
+export const firepassProvider = {
+	id: "firepass",
+	name: "Fire Pass (Fireworks Kimi K2.6 Turbo subscription)",
+	login: (cb: OAuthLoginCallbacks) => loginFirepass(cb),
+} as const satisfies ProviderDefinition;

package/src/{utils/oauth → registry}/fireworks.ts

@@ -1,5 +1,6 @@
-/** Fireworks login flow (API key paste against https://api.fireworks.ai/inference/v1). */
 import { createApiKeyLogin } from "./api-key-login";
+import type { OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
 
 export const loginFireworks = createApiKeyLogin({
 	providerLabel: "Fireworks",
@@ -13,3 +14,9 @@ export const loginFireworks = createApiKeyLogin({
 		modelsUrl: "https://api.fireworks.ai/inference/v1/models",
 	},
 });
+
+export const fireworksProvider = {
+	id: "fireworks",
+	name: "Fireworks",
+	login: (cb: OAuthLoginCallbacks) => loginFireworks(cb),
+} as const satisfies ProviderDefinition;

package/src/registry/github-copilot.ts

@@ -0,0 +1,22 @@
+import type { OAuthCredentials, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
+export const githubCopilotProvider = {
+	id: "github-copilot",
+	name: "GitHub Copilot",
+	login: async (cb: OAuthLoginCallbacks) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { loginGitHubCopilot } = await import("./oauth/github-copilot");
+		return loginGitHubCopilot({
+			onAuth: (url, instructions) => cb.onAuth({ url, instructions }),
+			onPrompt: cb.onPrompt,
+			onProgress: cb.onProgress,
+			signal: cb.signal,
+		});
+	},
+	refreshToken: async (credentials: OAuthCredentials) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { refreshGitHubCopilotToken } = await import("./oauth/github-copilot");
+		return refreshGitHubCopilotToken(credentials.refresh, credentials.enterpriseUrl);
+	},
+} as const satisfies ProviderDefinition;

package/src/registry/gitlab-duo.ts

@@ -0,0 +1,19 @@
+import type { OAuthCredentials, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
+export const gitlabDuoProvider = {
+	id: "gitlab-duo",
+	name: "GitLab Duo",
+	login: async (cb: OAuthLoginCallbacks) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { loginGitLabDuo } = await import("./oauth/gitlab-duo");
+		return loginGitLabDuo(cb);
+	},
+	refreshToken: async (credentials: OAuthCredentials) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { refreshGitLabDuoToken } = await import("./oauth/gitlab-duo");
+		return refreshGitLabDuoToken(credentials);
+	},
+	callbackPort: 8080,
+	pasteCodeFlow: true,
+} as const satisfies ProviderDefinition;

package/src/registry/google-antigravity.ts

@@ -0,0 +1,21 @@
+import type { OAuthCredentials, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
+export const googleAntigravityProvider = {
+	id: "google-antigravity",
+	name: "Antigravity (Gemini 3, Claude, GPT-OSS)",
+	login: async (cb: OAuthLoginCallbacks) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { loginAntigravity } = await import("./oauth/google-antigravity");
+		return loginAntigravity(cb);
+	},
+	refreshToken: async (credentials: OAuthCredentials) => {
+		if (!credentials.projectId) {
+			throw new Error("Antigravity credentials missing projectId");
+		}
+		const { refreshAntigravityToken } = await import("./oauth/google-antigravity");
+		return refreshAntigravityToken(credentials.refresh, credentials.projectId);
+	},
+	callbackPort: 51121,
+	pasteCodeFlow: true,
+} as const satisfies ProviderDefinition;

package/src/registry/google-gemini-cli.ts

@@ -0,0 +1,21 @@
+import type { OAuthCredentials, OAuthLoginCallbacks } from "./oauth/types";
+import type { ProviderDefinition } from "./types";
+
+export const googleGeminiCliProvider = {
+	id: "google-gemini-cli",
+	name: "Google Cloud Code Assist (Gemini CLI)",
+	login: async (cb: OAuthLoginCallbacks) => {
+		// Lazy import: keep heavy OAuth flow modules out of the eager registry graph.
+		const { loginGeminiCli } = await import("./oauth/google-gemini-cli");
+		return loginGeminiCli(cb);
+	},
+	refreshToken: async (credentials: OAuthCredentials) => {
+		if (!credentials.projectId) {
+			throw new Error("Google Cloud credentials missing projectId");
+		}
+		const { refreshGoogleCloudToken } = await import("./oauth/google-gemini-cli");
+		return refreshGoogleCloudToken(credentials.refresh, credentials.projectId);
+	},
+	callbackPort: 8085,
+	pasteCodeFlow: true,
+} as const satisfies ProviderDefinition;

package/src/registry/google-vertex.ts

@@ -0,0 +1,38 @@
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import { $env } from "@prometheus-ai/utils";
+import type { ProviderDefinition } from "./types";
+
+let cachedVertexAdcCredentialsExists: boolean | null = null;
+
+function hasVertexAdcCredentials(): boolean {
+	if (cachedVertexAdcCredentialsExists === null) {
+		const gacPath = $env.GOOGLE_APPLICATION_CREDENTIALS;
+		if (gacPath) {
+			cachedVertexAdcCredentialsExists = fs.existsSync(gacPath);
+		} else {
+			cachedVertexAdcCredentialsExists = fs.existsSync(
+				path.join(os.homedir(), ".config", "gcloud", "application_default_credentials.json"),
+			);
+		}
+	}
+	return cachedVertexAdcCredentialsExists;
+}
+
+export const googleVertexProvider = {
+	id: "google-vertex",
+	name: "Google Vertex AI",
+	// Vertex AI supports either GOOGLE_CLOUD_API_KEY or Application Default Credentials.
+	envKeys: () => {
+		if ($env.GOOGLE_CLOUD_API_KEY) {
+			return $env.GOOGLE_CLOUD_API_KEY;
+		}
+		const hasCredentials = hasVertexAdcCredentials();
+		const hasProject = !!($env.GOOGLE_CLOUD_PROJECT || $env.GCP_PROJECT || $env.GCLOUD_PROJECT);
+		const hasLocation = !!($env.GOOGLE_VERTEX_LOCATION || $env.GOOGLE_CLOUD_LOCATION || $env.VERTEX_LOCATION);
+		if (hasCredentials && hasProject && hasLocation) {
+			return "<authenticated>";
+		}
+	},
+} as const satisfies ProviderDefinition;

package/src/registry/google.ts

@@ -0,0 +1,6 @@
+import type { ProviderDefinition } from "./types";
+
+export const googleProvider = {
+	id: "google",
+	name: "Google Gemini",
+} as const satisfies ProviderDefinition;

package/src/registry/groq.ts

@@ -0,0 +1,6 @@
+import type { ProviderDefinition } from "./types";
+
+export const groqProvider = {
+	id: "groq",
+	name: "Groq",
+} as const satisfies ProviderDefinition;