From e99f34893ef934082b6dbcc3116628e6f71cad13 Mon Sep 17 00:00:00 2001
From: airwish <lubaiqi@outlook.com>
Date: Tue, 11 Feb 2025 16:16:47 +0000
Subject: [PATCH] feat: expanded web search models

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

diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts
index 2f8f9c00..5a0191b7 100644
--- a/src/renderer/src/config/models.ts
+++ b/src/renderer/src/config/models.ts
@@ -1191,7 +1191,12 @@ export function isWebSearchModel(model: Model): boolean {
   }
 
   if (provider.id === 'aihubmix') {
-    return model?.id === 'gemini-2.0-flash-exp-search'
+    const models = [
+      'gemini-2.0-flash-search',
+      'gemini-2.0-flash-exp-search',
+      'gemini-2.0-pro-exp-02-05-search'
+    ]
+    return models.includes(model?.id)
   }
 
   if (provider.id === 'zhipu') {