From 0219e5dea3d9390c77544feaf73cf39b1d325618 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 18 Sep 2024 13:45:43 +0800
Subject: [PATCH] remove code

---
 packages/service/core/dataset/training/controller.ts         | 2 +-
 .../service/core/workflow/dispatch/agent/classifyQuestion.ts | 2 +-
 packages/service/core/workflow/dispatch/dataset/search.ts    | 5 +++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/packages/service/core/dataset/training/controller.ts b/packages/service/core/dataset/training/controller.ts
index e293a8183e7..d09576bf4b6 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 0491009995e..3f2c7826742 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 04f119527a6..c355f4c9f6b 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()
     }))
   };
 }