Add error messaging for 429 errors (#1316)
This commit is contained in:
parent
21e6a36cf1
commit
dc76bcc433
|
@ -215,8 +215,7 @@ describe('runNonInteractive', () => {
|
||||||
await runNonInteractive(mockConfig, 'Initial fail');
|
await runNonInteractive(mockConfig, 'Initial fail');
|
||||||
|
|
||||||
expect(consoleErrorSpy).toHaveBeenCalledWith(
|
expect(consoleErrorSpy).toHaveBeenCalledWith(
|
||||||
'Error processing input:',
|
'[API Error: API connection failed]',
|
||||||
apiError,
|
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
|
@ -19,6 +19,8 @@ import {
|
||||||
GenerateContentResponse,
|
GenerateContentResponse,
|
||||||
} from '@google/genai';
|
} from '@google/genai';
|
||||||
|
|
||||||
|
import { parseAndFormatApiError } from './ui/utils/errorParsing.js';
|
||||||
|
|
||||||
function getResponseText(response: GenerateContentResponse): string | null {
|
function getResponseText(response: GenerateContentResponse): string | null {
|
||||||
if (response.candidates && response.candidates.length > 0) {
|
if (response.candidates && response.candidates.length > 0) {
|
||||||
const candidate = response.candidates[0];
|
const candidate = response.candidates[0];
|
||||||
|
@ -126,7 +128,7 @@ export async function runNonInteractive(
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error('Error processing input:', error);
|
console.error(parseAndFormatApiError(error));
|
||||||
process.exit(1);
|
process.exit(1);
|
||||||
} finally {
|
} finally {
|
||||||
if (isTelemetrySdkInitialized()) {
|
if (isTelemetrySdkInitialized()) {
|
||||||
|
|
|
@ -396,7 +396,10 @@ export const useGeminiStream = (
|
||||||
setPendingHistoryItem(null);
|
setPendingHistoryItem(null);
|
||||||
}
|
}
|
||||||
addItem(
|
addItem(
|
||||||
{ type: MessageType.ERROR, text: `[API Error: ${eventValue.message}]` },
|
{
|
||||||
|
type: MessageType.ERROR,
|
||||||
|
text: parseAndFormatApiError(eventValue.error),
|
||||||
|
},
|
||||||
userMessageTimestamp,
|
userMessageTimestamp,
|
||||||
);
|
);
|
||||||
},
|
},
|
||||||
|
@ -530,6 +533,10 @@ export const useGeminiStream = (
|
||||||
setPendingHistoryItem(null);
|
setPendingHistoryItem(null);
|
||||||
}
|
}
|
||||||
} catch (error: unknown) {
|
} catch (error: unknown) {
|
||||||
|
console.log(
|
||||||
|
'GEMINI_DEBUG: Caught error in useGeminiStream.ts:',
|
||||||
|
JSON.stringify(error),
|
||||||
|
);
|
||||||
if (isAuthError(error)) {
|
if (isAuthError(error)) {
|
||||||
onAuthError();
|
onAuthError();
|
||||||
} else if (!isNodeError(error) || error.name !== 'AbortError') {
|
} else if (!isNodeError(error) || error.name !== 'AbortError') {
|
||||||
|
|
|
@ -6,29 +6,46 @@
|
||||||
|
|
||||||
import { describe, it, expect } from 'vitest';
|
import { describe, it, expect } from 'vitest';
|
||||||
import { parseAndFormatApiError } from './errorParsing.js';
|
import { parseAndFormatApiError } from './errorParsing.js';
|
||||||
|
import { StructuredError } from '@gemini-cli/core';
|
||||||
|
|
||||||
describe('parseAndFormatApiError', () => {
|
describe('parseAndFormatApiError', () => {
|
||||||
|
const rateLimitMessage =
|
||||||
|
'Please wait and try again later. To increase your limits, upgrade to a plan with higher limits, or use /auth to switch to using a paid API key from AI Studio at https://aistudio.google.com/apikey';
|
||||||
|
|
||||||
it('should format a valid API error JSON', () => {
|
it('should format a valid API error JSON', () => {
|
||||||
const errorMessage =
|
const errorMessage =
|
||||||
'got status: 400 Bad Request. {"error":{"code":400,"message":"API key not valid. Please pass a valid API key.","status":"INVALID_ARGUMENT"}}';
|
'got status: 400 Bad Request. {"error":{"code":400,"message":"API key not valid. Please pass a valid API key.","status":"INVALID_ARGUMENT"}}';
|
||||||
const expected =
|
const expected =
|
||||||
'API Error: API key not valid. Please pass a valid API key. (Status: INVALID_ARGUMENT)';
|
'[API Error: API key not valid. Please pass a valid API key. (Status: INVALID_ARGUMENT)]';
|
||||||
|
expect(parseAndFormatApiError(errorMessage)).toBe(expected);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should format a 429 API error JSON with the custom message', () => {
|
||||||
|
const errorMessage =
|
||||||
|
'got status: 429 Too Many Requests. {"error":{"code":429,"message":"Rate limit exceeded","status":"RESOURCE_EXHAUSTED"}}';
|
||||||
|
const expected = `[API Error: Rate limit exceeded (Status: RESOURCE_EXHAUSTED)]\n${rateLimitMessage}`;
|
||||||
expect(parseAndFormatApiError(errorMessage)).toBe(expected);
|
expect(parseAndFormatApiError(errorMessage)).toBe(expected);
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return the original message if it is not a JSON error', () => {
|
it('should return the original message if it is not a JSON error', () => {
|
||||||
const errorMessage = 'This is a plain old error message';
|
const errorMessage = 'This is a plain old error message';
|
||||||
expect(parseAndFormatApiError(errorMessage)).toBe(errorMessage);
|
expect(parseAndFormatApiError(errorMessage)).toBe(
|
||||||
|
`[API Error: ${errorMessage}]`,
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should return the original message for malformed JSON', () => {
|
it('should return the original message for malformed JSON', () => {
|
||||||
const errorMessage = '[Stream Error: {"error": "malformed}';
|
const errorMessage = '[Stream Error: {"error": "malformed}';
|
||||||
expect(parseAndFormatApiError(errorMessage)).toBe(errorMessage);
|
expect(parseAndFormatApiError(errorMessage)).toBe(
|
||||||
|
`[API Error: ${errorMessage}]`,
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should handle JSON that does not match the ApiError structure', () => {
|
it('should handle JSON that does not match the ApiError structure', () => {
|
||||||
const errorMessage = '[Stream Error: {"not_an_error": "some other json"}]';
|
const errorMessage = '[Stream Error: {"not_an_error": "some other json"}]';
|
||||||
expect(parseAndFormatApiError(errorMessage)).toBe(errorMessage);
|
expect(parseAndFormatApiError(errorMessage)).toBe(
|
||||||
|
`[API Error: ${errorMessage}]`,
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
it('should format a nested API error', () => {
|
it('should format a nested API error', () => {
|
||||||
|
@ -49,8 +66,32 @@ describe('parseAndFormatApiError', () => {
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
expect(parseAndFormatApiError(errorMessage)).toBe(
|
const expected = `[API Error: Gemini 2.5 Pro Preview doesn't have a free quota tier. For more information on this error, head to: https://ai.google.dev/gemini-api/docs/rate-limits. (Status: Too Many Requests)]\n${rateLimitMessage}`;
|
||||||
"API Error: Gemini 2.5 Pro Preview doesn't have a free quota tier. For more information on this error, head to: https://ai.google.dev/gemini-api/docs/rate-limits. (Status: Too Many Requests)",
|
|
||||||
);
|
expect(parseAndFormatApiError(errorMessage)).toBe(expected);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should format a StructuredError', () => {
|
||||||
|
const error: StructuredError = {
|
||||||
|
message: 'A structured error occurred',
|
||||||
|
status: 500,
|
||||||
|
};
|
||||||
|
const expected = '[API Error: A structured error occurred]';
|
||||||
|
expect(parseAndFormatApiError(error)).toBe(expected);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should format a 429 StructuredError with the custom message', () => {
|
||||||
|
const error: StructuredError = {
|
||||||
|
message: 'Rate limit exceeded',
|
||||||
|
status: 429,
|
||||||
|
};
|
||||||
|
const expected = `[API Error: Rate limit exceeded]\n${rateLimitMessage}`;
|
||||||
|
expect(parseAndFormatApiError(error)).toBe(expected);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should handle an unknown error type', () => {
|
||||||
|
const error = 12345;
|
||||||
|
const expected = '[API Error: An unknown error occurred.]';
|
||||||
|
expect(parseAndFormatApiError(error)).toBe(expected);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
|
@ -4,6 +4,11 @@
|
||||||
* SPDX-License-Identifier: Apache-2.0
|
* SPDX-License-Identifier: Apache-2.0
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
import { StructuredError } from '@gemini-cli/core';
|
||||||
|
|
||||||
|
const RATE_LIMIT_ERROR_MESSAGE =
|
||||||
|
'\nPlease wait and try again later. To increase your limits, upgrade to a plan with higher limits, or use /auth to switch to using a paid API key from AI Studio at https://aistudio.google.com/apikey';
|
||||||
|
|
||||||
export interface ApiError {
|
export interface ApiError {
|
||||||
error: {
|
error: {
|
||||||
code: number;
|
code: number;
|
||||||
|
@ -23,20 +28,37 @@ function isApiError(error: unknown): error is ApiError {
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
export function parseAndFormatApiError(errorMessage: string): string {
|
function isStructuredError(error: unknown): error is StructuredError {
|
||||||
// The error message might be prefixed with some text, like "[Stream Error: ...]".
|
return (
|
||||||
// We want to find the start of the JSON object.
|
typeof error === 'object' &&
|
||||||
const jsonStart = errorMessage.indexOf('{');
|
error !== null &&
|
||||||
if (jsonStart === -1) {
|
'message' in error &&
|
||||||
return errorMessage; // Not a JSON error, return as is.
|
typeof (error as StructuredError).message === 'string'
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function parseAndFormatApiError(error: unknown): string {
|
||||||
|
if (isStructuredError(error)) {
|
||||||
|
let text = `[API Error: ${error.message}]`;
|
||||||
|
if (error.status === 429) {
|
||||||
|
text += RATE_LIMIT_ERROR_MESSAGE;
|
||||||
|
}
|
||||||
|
return text;
|
||||||
}
|
}
|
||||||
|
|
||||||
const jsonString = errorMessage.substring(jsonStart);
|
// 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 {
|
try {
|
||||||
const error = JSON.parse(jsonString) as unknown;
|
const parsedError = JSON.parse(jsonString) as unknown;
|
||||||
if (isApiError(error)) {
|
if (isApiError(parsedError)) {
|
||||||
let finalMessage = error.error.message;
|
let finalMessage = parsedError.error.message;
|
||||||
try {
|
try {
|
||||||
// See if the message is a stringified JSON with another error
|
// See if the message is a stringified JSON with another error
|
||||||
const nestedError = JSON.parse(finalMessage) as unknown;
|
const nestedError = JSON.parse(finalMessage) as unknown;
|
||||||
|
@ -46,11 +68,17 @@ export function parseAndFormatApiError(errorMessage: string): string {
|
||||||
} catch (_e) {
|
} catch (_e) {
|
||||||
// It's not a nested JSON error, so we just use the message as is.
|
// It's not a nested JSON error, so we just use the message as is.
|
||||||
}
|
}
|
||||||
return `API Error: ${finalMessage} (Status: ${error.error.status})`;
|
let text = `[API Error: ${finalMessage} (Status: ${parsedError.error.status})]`;
|
||||||
|
if (parsedError.error.code === 429) {
|
||||||
|
text += RATE_LIMIT_ERROR_MESSAGE;
|
||||||
|
}
|
||||||
|
return text;
|
||||||
}
|
}
|
||||||
} catch (_e) {
|
} catch (_e) {
|
||||||
// Not a valid JSON, fall through and return the original message.
|
// Not a valid JSON, fall through and return the original message.
|
||||||
}
|
}
|
||||||
|
return `[API Error: ${error}]`;
|
||||||
|
}
|
||||||
|
|
||||||
return errorMessage;
|
return '[API Error: An unknown error occurred.]';
|
||||||
}
|
}
|
||||||
|
|
|
@ -233,7 +233,9 @@ describe('Turn', () => {
|
||||||
expect(events.length).toBe(1);
|
expect(events.length).toBe(1);
|
||||||
const errorEvent = events[0] as ServerGeminiErrorEvent;
|
const errorEvent = events[0] as ServerGeminiErrorEvent;
|
||||||
expect(errorEvent.type).toBe(GeminiEventType.Error);
|
expect(errorEvent.type).toBe(GeminiEventType.Error);
|
||||||
expect(errorEvent.value).toEqual({ message: 'API Error' });
|
expect(errorEvent.value).toEqual({
|
||||||
|
error: { message: 'API Error', status: undefined },
|
||||||
|
});
|
||||||
expect(turn.getDebugResponses().length).toBe(0);
|
expect(turn.getDebugResponses().length).toBe(0);
|
||||||
expect(reportError).toHaveBeenCalledWith(
|
expect(reportError).toHaveBeenCalledWith(
|
||||||
error,
|
error,
|
||||||
|
|
|
@ -49,8 +49,13 @@ export enum GeminiEventType {
|
||||||
Thought = 'thought',
|
Thought = 'thought',
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface GeminiErrorEventValue {
|
export interface StructuredError {
|
||||||
message: string;
|
message: string;
|
||||||
|
status?: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface GeminiErrorEventValue {
|
||||||
|
error: StructuredError;
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface ToolCallRequestInfo {
|
export interface ToolCallRequestInfo {
|
||||||
|
@ -236,8 +241,18 @@ export class Turn {
|
||||||
contextForReport,
|
contextForReport,
|
||||||
'Turn.run-sendMessageStream',
|
'Turn.run-sendMessageStream',
|
||||||
);
|
);
|
||||||
const errorMessage = getErrorMessage(error);
|
const status =
|
||||||
yield { type: GeminiEventType.Error, value: { message: errorMessage } };
|
typeof error === 'object' &&
|
||||||
|
error !== null &&
|
||||||
|
'status' in error &&
|
||||||
|
typeof (error as { status: unknown }).status === 'number'
|
||||||
|
? (error as { status: number }).status
|
||||||
|
: undefined;
|
||||||
|
const structuredError: StructuredError = {
|
||||||
|
message: getErrorMessage(error),
|
||||||
|
status,
|
||||||
|
};
|
||||||
|
yield { type: GeminiEventType.Error, value: { error: structuredError } };
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue