From 5cdf71b72f9f815b03fffb673836789ae5e57b7e Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sun, 31 Mar 2024 12:03:49 +1100
Subject: [PATCH] feat(events): add missing events

These events weren't being emitted via socket.io:
- DownloadCancelledEvent
- DownloadCompleteEvent
- DownloadErrorEvent
- DownloadProgressEvent
- DownloadStartedEvent
- ModelInstallDownloadsCompleteEvent
---
 invokeai/app/api/sockets.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/invokeai/app/api/sockets.py b/invokeai/app/api/sockets.py
index 8fa34da078..bfe38a3e13 100644
--- a/invokeai/app/api/sockets.py
+++ b/invokeai/app/api/sockets.py
@@ -12,6 +12,11 @@ from invokeai.app.services.events.events_common import (
     BulkDownloadErrorEvent,
     BulkDownloadEventBase,
     BulkDownloadStartedEvent,
+    DownloadCancelledEvent,
+    DownloadCompleteEvent,
+    DownloadErrorEvent,
+    DownloadProgressEvent,
+    DownloadStartedEvent,
     FastAPIEvent,
     InvocationCompleteEvent,
     InvocationDenoiseProgressEvent,
@@ -21,6 +26,7 @@ from invokeai.app.services.events.events_common import (
     ModelInstallCancelledEvent,
     ModelInstallCompleteEvent,
     ModelInstallDownloadProgressEvent,
+    ModelInstallDownloadsCompleteEvent,
     ModelInstallErrorEvent,
     ModelInstallStartedEvent,
     ModelLoadCompleteEvent,
@@ -78,9 +84,15 @@ class SocketIO:
 
         register_events(
             {
+                DownloadCancelledEvent,
+                DownloadCompleteEvent,
+                DownloadErrorEvent,
+                DownloadProgressEvent,
+                DownloadStartedEvent,
                 ModelLoadStartedEvent,
                 ModelLoadCompleteEvent,
                 ModelInstallDownloadProgressEvent,
+                ModelInstallDownloadsCompleteEvent,
                 ModelInstallStartedEvent,
                 ModelInstallCompleteEvent,
                 ModelInstallCancelledEvent,