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

fix: Multiple recall error when the previous node has no data (#3665)

This commit is contained in:
shaohuzhang1 2025-07-18 11:34:20 +08:00 committed by GitHub
parent 638a09d4f4
commit 7e78b7fa81
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py
View File

@ -61,6 +61,12 @@ def reset_result_list(result_list: List[Document], document_list: List[Document]
return r
def get_none_result(question):
return NodeResult(
{'document_list': [], 'question': question,
'result_list': [], 'result': ''}, {})
class BaseRerankerNode(IRerankerNode):
def save_context(self, details, workflow_manage):
self.context['document_list'] = details.get('document_list', [])
@ -73,6 +79,9 @@ class BaseRerankerNode(IRerankerNode):
**kwargs) -> NodeResult:
self.context['show_knowledge'] = show_knowledge
documents = merge_reranker_list(reranker_list)
documents = [d for d in documents if d.page_content and len(d.page_content) > 0]
if len(documents) == 0:
return get_none_result(question)
top_n = reranker_setting.get('top_n', 3)
self.context['document_list'] = [{'page_content': document.page_content, 'metadata': document.metadata} for
document in documents]