fix: 腾讯混元的联网开关 #575

This commit is contained in:
kangfenmao 2025-01-02 16:26:24 +08:00
parent 7f92cb59a6
commit a782b2b4aa
2 changed files with 11 additions and 2 deletions

View File

@ -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
}

View File

@ -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)
})