feat(cli): handle invalid model names in useQuotaAndFallback (#19222)

This commit is contained in:
Sehoon Shon
2026-02-16 14:55:17 -05:00
committed by GitHub
parent 5a74f7a2eb
commit 15ef1cd797
2 changed files with 54 additions and 9 deletions

View File

@@ -83,16 +83,21 @@ export function useQuotaAndFallback({
`/auth to switch to API key.`,
].filter(Boolean);
message = messageLines.join('\n');
} else if (
error instanceof ModelNotFoundError &&
VALID_GEMINI_MODELS.has(failedModel)
) {
} else if (error instanceof ModelNotFoundError) {
isModelNotFoundError = true;
const messageLines = [
`It seems like you don't have access to ${failedModel}.`,
`Your admin might have disabled the access. Contact them to enable the Preview Release Channel.`,
];
message = messageLines.join('\n');
if (VALID_GEMINI_MODELS.has(failedModel)) {
const messageLines = [
`It seems like you don't have access to ${failedModel}.`,
`Your admin might have disabled the access. Contact them to enable the Preview Release Channel.`,
];
message = messageLines.join('\n');
} else {
const messageLines = [
`Model "${failedModel}" was not found or is invalid.`,
`/model to switch models.`,
];
message = messageLines.join('\n');
}
} else {
const messageLines = [
`We are currently experiencing high demand.`,