fix: DownloadManager onDownloadProgress callback type

Comfy-Org · Nov 19, 2024 · 56e958e · 56e958e
1 parent 42837e4
commit 56e958e
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/preload.ts b/src/preload.ts
@@ -26,6 +26,15 @@ export interface SystemPaths {
   defaultInstallPath: string;
 }
 
+export interface DownloadProgressUpdate {
+  url: string;
+  filename: string;
+  savePath: string;
+  progress: number;
+  status: DownloadStatus;
+  message?: string;
+}
+
 const electronAPI = {
   /**
    * Callback for progress updates from the main process for starting ComfyUI.
@@ -93,12 +102,7 @@ const electronAPI = {
   },
   DownloadManager: {
     onDownloadProgress: (
-      callback: (progress: {
-        url: string;
-        progress_percentage: number;
-        status: DownloadStatus;
-        message?: string;
-      }) => void
+      callback: (progress: DownloadProgressUpdate) => void
     ) => {
       ipcRenderer.on(IPC_CHANNELS.DOWNLOAD_PROGRESS, (_event, progress) => callback(progress));
     },