zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 多文档分割线展示为多个‘-’

Browse Source
This commit is contained in:
CaptainB 2024-11-25 11:37:21 +08:00 committed by 刘瑞斌
parent 639e1d7b6d
commit ce7b1d941b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py
View File

@ -14,8 +14,8 @@ class BaseDocumentExtractNode(IDocumentExtractNode):
get_buffer = FileBufferHandle().get_buffer get_buffer = FileBufferHandle().get_buffer
self.context['document_list'] = document self.context['document_list'] = document
content = '' content = []
splitter = '\n-----------------------------------\n' splitter = '\n`-----------------------------------`\n'
if document is None: if document is None:
return NodeResult({'content': content}, {}) return NodeResult({'content': content}, {})
@ -29,10 +29,10 @@ class BaseDocumentExtractNode(IDocumentExtractNode):
# 回到文件头 # 回到文件头
buffer.seek(0) buffer.seek(0)
file_content = split_handle.get_content(buffer) file_content = split_handle.get_content(buffer)
content += splitter + '## ' + doc['name'] + '\n' + file_content content.append( '## ' + doc['name'] + '\n' + file_content)
break break
return NodeResult({'content': content}, {}) return NodeResult({'content': splitter.join(content)}, {})
def get_details(self, index: int, **kwargs): def get_details(self, index: int, **kwargs):
return { return {