From 496b4684ea07aef699d1197c5f6746276fcf7286 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Fri, 7 Feb 2025 17:17:56 +0800
Subject: [PATCH] feat: expand supported Gemini models for web search #1096

---
 src/renderer/src/config/models.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts
index 639550a9..2f8f9c00 100644
--- a/src/renderer/src/config/models.ts
+++ b/src/renderer/src/config/models.ts
@@ -1176,7 +1176,14 @@ export function isWebSearchModel(model: Model): boolean {
   }
 
   if (provider.id === 'gemini' || provider?.type === 'gemini') {
-    return model?.id === 'gemini-2.0-flash-exp'
+    const models = [
+      'gemini-2.0-flash',
+      'gemini-2.0-flash-exp',
+      'gemini-2.0-flash-001',
+      'gemini-2.0-pro-exp-02-05',
+      'gemini-2.0-pro-exp'
+    ]
+    return models.includes(model?.id)
   }
 
   if (provider.id === 'hunyuan') {