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: 修复返回的分段结果排序不对，不是按照命中高低顺序排序[BUG] #746 (#770) 

(cherry picked from commit d554e0ae88)
This commit is contained in:
shaohuzhang1 2024-07-16 10:33:18 +08:00 committed by shaohuzhang1
parent a5ca97ac06
commit 16f490567e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py
View File

@ -40,6 +40,7 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode):
return NodeResult({'paragraph_list': [], 'is_hit_handling_method': []}, {})
paragraph_list = self.list_paragraph(embedding_list, vector)
result = [self.reset_paragraph(paragraph, embedding_list) for paragraph in paragraph_list]
result = sorted(result, key=lambda p: p.get('similarity'), reverse=True)
return NodeResult({'paragraph_list': result,
'is_hit_handling_method_list': [row for row in result if row.get('is_hit_handling_method')],
'data': '\n'.join([paragraph.get('content') for paragraph in paragraph_list]),