summaryrefslogtreecommitdiff
path: root/packages/core/src
diff options
context:
space:
mode:
Diffstat (limited to 'packages/core/src')
-rw-r--r--packages/core/src/config/models.ts2
-rw-r--r--packages/core/src/core/tokenLimits.ts1
2 files changed, 2 insertions, 1 deletions
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;