Skip to content

Commit 5e7879c

Browse files
committed
fix: 修复AI编排中,指定回复的内容,不能在任务流程中传递 #726 (#728)
(cherry picked from commit c182311)
1 parent 0ce3fc5 commit 5e7879c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,10 +65,10 @@ def execute(self, reply_type, stream, fields=None, content=None, **kwargs) -> No
6565
else:
6666
result = self.generate_reply_content(content)
6767
if stream:
68-
return NodeResult({'result': iter([AIMessageChunk(content=result)])}, {},
68+
return NodeResult({'result': iter([AIMessageChunk(content=result)]), 'answer': result}, {},
6969
_to_response=to_stream_response)
7070
else:
71-
return NodeResult({'result': AIMessage(content=result)}, {}, _to_response=to_response)
71+
return NodeResult({'result': AIMessage(content=result), 'answer': result}, {}, _to_response=to_response)
7272

7373
def generate_reply_content(self, prompt):
7474
return self.workflow_manage.generate_prompt(prompt)

0 commit comments

Comments
 (0)