diff --git a/packages/core/src/config/models.ts b/packages/core/src/config/models.ts index 2b47afa5..32a51d97 100644 --- a/packages/core/src/config/models.ts +++ b/packages/core/src/config/models.ts @@ -4,6 +4,6 @@ * SPDX-License-Identifier: Apache-2.0 */ -export const DEFAULT_GEMINI_MODEL = 'gemini-2.5-pro-preview-06-05'; +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_EMBEDDING_MODEL = 'gemini-embedding-001'; diff --git a/packages/core/src/core/tokenLimits.ts b/packages/core/src/core/tokenLimits.ts index 0fe99876..68901b3c 100644 --- a/packages/core/src/core/tokenLimits.ts +++ b/packages/core/src/core/tokenLimits.ts @@ -18,6 +18,7 @@ export function tokenLimit(model: Model): TokenCount { case 'gemini-1.5-flash': case 'gemini-2.5-pro-preview-05-06': case 'gemini-2.5-pro-preview-06-05': + case 'gemini-2.5-pro': case 'gemini-2.5-flash-preview-05-20': case 'gemini-2.0-flash': return 1_048_576;