From fcd305a87829b82497afb90b4e5f601de0203d8d Mon Sep 17 00:00:00 2001 From: CaptainB Date: Mon, 16 Dec 2024 14:55:09 +0800 Subject: [PATCH] refactor: Image generate model history message show images --- .../impl/base_image_generate_node.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py b/apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py index 77231b70d24..2972e276df2 100644 --- a/apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py +++ b/apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py @@ -57,7 +57,10 @@ def generate_history_ai_message(self, chat_record): if self.node.id == val['node_id'] and 'image_list' in val: if val['dialogue_type'] == 'WORKFLOW': return chat_record.get_ai_message() - return AIMessage(content=val['answer']) + image_list = val['image_list'] + return AIMessage(content=[ + *[{'type': 'image_url', 'image_url': {'url': f'{file_url}'}} for file_url in image_list] + ]) return chat_record.get_ai_message() def get_history_message(self, history_chat_record, dialogue_number):