zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
shaohuzhang1 2024-07-10 11:30:55 +08:00 committed by GitHub
parent fd6aa4fb39
commit c1823110d9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py
View File

@ -65,10 +65,10 @@ class BaseReplyNode(IReplyNode):
else: else:
result = self.generate_reply_content(content) result = self.generate_reply_content(content)
if stream: if stream:
return NodeResult({'result': iter([AIMessageChunk(content=result)])}, {}, return NodeResult({'result': iter([AIMessageChunk(content=result)]), 'answer': result}, {},
_to_response=to_stream_response) _to_response=to_stream_response)
else: 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): def generate_reply_content(self, prompt):
return self.workflow_manage.generate_prompt(prompt) return self.workflow_manage.generate_prompt(prompt)