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

This commit is contained in:
CaptainB 2024-11-25 11:37:21 +08:00 committed by 刘瑞斌
parent 639e1d7b6d
commit ce7b1d941b

View File

@ -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 {