copilot-api-plus 1.0.17 → 1.0.19
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/{auth-D57ZBKnz.js → auth-Ijg_vdBM.js} +1 -1
- package/dist/{auth-Cua-c0dq.js → auth-OZNLGzjK.js} +9 -5
- package/dist/auth-OZNLGzjK.js.map +1 -0
- package/dist/{auth-BO-mwvoU.js → auth-wHA1Oitm.js} +1 -1
- package/dist/{auth-CfSlMdNh.js → auth-y23hLerx.js} +9 -5
- package/dist/auth-y23hLerx.js.map +1 -0
- package/dist/get-models-DqEAMDmN.js +5 -0
- package/dist/{get-models-Bcu-Kwi6.js → get-models-uSazerPk.js} +32 -4
- package/dist/get-models-uSazerPk.js.map +1 -0
- package/dist/main.js +25 -13
- package/dist/main.js.map +1 -1
- package/package.json +1 -1
- package/dist/auth-CfSlMdNh.js.map +0 -1
- package/dist/auth-Cua-c0dq.js.map +0 -1
- package/dist/get-models-Bcu-Kwi6.js.map +0 -1
- package/dist/get-models-CtmoN5yg.js +0 -5
|
@@ -1,4 +1,4 @@
|
|
|
1
1
|
import "./paths-Ch0ixSo2.js";
|
|
2
|
-
import { addAntigravityAccount, clearAntigravityAuth, disableCurrentAccount, exchangeCodeForTokens, generateRandomProjectId, getAntigravityAuthPath, getApiKey, getCurrentAccount, getOAuthUrl, getValidAccessToken, hasApiKey, isTokenExpired, loadAntigravityAuth, refreshAccessToken, rotateAccount, saveAntigravityAuth, setOAuthCredentials, setupAntigravity, setupAntigravityManual, setupAntigravityWeb } from "./auth-
|
|
2
|
+
import { addAntigravityAccount, clearAntigravityAuth, disableCurrentAccount, exchangeCodeForTokens, generateRandomProjectId, getAntigravityAuthPath, getApiKey, getCurrentAccount, getOAuthUrl, getValidAccessToken, hasApiKey, isTokenExpired, loadAntigravityAuth, refreshAccessToken, rotateAccount, saveAntigravityAuth, setOAuthCredentials, setupAntigravity, setupAntigravityManual, setupAntigravityWeb } from "./auth-y23hLerx.js";
|
|
3
3
|
|
|
4
4
|
export { getApiKey, getCurrentAccount, hasApiKey, loadAntigravityAuth, setOAuthCredentials, setupAntigravity };
|
|
@@ -15,7 +15,7 @@ function getZenAuthPath() {
|
|
|
15
15
|
async function saveZenAuth(auth) {
|
|
16
16
|
await ensurePaths();
|
|
17
17
|
const authPath = getZenAuthPath();
|
|
18
|
-
await
|
|
18
|
+
await (await import("node:fs/promises")).writeFile(authPath, JSON.stringify(auth, null, 2), "utf-8");
|
|
19
19
|
consola.success("Zen API key saved to", authPath);
|
|
20
20
|
}
|
|
21
21
|
/**
|
|
@@ -24,9 +24,13 @@ async function saveZenAuth(auth) {
|
|
|
24
24
|
async function loadZenAuth() {
|
|
25
25
|
try {
|
|
26
26
|
const authPath = getZenAuthPath();
|
|
27
|
-
const
|
|
28
|
-
|
|
29
|
-
|
|
27
|
+
const fs = await import("node:fs/promises");
|
|
28
|
+
try {
|
|
29
|
+
await fs.access(authPath);
|
|
30
|
+
} catch {
|
|
31
|
+
return null;
|
|
32
|
+
}
|
|
33
|
+
const content = await fs.readFile(authPath, "utf-8");
|
|
30
34
|
return JSON.parse(content);
|
|
31
35
|
} catch {
|
|
32
36
|
return null;
|
|
@@ -70,4 +74,4 @@ async function setupZenApiKey(force = false) {
|
|
|
70
74
|
|
|
71
75
|
//#endregion
|
|
72
76
|
export { clearZenAuth, getZenAuthPath, loadZenAuth, saveZenAuth, setupZenApiKey };
|
|
73
|
-
//# sourceMappingURL=auth-
|
|
77
|
+
//# sourceMappingURL=auth-OZNLGzjK.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"auth-OZNLGzjK.js","names":[],"sources":["../src/services/zen/auth.ts"],"sourcesContent":["/**\n * OpenCode Zen Authentication\n *\n * Handles API key authentication for OpenCode Zen.\n * API keys are created at https://opencode.ai/zen\n */\n\nimport consola from \"consola\"\n\nimport { PATHS, ensurePaths } from \"~/lib/paths\"\n\nexport interface ZenAuth {\n apiKey: string\n}\n\nconst ZEN_AUTH_FILENAME = \"zen-auth.json\"\n\n/**\n * Get the path to the Zen auth file\n */\nexport function getZenAuthPath(): string {\n return `${PATHS.DATA_DIR}/${ZEN_AUTH_FILENAME}`\n}\n\n/**\n * Save Zen API key to file\n */\nexport async function saveZenAuth(auth: ZenAuth): Promise<void> {\n await ensurePaths()\n const authPath = getZenAuthPath()\n const fs = await import(\"node:fs/promises\")\n await fs.writeFile(authPath, JSON.stringify(auth, null, 2), \"utf-8\")\n consola.success(\"Zen API key saved to\", authPath)\n}\n\n/**\n * Load Zen API key from file\n */\nexport async function loadZenAuth(): Promise<ZenAuth | null> {\n try {\n const authPath = getZenAuthPath()\n const fs = await import(\"node:fs/promises\")\n\n // Check if file exists\n try {\n await fs.access(authPath)\n } catch {\n return null\n }\n\n const content = await fs.readFile(authPath, \"utf-8\")\n return JSON.parse(content) as ZenAuth\n } catch {\n return null\n }\n}\n\n/**\n * Clear Zen API key\n */\nexport async function clearZenAuth(): Promise<void> {\n try {\n const authPath = getZenAuthPath()\n const fs = await import(\"node:fs/promises\")\n await fs.unlink(authPath)\n consola.success(\"Zen API key cleared\")\n } catch {\n // File might not exist, ignore\n }\n}\n\n/**\n * Setup Zen API key interactively\n */\nexport async function setupZenApiKey(force = false): Promise<string> {\n const existingAuth = await loadZenAuth()\n\n if (existingAuth && !force) {\n consola.info(\"Using existing Zen API key\")\n return existingAuth.apiKey\n }\n\n consola.info(\"OpenCode Zen gives you access to all the best coding models\")\n consola.info(\"Get your API key at: https://opencode.ai/zen\")\n consola.info(\"\")\n\n const apiKey = await consola.prompt(\"Enter your OpenCode Zen API key:\", {\n type: \"text\",\n })\n\n if (!apiKey || typeof apiKey !== \"string\") {\n throw new Error(\"API key is required\")\n }\n\n // Validate the API key by fetching models\n try {\n const response = await fetch(\"https://opencode.ai/zen/v1/models\", {\n headers: {\n Authorization: `Bearer ${apiKey}`,\n },\n })\n\n if (!response.ok) {\n throw new Error(\n `Invalid API key: ${response.status} ${response.statusText}`,\n )\n }\n\n consola.success(\"API key validated successfully\")\n } catch (error) {\n consola.error(\"Failed to validate API key:\", error)\n throw error\n }\n\n await saveZenAuth({ apiKey })\n return apiKey\n}\n"],"mappings":";;;;AAeA,MAAM,oBAAoB;;;;AAK1B,SAAgB,iBAAyB;AACvC,QAAO,GAAG,MAAM,SAAS,GAAG;;;;;AAM9B,eAAsB,YAAY,MAA8B;AAC9D,OAAM,aAAa;CACnB,MAAM,WAAW,gBAAgB;AAEjC,QADW,MAAM,OAAO,qBACf,UAAU,UAAU,KAAK,UAAU,MAAM,MAAM,EAAE,EAAE,QAAQ;AACpE,SAAQ,QAAQ,wBAAwB,SAAS;;;;;AAMnD,eAAsB,cAAuC;AAC3D,KAAI;EACF,MAAM,WAAW,gBAAgB;EACjC,MAAM,KAAK,MAAM,OAAO;AAGxB,MAAI;AACF,SAAM,GAAG,OAAO,SAAS;UACnB;AACN,UAAO;;EAGT,MAAM,UAAU,MAAM,GAAG,SAAS,UAAU,QAAQ;AACpD,SAAO,KAAK,MAAM,QAAQ;SACpB;AACN,SAAO;;;;;;AAOX,eAAsB,eAA8B;AAClD,KAAI;EACF,MAAM,WAAW,gBAAgB;AAEjC,SADW,MAAM,OAAO,qBACf,OAAO,SAAS;AACzB,UAAQ,QAAQ,sBAAsB;SAChC;;;;;AAQV,eAAsB,eAAe,QAAQ,OAAwB;CACnE,MAAM,eAAe,MAAM,aAAa;AAExC,KAAI,gBAAgB,CAAC,OAAO;AAC1B,UAAQ,KAAK,6BAA6B;AAC1C,SAAO,aAAa;;AAGtB,SAAQ,KAAK,8DAA8D;AAC3E,SAAQ,KAAK,+CAA+C;AAC5D,SAAQ,KAAK,GAAG;CAEhB,MAAM,SAAS,MAAM,QAAQ,OAAO,oCAAoC,EACtE,MAAM,QACP,CAAC;AAEF,KAAI,CAAC,UAAU,OAAO,WAAW,SAC/B,OAAM,IAAI,MAAM,sBAAsB;AAIxC,KAAI;EACF,MAAM,WAAW,MAAM,MAAM,qCAAqC,EAChE,SAAS,EACP,eAAe,UAAU,UAC1B,EACF,CAAC;AAEF,MAAI,CAAC,SAAS,GACZ,OAAM,IAAI,MACR,oBAAoB,SAAS,OAAO,GAAG,SAAS,aACjD;AAGH,UAAQ,QAAQ,iCAAiC;UAC1C,OAAO;AACd,UAAQ,MAAM,+BAA+B,MAAM;AACnD,QAAM;;AAGR,OAAM,YAAY,EAAE,QAAQ,CAAC;AAC7B,QAAO"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
1
|
import "./paths-Ch0ixSo2.js";
|
|
2
|
-
import { clearZenAuth, getZenAuthPath, loadZenAuth, saveZenAuth, setupZenApiKey } from "./auth-
|
|
2
|
+
import { clearZenAuth, getZenAuthPath, loadZenAuth, saveZenAuth, setupZenApiKey } from "./auth-OZNLGzjK.js";
|
|
3
3
|
|
|
4
4
|
export { loadZenAuth, setupZenApiKey };
|
|
@@ -48,7 +48,7 @@ function getAntigravityAuthPath() {
|
|
|
48
48
|
async function saveAntigravityAuth(auth) {
|
|
49
49
|
await ensurePaths();
|
|
50
50
|
const authPath = getAntigravityAuthPath();
|
|
51
|
-
await
|
|
51
|
+
await (await import("node:fs/promises")).writeFile(authPath, JSON.stringify(auth, null, 2), "utf-8");
|
|
52
52
|
consola.success("Antigravity accounts saved to", authPath);
|
|
53
53
|
}
|
|
54
54
|
/**
|
|
@@ -57,9 +57,13 @@ async function saveAntigravityAuth(auth) {
|
|
|
57
57
|
async function loadAntigravityAuth() {
|
|
58
58
|
try {
|
|
59
59
|
const authPath = getAntigravityAuthPath();
|
|
60
|
-
const
|
|
61
|
-
|
|
62
|
-
|
|
60
|
+
const fs = await import("node:fs/promises");
|
|
61
|
+
try {
|
|
62
|
+
await fs.access(authPath);
|
|
63
|
+
} catch {
|
|
64
|
+
return null;
|
|
65
|
+
}
|
|
66
|
+
const content = await fs.readFile(authPath, "utf-8");
|
|
63
67
|
const data = JSON.parse(content);
|
|
64
68
|
if (Array.isArray(data)) return {
|
|
65
69
|
accounts: data,
|
|
@@ -408,4 +412,4 @@ async function setupAntigravity() {
|
|
|
408
412
|
|
|
409
413
|
//#endregion
|
|
410
414
|
export { addAntigravityAccount, clearAntigravityAuth, disableCurrentAccount, exchangeCodeForTokens, generateRandomProjectId, getAntigravityAuthPath, getApiKey, getCurrentAccount, getOAuthUrl, getValidAccessToken, hasApiKey, isTokenExpired, loadAntigravityAuth, refreshAccessToken, rotateAccount, saveAntigravityAuth, setOAuthCredentials, setupAntigravity, setupAntigravityManual, setupAntigravityWeb };
|
|
411
|
-
//# sourceMappingURL=auth-
|
|
415
|
+
//# sourceMappingURL=auth-y23hLerx.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"auth-y23hLerx.js","names":[],"sources":["../src/services/antigravity/auth.ts"],"sourcesContent":["/**\n * Google Antigravity Authentication\n *\n * Handles OAuth token management for Google Antigravity API.\n * Supports multiple accounts with auto-rotation and token refresh.\n */\n\n/* eslint-disable @typescript-eslint/no-unsafe-assignment */\n/* eslint-disable @typescript-eslint/no-unnecessary-condition */\n\nimport consola from \"consola\"\n\nimport { PATHS, ensurePaths } from \"~/lib/paths\"\n\nexport interface AntigravityAccount {\n access_token: string\n refresh_token: string\n expires_in: number\n timestamp: number\n enable: boolean\n project_id?: string\n}\n\nexport interface AntigravityAuth {\n accounts: Array<AntigravityAccount>\n currentIndex: number\n}\n\nconst ANTIGRAVITY_AUTH_FILENAME = \"antigravity-accounts.json\"\n\n// ============================================\n// Authentication Methods (choose one):\n// ============================================\n//\n// Method 1: API Key (Recommended - Simplest)\n// Set environment variable: GEMINI_API_KEY\n// Get your API key from: https://aistudio.google.com/apikey\n//\n// Method 2: OAuth Credentials\n// Set environment variables:\n// ANTIGRAVITY_CLIENT_ID - Google OAuth Client ID\n// ANTIGRAVITY_CLIENT_SECRET - Google OAuth Client Secret\n//\n// If you get \"invalid_client\" error, create your own OAuth app:\n// 1. Go to https://console.cloud.google.com/apis/credentials\n// 2. Create OAuth 2.0 Client ID (Desktop app or Web app type)\n// 3. Add redirect URI: http://localhost:8046/callback\n// 4. Set environment variables with your credentials\n// ============================================\n\n// API Key authentication (simplest method)\nconst GEMINI_API_KEY = process.env.GEMINI_API_KEY || \"\"\n\n/**\n * Check if API Key authentication is available\n */\nexport function hasApiKey(): boolean {\n return GEMINI_API_KEY.length > 0\n}\n\n/**\n * Get API Key for authentication\n */\nexport function getApiKey(): string | null {\n return GEMINI_API_KEY || null\n}\n\n// Default OAuth credentials (from reference projects: gcli2api, antigravity2api-nodejs, Antigravity-Manager)\nconst DEFAULT_CLIENT_ID = \"1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com\"\nconst DEFAULT_CLIENT_SECRET = \"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\"\n\n// OAuth credentials - can be set via env, CLI, or defaults\nlet GOOGLE_CLIENT_ID =\n process.env.ANTIGRAVITY_CLIENT_ID || DEFAULT_CLIENT_ID\nlet GOOGLE_CLIENT_SECRET =\n process.env.ANTIGRAVITY_CLIENT_SECRET || DEFAULT_CLIENT_SECRET\nconst GOOGLE_REDIRECT_URI = \"http://localhost:8046/callback\"\n\n// OAuth scopes required for Antigravity API access\nconst OAUTH_SCOPES = [\n \"https://www.googleapis.com/auth/cloud-platform\",\n \"https://www.googleapis.com/auth/userinfo.email\",\n \"https://www.googleapis.com/auth/userinfo.profile\",\n \"https://www.googleapis.com/auth/cclog\",\n \"https://www.googleapis.com/auth/experimentsandconfigs\",\n]\n\n/**\n * Set OAuth credentials from CLI arguments\n * This overrides environment variables and defaults\n */\nexport function setOAuthCredentials(clientId: string, clientSecret: string): void {\n GOOGLE_CLIENT_ID = clientId\n GOOGLE_CLIENT_SECRET = clientSecret\n}\n\n/**\n * Get the path to the Antigravity auth file\n */\nexport function getAntigravityAuthPath(): string {\n return `${PATHS.DATA_DIR}/${ANTIGRAVITY_AUTH_FILENAME}`\n}\n\n/**\n * Save Antigravity accounts to file\n */\nexport async function saveAntigravityAuth(\n auth: AntigravityAuth,\n): Promise<void> {\n await ensurePaths()\n const authPath = getAntigravityAuthPath()\n const fs = await import(\"node:fs/promises\")\n await fs.writeFile(authPath, JSON.stringify(auth, null, 2), \"utf-8\")\n consola.success(\"Antigravity accounts saved to\", authPath)\n}\n\n/**\n * Load Antigravity accounts from file\n */\nexport async function loadAntigravityAuth(): Promise<AntigravityAuth | null> {\n try {\n const authPath = getAntigravityAuthPath()\n const fs = await import(\"node:fs/promises\")\n\n // Check if file exists\n try {\n await fs.access(authPath)\n } catch {\n return null\n }\n\n const content = await fs.readFile(authPath, \"utf-8\")\n const data = JSON.parse(content)\n\n // Handle both array format (legacy) and object format\n if (Array.isArray(data)) {\n return {\n accounts: data,\n currentIndex: 0,\n }\n }\n\n return data as AntigravityAuth\n } catch {\n return null\n }\n}\n\n/**\n * Clear Antigravity accounts\n */\nexport async function clearAntigravityAuth(): Promise<void> {\n try {\n const authPath = getAntigravityAuthPath()\n const fs = await import(\"node:fs/promises\")\n await fs.unlink(authPath)\n consola.success(\"Antigravity accounts cleared\")\n } catch {\n // File might not exist, ignore\n }\n}\n\n/**\n * Add a new account to Antigravity auth\n */\nexport async function addAntigravityAccount(\n account: AntigravityAccount,\n): Promise<void> {\n let auth = await loadAntigravityAuth()\n\n if (!auth) {\n auth = {\n accounts: [],\n currentIndex: 0,\n }\n }\n\n auth.accounts.push(account)\n await saveAntigravityAuth(auth)\n consola.success(\"Added new Antigravity account\")\n}\n\n/**\n * Get the current active account\n */\nexport async function getCurrentAccount(): Promise<AntigravityAccount | null> {\n const auth = await loadAntigravityAuth()\n\n if (!auth || auth.accounts.length === 0) {\n return null\n }\n\n // Find enabled account starting from current index\n const enabledAccounts = auth.accounts.filter((a) => a.enable)\n\n if (enabledAccounts.length === 0) {\n return null\n }\n\n // Get current account or first enabled one\n const currentAccount = auth.accounts[auth.currentIndex]\n if (currentAccount && currentAccount.enable) {\n return currentAccount\n }\n\n return enabledAccounts[0]\n}\n\n/**\n * Rotate to the next account\n */\nexport async function rotateAccount(): Promise<void> {\n const auth = await loadAntigravityAuth()\n\n if (!auth || auth.accounts.length <= 1) {\n return\n }\n\n // Find next enabled account\n let nextIndex = (auth.currentIndex + 1) % auth.accounts.length\n let attempts = 0\n\n while (!auth.accounts[nextIndex].enable && attempts < auth.accounts.length) {\n nextIndex = (nextIndex + 1) % auth.accounts.length\n attempts++\n }\n\n auth.currentIndex = nextIndex\n await saveAntigravityAuth(auth)\n consola.info(`Rotated to account ${nextIndex}`)\n}\n\n/**\n * Disable current account\n */\nexport async function disableCurrentAccount(): Promise<void> {\n const auth = await loadAntigravityAuth()\n\n if (!auth || auth.accounts.length === 0) {\n return\n }\n\n auth.accounts[auth.currentIndex].enable = false\n await saveAntigravityAuth(auth)\n consola.warn(`Disabled account ${auth.currentIndex}`)\n\n // Rotate to next account\n await rotateAccount()\n}\n\n/**\n * Refresh access token using refresh token\n */\nexport async function refreshAccessToken(\n account: AntigravityAccount,\n): Promise<AntigravityAccount | null> {\n try {\n const response = await fetch(\"https://oauth2.googleapis.com/token\", {\n method: \"POST\",\n headers: {\n \"Content-Type\": \"application/x-www-form-urlencoded\",\n },\n body: new URLSearchParams({\n client_id: GOOGLE_CLIENT_ID,\n client_secret: GOOGLE_CLIENT_SECRET,\n refresh_token: account.refresh_token,\n grant_type: \"refresh_token\",\n }),\n })\n\n if (!response.ok) {\n const error = await response.text()\n consola.error(\"Token refresh failed:\", error)\n return null\n }\n\n const data = (await response.json()) as {\n access_token: string\n expires_in: number\n }\n\n return {\n ...account,\n access_token: data.access_token,\n expires_in: data.expires_in,\n timestamp: Date.now(),\n }\n } catch (error) {\n consola.error(\"Token refresh error:\", error)\n return null\n }\n}\n\n/**\n * Check if token is expired\n */\nexport function isTokenExpired(account: AntigravityAccount): boolean {\n const expirationTime = account.timestamp + account.expires_in * 1000\n // Refresh 5 minutes before expiration\n return Date.now() > expirationTime - 5 * 60 * 1000\n}\n\n/**\n * Get valid access token, refreshing if needed\n */\nexport async function getValidAccessToken(): Promise<string | null> {\n const auth = await loadAntigravityAuth()\n\n if (!auth || auth.accounts.length === 0) {\n return null\n }\n\n let account = auth.accounts[auth.currentIndex]\n\n if (!account || !account.enable) {\n const enabledAccount = auth.accounts.find((a) => a.enable)\n if (!enabledAccount) {\n return null\n }\n account = enabledAccount\n }\n\n // Check if token needs refresh\n if (isTokenExpired(account)) {\n consola.info(\"Access token expired, refreshing...\")\n const refreshedAccount = await refreshAccessToken(account)\n\n if (!refreshedAccount) {\n consola.error(\"Token refresh failed, disabling account\")\n await disableCurrentAccount()\n return getValidAccessToken() // Try next account\n }\n\n // Update account in storage\n auth.accounts[auth.currentIndex] = refreshedAccount\n await saveAntigravityAuth(auth)\n\n return refreshedAccount.access_token\n }\n\n return account.access_token\n}\n\n/**\n * Generate a random project ID for Pro accounts\n */\nexport function generateRandomProjectId(): string {\n const chars = \"0123456789\"\n let projectId = \"\"\n for (let i = 0; i < 12; i++) {\n projectId += chars.charAt(Math.floor(Math.random() * chars.length))\n }\n return projectId\n}\n\n/**\n * Get OAuth authorization URL\n */\nexport function getOAuthUrl(): string {\n const params = new URLSearchParams({\n client_id: GOOGLE_CLIENT_ID,\n redirect_uri: GOOGLE_REDIRECT_URI,\n response_type: \"code\",\n scope: OAUTH_SCOPES.join(\" \"),\n access_type: \"offline\",\n prompt: \"consent\",\n include_granted_scopes: \"true\",\n })\n\n return `https://accounts.google.com/o/oauth2/v2/auth?${params.toString()}`\n}\n\n/**\n * Exchange authorization code for tokens\n */\nexport async function exchangeCodeForTokens(\n code: string,\n): Promise<AntigravityAccount | null> {\n try {\n const response = await fetch(\"https://oauth2.googleapis.com/token\", {\n method: \"POST\",\n headers: {\n \"Content-Type\": \"application/x-www-form-urlencoded\",\n },\n body: new URLSearchParams({\n client_id: GOOGLE_CLIENT_ID,\n client_secret: GOOGLE_CLIENT_SECRET,\n code,\n redirect_uri: GOOGLE_REDIRECT_URI,\n grant_type: \"authorization_code\",\n }),\n })\n\n if (!response.ok) {\n const error = await response.text()\n consola.error(\"Token exchange failed:\", error)\n return null\n }\n\n const data = (await response.json()) as {\n access_token: string\n refresh_token: string\n expires_in: number\n }\n\n return {\n access_token: data.access_token,\n refresh_token: data.refresh_token,\n expires_in: data.expires_in,\n timestamp: Date.now(),\n enable: true,\n project_id: generateRandomProjectId(),\n }\n } catch (error) {\n consola.error(\"Token exchange error:\", error)\n return null\n }\n}\n\n/**\n * Start a local OAuth callback server and wait for authorization\n * This provides a seamless web-based login experience\n * Uses Node.js http module for compatibility with both Node.js and Bun\n *\n * @param onServerReady - Callback function called when server is ready to accept connections\n */\nasync function startOAuthCallbackServer(onServerReady?: () => void): Promise<string> {\n const http = await import(\"node:http\")\n\n return new Promise((resolve, reject) => {\n const server = http.createServer((req, res) => {\n const url = new URL(req.url || \"/\", `http://localhost:8046`)\n\n if (url.pathname === \"/callback\") {\n const code = url.searchParams.get(\"code\")\n const error = url.searchParams.get(\"error\")\n\n if (error) {\n res.writeHead(200, { \"Content-Type\": \"text/html\" })\n res.end(\n `<html><body><h1>Authorization Failed</h1><p>Error: ${error}</p><p>You can close this window.</p></body></html>`\n )\n setTimeout(() => server.close(), 100)\n reject(new Error(`OAuth error: ${error}`))\n return\n }\n\n if (code) {\n res.writeHead(200, { \"Content-Type\": \"text/html\" })\n res.end(\n `<html><body><h1>Authorization Successful!</h1><p>You can close this window and return to the terminal.</p></body></html>`\n )\n setTimeout(() => server.close(), 100)\n resolve(code)\n return\n }\n\n res.writeHead(400, { \"Content-Type\": \"text/plain\" })\n res.end(\"Missing authorization code\")\n return\n }\n\n res.writeHead(404, { \"Content-Type\": \"text/plain\" })\n res.end(\"Not found\")\n })\n\n server.listen(8046, () => {\n consola.info(`OAuth callback server started on http://localhost:8046`)\n\n // Call the callback to signal server is ready\n if (onServerReady) {\n onServerReady()\n }\n })\n\n // Timeout after 5 minutes\n setTimeout(() => {\n server.close()\n reject(new Error(\"OAuth timeout - no callback received within 5 minutes\"))\n }, 5 * 60 * 1000)\n })\n}\n\n/**\n * Setup Antigravity with web-based OAuth flow\n * Opens browser for login and automatically captures the callback\n */\nexport async function setupAntigravityWeb(): Promise<void> {\n const auth = await loadAntigravityAuth()\n\n if (auth && auth.accounts.length > 0) {\n const enabledCount = auth.accounts.filter((a) => a.enable).length\n consola.info(\n `Found ${auth.accounts.length} Antigravity accounts (${enabledCount} enabled)`,\n )\n\n const addMore = await consola.prompt(\"Add another account?\", {\n type: \"confirm\",\n initial: false,\n })\n\n if (!addMore) {\n return\n }\n }\n\n consola.info(\"\")\n consola.info(\"Google Antigravity OAuth Setup (Web Flow)\")\n consola.info(\"=========================================\")\n consola.info(\"\")\n\n const oauthUrl = getOAuthUrl()\n\n // Function to open browser - will be called after server is ready\n const openBrowser = async () => {\n consola.info(\"Opening browser for Google login...\")\n consola.info(`If browser doesn't open, visit: ${oauthUrl}`)\n consola.info(\"\")\n\n try {\n const { exec } = await import(\"node:child_process\")\n const platform = process.platform\n\n if (platform === \"win32\") {\n exec(`start \"\" \"${oauthUrl}\"`)\n } else if (platform === \"darwin\") {\n exec(`open \"${oauthUrl}\"`)\n } else {\n exec(`xdg-open \"${oauthUrl}\"`)\n }\n } catch {\n consola.warn(\"Could not open browser automatically\")\n }\n\n consola.info(\"Waiting for authorization...\")\n }\n\n try {\n // Start server first, then open browser when server is ready\n const code = await startOAuthCallbackServer(() => {\n // Use setImmediate to ensure this runs after the server is fully ready\n setImmediate(() => {\n openBrowser().catch(() => {\n // Ignore browser open errors\n })\n })\n })\n\n consola.info(\"Authorization code received, exchanging for tokens...\")\n\n const account = await exchangeCodeForTokens(code)\n\n if (!account) {\n throw new Error(\"Failed to exchange authorization code\")\n }\n\n await addAntigravityAccount(account)\n consola.success(\"Antigravity account added successfully!\")\n } catch (error) {\n consola.error(\"OAuth flow failed:\", error)\n throw error\n }\n}\n\n/**\n * Setup Antigravity interactively (manual URL copy method)\n */\nexport async function setupAntigravityManual(): Promise<void> {\n const auth = await loadAntigravityAuth()\n\n if (auth && auth.accounts.length > 0) {\n const enabledCount = auth.accounts.filter((a) => a.enable).length\n consola.info(\n `Found ${auth.accounts.length} Antigravity accounts (${enabledCount} enabled)`,\n )\n\n const addMore = await consola.prompt(\"Add another account?\", {\n type: \"confirm\",\n initial: false,\n })\n\n if (!addMore) {\n return\n }\n }\n\n consola.info(\"\")\n consola.info(\"Google Antigravity OAuth Setup (Manual)\")\n consola.info(\"=======================================\")\n consola.info(\"\")\n consola.info(\"You need to authorize with Google to use Antigravity API.\")\n consola.info(\"Please follow these steps:\")\n consola.info(\"\")\n consola.info(\"1. Open this URL in your browser:\")\n consola.info(` ${getOAuthUrl()}`)\n consola.info(\"\")\n consola.info(\"2. Complete the Google sign-in process\")\n consola.info(\"3. After authorization, you'll be redirected to a callback URL\")\n consola.info(\"4. Copy the full callback URL and paste it below\")\n consola.info(\"\")\n\n const callbackUrl = await consola.prompt(\"Enter the callback URL:\", {\n type: \"text\",\n })\n\n if (!callbackUrl || typeof callbackUrl !== \"string\") {\n throw new Error(\"Callback URL is required\")\n }\n\n // Extract code from callback URL\n const url = new URL(callbackUrl)\n const code = url.searchParams.get(\"code\")\n\n if (!code) {\n throw new Error(\"Authorization code not found in URL\")\n }\n\n consola.info(\"Exchanging authorization code for tokens...\")\n\n const account = await exchangeCodeForTokens(code)\n\n if (!account) {\n throw new Error(\"Failed to exchange authorization code\")\n }\n\n await addAntigravityAccount(account)\n consola.success(\"Antigravity account added successfully!\")\n}\n\n/**\n * Setup Antigravity interactively\n * Offers choice between web-based and manual OAuth flow\n */\nexport async function setupAntigravity(): Promise<void> {\n const method = await consola.prompt(\"Choose OAuth login method:\", {\n type: \"select\",\n options: [\n { value: \"web\", label: \"Web (auto-open browser, recommended)\" },\n { value: \"manual\", label: \"Manual (copy/paste callback URL)\" },\n ],\n })\n\n if (method === \"web\") {\n await setupAntigravityWeb()\n } else {\n await setupAntigravityManual()\n }\n}\n"],"mappings":";;;;AA4BA,MAAM,4BAA4B;AAuBlC,MAAM,iBAAiB,QAAQ,IAAI,kBAAkB;;;;AAKrD,SAAgB,YAAqB;AACnC,QAAO,eAAe,SAAS;;;;;AAMjC,SAAgB,YAA2B;AACzC,QAAO,kBAAkB;;AAI3B,MAAM,oBAAoB;AAC1B,MAAM,wBAAwB;AAG9B,IAAI,mBACF,QAAQ,IAAI,yBAAyB;AACvC,IAAI,uBACF,QAAQ,IAAI,6BAA6B;AAC3C,MAAM,sBAAsB;AAG5B,MAAM,eAAe;CACnB;CACA;CACA;CACA;CACA;CACD;;;;;AAMD,SAAgB,oBAAoB,UAAkB,cAA4B;AAChF,oBAAmB;AACnB,wBAAuB;;;;;AAMzB,SAAgB,yBAAiC;AAC/C,QAAO,GAAG,MAAM,SAAS,GAAG;;;;;AAM9B,eAAsB,oBACpB,MACe;AACf,OAAM,aAAa;CACnB,MAAM,WAAW,wBAAwB;AAEzC,QADW,MAAM,OAAO,qBACf,UAAU,UAAU,KAAK,UAAU,MAAM,MAAM,EAAE,EAAE,QAAQ;AACpE,SAAQ,QAAQ,iCAAiC,SAAS;;;;;AAM5D,eAAsB,sBAAuD;AAC3E,KAAI;EACF,MAAM,WAAW,wBAAwB;EACzC,MAAM,KAAK,MAAM,OAAO;AAGxB,MAAI;AACF,SAAM,GAAG,OAAO,SAAS;UACnB;AACN,UAAO;;EAGT,MAAM,UAAU,MAAM,GAAG,SAAS,UAAU,QAAQ;EACpD,MAAM,OAAO,KAAK,MAAM,QAAQ;AAGhC,MAAI,MAAM,QAAQ,KAAK,CACrB,QAAO;GACL,UAAU;GACV,cAAc;GACf;AAGH,SAAO;SACD;AACN,SAAO;;;;;;AAOX,eAAsB,uBAAsC;AAC1D,KAAI;EACF,MAAM,WAAW,wBAAwB;AAEzC,SADW,MAAM,OAAO,qBACf,OAAO,SAAS;AACzB,UAAQ,QAAQ,+BAA+B;SACzC;;;;;AAQV,eAAsB,sBACpB,SACe;CACf,IAAI,OAAO,MAAM,qBAAqB;AAEtC,KAAI,CAAC,KACH,QAAO;EACL,UAAU,EAAE;EACZ,cAAc;EACf;AAGH,MAAK,SAAS,KAAK,QAAQ;AAC3B,OAAM,oBAAoB,KAAK;AAC/B,SAAQ,QAAQ,gCAAgC;;;;;AAMlD,eAAsB,oBAAwD;CAC5E,MAAM,OAAO,MAAM,qBAAqB;AAExC,KAAI,CAAC,QAAQ,KAAK,SAAS,WAAW,EACpC,QAAO;CAIT,MAAM,kBAAkB,KAAK,SAAS,QAAQ,MAAM,EAAE,OAAO;AAE7D,KAAI,gBAAgB,WAAW,EAC7B,QAAO;CAIT,MAAM,iBAAiB,KAAK,SAAS,KAAK;AAC1C,KAAI,kBAAkB,eAAe,OACnC,QAAO;AAGT,QAAO,gBAAgB;;;;;AAMzB,eAAsB,gBAA+B;CACnD,MAAM,OAAO,MAAM,qBAAqB;AAExC,KAAI,CAAC,QAAQ,KAAK,SAAS,UAAU,EACnC;CAIF,IAAI,aAAa,KAAK,eAAe,KAAK,KAAK,SAAS;CACxD,IAAI,WAAW;AAEf,QAAO,CAAC,KAAK,SAAS,WAAW,UAAU,WAAW,KAAK,SAAS,QAAQ;AAC1E,eAAa,YAAY,KAAK,KAAK,SAAS;AAC5C;;AAGF,MAAK,eAAe;AACpB,OAAM,oBAAoB,KAAK;AAC/B,SAAQ,KAAK,sBAAsB,YAAY;;;;;AAMjD,eAAsB,wBAAuC;CAC3D,MAAM,OAAO,MAAM,qBAAqB;AAExC,KAAI,CAAC,QAAQ,KAAK,SAAS,WAAW,EACpC;AAGF,MAAK,SAAS,KAAK,cAAc,SAAS;AAC1C,OAAM,oBAAoB,KAAK;AAC/B,SAAQ,KAAK,oBAAoB,KAAK,eAAe;AAGrD,OAAM,eAAe;;;;;AAMvB,eAAsB,mBACpB,SACoC;AACpC,KAAI;EACF,MAAM,WAAW,MAAM,MAAM,uCAAuC;GAClE,QAAQ;GACR,SAAS,EACP,gBAAgB,qCACjB;GACD,MAAM,IAAI,gBAAgB;IACxB,WAAW;IACX,eAAe;IACf,eAAe,QAAQ;IACvB,YAAY;IACb,CAAC;GACH,CAAC;AAEF,MAAI,CAAC,SAAS,IAAI;GAChB,MAAM,QAAQ,MAAM,SAAS,MAAM;AACnC,WAAQ,MAAM,yBAAyB,MAAM;AAC7C,UAAO;;EAGT,MAAM,OAAQ,MAAM,SAAS,MAAM;AAKnC,SAAO;GACL,GAAG;GACH,cAAc,KAAK;GACnB,YAAY,KAAK;GACjB,WAAW,KAAK,KAAK;GACtB;UACM,OAAO;AACd,UAAQ,MAAM,wBAAwB,MAAM;AAC5C,SAAO;;;;;;AAOX,SAAgB,eAAe,SAAsC;CACnE,MAAM,iBAAiB,QAAQ,YAAY,QAAQ,aAAa;AAEhE,QAAO,KAAK,KAAK,GAAG,iBAAiB,MAAS;;;;;AAMhD,eAAsB,sBAA8C;CAClE,MAAM,OAAO,MAAM,qBAAqB;AAExC,KAAI,CAAC,QAAQ,KAAK,SAAS,WAAW,EACpC,QAAO;CAGT,IAAI,UAAU,KAAK,SAAS,KAAK;AAEjC,KAAI,CAAC,WAAW,CAAC,QAAQ,QAAQ;EAC/B,MAAM,iBAAiB,KAAK,SAAS,MAAM,MAAM,EAAE,OAAO;AAC1D,MAAI,CAAC,eACH,QAAO;AAET,YAAU;;AAIZ,KAAI,eAAe,QAAQ,EAAE;AAC3B,UAAQ,KAAK,sCAAsC;EACnD,MAAM,mBAAmB,MAAM,mBAAmB,QAAQ;AAE1D,MAAI,CAAC,kBAAkB;AACrB,WAAQ,MAAM,0CAA0C;AACxD,SAAM,uBAAuB;AAC7B,UAAO,qBAAqB;;AAI9B,OAAK,SAAS,KAAK,gBAAgB;AACnC,QAAM,oBAAoB,KAAK;AAE/B,SAAO,iBAAiB;;AAG1B,QAAO,QAAQ;;;;;AAMjB,SAAgB,0BAAkC;CAChD,MAAM,QAAQ;CACd,IAAI,YAAY;AAChB,MAAK,IAAI,IAAI,GAAG,IAAI,IAAI,IACtB,cAAa,MAAM,OAAO,KAAK,MAAM,KAAK,QAAQ,GAAG,GAAa,CAAC;AAErE,QAAO;;;;;AAMT,SAAgB,cAAsB;AAWpC,QAAO,gDAVQ,IAAI,gBAAgB;EACjC,WAAW;EACX,cAAc;EACd,eAAe;EACf,OAAO,aAAa,KAAK,IAAI;EAC7B,aAAa;EACb,QAAQ;EACR,wBAAwB;EACzB,CAAC,CAE4D,UAAU;;;;;AAM1E,eAAsB,sBACpB,MACoC;AACpC,KAAI;EACF,MAAM,WAAW,MAAM,MAAM,uCAAuC;GAClE,QAAQ;GACR,SAAS,EACP,gBAAgB,qCACjB;GACD,MAAM,IAAI,gBAAgB;IACxB,WAAW;IACX,eAAe;IACf;IACA,cAAc;IACd,YAAY;IACb,CAAC;GACH,CAAC;AAEF,MAAI,CAAC,SAAS,IAAI;GAChB,MAAM,QAAQ,MAAM,SAAS,MAAM;AACnC,WAAQ,MAAM,0BAA0B,MAAM;AAC9C,UAAO;;EAGT,MAAM,OAAQ,MAAM,SAAS,MAAM;AAMnC,SAAO;GACL,cAAc,KAAK;GACnB,eAAe,KAAK;GACpB,YAAY,KAAK;GACjB,WAAW,KAAK,KAAK;GACrB,QAAQ;GACR,YAAY,yBAAyB;GACtC;UACM,OAAO;AACd,UAAQ,MAAM,yBAAyB,MAAM;AAC7C,SAAO;;;;;;;;;;AAWX,eAAe,yBAAyB,eAA6C;CACnF,MAAM,OAAO,MAAM,OAAO;AAE1B,QAAO,IAAI,SAAS,SAAS,WAAW;EACtC,MAAM,SAAS,KAAK,cAAc,KAAK,QAAQ;GAC7C,MAAM,MAAM,IAAI,IAAI,IAAI,OAAO,KAAK,wBAAwB;AAE5D,OAAI,IAAI,aAAa,aAAa;IAChC,MAAM,OAAO,IAAI,aAAa,IAAI,OAAO;IACzC,MAAM,QAAQ,IAAI,aAAa,IAAI,QAAQ;AAE3C,QAAI,OAAO;AACT,SAAI,UAAU,KAAK,EAAE,gBAAgB,aAAa,CAAC;AACnD,SAAI,IACF,sDAAsD,MAAM,qDAC7D;AACD,sBAAiB,OAAO,OAAO,EAAE,IAAI;AACrC,4BAAO,IAAI,MAAM,gBAAgB,QAAQ,CAAC;AAC1C;;AAGF,QAAI,MAAM;AACR,SAAI,UAAU,KAAK,EAAE,gBAAgB,aAAa,CAAC;AACnD,SAAI,IACF,2HACD;AACD,sBAAiB,OAAO,OAAO,EAAE,IAAI;AACrC,aAAQ,KAAK;AACb;;AAGF,QAAI,UAAU,KAAK,EAAE,gBAAgB,cAAc,CAAC;AACpD,QAAI,IAAI,6BAA6B;AACrC;;AAGF,OAAI,UAAU,KAAK,EAAE,gBAAgB,cAAc,CAAC;AACpD,OAAI,IAAI,YAAY;IACpB;AAEF,SAAO,OAAO,YAAY;AACxB,WAAQ,KAAK,yDAAyD;AAGtE,OAAI,cACF,gBAAe;IAEjB;AAGF,mBAAiB;AACf,UAAO,OAAO;AACd,0BAAO,IAAI,MAAM,wDAAwD,CAAC;KACzE,MAAS,IAAK;GACjB;;;;;;AAOJ,eAAsB,sBAAqC;CACzD,MAAM,OAAO,MAAM,qBAAqB;AAExC,KAAI,QAAQ,KAAK,SAAS,SAAS,GAAG;EACpC,MAAM,eAAe,KAAK,SAAS,QAAQ,MAAM,EAAE,OAAO,CAAC;AAC3D,UAAQ,KACN,SAAS,KAAK,SAAS,OAAO,yBAAyB,aAAa,WACrE;AAOD,MAAI,CALY,MAAM,QAAQ,OAAO,wBAAwB;GAC3D,MAAM;GACN,SAAS;GACV,CAAC,CAGA;;AAIJ,SAAQ,KAAK,GAAG;AAChB,SAAQ,KAAK,4CAA4C;AACzD,SAAQ,KAAK,4CAA4C;AACzD,SAAQ,KAAK,GAAG;CAEhB,MAAM,WAAW,aAAa;CAG9B,MAAM,cAAc,YAAY;AAC9B,UAAQ,KAAK,sCAAsC;AACnD,UAAQ,KAAK,mCAAmC,WAAW;AAC3D,UAAQ,KAAK,GAAG;AAEhB,MAAI;GACF,MAAM,EAAE,SAAS,MAAM,OAAO;GAC9B,MAAM,WAAW,QAAQ;AAEzB,OAAI,aAAa,QACf,MAAK,aAAa,SAAS,GAAG;YACrB,aAAa,SACtB,MAAK,SAAS,SAAS,GAAG;OAE1B,MAAK,aAAa,SAAS,GAAG;UAE1B;AACN,WAAQ,KAAK,uCAAuC;;AAGtD,UAAQ,KAAK,+BAA+B;;AAG9C,KAAI;EAEF,MAAM,OAAO,MAAM,+BAA+B;AAEhD,sBAAmB;AACjB,iBAAa,CAAC,YAAY,GAExB;KACF;IACF;AAEF,UAAQ,KAAK,wDAAwD;EAErE,MAAM,UAAU,MAAM,sBAAsB,KAAK;AAEjD,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,wCAAwC;AAG1D,QAAM,sBAAsB,QAAQ;AACpC,UAAQ,QAAQ,0CAA0C;UACnD,OAAO;AACd,UAAQ,MAAM,sBAAsB,MAAM;AAC1C,QAAM;;;;;;AAOV,eAAsB,yBAAwC;CAC5D,MAAM,OAAO,MAAM,qBAAqB;AAExC,KAAI,QAAQ,KAAK,SAAS,SAAS,GAAG;EACpC,MAAM,eAAe,KAAK,SAAS,QAAQ,MAAM,EAAE,OAAO,CAAC;AAC3D,UAAQ,KACN,SAAS,KAAK,SAAS,OAAO,yBAAyB,aAAa,WACrE;AAOD,MAAI,CALY,MAAM,QAAQ,OAAO,wBAAwB;GAC3D,MAAM;GACN,SAAS;GACV,CAAC,CAGA;;AAIJ,SAAQ,KAAK,GAAG;AAChB,SAAQ,KAAK,0CAA0C;AACvD,SAAQ,KAAK,0CAA0C;AACvD,SAAQ,KAAK,GAAG;AAChB,SAAQ,KAAK,4DAA4D;AACzE,SAAQ,KAAK,6BAA6B;AAC1C,SAAQ,KAAK,GAAG;AAChB,SAAQ,KAAK,oCAAoC;AACjD,SAAQ,KAAK,MAAM,aAAa,GAAG;AACnC,SAAQ,KAAK,GAAG;AAChB,SAAQ,KAAK,yCAAyC;AACtD,SAAQ,KAAK,iEAAiE;AAC9E,SAAQ,KAAK,mDAAmD;AAChE,SAAQ,KAAK,GAAG;CAEhB,MAAM,cAAc,MAAM,QAAQ,OAAO,2BAA2B,EAClE,MAAM,QACP,CAAC;AAEF,KAAI,CAAC,eAAe,OAAO,gBAAgB,SACzC,OAAM,IAAI,MAAM,2BAA2B;CAK7C,MAAM,OADM,IAAI,IAAI,YAAY,CACf,aAAa,IAAI,OAAO;AAEzC,KAAI,CAAC,KACH,OAAM,IAAI,MAAM,sCAAsC;AAGxD,SAAQ,KAAK,8CAA8C;CAE3D,MAAM,UAAU,MAAM,sBAAsB,KAAK;AAEjD,KAAI,CAAC,QACH,OAAM,IAAI,MAAM,wCAAwC;AAG1D,OAAM,sBAAsB,QAAQ;AACpC,SAAQ,QAAQ,0CAA0C;;;;;;AAO5D,eAAsB,mBAAkC;AAStD,KARe,MAAM,QAAQ,OAAO,8BAA8B;EAChE,MAAM;EACN,SAAS,CACP;GAAE,OAAO;GAAO,OAAO;GAAwC,EAC/D;GAAE,OAAO;GAAU,OAAO;GAAoC,CAC/D;EACF,CAAC,KAEa,MACb,OAAM,qBAAqB;KAE3B,OAAM,wBAAwB"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { getValidAccessToken } from "./auth-
|
|
1
|
+
import { getValidAccessToken } from "./auth-y23hLerx.js";
|
|
2
2
|
import consola from "consola";
|
|
3
3
|
|
|
4
4
|
//#region src/services/antigravity/get-models.ts
|
|
@@ -170,7 +170,11 @@ async function fetchModelsFromApi() {
|
|
|
170
170
|
id: modelId,
|
|
171
171
|
object: "model",
|
|
172
172
|
created: 17e8,
|
|
173
|
-
owned_by: isGoogle ? "google" : "anthropic"
|
|
173
|
+
owned_by: isGoogle ? "google" : "anthropic",
|
|
174
|
+
quotaInfo: m.quotaInfo ? {
|
|
175
|
+
remainingFraction: m.quotaInfo.remainingFraction ?? 1,
|
|
176
|
+
resetTime: m.quotaInfo.resetTime ?? ""
|
|
177
|
+
} : void 0
|
|
174
178
|
};
|
|
175
179
|
});
|
|
176
180
|
consola.debug(`Fetched ${models.length} models from Antigravity API`);
|
|
@@ -207,6 +211,30 @@ async function getAntigravityModels() {
|
|
|
207
211
|
};
|
|
208
212
|
}
|
|
209
213
|
/**
|
|
214
|
+
* Get Antigravity usage/quota information
|
|
215
|
+
*/
|
|
216
|
+
async function getAntigravityUsage() {
|
|
217
|
+
cachedModels = null;
|
|
218
|
+
cacheTimestamp = 0;
|
|
219
|
+
const modelsResponse = await getAntigravityModels();
|
|
220
|
+
let earliestResetTime = "";
|
|
221
|
+
const modelsQuota = {};
|
|
222
|
+
for (const model of modelsResponse.data) if (model.quotaInfo) {
|
|
223
|
+
const resetTime = model.quotaInfo.resetTime;
|
|
224
|
+
if (!earliestResetTime || resetTime && resetTime < earliestResetTime) earliestResetTime = resetTime;
|
|
225
|
+
modelsQuota[model.id] = {
|
|
226
|
+
remaining_fraction: model.quotaInfo.remainingFraction,
|
|
227
|
+
reset_time: model.quotaInfo.resetTime,
|
|
228
|
+
percent_remaining: Math.round(model.quotaInfo.remainingFraction * 100)
|
|
229
|
+
};
|
|
230
|
+
}
|
|
231
|
+
return {
|
|
232
|
+
copilot_plan: "antigravity",
|
|
233
|
+
quota_reset_date: earliestResetTime,
|
|
234
|
+
quota_snapshots: { models: modelsQuota }
|
|
235
|
+
};
|
|
236
|
+
}
|
|
237
|
+
/**
|
|
210
238
|
* Check if a model is a thinking/reasoning model
|
|
211
239
|
*/
|
|
212
240
|
function isThinkingModel(modelId) {
|
|
@@ -214,5 +242,5 @@ function isThinkingModel(modelId) {
|
|
|
214
242
|
}
|
|
215
243
|
|
|
216
244
|
//#endregion
|
|
217
|
-
export { getAntigravityModels, isThinkingModel };
|
|
218
|
-
//# sourceMappingURL=get-models-
|
|
245
|
+
export { getAntigravityModels, getAntigravityUsage, isThinkingModel };
|
|
246
|
+
//# sourceMappingURL=get-models-uSazerPk.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"get-models-uSazerPk.js","names":["FALLBACK_MODELS: Array<AntigravityModel>","cachedModels: Array<AntigravityModel> | null","cacheTimestamp: number","models: Array<AntigravityModel>","modelsQuota: Record<string, {\n remaining_fraction: number\n reset_time: string\n percent_remaining: number\n }>"],"sources":["../src/services/antigravity/get-models.ts"],"sourcesContent":["/**\n * Google Antigravity Models\n *\n * Provides list of available models from Antigravity.\n * Based on: https://github.com/liuw1535/antigravity2api-nodejs\n */\n\n/* eslint-disable require-atomic-updates */\n\nimport consola from \"consola\"\n\nimport { getValidAccessToken } from \"./auth\"\n\n// Antigravity API endpoints\nconst ANTIGRAVITY_API_HOST = \"daily-cloudcode-pa.sandbox.googleapis.com\"\nconst ANTIGRAVITY_MODELS_URL = `https://${ANTIGRAVITY_API_HOST}/v1internal:fetchAvailableModels`\nconst ANTIGRAVITY_USER_AGENT = \"antigravity/1.11.3 windows/amd64\"\n\nexport interface AntigravityQuotaInfo {\n remainingFraction: number\n resetTime: string\n}\n\nexport interface AntigravityModel {\n id: string\n object: string\n created: number\n owned_by: string\n quotaInfo?: AntigravityQuotaInfo\n}\n\nexport interface AntigravityModelsResponse {\n object: string\n data: Array<AntigravityModel>\n}\n\n/**\n * Fallback Antigravity models when API is unavailable\n * Based on antigravity2api-nodejs model list\n */\nconst FALLBACK_MODELS: Array<AntigravityModel> = [\n // Gemini models\n {\n id: \"gemini-2.5-pro-exp-03-25\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.5-pro-preview-05-06\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.0-flash-exp\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.0-flash-001\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.0-flash-thinking-exp-1219\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.0-flash-thinking-exp\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.0-flash-thinking-exp-01-21\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-2.0-pro-exp-02-05\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-1.5-pro\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-1.5-flash\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-1.5-flash-8b\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"gemini-exp-1206\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n {\n id: \"learnlm-1.5-pro-experimental\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"google\",\n },\n\n // Claude models (via Antigravity)\n {\n id: \"claude-opus-4-5\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n {\n id: \"claude-sonnet-4-5\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n {\n id: \"claude-3-5-sonnet-20241022\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n {\n id: \"claude-3-5-haiku-20241022\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n {\n id: \"claude-3-opus-20240229\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n {\n id: \"claude-3-sonnet-20240229\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n {\n id: \"claude-3-haiku-20240307\",\n object: \"model\",\n created: 1700000000,\n owned_by: \"anthropic\",\n },\n]\n\n// Cache for fetched models\nlet cachedModels: Array<AntigravityModel> | null = null\nlet cacheTimestamp: number = 0\nconst CACHE_TTL = 5 * 60 * 1000 // 5 minutes\n\n/**\n * Fetch models from Antigravity API\n */\nasync function fetchModelsFromApi(): Promise<Array<AntigravityModel> | null> {\n const accessToken = await getValidAccessToken()\n\n if (!accessToken) {\n consola.debug(\"No access token available, using fallback models\")\n return null\n }\n\n try {\n const response = await fetch(ANTIGRAVITY_MODELS_URL, {\n method: \"POST\",\n headers: {\n Host: ANTIGRAVITY_API_HOST,\n \"User-Agent\": ANTIGRAVITY_USER_AGENT,\n Authorization: `Bearer ${accessToken}`,\n \"Content-Type\": \"application/json\",\n \"Accept-Encoding\": \"gzip\",\n },\n body: JSON.stringify({}),\n })\n\n if (!response.ok) {\n consola.warn(`Failed to fetch Antigravity models: ${response.status}`)\n return null\n }\n\n const data = (await response.json()) as {\n models?: Array<{\n name: string\n quotaInfo?: {\n remainingFraction?: number\n resetTime?: string\n }\n }>\n }\n\n if (!data.models || !Array.isArray(data.models)) {\n return null\n }\n\n // Convert to OpenAI format\n const models: Array<AntigravityModel> = data.models\n .filter((m) => {\n // Filter out models with no remaining quota\n const remaining = m.quotaInfo?.remainingFraction ?? 1\n return remaining > 0\n })\n .map((m) => {\n // Extract model ID from name (e.g., \"models/gemini-2.0-flash\" -> \"gemini-2.0-flash\")\n const modelId = m.name.replace(\"models/\", \"\")\n const isGoogle =\n modelId.startsWith(\"gemini\") || modelId.startsWith(\"learnlm\")\n\n return {\n id: modelId,\n object: \"model\",\n created: 1700000000,\n owned_by: isGoogle ? \"google\" : \"anthropic\",\n quotaInfo: m.quotaInfo ? {\n remainingFraction: m.quotaInfo.remainingFraction ?? 1,\n resetTime: m.quotaInfo.resetTime ?? \"\",\n } : undefined,\n }\n })\n\n consola.debug(`Fetched ${models.length} models from Antigravity API`)\n return models\n } catch (error) {\n consola.warn(\"Error fetching Antigravity models:\", error)\n return null\n }\n}\n\n/**\n * Get available Antigravity models\n */\nexport async function getAntigravityModels(): Promise<AntigravityModelsResponse> {\n // Check cache\n if (cachedModels && Date.now() - cacheTimestamp < CACHE_TTL) {\n consola.debug(`Returning ${cachedModels.length} cached Antigravity models`)\n return {\n object: \"list\",\n data: cachedModels,\n }\n }\n\n // Try to fetch from API\n const apiModels = await fetchModelsFromApi()\n\n if (apiModels && apiModels.length > 0) {\n cachedModels = apiModels\n cacheTimestamp = Date.now()\n\n return {\n object: \"list\",\n data: apiModels,\n }\n }\n\n // Use fallback models\n consola.debug(\n `Returning ${FALLBACK_MODELS.length} fallback Antigravity models`,\n )\n\n return {\n object: \"list\",\n data: FALLBACK_MODELS,\n }\n}\n\n/**\n * Antigravity usage response format (compatible with Copilot usage viewer)\n */\nexport interface AntigravityUsageResponse {\n copilot_plan: string\n quota_reset_date: string\n quota_snapshots: {\n models: Record<string, {\n remaining_fraction: number\n reset_time: string\n percent_remaining: number\n }>\n }\n}\n\n/**\n * Get Antigravity usage/quota information\n */\nexport async function getAntigravityUsage(): Promise<AntigravityUsageResponse> {\n // Force refresh models to get latest quota\n cachedModels = null\n cacheTimestamp = 0\n\n const modelsResponse = await getAntigravityModels()\n\n // Find earliest reset time\n let earliestResetTime = \"\"\n const modelsQuota: Record<string, {\n remaining_fraction: number\n reset_time: string\n percent_remaining: number\n }> = {}\n\n for (const model of modelsResponse.data) {\n if (model.quotaInfo) {\n const resetTime = model.quotaInfo.resetTime\n if (!earliestResetTime || (resetTime && resetTime < earliestResetTime)) {\n earliestResetTime = resetTime\n }\n\n modelsQuota[model.id] = {\n remaining_fraction: model.quotaInfo.remainingFraction,\n reset_time: model.quotaInfo.resetTime,\n percent_remaining: Math.round(model.quotaInfo.remainingFraction * 100),\n }\n }\n }\n\n return {\n copilot_plan: \"antigravity\",\n quota_reset_date: earliestResetTime,\n quota_snapshots: {\n models: modelsQuota,\n },\n }\n}\n\n/**\n * Check if a model is a Claude model\n */\nexport function isClaudeModel(modelId: string): boolean {\n return modelId.startsWith(\"claude-\")\n}\n\n/**\n * Check if a model is a thinking/reasoning model\n */\nexport function isThinkingModel(modelId: string): boolean {\n return modelId.includes(\"thinking\")\n}\n\n/**\n * Check if a model is an image generation model\n */\nexport function isImageModel(modelId: string): boolean {\n return modelId.includes(\"image\")\n}\n"],"mappings":";;;;AAcA,MAAM,uBAAuB;AAC7B,MAAM,yBAAyB,WAAW,qBAAqB;AAC/D,MAAM,yBAAyB;;;;;AAwB/B,MAAMA,kBAA2C;CAE/C;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CAGD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACD;EACE,IAAI;EACJ,QAAQ;EACR,SAAS;EACT,UAAU;EACX;CACF;AAGD,IAAIC,eAA+C;AACnD,IAAIC,iBAAyB;AAC7B,MAAM,YAAY,MAAS;;;;AAK3B,eAAe,qBAA8D;CAC3E,MAAM,cAAc,MAAM,qBAAqB;AAE/C,KAAI,CAAC,aAAa;AAChB,UAAQ,MAAM,mDAAmD;AACjE,SAAO;;AAGT,KAAI;EACF,MAAM,WAAW,MAAM,MAAM,wBAAwB;GACnD,QAAQ;GACR,SAAS;IACP,MAAM;IACN,cAAc;IACd,eAAe,UAAU;IACzB,gBAAgB;IAChB,mBAAmB;IACpB;GACD,MAAM,KAAK,UAAU,EAAE,CAAC;GACzB,CAAC;AAEF,MAAI,CAAC,SAAS,IAAI;AAChB,WAAQ,KAAK,uCAAuC,SAAS,SAAS;AACtE,UAAO;;EAGT,MAAM,OAAQ,MAAM,SAAS,MAAM;AAUnC,MAAI,CAAC,KAAK,UAAU,CAAC,MAAM,QAAQ,KAAK,OAAO,CAC7C,QAAO;EAIT,MAAMC,SAAkC,KAAK,OAC1C,QAAQ,MAAM;AAGb,WADkB,EAAE,WAAW,qBAAqB,KACjC;IACnB,CACD,KAAK,MAAM;GAEV,MAAM,UAAU,EAAE,KAAK,QAAQ,WAAW,GAAG;GAC7C,MAAM,WACJ,QAAQ,WAAW,SAAS,IAAI,QAAQ,WAAW,UAAU;AAE/D,UAAO;IACL,IAAI;IACJ,QAAQ;IACR,SAAS;IACT,UAAU,WAAW,WAAW;IAChC,WAAW,EAAE,YAAY;KACvB,mBAAmB,EAAE,UAAU,qBAAqB;KACpD,WAAW,EAAE,UAAU,aAAa;KACrC,GAAG;IACL;IACD;AAEJ,UAAQ,MAAM,WAAW,OAAO,OAAO,8BAA8B;AACrE,SAAO;UACA,OAAO;AACd,UAAQ,KAAK,sCAAsC,MAAM;AACzD,SAAO;;;;;;AAOX,eAAsB,uBAA2D;AAE/E,KAAI,gBAAgB,KAAK,KAAK,GAAG,iBAAiB,WAAW;AAC3D,UAAQ,MAAM,aAAa,aAAa,OAAO,4BAA4B;AAC3E,SAAO;GACL,QAAQ;GACR,MAAM;GACP;;CAIH,MAAM,YAAY,MAAM,oBAAoB;AAE5C,KAAI,aAAa,UAAU,SAAS,GAAG;AACrC,iBAAe;AACf,mBAAiB,KAAK,KAAK;AAE3B,SAAO;GACL,QAAQ;GACR,MAAM;GACP;;AAIH,SAAQ,MACN,aAAa,gBAAgB,OAAO,8BACrC;AAED,QAAO;EACL,QAAQ;EACR,MAAM;EACP;;;;;AAqBH,eAAsB,sBAAyD;AAE7E,gBAAe;AACf,kBAAiB;CAEjB,MAAM,iBAAiB,MAAM,sBAAsB;CAGnD,IAAI,oBAAoB;CACxB,MAAMC,cAID,EAAE;AAEP,MAAK,MAAM,SAAS,eAAe,KACjC,KAAI,MAAM,WAAW;EACnB,MAAM,YAAY,MAAM,UAAU;AAClC,MAAI,CAAC,qBAAsB,aAAa,YAAY,kBAClD,qBAAoB;AAGtB,cAAY,MAAM,MAAM;GACtB,oBAAoB,MAAM,UAAU;GACpC,YAAY,MAAM,UAAU;GAC5B,mBAAmB,KAAK,MAAM,MAAM,UAAU,oBAAoB,IAAI;GACvE;;AAIL,QAAO;EACL,cAAc;EACd,kBAAkB;EAClB,iBAAiB,EACf,QAAQ,aACT;EACF;;;;;AAaH,SAAgB,gBAAgB,SAA0B;AACxD,QAAO,QAAQ,SAAS,WAAW"}
|
package/dist/main.js
CHANGED
|
@@ -4,9 +4,9 @@ import { state } from "./state-CcLGr8VN.js";
|
|
|
4
4
|
import { GITHUB_API_BASE_URL, copilotBaseUrl, copilotHeaders, githubHeaders } from "./get-user-BzIEATcF.js";
|
|
5
5
|
import { HTTPError, forwardError } from "./error-CvU5otz-.js";
|
|
6
6
|
import { cacheModels, cacheVSCodeVersion, clearGithubToken, isNullish, setupCopilotToken, setupGitHubToken, sleep } from "./token-BUmQ_BcM.js";
|
|
7
|
-
import { clearAntigravityAuth, disableCurrentAccount, getAntigravityAuthPath, getApiKey, getValidAccessToken, rotateAccount } from "./auth-
|
|
8
|
-
import { clearZenAuth, getZenAuthPath } from "./auth-
|
|
9
|
-
import { getAntigravityModels, isThinkingModel } from "./get-models-
|
|
7
|
+
import { clearAntigravityAuth, disableCurrentAccount, getAntigravityAuthPath, getApiKey, getValidAccessToken, rotateAccount } from "./auth-y23hLerx.js";
|
|
8
|
+
import { clearZenAuth, getZenAuthPath } from "./auth-OZNLGzjK.js";
|
|
9
|
+
import { getAntigravityModels, getAntigravityUsage, isThinkingModel } from "./get-models-uSazerPk.js";
|
|
10
10
|
import { createRequire } from "node:module";
|
|
11
11
|
import { defineCommand, runMain } from "citty";
|
|
12
12
|
import consola from "consola";
|
|
@@ -2869,10 +2869,21 @@ tokenRoute.get("/", (c) => {
|
|
|
2869
2869
|
const usageRoute = new Hono();
|
|
2870
2870
|
usageRoute.get("/", async (c) => {
|
|
2871
2871
|
try {
|
|
2872
|
+
if (state.antigravityMode) {
|
|
2873
|
+
const usage$1 = await getAntigravityUsage();
|
|
2874
|
+
return c.json(usage$1);
|
|
2875
|
+
}
|
|
2876
|
+
if (state.zenMode) return c.json({
|
|
2877
|
+
error: "Usage statistics not available for Zen mode",
|
|
2878
|
+
message: "Please check your usage at https://console.opencode.ai",
|
|
2879
|
+
mode: "zen"
|
|
2880
|
+
}, 200);
|
|
2872
2881
|
const usage = await getCopilotUsage();
|
|
2873
2882
|
return c.json(usage);
|
|
2874
2883
|
} catch (error) {
|
|
2875
|
-
consola.error("Error fetching
|
|
2884
|
+
consola.error("Error fetching usage:", error);
|
|
2885
|
+
if (state.antigravityMode) return c.json({ error: "Failed to fetch Antigravity usage" }, 500);
|
|
2886
|
+
if (state.zenMode) return c.json({ error: "Failed to fetch Zen usage" }, 500);
|
|
2876
2887
|
return c.json({ error: "Failed to fetch Copilot usage" }, 500);
|
|
2877
2888
|
}
|
|
2878
2889
|
});
|
|
@@ -3148,7 +3159,7 @@ async function runServer(options$1) {
|
|
|
3148
3159
|
state.zenApiKey = options$1.zenApiKey;
|
|
3149
3160
|
consola.info("Using provided Zen API key");
|
|
3150
3161
|
} else {
|
|
3151
|
-
const { setupZenApiKey, loadZenAuth } = await import("./auth-
|
|
3162
|
+
const { setupZenApiKey, loadZenAuth } = await import("./auth-wHA1Oitm.js");
|
|
3152
3163
|
const existingAuth = await loadZenAuth();
|
|
3153
3164
|
if (existingAuth) {
|
|
3154
3165
|
state.zenApiKey = existingAuth.apiKey;
|
|
@@ -3161,7 +3172,7 @@ async function runServer(options$1) {
|
|
|
3161
3172
|
} else if (options$1.antigravity) {
|
|
3162
3173
|
consola.info("Google Antigravity mode enabled");
|
|
3163
3174
|
state.antigravityMode = true;
|
|
3164
|
-
const { loadAntigravityAuth, setupAntigravity, getCurrentAccount, hasApiKey, getApiKey: getApiKey$1, setOAuthCredentials } = await import("./auth-
|
|
3175
|
+
const { loadAntigravityAuth, setupAntigravity, getCurrentAccount, hasApiKey, getApiKey: getApiKey$1, setOAuthCredentials } = await import("./auth-Ijg_vdBM.js");
|
|
3165
3176
|
if (options$1.antigravityClientId && options$1.antigravityClientSecret) {
|
|
3166
3177
|
setOAuthCredentials(options$1.antigravityClientId, options$1.antigravityClientSecret);
|
|
3167
3178
|
consola.info("Using provided OAuth credentials from CLI");
|
|
@@ -3190,7 +3201,7 @@ async function runServer(options$1) {
|
|
|
3190
3201
|
}
|
|
3191
3202
|
if (!await getCurrentAccount() && !hasApiKey()) throw new Error("No enabled Antigravity accounts available");
|
|
3192
3203
|
}
|
|
3193
|
-
const { getAntigravityModels: getAntigravityModels$1 } = await import("./get-models-
|
|
3204
|
+
const { getAntigravityModels: getAntigravityModels$1 } = await import("./get-models-DqEAMDmN.js");
|
|
3194
3205
|
const models = await getAntigravityModels$1();
|
|
3195
3206
|
state.antigravityModels = models;
|
|
3196
3207
|
consola.info(`Available Antigravity models: \n${models.data.map((model) => `- ${model.id}`).join("\n")}`);
|
|
@@ -3225,17 +3236,18 @@ async function runServer(options$1) {
|
|
|
3225
3236
|
}
|
|
3226
3237
|
const serverUrl = `http://localhost:${options$1.port}`;
|
|
3227
3238
|
if (options$1.claudeCode) {
|
|
3228
|
-
let
|
|
3229
|
-
if (state.zenMode)
|
|
3230
|
-
else if (state.antigravityMode)
|
|
3231
|
-
|
|
3239
|
+
let modelList;
|
|
3240
|
+
if (state.zenMode) modelList = state.zenModels?.data;
|
|
3241
|
+
else if (state.antigravityMode) modelList = state.antigravityModels?.data;
|
|
3242
|
+
else modelList = state.models?.data;
|
|
3243
|
+
invariant(modelList, "Models should be loaded by now");
|
|
3232
3244
|
const selectedModel = await consola.prompt("Select a model to use with Claude Code", {
|
|
3233
3245
|
type: "select",
|
|
3234
|
-
options:
|
|
3246
|
+
options: modelList.map((model) => model.id)
|
|
3235
3247
|
});
|
|
3236
3248
|
const selectedSmallModel = await consola.prompt("Select a small model to use with Claude Code", {
|
|
3237
3249
|
type: "select",
|
|
3238
|
-
options:
|
|
3250
|
+
options: modelList.map((model) => model.id)
|
|
3239
3251
|
});
|
|
3240
3252
|
const command = generateEnvScript({
|
|
3241
3253
|
ANTHROPIC_BASE_URL: serverUrl,
|