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

fix: reranker添加检索内容数据

Browse Source
This commit is contained in:
shaohuzhang1 2024-09-23 10:31:31 +08:00 committed by shaohuzhang1
parent 2dc06191ca
commit 1321ed31ee
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py
View File

@ -48,6 +48,7 @@ class BaseRerankerNode(IRerankerNode):
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
documents = merge_reranker_list(reranker_list) documents = merge_reranker_list(reranker_list)
self.context['document_list'] = documents self.context['document_list'] = documents
self.context['question'] = question
reranker_model = get_model_instance_by_model_user_id(reranker_model_id, reranker_model = get_model_instance_by_model_user_id(reranker_model_id,
self.flow_params_serializer.data.get('user_id')) self.flow_params_serializer.data.get('user_id'))
result = reranker_model.compress_documents( result = reranker_model.compress_documents(
@ -64,7 +65,7 @@ class BaseRerankerNode(IRerankerNode):
'name': self.node.properties.get('stepName'), 'name': self.node.properties.get('stepName'),
"index": index, "index": index,
'document_list': self.context.get('document_list'), 'document_list': self.context.get('document_list'),
"question": self.node_params_serializer.data.get('question'), "question": self.context.get('question'),
'run_time': self.context.get('run_time'), 'run_time': self.context.get('run_time'),
'type': self.node.type, 'type': self.node.type,
'reranker_setting': self.node_params_serializer.data.get('reranker_setting'), 'reranker_setting': self.node_params_serializer.data.get('reranker_setting'),