From 88a969df07160a04d28af268c62f82555d07c0b2 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Sat, 8 Feb 2025 17:51:37 -0500
Subject: [PATCH] Update litegraph 0.8.72 (#2470)

---
 package-lock.json                  | 8 ++++----
 package.json                       | 2 +-
 src/extensions/core/load3d.ts      | 5 +++--
 src/extensions/core/uploadAudio.ts | 7 ++++---
 src/scripts/widgets.ts             | 8 ++++++--
 5 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index a7c04cba8..3843d1adb 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -11,7 +11,7 @@
       "dependencies": {
         "@atlaskit/pragmatic-drag-and-drop": "^1.3.1",
         "@comfyorg/comfyui-electron-types": "^0.4.16",
-        "@comfyorg/litegraph": "^0.8.71",
+        "@comfyorg/litegraph": "^0.8.72",
         "@primevue/forms": "^4.2.5",
         "@primevue/themes": "^4.2.5",
         "@sentry/vue": "^8.48.0",
@@ -1944,9 +1944,9 @@
       "license": "GPL-3.0-only"
     },
     "node_modules/@comfyorg/litegraph": {
-      "version": "0.8.71",
-      "resolved": "https://registry.npmjs.org/@comfyorg/litegraph/-/litegraph-0.8.71.tgz",
-      "integrity": "sha512-BHe8BbrEC8hvLqRRv9vT7kkfn5CGY+w13DEhJOwXPj72PBtISa4ESfeOIZAMAASpjPWyagwWeSr4BzJTspSl6w==",
+      "version": "0.8.72",
+      "resolved": "https://registry.npmjs.org/@comfyorg/litegraph/-/litegraph-0.8.72.tgz",
+      "integrity": "sha512-/DCbV+zYjICC0uQWPqcj0KioAJ3j89y6eEnbpGgDSpH6SmbSks+Hpko7HWokWFMOKqriRa/NT5xMcHAiRts2cA==",
       "license": "MIT"
     },
     "node_modules/@cspotcode/source-map-support": {
diff --git a/package.json b/package.json
index acbf095e1..e1b4d6073 100644
--- a/package.json
+++ b/package.json
@@ -84,7 +84,7 @@
   "dependencies": {
     "@atlaskit/pragmatic-drag-and-drop": "^1.3.1",
     "@comfyorg/comfyui-electron-types": "^0.4.16",
-    "@comfyorg/litegraph": "^0.8.71",
+    "@comfyorg/litegraph": "^0.8.72",
     "@primevue/forms": "^4.2.5",
     "@primevue/themes": "^4.2.5",
     "@sentry/vue": "^8.48.0",
diff --git a/src/extensions/core/load3d.ts b/src/extensions/core/load3d.ts
index 4110ae357..5a78e79fd 100644
--- a/src/extensions/core/load3d.ts
+++ b/src/extensions/core/load3d.ts
@@ -1,5 +1,6 @@
 // @ts-strict-ignore
 import { IWidget } from '@comfyorg/litegraph'
+import type { IStringWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import { nextTick } from 'vue'
 
 import Load3DConfiguration from '@/extensions/core/load3d/Load3DConfiguration'
@@ -65,7 +66,7 @@ app.registerExtension({
           if (fileInput.files?.length) {
             const modelWidget = node.widgets?.find(
               (w: IWidget) => w.name === 'model_file'
-            )
+            ) as IStringWidget
             const uploadPath = await Load3dUtils.uploadFile(
               load3d,
               fileInput.files[0],
@@ -234,7 +235,7 @@ app.registerExtension({
           if (fileInput.files?.length) {
             const modelWidget = node.widgets?.find(
               (w: IWidget) => w.name === 'model_file'
-            )
+            ) as IStringWidget
             const uploadPath = await Load3dUtils.uploadFile(
               load3d,
               fileInput.files[0],
diff --git a/src/extensions/core/uploadAudio.ts b/src/extensions/core/uploadAudio.ts
index 194a5caf3..985dcb82b 100644
--- a/src/extensions/core/uploadAudio.ts
+++ b/src/extensions/core/uploadAudio.ts
@@ -1,5 +1,6 @@
 // @ts-strict-ignore
 import type { IWidget } from '@comfyorg/litegraph'
+import type { IStringWidget } from '@comfyorg/litegraph/dist/types/widgets'
 
 import type { DOMWidget } from '@/scripts/domWidget'
 import { useToastStore } from '@/stores/toastStore'
@@ -37,7 +38,7 @@ function getResourceURL(
 }
 
 async function uploadFile(
-  audioWidget: IWidget,
+  audioWidget: IStringWidget,
   audioUIWidget: DOMWidget<HTMLAudioElement, string>,
   file: File,
   updateNode: boolean,
@@ -151,9 +152,9 @@ app.registerExtension({
     return {
       AUDIOUPLOAD(node, inputName: string) {
         // The widget that allows user to select file.
-        const audioWidget: IWidget = node.widgets.find(
+        const audioWidget = node.widgets.find(
           (w: IWidget) => w.name === 'audio'
-        )
+        ) as IStringWidget
         const audioUIWidget = node.widgets.find(
           (w: IWidget) => w.name === 'audioUI'
         ) as unknown as DOMWidget<HTMLAudioElement, string>
diff --git a/src/scripts/widgets.ts b/src/scripts/widgets.ts
index fd0583db3..9e576a22c 100644
--- a/src/scripts/widgets.ts
+++ b/src/scripts/widgets.ts
@@ -1,6 +1,10 @@
 // @ts-strict-ignore
 import type { LGraphNode } from '@comfyorg/litegraph'
 import type { IWidget } from '@comfyorg/litegraph'
+import type {
+  IComboWidget,
+  IStringWidget
+} from '@comfyorg/litegraph/dist/types/widgets'
 import { Editor as TiptapEditor } from '@tiptap/core'
 import TiptapLink from '@tiptap/extension-link'
 import TiptapTable from '@tiptap/extension-table'
@@ -568,7 +572,7 @@ export const ComfyWidgets: Record<string, ComfyWidgetConstructor> = {
     const res = {
       widget: node.addWidget('combo', inputName, defaultValue, () => {}, {
         values: options ?? inputData[0]
-      })
+      }) as IComboWidget
     }
 
     if (type === 'remote') {
@@ -617,7 +621,7 @@ export const ComfyWidgets: Record<string, ComfyWidgetConstructor> = {
     // TODO make image upload handle a custom node type?
     const imageWidget = node.widgets.find(
       (w) => w.name === (inputData[1]?.widget ?? 'image')
-    )
+    ) as IStringWidget
     let uploadWidget
 
     function showImage(name) {