From 80aeecfe01ed96036cae65d64b354a09ee16c0bb Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Thu, 28 Nov 2024 16:30:44 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E6=89=A7=E8=A1=8C=E8=AF=A6?=
 =?UTF-8?q?=E6=83=85=E9=87=8C=E5=B1=95=E7=A4=BA=E6=96=87=E6=A1=A3=E6=8F=90?=
 =?UTF-8?q?=E5=8F=96=E6=97=A5=E5=BF=97=E6=97=B6=E5=A4=9A=E4=B8=AA=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E5=8D=95=E7=8B=AC=E6=98=BE=E7=A4=BA=EF=BC=8C=E6=9C=80?=
 =?UTF-8?q?=E5=A4=9A500=E5=AD=97=E7=AC=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/base_document_extract_node.py                 | 10 +++++-----
 ui/src/components/ai-chat/ExecutionDetailDialog.vue    |  7 ++++---
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py b/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py
index 2cfbf3c4b..620763aef 100644
--- a/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py
+++ b/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py
@@ -7,7 +7,6 @@ from django.db.models import QuerySet
 
 from application.flow.i_step_node import NodeResult
 from application.flow.step_node.document_extract_node.i_document_extract_node import IDocumentExtractNode
-from application.models import Chat
 from dataset.models import File
 from dataset.serializers.document_serializers import split_handles, parse_table_handle_list, FileBufferHandle
 from dataset.serializers.file_serializers import FileSerializer
@@ -36,15 +35,16 @@ def bytes_to_uploaded_file(file_bytes, file_name="file.txt"):
     return uploaded_file
 
 
+splitter = '\n`-----------------------------------`\n'
+
 class BaseDocumentExtractNode(IDocumentExtractNode):
     def execute(self, document, chat_id, **kwargs):
         get_buffer = FileBufferHandle().get_buffer
 
         self.context['document_list'] = document
         content = []
-        splitter = '\n`-----------------------------------`\n'
         if document is None or not isinstance(document, list):
-            return NodeResult({'content': content}, {})
+            return NodeResult({'content': ''}, {})
 
         application = self.workflow_manage.work_flow_post_handler.chat_info.application
 
@@ -76,14 +76,14 @@ class BaseDocumentExtractNode(IDocumentExtractNode):
         return NodeResult({'content': splitter.join(content)}, {})
 
     def get_details(self, index: int, **kwargs):
+        content = self.context.get('content', '').split(splitter)
         # 不保存content全部内容，因为content内容可能会很大
-        content = (self.context.get('content')[:500] + '...') if len(self.context.get('content')) > 0 else ''
         return {
             'name': self.node.properties.get('stepName'),
             "index": index,
             'run_time': self.context.get('run_time'),
             'type': self.node.type,
-            'content': content,
+            'content': [file_content[:500] for file_content in content],
             'status': self.status,
             'err_message': self.err_message,
             'document_list': self.context.get('document_list')
diff --git a/ui/src/components/ai-chat/ExecutionDetailDialog.vue b/ui/src/components/ai-chat/ExecutionDetailDialog.vue
index b76ec0138..c39e95c60 100644
--- a/ui/src/components/ai-chat/ExecutionDetailDialog.vue
+++ b/ui/src/components/ai-chat/ExecutionDetailDialog.vue
@@ -231,12 +231,13 @@
                       </h5>
                       <div class="p-8-12 border-t-dashed lighter">
                         <el-scrollbar height="150">
-                          <el-card shadow="never" style="--el-card-padding: 8px">
+                          <el-card shadow="never" style="--el-card-padding: 8px" v-for="(file_content, index) in item.content"
+                                   :key="index" class="mb-8">
                             <MdPreview
-                              v-if="item.content"
+                              v-if="file_content"
                               ref="editorRef"
                               editorId="preview-only"
-                              :modelValue="item.content"
+                              :modelValue="file_content"
                               style="background: none"
                             />
                             <template v-else> - </template>