From 5e7879c582eadae3089095fda1ea7ff189605306 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Wed, 10 Jul 2024 11:30:55 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8DAI=E7=BC=96=E6=8E=92?= =?UTF-8?q?=E4=B8=AD=EF=BC=8C=E6=8C=87=E5=AE=9A=E5=9B=9E=E5=A4=8D=E7=9A=84?= =?UTF-8?q?=E5=86=85=E5=AE=B9=EF=BC=8C=E4=B8=8D=E8=83=BD=E5=9C=A8=E4=BB=BB?= =?UTF-8?q?=E5=8A=A1=E6=B5=81=E7=A8=8B=E4=B8=AD=E4=BC=A0=E9=80=92=20#726?= =?UTF-8?q?=20(#728)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit (cherry picked from commit c1823110d9cc1a49d8bf9e980c2e8e21212e4689) --- .../flow/step_node/direct_reply_node/impl/base_reply_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py b/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py index d266265be..717dce161 100644 --- a/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py +++ b/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py @@ -65,10 +65,10 @@ class BaseReplyNode(IReplyNode): else: result = self.generate_reply_content(content) if stream: - return NodeResult({'result': iter([AIMessageChunk(content=result)])}, {}, + return NodeResult({'result': iter([AIMessageChunk(content=result)]), 'answer': result}, {}, _to_response=to_stream_response) else: - return NodeResult({'result': AIMessage(content=result)}, {}, _to_response=to_response) + return NodeResult({'result': AIMessage(content=result), 'answer': result}, {}, _to_response=to_response) def generate_reply_content(self, prompt): return self.workflow_manage.generate_prompt(prompt)