From 012e032315d23931f6fd232b71245eaa33b8219d Mon Sep 17 00:00:00 2001
From: Kiyoto Ryuman <46114742+go-to-the-future@users.noreply.github.com>
Date: Mon, 28 Apr 2025 14:19:29 +0900
Subject: [PATCH] add gemini 2.5 pro and flash models

---
 .changeset/cold-planets-confess.md |  5 +++++
 docs/providers/gemini.md           |  2 ++
 src/models.ts                      | 10 ++++++++++
 3 files changed, 17 insertions(+)
 create mode 100644 .changeset/cold-planets-confess.md

diff --git a/.changeset/cold-planets-confess.md b/.changeset/cold-planets-confess.md
new file mode 100644
index 0000000..d28fb77
--- /dev/null
+++ b/.changeset/cold-planets-confess.md
@@ -0,0 +1,5 @@
+---
+'token.js': patch
+---
+
+Add gemini 2.5 pro and flash preview models
diff --git a/docs/providers/gemini.md b/docs/providers/gemini.md
index 5f93c84..6be8d57 100644
--- a/docs/providers/gemini.md
+++ b/docs/providers/gemini.md
@@ -46,6 +46,8 @@ main()
 | gemini-1.5-flash                    | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
 | gemini-1.5-flash-8b                 | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
 | gemini-1.0-pro                      | ✅               | ✅         | ➖           | ➖           | ✅                | ✅     |
+| gemini-2.5-flash-preview-04-17      | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
+| gemini-2.5-pro-preview-03-25        | ✅               | ✅         | ✅           | ✅           | ✅                | ✅     |
 
 ### Legend
 | Symbol             | Description                           |
diff --git a/src/models.ts b/src/models.ts
index f506c6a..dcf270b 100644
--- a/src/models.ts
+++ b/src/models.ts
@@ -205,6 +205,8 @@ export const models = {
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
       'gemini-1.0-pro',
+      'gemini-2.5-flash-preview-04-17',
+      'gemini-2.5-pro-preview-03-25',
     ] as const,
     supportsCompletion: true,
     supportsStreaming: [
@@ -214,6 +216,8 @@ export const models = {
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
       'gemini-1.0-pro',
+      'gemini-2.5-flash-preview-04-17',
+      'gemini-2.5-pro-preview-03-25',
     ] as const,
     supportsJSON: [
       'gemini-2.0-flash-001',
@@ -221,6 +225,8 @@ export const models = {
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
+      'gemini-2.5-flash-preview-04-17',
+      'gemini-2.5-pro-preview-03-25',
     ] as const,
     supportsImages: [
       'gemini-2.0-flash-001',
@@ -228,6 +234,8 @@ export const models = {
       'gemini-1.5-pro',
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
+      'gemini-2.5-flash-preview-04-17',
+      'gemini-2.5-pro-preview-03-25',
     ] as const,
     supportsToolCalls: [
       'gemini-2.0-flash-001',
@@ -235,6 +243,8 @@ export const models = {
       'gemini-1.5-flash',
       'gemini-1.5-flash-8b',
       'gemini-1.0-pro',
+      'gemini-2.5-flash-preview-04-17',
+      'gemini-2.5-pro-preview-03-25',
     ] as const,
     supportsN: true,
     generateDocs: true,