diff --git a/packages/service/core/dataset/training/controller.ts b/packages/service/core/dataset/training/controller.ts index e293a8183..d09576bf4 100644 --- a/packages/service/core/dataset/training/controller.ts +++ b/packages/service/core/dataset/training/controller.ts @@ -183,7 +183,7 @@ export async function pushDataListToTrainingQueue({ }); console.log('failed', failedDocuments); } - console.log(startIndex, '==='); + // 对于失败的文档,尝试单独插入 await MongoDatasetTraining.create(failedDocuments, { session }); diff --git a/packages/service/core/workflow/dispatch/agent/classifyQuestion.ts b/packages/service/core/workflow/dispatch/agent/classifyQuestion.ts index 049100999..3f2c78267 100644 --- a/packages/service/core/workflow/dispatch/agent/classifyQuestion.ts +++ b/packages/service/core/workflow/dispatch/agent/classifyQuestion.ts @@ -107,7 +107,7 @@ const completions = async ({ .join('\n------\n'), history: histories .map((item) => `${item.obj}:${chatValue2RuntimePrompt(item.value).text}`) - .join('------'), + .join('\n------\n'), question: userChatInput }) } diff --git a/packages/service/core/workflow/dispatch/dataset/search.ts b/packages/service/core/workflow/dispatch/dataset/search.ts index 04f119527..c355f4c9f 100644 --- a/packages/service/core/workflow/dispatch/dataset/search.ts +++ b/packages/service/core/workflow/dispatch/dataset/search.ts @@ -159,8 +159,9 @@ export async function dispatchDatasetSearch( [DispatchNodeResponseKeyEnum.nodeResponse]: responseData, nodeDispatchUsages, [DispatchNodeResponseKeyEnum.toolResponses]: searchRes.map((item) => ({ - id: item.id, - text: `${item.q}\n${item.a}`.trim() + sourceName: item.sourceName, + updateTime: item.updateTime, + content: `${item.q}\n${item.a}`.trim() })) }; }