feat(aiCore): 添加对智谱AI模型的支持

在getReasoningEffort函数中新增对智谱AI模型的支持逻辑,包括判断模型类型及设置对应的推理参数
This commit is contained in:
icarus 2025-08-29 19:19:54 +08:00
parent a7b8b40301
commit 58874a954c

View File

@ -20,6 +20,7 @@ import {
isSupportedThinkingTokenHunyuanModel,
isSupportedThinkingTokenModel,
isSupportedThinkingTokenQwenModel,
isSupportedThinkingTokenZhipuModel,
MODEL_SUPPORTED_REASONING_EFFORT
} from '@renderer/config/models'
import { isSupportEnableThinkingProvider } from '@renderer/config/providers'
@ -88,7 +89,8 @@ export function getReasoningEffort(assistant: Assistant, model: Model): Reasonin
return {}
}
if (isSupportedThinkingTokenDoubaoModel(model)) {
// use thinking, doubao, zhipu, etc.
if (isSupportedThinkingTokenDoubaoModel(model) || isSupportedThinkingTokenZhipuModel(model)) {
return { thinking: { type: 'disabled' } }
}
@ -254,7 +256,7 @@ export function getReasoningEffort(assistant: Assistant, model: Model): Reasonin
}
}
// Doubao models, use thinking
// Use thinking, doubao, zhipu, etc.
if (isSupportedThinkingTokenDoubaoModel(model)) {
if (assistant.settings?.reasoning_effort === 'high') {
return {
@ -264,6 +266,9 @@ export function getReasoningEffort(assistant: Assistant, model: Model): Reasonin
}
}
}
if (isSupportedThinkingTokenZhipuModel(model)) {
return { thinking: { type: 'enabled' } }
}
// Default case: no special thinking settings
return {}