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 d266265be8a..717dce161e7 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 @@ def execute(self, reply_type, stream, fields=None, content=None, **kwargs) -> No 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)