From 82b7e8d69e4182229407e1b4892220296a6d41db Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Thu, 12 Sep 2024 14:53:59 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E9=87=8D=E6=8E=92?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E6=B7=BB=E5=8A=A0=E6=A3=80=E7=B4=A2=E5=86=85?= =?UTF-8?q?=E5=AE=B9=E5=AD=97=E6=AE=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../flow/step_node/reranker_node/impl/base_reranker_node.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py b/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py index 129fe3ff9..513d621a3 100644 --- a/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py +++ b/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py @@ -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,