summaryrefslogtreecommitdiff
path: root/packages/cli/src/ui/utils/errorParsing.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/ui/utils/errorParsing.ts')
-rw-r--r--packages/cli/src/ui/utils/errorParsing.ts164
1 files changed, 0 insertions, 164 deletions
diff --git a/packages/cli/src/ui/utils/errorParsing.ts b/packages/cli/src/ui/utils/errorParsing.ts
deleted file mode 100644
index 5031bc0a..00000000
--- a/packages/cli/src/ui/utils/errorParsing.ts
+++ /dev/null
@@ -1,164 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-
-import {
- AuthType,
- UserTierId,
- DEFAULT_GEMINI_FLASH_MODEL,
- DEFAULT_GEMINI_MODEL,
- isProQuotaExceededError,
- isGenericQuotaExceededError,
- isApiError,
- isStructuredError,
-} from '@google/gemini-cli-core';
-
-// Free Tier message functions
-const getRateLimitErrorMessageGoogleFree = (
- fallbackModel: string = DEFAULT_GEMINI_FLASH_MODEL,
-) =>
- `\nPossible quota limitations in place or slow response times detected. Switching to the ${fallbackModel} model for the rest of this session.`;
-
-const getRateLimitErrorMessageGoogleProQuotaFree = (
- currentModel: string = DEFAULT_GEMINI_MODEL,
- fallbackModel: string = DEFAULT_GEMINI_FLASH_MODEL,
-) =>
- `\nYou have reached your daily ${currentModel} quota limit. You will be switched to the ${fallbackModel} model for the rest of this session. To increase your limits, upgrade to a Gemini Code Assist Standard or Enterprise plan with higher limits at https://goo.gle/set-up-gemini-code-assist, or use /auth to switch to using a paid API key from AI Studio at https://aistudio.google.com/apikey`;
-
-const getRateLimitErrorMessageGoogleGenericQuotaFree = () =>
- `\nYou have reached your daily quota limit. To increase your limits, upgrade to a Gemini Code Assist Standard or Enterprise plan with higher limits at https://goo.gle/set-up-gemini-code-assist, or use /auth to switch to using a paid API key from AI Studio at https://aistudio.google.com/apikey`;
-
-// Legacy/Standard Tier message functions
-const getRateLimitErrorMessageGooglePaid = (
- fallbackModel: string = DEFAULT_GEMINI_FLASH_MODEL,
-) =>
- `\nPossible quota limitations in place or slow response times detected. Switching to the ${fallbackModel} model for the rest of this session. We appreciate you for choosing Gemini Code Assist and the Gemini CLI.`;
-
-const getRateLimitErrorMessageGoogleProQuotaPaid = (
- currentModel: string = DEFAULT_GEMINI_MODEL,
- fallbackModel: string = DEFAULT_GEMINI_FLASH_MODEL,
-) =>
- `\nYou have reached your daily ${currentModel} quota limit. You will be switched to the ${fallbackModel} model for the rest of this session. We appreciate you for choosing Gemini Code Assist and the Gemini CLI. To continue accessing the ${currentModel} model today, consider using /auth to switch to using a paid API key from AI Studio at https://aistudio.google.com/apikey`;
-
-const getRateLimitErrorMessageGoogleGenericQuotaPaid = (
- currentModel: string = DEFAULT_GEMINI_MODEL,
-) =>
- `\nYou have reached your daily quota limit. We appreciate you for choosing Gemini Code Assist and the Gemini CLI. To continue accessing the ${currentModel} model today, consider using /auth to switch to using a paid API key from AI Studio at https://aistudio.google.com/apikey`;
-const RATE_LIMIT_ERROR_MESSAGE_USE_GEMINI =
- '\nPlease wait and try again later. To increase your limits, request a quota increase through AI Studio, or switch to another /auth method';
-const RATE_LIMIT_ERROR_MESSAGE_VERTEX =
- '\nPlease wait and try again later. To increase your limits, request a quota increase through Vertex, or switch to another /auth method';
-const getRateLimitErrorMessageDefault = (
- fallbackModel: string = DEFAULT_GEMINI_FLASH_MODEL,
-) =>
- `\nPossible quota limitations in place or slow response times detected. Switching to the ${fallbackModel} model for the rest of this session.`;
-
-function getRateLimitMessage(
- authType?: AuthType,
- error?: unknown,
- userTier?: UserTierId,
- currentModel?: string,
- fallbackModel?: string,
-): string {
- switch (authType) {
- case AuthType.LOGIN_WITH_GOOGLE: {
- // Determine if user is on a paid tier (Legacy or Standard) - default to FREE if not specified
- const isPaidTier =
- userTier === UserTierId.LEGACY || userTier === UserTierId.STANDARD;
-
- if (isProQuotaExceededError(error)) {
- return isPaidTier
- ? getRateLimitErrorMessageGoogleProQuotaPaid(
- currentModel || DEFAULT_GEMINI_MODEL,
- fallbackModel,
- )
- : getRateLimitErrorMessageGoogleProQuotaFree(
- currentModel || DEFAULT_GEMINI_MODEL,
- fallbackModel,
- );
- } else if (isGenericQuotaExceededError(error)) {
- return isPaidTier
- ? getRateLimitErrorMessageGoogleGenericQuotaPaid(
- currentModel || DEFAULT_GEMINI_MODEL,
- )
- : getRateLimitErrorMessageGoogleGenericQuotaFree();
- } else {
- return isPaidTier
- ? getRateLimitErrorMessageGooglePaid(fallbackModel)
- : getRateLimitErrorMessageGoogleFree(fallbackModel);
- }
- }
- case AuthType.USE_GEMINI:
- return RATE_LIMIT_ERROR_MESSAGE_USE_GEMINI;
- case AuthType.USE_VERTEX_AI:
- return RATE_LIMIT_ERROR_MESSAGE_VERTEX;
- default:
- return getRateLimitErrorMessageDefault(fallbackModel);
- }
-}
-
-export function parseAndFormatApiError(
- error: unknown,
- authType?: AuthType,
- userTier?: UserTierId,
- currentModel?: string,
- fallbackModel?: string,
-): string {
- if (isStructuredError(error)) {
- let text = `[API Error: ${error.message}]`;
- if (error.status === 429) {
- text += getRateLimitMessage(
- authType,
- error,
- userTier,
- currentModel,
- fallbackModel,
- );
- }
- return text;
- }
-
- // The error message might be a string containing a JSON object.
- if (typeof error === 'string') {
- const jsonStart = error.indexOf('{');
- if (jsonStart === -1) {
- return `[API Error: ${error}]`; // Not a JSON error, return as is.
- }
-
- const jsonString = error.substring(jsonStart);
-
- try {
- const parsedError = JSON.parse(jsonString) as unknown;
- if (isApiError(parsedError)) {
- let finalMessage = parsedError.error.message;
- try {
- // See if the message is a stringified JSON with another error
- const nestedError = JSON.parse(finalMessage) as unknown;
- if (isApiError(nestedError)) {
- finalMessage = nestedError.error.message;
- }
- } catch (_e) {
- // It's not a nested JSON error, so we just use the message as is.
- }
- let text = `[API Error: ${finalMessage} (Status: ${parsedError.error.status})]`;
- if (parsedError.error.code === 429) {
- text += getRateLimitMessage(
- authType,
- parsedError,
- userTier,
- currentModel,
- fallbackModel,
- );
- }
- return text;
- }
- } catch (_e) {
- // Not a valid JSON, fall through and return the original message.
- }
- return `[API Error: ${error}]`;
- }
-
- return '[API Error: An unknown error occurred.]';
-}