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): fix session processor event handling

Browse Source
This commit is contained in:
psychedelicious 2024-05-20 15:22:10 +10:00
parent 9c45cbe8f7
commit 87f2d04ddd
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/services/session_processor/session_processor_default.py
View File

@ -10,6 +10,7 @@ from invokeai.app.services.events.events_common import (
FastAPIEvent, FastAPIEvent,
QueueClearedEvent, QueueClearedEvent,
QueueEventBase, QueueEventBase,
QueueItemStatusChangedEvent,
SessionCanceledEvent, SessionCanceledEvent,
register_events, register_events,
) )
@ -89,11 +90,7 @@ class DefaultSessionProcessor(SessionProcessorBase):
self._poll_now() self._poll_now()
elif isinstance(payload, BatchEnqueuedEvent): elif isinstance(payload, BatchEnqueuedEvent):
self._poll_now() self._poll_now()
elif event_name == "queue_item_status_changed" and event[1]["data"]["queue_item"]["status"] in [ elif isinstance(payload, QueueItemStatusChangedEvent) and payload.status in ("completed", "failed", "canceled"):
"completed",
"failed",
"canceled",
]:
self._poll_now() self._poll_now()
def resume(self) -> SessionProcessorStatus: def resume(self) -> SessionProcessorStatus: