Skip to content

Commit 6a55750

Browse files
committed
refactor: Image generate model history message show images
1 parent 64443ee commit 6a55750

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,10 @@ def generate_history_ai_message(self, chat_record):
5757
if self.node.id == val['node_id'] and 'image_list' in val:
5858
if val['dialogue_type'] == 'WORKFLOW':
5959
return chat_record.get_ai_message()
60-
return AIMessage(content=val['answer'])
60+
image_list = val['image_list']
61+
return AIMessage(content=[
62+
*[{'type': 'image_url', 'image_url': {'url': f'{file_url}'}} for file_url in image_list]
63+
])
6164
return chat_record.get_ai_message()
6265

6366
def get_history_message(self, history_chat_record, dialogue_number):

0 commit comments

Comments
 (0)