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

fix: 修复返回的分段结果排序不对，不是按照命中高低顺序排序[BUG] #746 (#770)

Browse Source
This commit is contained in:
shaohuzhang1 2024-07-16 10:33:18 +08:00 committed by GitHub
parent 5308ad9c89
commit d554e0ae88
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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': []}, {}) return NodeResult({'paragraph_list': [], 'is_hit_handling_method': []}, {})
paragraph_list = self.list_paragraph(embedding_list, vector) paragraph_list = self.list_paragraph(embedding_list, vector)
result = [self.reset_paragraph(paragraph, embedding_list) for paragraph in paragraph_list] 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, return NodeResult({'paragraph_list': result,
'is_hit_handling_method_list': [row for row in result if row.get('is_hit_handling_method')], '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]), 'data': '\n'.join([paragraph.get('content') for paragraph in paragraph_list]),