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

feat(events): add missing events

Browse Source 
These events weren't being emitted via socket.io:
- DownloadCancelledEvent
- DownloadCompleteEvent
- DownloadErrorEvent
- DownloadProgressEvent
- DownloadStartedEvent
- ModelInstallDownloadsCompleteEvent
This commit is contained in:
psychedelicious 2024-03-31 12:03:49 +11:00
parent 88a2340b95
commit 5cdf71b72f
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/api/sockets.py
View File

@ -12,6 +12,11 @@ from invokeai.app.services.events.events_common import (
BulkDownloadErrorEvent, BulkDownloadErrorEvent,
BulkDownloadEventBase, BulkDownloadEventBase,
BulkDownloadStartedEvent, BulkDownloadStartedEvent,
DownloadCancelledEvent,
DownloadCompleteEvent,
DownloadErrorEvent,
DownloadProgressEvent,
DownloadStartedEvent,
FastAPIEvent, FastAPIEvent,
InvocationCompleteEvent, InvocationCompleteEvent,
InvocationDenoiseProgressEvent, InvocationDenoiseProgressEvent,
@ -21,6 +26,7 @@ from invokeai.app.services.events.events_common import (
ModelInstallCancelledEvent, ModelInstallCancelledEvent,
ModelInstallCompleteEvent, ModelInstallCompleteEvent,
ModelInstallDownloadProgressEvent, ModelInstallDownloadProgressEvent,
ModelInstallDownloadsCompleteEvent,
ModelInstallErrorEvent, ModelInstallErrorEvent,
ModelInstallStartedEvent, ModelInstallStartedEvent,
ModelLoadCompleteEvent, ModelLoadCompleteEvent,
@ -78,9 +84,15 @@ class SocketIO:
register_events( register_events(
{ {
DownloadCancelledEvent,
DownloadCompleteEvent,
DownloadErrorEvent,
DownloadProgressEvent,
DownloadStartedEvent,
ModelLoadStartedEvent, ModelLoadStartedEvent,
ModelLoadCompleteEvent, ModelLoadCompleteEvent,
ModelInstallDownloadProgressEvent, ModelInstallDownloadProgressEvent,
ModelInstallDownloadsCompleteEvent,
ModelInstallStartedEvent, ModelInstallStartedEvent,
ModelInstallCompleteEvent, ModelInstallCompleteEvent,
ModelInstallCancelledEvent, ModelInstallCancelledEvent,