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

feat: 高级编排检索结果字符串添加标题字段

Browse Source
This commit is contained in:
shaohuzhang1 2024-10-15 16:16:58 +08:00 committed by shaohuzhang1
parent 464aa13d2e
commit 451be0c81c
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -62,9 +62,12 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode):
result = sorted(result, key=lambda p: p.get('similarity'), reverse=True) 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(
'directly_return': '\n'.join([paragraph.get('content') for paragraph in result if [f"{paragraph.get('title', '')}:{paragraph.get('content')}" for paragraph in
paragraph.get('is_hit_handling_method')]), paragraph_list]),
'directly_return': '\n'.join(
[f"{paragraph.get('title', '')}:{paragraph.get('content')}" for paragraph in result if
paragraph.get('is_hit_handling_method')]),
'question': question}, 'question': question},
{}) {})