From 3aa2c68ce6a6ac5262d27c91b1393c3c44fef816 Mon Sep 17 00:00:00 2001
From: Davidlasky <davidjiangxc@gmail.com>
Date: Thu, 17 Apr 2025 23:47:45 -0500
Subject: [PATCH] add gemini-2.5-flash-preview-04-17 and replace
 gemini-2.5-pro-exp-03-25 with preview one

---
 app/constant.ts                   | 9 ++++++---
 test/vision-model-checker.test.ts | 4 ----
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/app/constant.ts b/app/constant.ts
index f9153e159..521f3dacf 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -416,8 +416,8 @@ export const KnowledgeCutOffDate: Record<string, string> = {
   default: "2023-10",
   // After improvements,
   // it's now easier to add "KnowledgeCutOffDate" instead of stupid hardcoding it, as was done previously.
-  "gemini-2.5-pro-exp-03-25": "2025-01",
-  "gemini-2.0-flash": "2024-08",
+  "gemini-2.5-pro-preview-03-25": "2025-01",
+  "gemini-2.5-flash-preview-04-17": "2025-01",
   "claude-3-7-sonnet-latest": "2024-10",
   "claude-3-5-haiku-latest": "2024-10",
   "gpt-4.1": "2024-06",
@@ -456,7 +456,10 @@ export const EXCLUDE_VISION_MODEL_REGEXES = [/claude-3-5-haiku-20241022/];
 
 const openaiModels = ["dall-e-3", "o1", "o4-mini", "gpt-4.1", "gpt-4.1-mini"];
 
-const googleModels = ["gemini-2.0-flash", "gemini-2.5-pro-exp-03-25"];
+const googleModels = [
+  "gemini-2.5-flash-preview-04-17",
+  "gemini-2.5-pro-preview-03-25",
+];
 
 const anthropicModels = [
   "claude-3-5-sonnet-latest",
diff --git a/test/vision-model-checker.test.ts b/test/vision-model-checker.test.ts
index fa255e30b..98b594d49 100644
--- a/test/vision-model-checker.test.ts
+++ b/test/vision-model-checker.test.ts
@@ -17,11 +17,7 @@ describe("isVisionModel", () => {
     const visionModels = [
       "gpt-4.1",
       "claude-3-opus",
-      "gemini-2.0",
       "gemini-2.5-pro",
-      "learnlm-vision",
-      "qwen-vl-max",
-      "qwen2-vl-max",
       "o4-mini",
       "dall-e-3",
       "o1",