diff --git a/packages/core/src/config/models.ts b/packages/core/src/config/models.ts index 32a51d97..07e0e051 100644 --- a/packages/core/src/config/models.ts +++ b/packages/core/src/config/models.ts @@ -5,5 +5,5 @@ */ export const DEFAULT_GEMINI_MODEL = 'gemini-2.5-pro'; -export const DEFAULT_GEMINI_FLASH_MODEL = 'gemini-2.5-flash-preview-05-20'; +export const DEFAULT_GEMINI_FLASH_MODEL = 'gemini-2.5-flash'; export const DEFAULT_GEMINI_EMBEDDING_MODEL = 'gemini-embedding-001'; diff --git a/packages/core/src/core/tokenLimits.ts b/packages/core/src/core/tokenLimits.ts index 68901b3c..1c7fbde9 100644 --- a/packages/core/src/core/tokenLimits.ts +++ b/packages/core/src/core/tokenLimits.ts @@ -20,6 +20,7 @@ export function tokenLimit(model: Model): TokenCount { case 'gemini-2.5-pro-preview-06-05': case 'gemini-2.5-pro': case 'gemini-2.5-flash-preview-05-20': + case 'gemini-2.5-flash': case 'gemini-2.0-flash': return 1_048_576; case 'gemini-2.0-flash-preview-image-generation':