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

feat: 高级编排检索结果字符串添加标题字段
Some checks are pending
sync2gitee / repo-sync (push) Waiting to run
Details
Typos Check / Spell Check with Typos (push) Waiting to run
Details

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)
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]),
'directly_return': '\n'.join([paragraph.get('content') for paragraph in result if
paragraph.get('is_hit_handling_method')]),
'data': '\n'.join(
[f"{paragraph.get('title', '')}:{paragraph.get('content')}" for paragraph in
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},
{})