fix: 修复AI编排中，指定回复的内容，不能在任务流程中传递 #726 (#728)

1Panel-dev · Jul 10, 2024 · c182311 · c182311
1 parent fd6aa4f
commit c182311
Showing 1 changed file with 2 additions and 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
@@ -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)