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: 知识库检索添加question

This commit is contained in:
shaohuzhang1 2024-06-25 09:49:59 +08:00
parent 226759e668
commit fa2afd8e52
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -44,7 +44,9 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode):
'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 paragraph_list if
paragraph.get('is_hit_handling_method')])},
paragraph.get('is_hit_handling_method')]),
'question': question},
{})
@staticmethod
@ -81,6 +83,7 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode):
def get_details(self, index: int, **kwargs):
return {
'name': self.node.properties.get('stepName'),
'question': self.context.get('question'),
"index": index,
'run_time': self.context.get('run_time'),
'paragraph_list': self.context.get('paragraph_list'),