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

fix: 修复重排模型添加检索内容字段

This commit is contained in:
shaohuzhang1 2024-09-12 14:53:59 +08:00 committed by shaohuzhang1
parent f388cb5695
commit 82b7e8d69e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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