fix: enhance web search recognization in AI providers (#6423)

This commit is contained in:
SuYao 2025-05-25 21:13:29 +08:00 committed by GitHub
parent 47b1133cf1
commit 2b6d02f470
3 changed files with 6 additions and 4 deletions

View File

@ -254,7 +254,7 @@ export default class AnthropicProvider extends BaseProvider {
}
}
const isEnabledBuiltinWebSearch = assistant.enableWebSearch
const isEnabledBuiltinWebSearch = assistant.enableWebSearch && isWebSearchModel(model)
if (isEnabledBuiltinWebSearch) {
const webSearchTool = await this.getWebSearchParams(model)

View File

@ -15,6 +15,7 @@ import {
isSupportedThinkingTokenModel,
isSupportedThinkingTokenQwenModel,
isVisionModel,
isWebSearchModel,
isZhipuModel
} from '@renderer/config/models'
import { getStoreSetting } from '@renderer/hooks/useSettings'
@ -371,7 +372,7 @@ export default class OpenAIProvider extends BaseOpenAIProvider {
const model = assistant.model || defaultModel
const { contextCount, maxTokens, streamOutput } = getAssistantSettings(assistant)
const isEnabledBultinWebSearch = assistant.enableWebSearch
const isEnabledBultinWebSearch = assistant.enableWebSearch && isWebSearchModel(model)
messages = addImageFileToContents(messages)
const enableReasoning =
((isSupportedThinkingTokenModel(model) || isSupportedReasoningEffortModel(model)) &&

View File

@ -5,7 +5,8 @@ import {
isSupportedFlexServiceTier,
isSupportedModel,
isSupportedReasoningEffortOpenAIModel,
isVisionModel
isVisionModel,
isWebSearchModel
} from '@renderer/config/models'
import { getStoreSetting } from '@renderer/hooks/useSettings'
import i18n from '@renderer/i18n'
@ -317,7 +318,7 @@ export abstract class BaseOpenAIProvider extends BaseProvider {
const model = assistant.model || defaultModel
const { contextCount, maxTokens, streamOutput } = getAssistantSettings(assistant)
const isEnabledBuiltinWebSearch = assistant.enableWebSearch
const isEnabledBuiltinWebSearch = assistant.enableWebSearch && isWebSearchModel(model)
let tools: OpenAI.Responses.Tool[] = []
const toolChoices: OpenAI.Responses.ToolChoiceTypes = {