refactor: Image generate model history message show images

1Panel-dev · Dec 16, 2024 · fcd305a · fcd305a
1 parent 64443ee
commit fcd305a
Showing 1 changed file with 4 additions and 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
@@ -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):