From a782b2b4aa41097b0684d2e7278b9e571de2a69b Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Thu, 2 Jan 2025 16:26:24 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E8=85=BE=E8=AE=AF=E6=B7=B7=E5=85=83?=
 =?UTF-8?q?=E7=9A=84=E8=81=94=E7=BD=91=E5=BC=80=E5=85=B3=20#575?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/renderer/src/config/models.ts            | 10 +++++++++-
 src/renderer/src/providers/OpenAIProvider.ts |  3 ++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts
index c9e6174c..0556c0f6 100644
--- a/src/renderer/src/config/models.ts
+++ b/src/renderer/src/config/models.ts
@@ -1009,5 +1009,13 @@ export function isWebSearchModel(model: Model): boolean {
     return false
   }
 
-  return (provider.id === 'gemini' || provider?.type === 'gemini') && model?.id === 'gemini-2.0-flash-exp'
+  if (provider.id === 'gemini' || provider?.type === 'gemini') {
+    return model?.id === 'gemini-2.0-flash-exp'
+  }
+
+  if (provider.id === 'hunyuan') {
+    return model?.id !== 'hunyuan-lite'
+  }
+
+  return false
 }
diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts
index 3c19be48..6110ff7d 100644
--- a/src/renderer/src/providers/OpenAIProvider.ts
+++ b/src/renderer/src/providers/OpenAIProvider.ts
@@ -1,4 +1,4 @@
-import { isEmbeddingModel, isSupportedModel, isVisionModel } from '@renderer/config/models'
+import { isEmbeddingModel, isSupportedModel, isVisionModel, isWebSearchModel } from '@renderer/config/models'
 import { getStoreSetting } from '@renderer/hooks/useSettings'
 import i18n from '@renderer/i18n'
 import { getAssistantSettings, getDefaultModel, getTopNamingModel } from '@renderer/services/AssistantService'
@@ -149,6 +149,7 @@ export default class OpenAIProvider extends BaseProvider {
       max_tokens: maxTokens,
       keep_alive: this.keepAliveTime,
       stream: isSupportStreamOutput,
+      ...(isWebSearchModel(model) ? { enable_enhancement: true } : {}),
       ...this.getCustomParameters(assistant)
     })