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

refactor: Image generate model history message show images

Browse Source
This commit is contained in:
CaptainB 2024-12-16 14:55:09 +08:00 committed by 刘瑞斌
parent 64443ee136
commit 6a55750e8d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py
View File

@ -57,7 +57,10 @@ class BaseImageGenerateNode(IImageGenerateNode):
if self.node.id == val['node_id'] and 'image_list' in val: if self.node.id == val['node_id'] and 'image_list' in val:
if val['dialogue_type'] == 'WORKFLOW': if val['dialogue_type'] == 'WORKFLOW':
return chat_record.get_ai_message() 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() return chat_record.get_ai_message()
def get_history_message(self, history_chat_record, dialogue_number): def get_history_message(self, history_chat_record, dialogue_number):