From ce7b1d941b668b590d8e4ae73a2e45c72a06848e Mon Sep 17 00:00:00 2001 From: CaptainB Date: Mon, 25 Nov 2024 11:37:21 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E5=A4=9A=E6=96=87=E6=A1=A3?= =?UTF-8?q?=E5=88=86=E5=89=B2=E7=BA=BF=E5=B1=95=E7=A4=BA=E4=B8=BA=E5=A4=9A?= =?UTF-8?q?=E4=B8=AA=E2=80=98-=E2=80=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../impl/base_document_extract_node.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 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 2663af097..f450a1dd0 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 @@ -14,8 +14,8 @@ class BaseDocumentExtractNode(IDocumentExtractNode): get_buffer = FileBufferHandle().get_buffer self.context['document_list'] = document - content = '' - splitter = '\n-----------------------------------\n' + content = [] + splitter = '\n`-----------------------------------`\n' if document is None: return NodeResult({'content': content}, {}) @@ -29,10 +29,10 @@ class BaseDocumentExtractNode(IDocumentExtractNode): # 回到文件头 buffer.seek(0) file_content = split_handle.get_content(buffer) - content += splitter + '## ' + doc['name'] + '\n' + file_content + content.append( '## ' + doc['name'] + '\n' + file_content) break - return NodeResult({'content': content}, {}) + return NodeResult({'content': splitter.join(content)}, {}) def get_details(self, index: int, **kwargs): return {