From 1fc1e3fa808228dda642e684f24ed04e10992a8b Mon Sep 17 00:00:00 2001 From: Archer <545436317@qq.com> Date: Fri, 8 Aug 2025 10:38:57 +0800 Subject: [PATCH] fix: max tokens config (#5409) --- packages/service/core/ai/config/utils.ts | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/packages/service/core/ai/config/utils.ts b/packages/service/core/ai/config/utils.ts index 9aabd64cb..af38fc1b4 100644 --- a/packages/service/core/ai/config/utils.ts +++ b/packages/service/core/ai/config/utils.ts @@ -113,13 +113,18 @@ export const loadSystemModels = async (init = false) => { const modelData: any = { ...model, ...dbModel?.metadata, - // @ts-ignore - defaultConfig: mergeObject(model.defaultConfig, dbModel?.metadata?.defaultConfig), - // @ts-ignore - fieldMap: mergeObject(model.fieldMap, dbModel?.metadata?.fieldMap), provider: getModelProvider(dbModel?.metadata?.provider || (model.provider as any)).id, type: dbModel?.metadata?.type || model.type, - isCustom: false + isCustom: false, + + ...(model.type === ModelTypeEnum.llm && dbModel?.metadata?.type === ModelTypeEnum.llm + ? { + maxResponse: model.maxTokens ?? dbModel?.metadata?.maxResponse ?? 1000, + defaultConfig: mergeObject(model.defaultConfig, dbModel?.metadata?.defaultConfig), + fieldMap: mergeObject(model.fieldMap, dbModel?.metadata?.fieldMap), + maxTokens: undefined + } + : {}) }; pushModel(modelData); })