feat: Enhance web search parameter handling for assistants #575

src/renderer/src/config/models.ts

@@ -123,7 +123,7 @@ import WenxinModelLogoDark from '@renderer/assets/images/models/wenxin_dark.png'
 import YiModelLogo from '@renderer/assets/images/models/yi.png'
 import YiModelLogoDark from '@renderer/assets/images/models/yi_dark.png'
 import { getProviderByModel } from '@renderer/services/AssistantService'
-import { Model } from '@renderer/types'
+import { Assistant, Model } from '@renderer/types'
 import OpenAI from 'openai'
 
 import { getWebSearchTools } from './tools'
@@ -1182,16 +1182,22 @@ export function isWebSearchModel(model: Model): boolean {
   return false
 }
 
-export function getOpenAIWebSearchParams(model: Model): Record<string, any> {
+export function getOpenAIWebSearchParams(assistant: Assistant, model: Model): Record<string, any> {
   if (isWebSearchModel(model)) {
-    const webSearchTools = getWebSearchTools(model)
+    if (assistant.enableWebSearch) {
+      const webSearchTools = getWebSearchTools(model)
 
-    if (model.provider === 'hunyuan') {
+      if (model.provider === 'hunyuan') {
-      return { enable_enhancement: true }
+        return { enable_enhancement: true }
-    }
+      }
 
-    return {
+      return {
-      tools: webSearchTools
+        tools: webSearchTools
+      }
+    } else {
+      if (model.provider === 'hunyuan') {
+        return { enable_enhancement: false }
+      }
     }
   }
 

src/renderer/src/providers/OpenAIProvider.ts

@@ -193,7 +193,7 @@ export default class OpenAIProvider extends BaseProvider {
       keep_alive: this.keepAliveTime,
       stream: isSupportStreamOutput(),
       reasoning_effort: this.getReasoningEffort(assistant, model),
-      ...(assistant.enableWebSearch ? getOpenAIWebSearchParams(model) : {}),
+      ...getOpenAIWebSearchParams(assistant, model),
       ...this.getProviderSpecificParameters(model),
       ...this.getCustomParameters(assistant)
     })

src/renderer/src/providers/QwenLMProvider.ts

@@ -120,7 +120,7 @@ class QwenLMProvider extends OpenAIProvider {
       top_p: assistant?.settings?.topP,
       max_tokens: maxTokens,
       stream: true,
-      ...(assistant.enableWebSearch ? getOpenAIWebSearchParams(model) : {}),
+      ...getOpenAIWebSearchParams(assistant, model),
       ...this.getCustomParameters(assistant)
     })