From 669ee2633aca26ecaee4af68f82c924050b3b9c7 Mon Sep 17 00:00:00 2001
From: Matthew Meredith <matthewmeredith775@gmail.com>
Date: Sun, 3 Aug 2025 18:55:29 -0400
Subject: [PATCH] include litegraph augmentation in generated declarations
 (#4614)

---
 src/types/index.ts                    | 1 +
 src/types/litegraph-augmentation.d.ts | 5 ++---
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/types/index.ts b/src/types/index.ts
index 377581465a..c00c4dccce 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -41,6 +41,7 @@ export type {
   parseNodeExecutionId,
   createNodeExecutionId
 } from './nodeIdentification'
+export type { DOMWidget, DOMWidgetOptions } from '@/scripts/domWidget'
 export type {
   EmbeddingsResponse,
   ExtensionsResponse,
diff --git a/src/types/litegraph-augmentation.d.ts b/src/types/litegraph-augmentation.d.ts
index 68e2ac8f42..8518de4877 100644
--- a/src/types/litegraph-augmentation.d.ts
+++ b/src/types/litegraph-augmentation.d.ts
@@ -1,5 +1,7 @@
 import '@comfyorg/litegraph'
 import type { LLink, Size } from '@comfyorg/litegraph'
+import type { ExecutableLGraphNode, ExecutionId } from '@comfyorg/litegraph'
+import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
 
 import type { ComfyNodeDef as ComfyNodeDefV2 } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import type { ComfyNodeDef as ComfyNodeDefV1 } from '@/schemas/nodeDefSchema'
@@ -69,9 +71,6 @@ declare module '@comfyorg/litegraph/dist/interfaces' {
  *  ComfyUI extensions of litegraph
  */
 declare module '@comfyorg/litegraph' {
-  import type { ExecutableLGraphNode, ExecutionId } from '@comfyorg/litegraph'
-  import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
   interface LGraphNodeConstructor<T extends LGraphNode = LGraphNode> {
     type?: string
     comfyClass: string