15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

chore: limit paragraph list to first 5 segments in get_details method

This commit is contained in:
CaptainB 2025-12-11 16:21:50 +08:00 committed by 刘瑞斌
parent a0d5b17fab
commit 638f90e69f
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/application/flow/step_node/document_split_node/impl/base_document_split_node.py
View File

@ -161,6 +161,18 @@ class BaseDocumentSplitNode(IDocumentSplitNode):
return list(set(problem_list))
def get_details(self, index: int, **kwargs):
paragraph_list = self.context.get('paragraph_list', [])
# 每个文档保留前5个分段
limited_paragraph_list = []
for doc in paragraph_list:
if doc.get('paragraphs'):
doc_copy = doc.copy()
doc_copy['paragraphs'] = doc['paragraphs'][:5]
limited_paragraph_list.append(doc_copy)
else:
limited_paragraph_list.append(doc)
paragraph_list = limited_paragraph_list
return {
'name': self.node.properties.get('stepName'),
"index": index,
@ -168,11 +180,11 @@ class BaseDocumentSplitNode(IDocumentSplitNode):
'type': self.node.type,
'status': self.status,
'err_message': self.err_message,
'paragraph_list': self.context.get('paragraph_list', []),
'paragraph_list': paragraph_list,
'limit': self.context.get('limit'),
'chunk_size': self.context.get('chunk_size'),
'with_filter': self.context.get('with_filter'),
'patterns': self.context.get('patterns'),
'split_strategy': self.context.get('split_strategy'),
'document_list': self.context.get('document_list', []),
# 'document_list': self.context.get('document_list', []),
}