Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(events): dump event payloads to serializable format

Browse Source
This commit is contained in:
psychedelicious 2024-03-14 17:38:19 +11:00 committed by blessedcoolant
parent b07b7af710
commit ba3d8af161
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/app/services/events/events_base.py
View File

@ -80,7 +80,7 @@ class EventServiceBase:
"graph_execution_state_id": graph_execution_state_id, "graph_execution_state_id": graph_execution_state_id,
"node_id": node_id, "node_id": node_id,
"source_node_id": source_node_id, "source_node_id": source_node_id,
"progress_image": progress_image.model_dump() if progress_image is not None else None, "progress_image": progress_image.model_dump(mode="json") if progress_image is not None else None,
"step": step, "step": step,
"order": order, "order": order,
"total_steps": total_steps, "total_steps": total_steps,
@ -189,7 +189,7 @@ class EventServiceBase:
"queue_item_id": queue_item_id, "queue_item_id": queue_item_id,
"queue_batch_id": queue_batch_id, "queue_batch_id": queue_batch_id,
"graph_execution_state_id": graph_execution_state_id, "graph_execution_state_id": graph_execution_state_id,
"model_config": model_config.model_dump(), "model_config": model_config.model_dump(mode="json"),
}, },
) )
@ -209,7 +209,7 @@ class EventServiceBase:
"queue_item_id": queue_item_id, "queue_item_id": queue_item_id,
"queue_batch_id": queue_batch_id, "queue_batch_id": queue_batch_id,
"graph_execution_state_id": graph_execution_state_id, "graph_execution_state_id": graph_execution_state_id,
"model_config": model_config.model_dump(), "model_config": model_config.model_dump(mode="json"),
}, },
) )
@ -254,8 +254,8 @@ class EventServiceBase:
"started_at": str(session_queue_item.started_at) if session_queue_item.started_at else None, "started_at": str(session_queue_item.started_at) if session_queue_item.started_at else None,
"completed_at": str(session_queue_item.completed_at) if session_queue_item.completed_at else None, "completed_at": str(session_queue_item.completed_at) if session_queue_item.completed_at else None,
}, },
"batch_status": batch_status.model_dump(), "batch_status": batch_status.model_dump(mode="json"),
"queue_status": queue_status.model_dump(), "queue_status": queue_status.model_dump(mode="json"),
}, },
) )