From fef02e5f563eb5d0e71eecdfad7dbb2fe1588dc7 Mon Sep 17 00:00:00 2001
From: Benjamin Lu <benjaminlu1107@gmail.com>
Date: Sun, 3 Aug 2025 22:06:29 -0400
Subject: [PATCH] [refactor] Migrate litegraph imports from npm package to
 local subtree

- Updated all imports from '@comfyorg/litegraph' to '@/lib/litegraph/src/'
- Replaced deep dist imports with direct source paths
- Updated CSS import in main.ts
- All imports now use the @ alias consistently
---
 .../content/MissingCoreNodesMessage.vue       |  2 +-
 src/components/graph/GraphCanvas.vue          |  2 +-
 src/components/graph/GraphCanvasMenu.vue      |  2 +-
 src/components/graph/NodeTooltip.vue          | 10 +++++-----
 src/components/graph/SelectionOverlay.vue     |  2 +-
 src/components/graph/TitleEditor.vue          |  8 ++++++--
 .../ColorPickerButton.spec.ts                 |  4 ++--
 .../selectionToolbox/ColorPickerButton.vue    |  8 ++++++--
 .../graph/selectionToolbox/ExecuteButton.vue  |  2 +-
 .../graph/widgets/TextPreviewWidget.vue       |  2 +-
 src/components/load3d/Load3DScene.vue         |  2 +-
 .../load3d/controls/RecordingControls.vue     |  2 +-
 .../searchbox/NodeSearchBoxPopover.vue        | 14 ++++++-------
 .../canvas/useCanvasTransformSync.ts          |  2 +-
 .../canvas/useSelectedLiteGraphItems.ts       |  3 +--
 .../element/useAbsolutePosition.ts            |  2 +-
 .../element/useCanvasPositionConversion.ts    |  3 ++-
 src/composables/node/useNodeAnimatedImage.ts  |  5 ++---
 src/composables/node/useNodeBadge.ts          | 10 +++++-----
 .../node/useNodeCanvasImagePreview.ts         |  3 +--
 src/composables/node/useNodeChatHistory.ts    |  3 +--
 src/composables/node/useNodeDragAndDrop.ts    |  2 +-
 src/composables/node/useNodeFileInput.ts      |  3 +--
 src/composables/node/useNodeImage.ts          |  3 +--
 src/composables/node/useNodeImageUpload.ts    |  3 +--
 src/composables/node/useNodePaste.ts          |  2 +-
 src/composables/node/useNodePricing.ts        |  4 ++--
 src/composables/node/useNodeProgressText.ts   |  3 +--
 src/composables/node/useWatchWidget.ts        |  2 +-
 src/composables/nodePack/useMissingNodes.ts   |  4 ++--
 src/composables/nodePack/useWorkflowPacks.ts  |  2 +-
 src/composables/useCanvasDrop.ts              |  4 ++--
 src/composables/useContextMenuTranslation.ts  |  7 +++----
 src/composables/useCoreCommands.ts            | 15 +++++++-------
 src/composables/useGlobalLitegraph.ts         |  2 +-
 src/composables/useLitegraphSettings.ts       |  6 +++++-
 src/composables/useMinimap.ts                 |  2 +-
 src/composables/usePaste.ts                   |  4 ++--
 src/composables/useRefreshableSelection.ts    |  2 +-
 src/composables/useWorkflowValidation.ts      |  3 +--
 src/composables/widgets/useBooleanWidget.ts   |  3 +--
 .../widgets/useChatHistoryWidget.ts           |  2 +-
 src/composables/widgets/useComboWidget.ts     |  4 ++--
 src/composables/widgets/useFloatWidget.ts     |  4 ++--
 .../widgets/useImagePreviewWidget.ts          |  5 ++---
 .../widgets/useImageUploadWidget.ts           |  5 ++---
 src/composables/widgets/useIntWidget.ts       |  5 ++---
 src/composables/widgets/useMarkdownWidget.ts  |  2 +-
 .../widgets/useProgressTextWidget.ts          |  2 +-
 src/composables/widgets/useRemoteWidget.ts    |  4 ++--
 src/composables/widgets/useStringWidget.ts    |  3 +--
 src/constants/coreSettings.ts                 |  3 +--
 src/extensions/core/contextMenuFilter.ts      |  6 +++++-
 src/extensions/core/groupNode.ts              |  7 +++----
 src/extensions/core/groupNodeManage.ts        |  3 +--
 src/extensions/core/groupOptions.ts           |  9 ++++-----
 src/extensions/core/load3d.ts                 |  2 +-
 .../core/load3d/Load3DConfiguration.ts        |  3 +--
 src/extensions/core/load3d/Load3d.ts          |  2 +-
 src/extensions/core/load3d/Load3dAnimation.ts |  3 ++-
 src/extensions/core/load3d/NodeStorage.ts     |  2 +-
 src/extensions/core/load3d/interfaces.ts      |  2 +-
 src/extensions/core/nodeTemplates.ts          |  3 +--
 src/extensions/core/noteNode.ts               |  4 ++--
 src/extensions/core/rerouteNode.ts            |  8 ++++++--
 src/extensions/core/simpleTouchSupport.ts     |  2 +-
 src/extensions/core/slotDefaults.ts           |  2 +-
 src/extensions/core/uploadAudio.ts            | 10 +++++-----
 src/extensions/core/widgetInputs.ts           | 17 ++++++++--------
 src/lib/litegraph/lint-staged.config.js       |  5 +----
 src/main.ts                                   |  2 +-
 src/schemas/apiSchema.ts                      |  2 +-
 src/schemas/colorPaletteSchema.ts             |  3 ++-
 src/scripts/app.ts                            | 18 ++++++++---------
 src/scripts/changeTracker.ts                  |  2 +-
 src/scripts/domWidget.ts                      | 14 ++++++++-----
 src/scripts/errorNodeWidgets.ts               |  5 ++---
 src/scripts/fluxKontextEditNode.ts            |  8 ++++----
 src/scripts/pnginfo.ts                        |  2 +-
 src/scripts/widgets.ts                        | 13 ++++++------
 src/services/colorPaletteService.ts           |  4 ++--
 src/services/litegraphService.ts              | 16 +++++++--------
 src/services/load3dService.ts                 |  2 +-
 src/services/subgraphService.ts               |  3 +--
 src/services/workflowService.ts               |  4 ++--
 src/stores/executionStore.ts                  |  2 +-
 src/stores/graphStore.ts                      |  8 ++++++--
 src/stores/imagePreviewStore.ts               |  2 +-
 src/stores/nodeDefStore.ts                    |  2 +-
 src/stores/subgraphNavigationStore.ts         |  4 ++--
 src/stores/workflowStore.ts                   |  2 +-
 src/types/comfy.ts                            |  5 ++---
 src/types/index.ts                            |  3 +--
 src/types/litegraph-augmentation.d.ts         | 20 ++++++++++---------
 src/utils/errorReportUtil.ts                  |  3 +--
 src/utils/executableGroupNodeChildDTO.ts      |  5 ++---
 src/utils/executableGroupNodeDto.ts           |  2 +-
 src/utils/executionUtil.ts                    |  5 ++---
 src/utils/graphTraversalUtil.ts               |  7 +++++--
 src/utils/linkFixer.ts                        |  8 ++++----
 src/utils/litegraphUtil.ts                    | 16 +++++++++------
 src/utils/nodeFilterUtil.ts                   |  2 +-
 src/utils/searchAndReplace.ts                 |  3 +--
 src/utils/typeGuardUtil.ts                    |  5 ++---
 src/utils/vintageClipboard.ts                 |  8 ++++++--
 .../content/MissingCoreNodesMessage.spec.ts   |  2 +-
 .../canvas/useSelectedLiteGraphItems.test.ts  | 12 +++++------
 .../composables/node/useNodePricing.test.ts   |  4 ++--
 .../tests/composables/useMissingNodes.test.ts |  2 +-
 .../composables/useNodeChatHistory.test.ts    |  2 +-
 .../tests/composables/useWatchWidget.test.ts  |  2 +-
 tests-ui/tests/litegraph.test.ts              |  3 ++-
 .../tests/store/imagePreviewStore.test.ts     |  2 +-
 tests-ui/tests/store/workflowStore.test.ts    |  2 +-
 .../tests/utils/graphTraversalUtil.test.ts    |  6 +++++-
 tests-ui/tests/utils/litegraphUtil.test.ts    |  4 ++--
 tests-ui/tests/utils/nodeFilterUtil.test.ts   |  2 +-
 tests-ui/tests/utils/serachAndReplace.test.ts |  4 ++--
 tests-ui/tests/widgets/domWidget.test.ts      |  2 +-
 119 files changed, 284 insertions(+), 271 deletions(-)

diff --git a/src/components/dialog/content/MissingCoreNodesMessage.vue b/src/components/dialog/content/MissingCoreNodesMessage.vue
index 40347061e..036f088b3 100644
--- a/src/components/dialog/content/MissingCoreNodesMessage.vue
+++ b/src/components/dialog/content/MissingCoreNodesMessage.vue
@@ -42,11 +42,11 @@
 </template>
 
 <script setup lang="ts">
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { whenever } from '@vueuse/core'
 import Message from 'primevue/message'
 import { computed, ref } from 'vue'
 
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useSystemStatsStore } from '@/stores/systemStatsStore'
 import { compareVersions } from '@/utils/formatUtil'
 
diff --git a/src/components/graph/GraphCanvas.vue b/src/components/graph/GraphCanvas.vue
index a019f583e..1b0cf55ee 100644
--- a/src/components/graph/GraphCanvas.vue
+++ b/src/components/graph/GraphCanvas.vue
@@ -49,7 +49,6 @@
 </template>
 
 <script setup lang="ts">
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { useEventListener, whenever } from '@vueuse/core'
 import { computed, onMounted, ref, watch, watchEffect } from 'vue'
 
@@ -78,6 +77,7 @@ import { useWorkflowAutoSave } from '@/composables/useWorkflowAutoSave'
 import { useWorkflowPersistence } from '@/composables/useWorkflowPersistence'
 import { CORE_SETTINGS } from '@/constants/coreSettings'
 import { i18n, t } from '@/i18n'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { UnauthorizedError, api } from '@/scripts/api'
 import { app as comfyApp } from '@/scripts/app'
 import { ChangeTracker } from '@/scripts/changeTracker'
diff --git a/src/components/graph/GraphCanvasMenu.vue b/src/components/graph/GraphCanvasMenu.vue
index 70c44c2c5..0ddd95ef9 100644
--- a/src/components/graph/GraphCanvasMenu.vue
+++ b/src/components/graph/GraphCanvasMenu.vue
@@ -70,13 +70,13 @@
 </template>
 
 <script setup lang="ts">
-import { LiteGraph } from '@comfyorg/litegraph'
 import Button from 'primevue/button'
 import ButtonGroup from 'primevue/buttongroup'
 import { computed } from 'vue'
 import { useI18n } from 'vue-i18n'
 
 import { useCanvasInteractions } from '@/composables/graph/useCanvasInteractions'
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
 import { useCommandStore } from '@/stores/commandStore'
 import { useCanvasStore } from '@/stores/graphStore'
 import { useSettingStore } from '@/stores/settingStore'
diff --git a/src/components/graph/NodeTooltip.vue b/src/components/graph/NodeTooltip.vue
index 48155d0eb..647987000 100644
--- a/src/components/graph/NodeTooltip.vue
+++ b/src/components/graph/NodeTooltip.vue
@@ -10,15 +10,15 @@
 </template>
 
 <script setup lang="ts">
-import {
-  LiteGraph,
-  isOverNodeInput,
-  isOverNodeOutput
-} from '@comfyorg/litegraph'
 import { useEventListener } from '@vueuse/core'
 import { nextTick, ref } from 'vue'
 
 import { st } from '@/i18n'
+import {
+  LiteGraph,
+  isOverNodeInput,
+  isOverNodeOutput
+} from '@/lib/litegraph/src/litegraph'
 import { app as comfyApp } from '@/scripts/app'
 import { isDOMWidget } from '@/scripts/domWidget'
 import { useNodeDefStore } from '@/stores/nodeDefStore'
diff --git a/src/components/graph/SelectionOverlay.vue b/src/components/graph/SelectionOverlay.vue
index d998dad06..d0b6e9eb6 100644
--- a/src/components/graph/SelectionOverlay.vue
+++ b/src/components/graph/SelectionOverlay.vue
@@ -13,12 +13,12 @@
 </template>
 
 <script setup lang="ts">
-import { createBounds } from '@comfyorg/litegraph'
 import { whenever } from '@vueuse/core'
 import { ref, watch } from 'vue'
 
 import { useSelectedLiteGraphItems } from '@/composables/canvas/useSelectedLiteGraphItems'
 import { useAbsolutePosition } from '@/composables/element/useAbsolutePosition'
+import { createBounds } from '@/lib/litegraph/src/litegraph'
 import { useCanvasStore } from '@/stores/graphStore'
 
 const canvasStore = useCanvasStore()
diff --git a/src/components/graph/TitleEditor.vue b/src/components/graph/TitleEditor.vue
index f3e49e77f..220b549fb 100644
--- a/src/components/graph/TitleEditor.vue
+++ b/src/components/graph/TitleEditor.vue
@@ -13,13 +13,17 @@
 </template>
 
 <script setup lang="ts">
-import { LGraphGroup, LGraphNode, LiteGraph } from '@comfyorg/litegraph'
-import type { LiteGraphCanvasEvent } from '@comfyorg/litegraph'
 import { useEventListener } from '@vueuse/core'
 import { type CSSProperties, computed, ref, watch } from 'vue'
 
 import EditableText from '@/components/common/EditableText.vue'
 import { useAbsolutePosition } from '@/composables/element/useAbsolutePosition'
+import {
+  LGraphGroup,
+  LGraphNode,
+  LiteGraph
+} from '@/lib/litegraph/src/litegraph'
+import type { LiteGraphCanvasEvent } from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { useCanvasStore, useTitleEditorStore } from '@/stores/graphStore'
 import { useSettingStore } from '@/stores/settingStore'
diff --git a/src/components/graph/selectionToolbox/ColorPickerButton.spec.ts b/src/components/graph/selectionToolbox/ColorPickerButton.spec.ts
index 1989352b7..dac0ed084 100644
--- a/src/components/graph/selectionToolbox/ColorPickerButton.spec.ts
+++ b/src/components/graph/selectionToolbox/ColorPickerButton.spec.ts
@@ -11,8 +11,8 @@ import { useCanvasStore } from '@/stores/graphStore'
 import { useWorkflowStore } from '@/stores/workflowStore'
 
 // Mock the litegraph module
-vi.mock('@comfyorg/litegraph', async () => {
-  const actual = await vi.importActual('@comfyorg/litegraph')
+vi.mock('@/lib/litegraph/src/litegraph', async () => {
+  const actual = await vi.importActual('@/lib/litegraph/src/litegraph')
   return {
     ...actual,
     LGraphCanvas: {
diff --git a/src/components/graph/selectionToolbox/ColorPickerButton.vue b/src/components/graph/selectionToolbox/ColorPickerButton.vue
index b35bc4aa4..6f930536a 100644
--- a/src/components/graph/selectionToolbox/ColorPickerButton.vue
+++ b/src/components/graph/selectionToolbox/ColorPickerButton.vue
@@ -44,13 +44,17 @@
 </template>
 
 <script setup lang="ts">
-import type { ColorOption as CanvasColorOption } from '@comfyorg/litegraph'
-import { LGraphCanvas, LiteGraph, isColorable } from '@comfyorg/litegraph'
 import Button from 'primevue/button'
 import SelectButton from 'primevue/selectbutton'
 import { computed, ref, watch } from 'vue'
 import { useI18n } from 'vue-i18n'
 
+import type { ColorOption as CanvasColorOption } from '@/lib/litegraph/src/litegraph'
+import {
+  LGraphCanvas,
+  LiteGraph,
+  isColorable
+} from '@/lib/litegraph/src/litegraph'
 import { useCanvasStore } from '@/stores/graphStore'
 import { useWorkflowStore } from '@/stores/workflowStore'
 import { useColorPaletteStore } from '@/stores/workspace/colorPaletteStore'
diff --git a/src/components/graph/selectionToolbox/ExecuteButton.vue b/src/components/graph/selectionToolbox/ExecuteButton.vue
index 2f2717a69..39ca28d5e 100644
--- a/src/components/graph/selectionToolbox/ExecuteButton.vue
+++ b/src/components/graph/selectionToolbox/ExecuteButton.vue
@@ -19,11 +19,11 @@
 </template>
 
 <script setup lang="ts">
-import type { LGraphNode } from '@comfyorg/litegraph'
 import Button from 'primevue/button'
 import { computed, ref } from 'vue'
 import { useI18n } from 'vue-i18n'
 
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useCommandStore } from '@/stores/commandStore'
 import { useCanvasStore } from '@/stores/graphStore'
 import { isLGraphNode } from '@/utils/litegraphUtil'
diff --git a/src/components/graph/widgets/TextPreviewWidget.vue b/src/components/graph/widgets/TextPreviewWidget.vue
index 7244634a4..adf7c042b 100644
--- a/src/components/graph/widgets/TextPreviewWidget.vue
+++ b/src/components/graph/widgets/TextPreviewWidget.vue
@@ -12,10 +12,10 @@
 </template>
 
 <script setup lang="ts">
-import { NodeId } from '@comfyorg/litegraph'
 import Skeleton from 'primevue/skeleton'
 import { computed, onMounted, ref, watch } from 'vue'
 
+import { NodeId } from '@/lib/litegraph/src/litegraph'
 import { useExecutionStore } from '@/stores/executionStore'
 import { linkifyHtml, nl2br } from '@/utils/formatUtil'
 
diff --git a/src/components/load3d/Load3DScene.vue b/src/components/load3d/Load3DScene.vue
index d7c7dcae1..2bf38611a 100644
--- a/src/components/load3d/Load3DScene.vue
+++ b/src/components/load3d/Load3DScene.vue
@@ -5,7 +5,6 @@
 </template>
 
 <script setup lang="ts">
-import { LGraphNode } from '@comfyorg/litegraph'
 import { onMounted, onUnmounted, ref, toRaw, watch } from 'vue'
 
 import LoadingOverlay from '@/components/load3d/LoadingOverlay.vue'
@@ -17,6 +16,7 @@ import {
   UpDirection
 } from '@/extensions/core/load3d/interfaces'
 import { t } from '@/i18n'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { CustomInputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import { useLoad3dService } from '@/services/load3dService'
 
diff --git a/src/components/load3d/controls/RecordingControls.vue b/src/components/load3d/controls/RecordingControls.vue
index 9eb305a47..92fbf8c24 100644
--- a/src/components/load3d/controls/RecordingControls.vue
+++ b/src/components/load3d/controls/RecordingControls.vue
@@ -75,11 +75,11 @@
 </template>
 
 <script setup lang="ts">
-import { LGraphNode } from '@comfyorg/litegraph'
 import { Tooltip } from 'primevue'
 import Button from 'primevue/button'
 
 import { t } from '@/i18n'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useLoad3dService } from '@/services/load3dService'
 
 const vTooltip = Tooltip
diff --git a/src/components/searchbox/NodeSearchBoxPopover.vue b/src/components/searchbox/NodeSearchBoxPopover.vue
index ad4618989..f1a9be92d 100644
--- a/src/components/searchbox/NodeSearchBoxPopover.vue
+++ b/src/components/searchbox/NodeSearchBoxPopover.vue
@@ -33,18 +33,18 @@
 </template>
 
 <script setup lang="ts">
-import {
-  LGraphNode,
-  LiteGraph,
-  LiteGraphCanvasEvent
-} from '@comfyorg/litegraph'
-import { Point } from '@comfyorg/litegraph/dist/interfaces'
-import type { CanvasPointerEvent } from '@comfyorg/litegraph/dist/types/events'
 import { useEventListener } from '@vueuse/core'
 import { storeToRefs } from 'pinia'
 import Dialog from 'primevue/dialog'
 import { computed, ref, toRaw, watch, watchEffect } from 'vue'
 
+import { Point } from '@/lib/litegraph/src/interfaces'
+import {
+  LGraphNode,
+  LiteGraph,
+  LiteGraphCanvasEvent
+} from '@/lib/litegraph/src/litegraph'
+import type { CanvasPointerEvent } from '@/lib/litegraph/src/types/events'
 import { useLitegraphService } from '@/services/litegraphService'
 import { useCanvasStore } from '@/stores/graphStore'
 import { ComfyNodeDefImpl, useNodeDefStore } from '@/stores/nodeDefStore'
diff --git a/src/composables/canvas/useCanvasTransformSync.ts b/src/composables/canvas/useCanvasTransformSync.ts
index cddb4c7f7..fe0d2b457 100644
--- a/src/composables/canvas/useCanvasTransformSync.ts
+++ b/src/composables/canvas/useCanvasTransformSync.ts
@@ -1,6 +1,6 @@
-import { LGraphCanvas } from '@comfyorg/litegraph'
 import { onUnmounted, ref } from 'vue'
 
+import { LGraphCanvas } from '@/lib/litegraph/src/litegraph'
 import { useCanvasStore } from '@/stores/graphStore'
 
 interface CanvasTransformSyncOptions {
diff --git a/src/composables/canvas/useSelectedLiteGraphItems.ts b/src/composables/canvas/useSelectedLiteGraphItems.ts
index 32c29a1b2..0400bc833 100644
--- a/src/composables/canvas/useSelectedLiteGraphItems.ts
+++ b/src/composables/canvas/useSelectedLiteGraphItems.ts
@@ -3,8 +3,7 @@ import {
   LGraphNode,
   Positionable,
   Reroute
-} from '@comfyorg/litegraph'
-
+} from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { useCanvasStore } from '@/stores/graphStore'
 import {
diff --git a/src/composables/element/useAbsolutePosition.ts b/src/composables/element/useAbsolutePosition.ts
index bca981669..4eb0f4d82 100644
--- a/src/composables/element/useAbsolutePosition.ts
+++ b/src/composables/element/useAbsolutePosition.ts
@@ -1,7 +1,7 @@
-import type { Size, Vector2 } from '@comfyorg/litegraph'
 import { CSSProperties, ref, watch } from 'vue'
 
 import { useCanvasPositionConversion } from '@/composables/element/useCanvasPositionConversion'
+import type { Size, Vector2 } from '@/lib/litegraph/src/litegraph'
 import { useCanvasStore } from '@/stores/graphStore'
 import { useSettingStore } from '@/stores/settingStore'
 
diff --git a/src/composables/element/useCanvasPositionConversion.ts b/src/composables/element/useCanvasPositionConversion.ts
index 2c0f7198e..e7fb9ad9b 100644
--- a/src/composables/element/useCanvasPositionConversion.ts
+++ b/src/composables/element/useCanvasPositionConversion.ts
@@ -1,6 +1,7 @@
-import type { LGraphCanvas, Vector2 } from '@comfyorg/litegraph'
 import { useElementBounding } from '@vueuse/core'
 
+import type { LGraphCanvas, Vector2 } from '@/lib/litegraph/src/litegraph'
+
 /**
  * Convert between canvas and client positions
  * @param canvasElement - The canvas element
diff --git a/src/composables/node/useNodeAnimatedImage.ts b/src/composables/node/useNodeAnimatedImage.ts
index 3dabb8e9b..ac2e9b161 100644
--- a/src/composables/node/useNodeAnimatedImage.ts
+++ b/src/composables/node/useNodeAnimatedImage.ts
@@ -1,6 +1,5 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { IWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { IWidget } from '@/lib/litegraph/src/types/widgets'
 import { ANIM_PREVIEW_WIDGET } from '@/scripts/app'
 import { createImageHost } from '@/scripts/ui/imagePreview'
 import { fitDimensionsToNodeWidth } from '@/utils/imageUtil'
diff --git a/src/composables/node/useNodeBadge.ts b/src/composables/node/useNodeBadge.ts
index 736349853..2affec44b 100644
--- a/src/composables/node/useNodeBadge.ts
+++ b/src/composables/node/useNodeBadge.ts
@@ -1,13 +1,13 @@
-import {
-  BadgePosition,
-  LGraphBadge,
-  type LGraphNode
-} from '@comfyorg/litegraph'
 import _ from 'lodash'
 import { computed, onMounted, watch } from 'vue'
 
 import { useNodePricing } from '@/composables/node/useNodePricing'
 import { useComputedWithWidgetWatch } from '@/composables/node/useWatchWidget'
+import {
+  BadgePosition,
+  LGraphBadge,
+  type LGraphNode
+} from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { useExtensionStore } from '@/stores/extensionStore'
 import { ComfyNodeDefImpl, useNodeDefStore } from '@/stores/nodeDefStore'
diff --git a/src/composables/node/useNodeCanvasImagePreview.ts b/src/composables/node/useNodeCanvasImagePreview.ts
index 52759b9d9..98d49b485 100644
--- a/src/composables/node/useNodeCanvasImagePreview.ts
+++ b/src/composables/node/useNodeCanvasImagePreview.ts
@@ -1,6 +1,5 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-
 import { useImagePreviewWidget } from '@/composables/widgets/useImagePreviewWidget'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 const CANVAS_IMAGE_PREVIEW_WIDGET = '$$canvas-image-preview'
 
diff --git a/src/composables/node/useNodeChatHistory.ts b/src/composables/node/useNodeChatHistory.ts
index 135a48b66..8fbe78895 100644
--- a/src/composables/node/useNodeChatHistory.ts
+++ b/src/composables/node/useNodeChatHistory.ts
@@ -1,7 +1,6 @@
-import { LGraphNode } from '@comfyorg/litegraph'
-
 import type ChatHistoryWidget from '@/components/graph/widgets/ChatHistoryWidget.vue'
 import { useChatHistoryWidget } from '@/composables/widgets/useChatHistoryWidget'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 const CHAT_HISTORY_WIDGET_NAME = '$$node-chat-history'
 
diff --git a/src/composables/node/useNodeDragAndDrop.ts b/src/composables/node/useNodeDragAndDrop.ts
index cb603814c..62d15830e 100644
--- a/src/composables/node/useNodeDragAndDrop.ts
+++ b/src/composables/node/useNodeDragAndDrop.ts
@@ -1,4 +1,4 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 type DragHandler = (e: DragEvent) => boolean
 type DropHandler<T> = (files: File[]) => Promise<T[]>
diff --git a/src/composables/node/useNodeFileInput.ts b/src/composables/node/useNodeFileInput.ts
index ef61aaa0a..977178284 100644
--- a/src/composables/node/useNodeFileInput.ts
+++ b/src/composables/node/useNodeFileInput.ts
@@ -1,6 +1,5 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-
 import { useChainCallback } from '@/composables/functional/useChainCallback'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 interface FileInputOptions {
   accept?: string
diff --git a/src/composables/node/useNodeImage.ts b/src/composables/node/useNodeImage.ts
index 750459229..ab35111d7 100644
--- a/src/composables/node/useNodeImage.ts
+++ b/src/composables/node/useNodeImage.ts
@@ -1,5 +1,4 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useNodeOutputStore } from '@/stores/imagePreviewStore'
 import { fitDimensionsToNodeWidth } from '@/utils/imageUtil'
 
diff --git a/src/composables/node/useNodeImageUpload.ts b/src/composables/node/useNodeImageUpload.ts
index 2abf3b706..e586a9c0b 100644
--- a/src/composables/node/useNodeImageUpload.ts
+++ b/src/composables/node/useNodeImageUpload.ts
@@ -1,8 +1,7 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-
 import { useNodeDragAndDrop } from '@/composables/node/useNodeDragAndDrop'
 import { useNodeFileInput } from '@/composables/node/useNodeFileInput'
 import { useNodePaste } from '@/composables/node/useNodePaste'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { ResultItemType } from '@/schemas/apiSchema'
 import { api } from '@/scripts/api'
 import { useToastStore } from '@/stores/toastStore'
diff --git a/src/composables/node/useNodePaste.ts b/src/composables/node/useNodePaste.ts
index d198db926..3559ad119 100644
--- a/src/composables/node/useNodePaste.ts
+++ b/src/composables/node/useNodePaste.ts
@@ -1,4 +1,4 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 type PasteHandler<T> = (files: File[]) => Promise<T>
 
diff --git a/src/composables/node/useNodePricing.ts b/src/composables/node/useNodePricing.ts
index 36aaff5a7..8d521b81e 100644
--- a/src/composables/node/useNodePricing.ts
+++ b/src/composables/node/useNodePricing.ts
@@ -1,5 +1,5 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { IComboWidget } from '@comfyorg/litegraph/dist/types/widgets'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { IComboWidget } from '@/lib/litegraph/src/types/widgets'
 
 /**
  * Function that calculates dynamic pricing based on node widget values
diff --git a/src/composables/node/useNodeProgressText.ts b/src/composables/node/useNodeProgressText.ts
index 764b3413f..12e09bd5e 100644
--- a/src/composables/node/useNodeProgressText.ts
+++ b/src/composables/node/useNodeProgressText.ts
@@ -1,6 +1,5 @@
-import { LGraphNode } from '@comfyorg/litegraph'
-
 import { useTextPreviewWidget } from '@/composables/widgets/useProgressTextWidget'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 const TEXT_PREVIEW_WIDGET_NAME = '$$node-text-preview'
 
diff --git a/src/composables/node/useWatchWidget.ts b/src/composables/node/useWatchWidget.ts
index e8827d6ec..f30325696 100644
--- a/src/composables/node/useWatchWidget.ts
+++ b/src/composables/node/useWatchWidget.ts
@@ -1,8 +1,8 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { computedWithControl } from '@vueuse/core'
 import { type ComputedRef, ref } from 'vue'
 
 import { useChainCallback } from '@/composables/functional/useChainCallback'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 export interface UseComputedWithWidgetWatchOptions {
   /**
diff --git a/src/composables/nodePack/useMissingNodes.ts b/src/composables/nodePack/useMissingNodes.ts
index a7008ff08..1821a4090 100644
--- a/src/composables/nodePack/useMissingNodes.ts
+++ b/src/composables/nodePack/useMissingNodes.ts
@@ -1,9 +1,9 @@
-import { LGraphNode } from '@comfyorg/litegraph'
-import { NodeProperty } from '@comfyorg/litegraph/dist/LGraphNode'
 import { groupBy } from 'lodash'
 import { computed, onMounted } from 'vue'
 
 import { useWorkflowPacks } from '@/composables/nodePack/useWorkflowPacks'
+import { NodeProperty } from '@/lib/litegraph/src/LGraphNode'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { useComfyManagerStore } from '@/stores/comfyManagerStore'
 import { useNodeDefStore } from '@/stores/nodeDefStore'
diff --git a/src/composables/nodePack/useWorkflowPacks.ts b/src/composables/nodePack/useWorkflowPacks.ts
index 8096d1d49..32770ff55 100644
--- a/src/composables/nodePack/useWorkflowPacks.ts
+++ b/src/composables/nodePack/useWorkflowPacks.ts
@@ -1,7 +1,7 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import { computed, onUnmounted, ref } from 'vue'
 
 import { useNodePacks } from '@/composables/nodePack/useNodePacks'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import { app } from '@/scripts/app'
 import { useComfyRegistryStore } from '@/stores/comfyRegistryStore'
diff --git a/src/composables/useCanvasDrop.ts b/src/composables/useCanvasDrop.ts
index 1948b13d3..00e4f3ca8 100644
--- a/src/composables/useCanvasDrop.ts
+++ b/src/composables/useCanvasDrop.ts
@@ -1,8 +1,8 @@
-import { LGraphNode } from '@comfyorg/litegraph'
-import { LiteGraph } from '@comfyorg/litegraph'
 import { Ref } from 'vue'
 
 import { usePragmaticDroppable } from '@/composables/usePragmaticDragAndDrop'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
 import { app as comfyApp } from '@/scripts/app'
 import { useLitegraphService } from '@/services/litegraphService'
 import { useWorkflowService } from '@/services/workflowService'
diff --git a/src/composables/useContextMenuTranslation.ts b/src/composables/useContextMenuTranslation.ts
index 80a47749b..1fd851420 100644
--- a/src/composables/useContextMenuTranslation.ts
+++ b/src/composables/useContextMenuTranslation.ts
@@ -1,12 +1,11 @@
+import { st, te } from '@/i18n'
 import type {
   IContextMenuOptions,
   IContextMenuValue,
   INodeInputSlot,
   IWidget
-} from '@comfyorg/litegraph'
-import { LGraphCanvas, LiteGraph } from '@comfyorg/litegraph'
-
-import { st, te } from '@/i18n'
+} from '@/lib/litegraph/src/litegraph'
+import { LGraphCanvas, LiteGraph } from '@/lib/litegraph/src/litegraph'
 import { normalizeI18nKey } from '@/utils/formatUtil'
 
 /**
diff --git a/src/composables/useCoreCommands.ts b/src/composables/useCoreCommands.ts
index 26b795b62..9752321f9 100644
--- a/src/composables/useCoreCommands.ts
+++ b/src/composables/useCoreCommands.ts
@@ -1,11 +1,3 @@
-import {
-  LGraphEventMode,
-  LGraphGroup,
-  LGraphNode,
-  LiteGraph
-} from '@comfyorg/litegraph'
-import { Point } from '@comfyorg/litegraph'
-
 import { useFirebaseAuthActions } from '@/composables/auth/useFirebaseAuthActions'
 import { useSelectedLiteGraphItems } from '@/composables/canvas/useSelectedLiteGraphItems'
 import {
@@ -13,6 +5,13 @@ import {
   DEFAULT_LIGHT_COLOR_PALETTE
 } from '@/constants/coreColorPalettes'
 import { t } from '@/i18n'
+import {
+  LGraphEventMode,
+  LGraphGroup,
+  LGraphNode,
+  LiteGraph
+} from '@/lib/litegraph/src/litegraph'
+import { Point } from '@/lib/litegraph/src/litegraph'
 import { api } from '@/scripts/api'
 import { app } from '@/scripts/app'
 import { addFluxKontextGroupNode } from '@/scripts/fluxKontextEditNode'
diff --git a/src/composables/useGlobalLitegraph.ts b/src/composables/useGlobalLitegraph.ts
index 7a64588c6..107f8e289 100644
--- a/src/composables/useGlobalLitegraph.ts
+++ b/src/composables/useGlobalLitegraph.ts
@@ -8,7 +8,7 @@ import {
   LGraphNode,
   LLink,
   LiteGraph
-} from '@comfyorg/litegraph'
+} from '@/lib/litegraph/src/litegraph'
 
 /**
  * Assign all properties of LiteGraph to window to make it backward compatible.
diff --git a/src/composables/useLitegraphSettings.ts b/src/composables/useLitegraphSettings.ts
index 79eaee1c3..823ab027b 100644
--- a/src/composables/useLitegraphSettings.ts
+++ b/src/composables/useLitegraphSettings.ts
@@ -1,6 +1,10 @@
-import { CanvasPointer, LGraphNode, LiteGraph } from '@comfyorg/litegraph'
 import { watchEffect } from 'vue'
 
+import {
+  CanvasPointer,
+  LGraphNode,
+  LiteGraph
+} from '@/lib/litegraph/src/litegraph'
 import { useCanvasStore } from '@/stores/graphStore'
 import { useSettingStore } from '@/stores/settingStore'
 
diff --git a/src/composables/useMinimap.ts b/src/composables/useMinimap.ts
index da4f6bcb0..9f4f06e65 100644
--- a/src/composables/useMinimap.ts
+++ b/src/composables/useMinimap.ts
@@ -1,8 +1,8 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { useRafFn, useThrottleFn } from '@vueuse/core'
 import { computed, nextTick, ref, watch } from 'vue'
 
 import { useCanvasTransformSync } from '@/composables/canvas/useCanvasTransformSync'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { NodeId } from '@/schemas/comfyWorkflowSchema'
 import { api } from '@/scripts/api'
 import { app } from '@/scripts/app'
diff --git a/src/composables/usePaste.ts b/src/composables/usePaste.ts
index ce6c68d82..2a4d2d8ad 100644
--- a/src/composables/usePaste.ts
+++ b/src/composables/usePaste.ts
@@ -1,7 +1,7 @@
-import { LiteGraph } from '@comfyorg/litegraph'
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { useEventListener } from '@vueuse/core'
 
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import { app } from '@/scripts/app'
 import { useCanvasStore } from '@/stores/graphStore'
diff --git a/src/composables/useRefreshableSelection.ts b/src/composables/useRefreshableSelection.ts
index 20a7c5e75..bf2a6660f 100644
--- a/src/composables/useRefreshableSelection.ts
+++ b/src/composables/useRefreshableSelection.ts
@@ -1,6 +1,6 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { computed, ref, watchEffect } from 'vue'
 
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useCanvasStore } from '@/stores/graphStore'
 import { isLGraphNode } from '@/utils/litegraphUtil'
 
diff --git a/src/composables/useWorkflowValidation.ts b/src/composables/useWorkflowValidation.ts
index d8823098d..8577734a5 100644
--- a/src/composables/useWorkflowValidation.ts
+++ b/src/composables/useWorkflowValidation.ts
@@ -1,5 +1,4 @@
-import type { ISerialisedGraph } from '@comfyorg/litegraph/dist/types/serialisation'
-
+import type { ISerialisedGraph } from '@/lib/litegraph/src/types/serialisation'
 import type { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import { validateComfyWorkflow } from '@/schemas/comfyWorkflowSchema'
 import { useToastStore } from '@/stores/toastStore'
diff --git a/src/composables/widgets/useBooleanWidget.ts b/src/composables/widgets/useBooleanWidget.ts
index 233c04aa5..bc307e3b0 100644
--- a/src/composables/widgets/useBooleanWidget.ts
+++ b/src/composables/widgets/useBooleanWidget.ts
@@ -1,5 +1,4 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import {
   type InputSpec,
   isBooleanInputSpec
diff --git a/src/composables/widgets/useChatHistoryWidget.ts b/src/composables/widgets/useChatHistoryWidget.ts
index a3568fb21..53fe6ce6e 100644
--- a/src/composables/widgets/useChatHistoryWidget.ts
+++ b/src/composables/widgets/useChatHistoryWidget.ts
@@ -1,7 +1,7 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { ref } from 'vue'
 
 import ChatHistoryWidget from '@/components/graph/widgets/ChatHistoryWidget.vue'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import {
   ComponentWidgetImpl,
diff --git a/src/composables/widgets/useComboWidget.ts b/src/composables/widgets/useComboWidget.ts
index 25c75fb5c..ad973325d 100644
--- a/src/composables/widgets/useComboWidget.ts
+++ b/src/composables/widgets/useComboWidget.ts
@@ -1,8 +1,8 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { IComboWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import { ref } from 'vue'
 
 import MultiSelectWidget from '@/components/graph/widgets/MultiSelectWidget.vue'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { IComboWidget } from '@/lib/litegraph/src/types/widgets'
 import { transformInputSpecV2ToV1 } from '@/schemas/nodeDef/migration'
 import {
   ComboInputSpec,
diff --git a/src/composables/widgets/useFloatWidget.ts b/src/composables/widgets/useFloatWidget.ts
index bdf059742..cfd40bf8f 100644
--- a/src/composables/widgets/useFloatWidget.ts
+++ b/src/composables/widgets/useFloatWidget.ts
@@ -1,7 +1,7 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { INumericWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import _ from 'lodash'
 
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { INumericWidget } from '@/lib/litegraph/src/types/widgets'
 import {
   type InputSpec,
   isFloatInputSpec
diff --git a/src/composables/widgets/useImagePreviewWidget.ts b/src/composables/widgets/useImagePreviewWidget.ts
index 1230873b2..444a9409f 100644
--- a/src/composables/widgets/useImagePreviewWidget.ts
+++ b/src/composables/widgets/useImagePreviewWidget.ts
@@ -3,12 +3,11 @@ import {
   type CanvasPointer,
   type LGraphNode,
   LiteGraph
-} from '@comfyorg/litegraph'
+} from '@/lib/litegraph/src/litegraph'
 import type {
   IBaseWidget,
   IWidgetOptions
-} from '@comfyorg/litegraph/dist/types/widgets'
-
+} from '@/lib/litegraph/src/types/widgets'
 import type { InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import { app } from '@/scripts/app'
 import { calculateImageGrid } from '@/scripts/ui/imagePreview'
diff --git a/src/composables/widgets/useImageUploadWidget.ts b/src/composables/widgets/useImageUploadWidget.ts
index 137b2f42c..bf4e18fcb 100644
--- a/src/composables/widgets/useImageUploadWidget.ts
+++ b/src/composables/widgets/useImageUploadWidget.ts
@@ -1,10 +1,9 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import { IComboWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
 import { useNodeImage, useNodeVideo } from '@/composables/node/useNodeImage'
 import { useNodeImageUpload } from '@/composables/node/useNodeImageUpload'
 import { useValueTransform } from '@/composables/useValueTransform'
 import { t } from '@/i18n'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import { IComboWidget } from '@/lib/litegraph/src/types/widgets'
 import type { ResultItem, ResultItemType } from '@/schemas/apiSchema'
 import type { InputSpec } from '@/schemas/nodeDefSchema'
 import type { ComfyWidgetConstructor } from '@/scripts/widgets'
diff --git a/src/composables/widgets/useIntWidget.ts b/src/composables/widgets/useIntWidget.ts
index ee1f51fab..d7b0719e9 100644
--- a/src/composables/widgets/useIntWidget.ts
+++ b/src/composables/widgets/useIntWidget.ts
@@ -1,6 +1,5 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { INumericWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { INumericWidget } from '@/lib/litegraph/src/types/widgets'
 import { transformInputSpecV2ToV1 } from '@/schemas/nodeDef/migration'
 import {
   type InputSpec,
diff --git a/src/composables/widgets/useMarkdownWidget.ts b/src/composables/widgets/useMarkdownWidget.ts
index b19d3d9a0..fd91330be 100644
--- a/src/composables/widgets/useMarkdownWidget.ts
+++ b/src/composables/widgets/useMarkdownWidget.ts
@@ -1,4 +1,3 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { Editor as TiptapEditor } from '@tiptap/core'
 import TiptapLink from '@tiptap/extension-link'
 import TiptapTable from '@tiptap/extension-table'
@@ -8,6 +7,7 @@ import TiptapTableRow from '@tiptap/extension-table-row'
 import TiptapStarterKit from '@tiptap/starter-kit'
 import { Markdown as TiptapMarkdown } from 'tiptap-markdown'
 
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { type InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import { app } from '@/scripts/app'
 import { type ComfyWidgetConstructorV2 } from '@/scripts/widgets'
diff --git a/src/composables/widgets/useProgressTextWidget.ts b/src/composables/widgets/useProgressTextWidget.ts
index 1e7dc45ce..dd3f1d0c2 100644
--- a/src/composables/widgets/useProgressTextWidget.ts
+++ b/src/composables/widgets/useProgressTextWidget.ts
@@ -1,7 +1,7 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { ref } from 'vue'
 
 import TextPreviewWidget from '@/components/graph/widgets/TextPreviewWidget.vue'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import {
   ComponentWidgetImpl,
diff --git a/src/composables/widgets/useRemoteWidget.ts b/src/composables/widgets/useRemoteWidget.ts
index 8cfcfda1f..0fde7592a 100644
--- a/src/composables/widgets/useRemoteWidget.ts
+++ b/src/composables/widgets/useRemoteWidget.ts
@@ -1,8 +1,8 @@
-import { LGraphNode } from '@comfyorg/litegraph'
-import { IWidget } from '@comfyorg/litegraph'
 import axios from 'axios'
 
 import { useChainCallback } from '@/composables/functional/useChainCallback'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import { IWidget } from '@/lib/litegraph/src/litegraph'
 import type { RemoteWidgetConfig } from '@/schemas/nodeDefSchema'
 import { api } from '@/scripts/api'
 
diff --git a/src/composables/widgets/useStringWidget.ts b/src/composables/widgets/useStringWidget.ts
index b3f438a96..4766925ee 100644
--- a/src/composables/widgets/useStringWidget.ts
+++ b/src/composables/widgets/useStringWidget.ts
@@ -1,5 +1,4 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import {
   type InputSpec,
   isStringInputSpec
diff --git a/src/constants/coreSettings.ts b/src/constants/coreSettings.ts
index f79d779b7..0765bf61a 100644
--- a/src/constants/coreSettings.ts
+++ b/src/constants/coreSettings.ts
@@ -1,5 +1,4 @@
-import { LinkMarkerShape, LiteGraph } from '@comfyorg/litegraph'
-
+import { LinkMarkerShape, LiteGraph } from '@/lib/litegraph/src/litegraph'
 import type { ColorPalettes } from '@/schemas/colorPaletteSchema'
 import type { Keybinding } from '@/schemas/keyBindingSchema'
 import { NodeBadgeMode } from '@/types/nodeSource'
diff --git a/src/extensions/core/contextMenuFilter.ts b/src/extensions/core/contextMenuFilter.ts
index dec9fe4c7..5cb0e98c5 100644
--- a/src/extensions/core/contextMenuFilter.ts
+++ b/src/extensions/core/contextMenuFilter.ts
@@ -1,4 +1,8 @@
-import { LGraphCanvas, LiteGraph, isComboWidget } from '@comfyorg/litegraph'
+import {
+  LGraphCanvas,
+  LiteGraph,
+  isComboWidget
+} from '@/lib/litegraph/src/litegraph'
 
 import { app } from '../../scripts/app'
 
diff --git a/src/extensions/core/groupNode.ts b/src/extensions/core/groupNode.ts
index 7aa714133..0b7cbb700 100644
--- a/src/extensions/core/groupNode.ts
+++ b/src/extensions/core/groupNode.ts
@@ -1,13 +1,12 @@
+import { t } from '@/i18n'
+import { type NodeId } from '@/lib/litegraph/src/LGraphNode'
 import {
   type ExecutableLGraphNode,
   type ExecutionId,
   LGraphNode,
   LiteGraph,
   SubgraphNode
-} from '@comfyorg/litegraph'
-import { type NodeId } from '@comfyorg/litegraph/dist/LGraphNode'
-
-import { t } from '@/i18n'
+} from '@/lib/litegraph/src/litegraph'
 import {
   ComfyLink,
   ComfyNode,
diff --git a/src/extensions/core/groupNodeManage.ts b/src/extensions/core/groupNodeManage.ts
index 51bec17f4..fbf04033f 100644
--- a/src/extensions/core/groupNodeManage.ts
+++ b/src/extensions/core/groupNodeManage.ts
@@ -2,8 +2,7 @@ import {
   type LGraphNode,
   type LGraphNodeConstructor,
   LiteGraph
-} from '@comfyorg/litegraph'
-
+} from '@/lib/litegraph/src/litegraph'
 import { useToastStore } from '@/stores/toastStore'
 
 import { type ComfyApp, app } from '../../scripts/app'
diff --git a/src/extensions/core/groupOptions.ts b/src/extensions/core/groupOptions.ts
index 0a044d1f5..945f5bd90 100644
--- a/src/extensions/core/groupOptions.ts
+++ b/src/extensions/core/groupOptions.ts
@@ -1,8 +1,7 @@
-import { LGraphGroup } from '@comfyorg/litegraph'
-import { LGraphCanvas } from '@comfyorg/litegraph'
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { Positionable } from '@comfyorg/litegraph/dist/interfaces'
-
+import type { Positionable } from '@/lib/litegraph/src/interfaces'
+import { LGraphGroup } from '@/lib/litegraph/src/litegraph'
+import { LGraphCanvas } from '@/lib/litegraph/src/litegraph'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useSettingStore } from '@/stores/settingStore'
 
 import { app } from '../../scripts/app'
diff --git a/src/extensions/core/load3d.ts b/src/extensions/core/load3d.ts
index 373eb701f..5dc97ae75 100644
--- a/src/extensions/core/load3d.ts
+++ b/src/extensions/core/load3d.ts
@@ -1,4 +1,3 @@
-import type { IStringWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import { nextTick } from 'vue'
 
 import Load3D from '@/components/load3d/Load3D.vue'
@@ -7,6 +6,7 @@ import Load3DConfiguration from '@/extensions/core/load3d/Load3DConfiguration'
 import Load3dAnimation from '@/extensions/core/load3d/Load3dAnimation'
 import Load3dUtils from '@/extensions/core/load3d/Load3dUtils'
 import { t } from '@/i18n'
+import type { IStringWidget } from '@/lib/litegraph/src/types/widgets'
 import { CustomInputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import { api } from '@/scripts/api'
 import { ComponentWidgetImpl, addWidget } from '@/scripts/domWidget'
diff --git a/src/extensions/core/load3d/Load3DConfiguration.ts b/src/extensions/core/load3d/Load3DConfiguration.ts
index ec029298c..a79ae3fb2 100644
--- a/src/extensions/core/load3d/Load3DConfiguration.ts
+++ b/src/extensions/core/load3d/Load3DConfiguration.ts
@@ -1,7 +1,6 @@
-import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
 import Load3d from '@/extensions/core/load3d/Load3d'
 import Load3dUtils from '@/extensions/core/load3d/Load3dUtils'
+import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
 import { api } from '@/scripts/api'
 import { useSettingStore } from '@/stores/settingStore'
 
diff --git a/src/extensions/core/load3d/Load3d.ts b/src/extensions/core/load3d/Load3d.ts
index d33f61031..cacece808 100644
--- a/src/extensions/core/load3d/Load3d.ts
+++ b/src/extensions/core/load3d/Load3d.ts
@@ -1,6 +1,6 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import * as THREE from 'three'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { CustomInputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 
 import { CameraManager } from './CameraManager'
diff --git a/src/extensions/core/load3d/Load3dAnimation.ts b/src/extensions/core/load3d/Load3dAnimation.ts
index 207b27497..849cdcf31 100644
--- a/src/extensions/core/load3d/Load3dAnimation.ts
+++ b/src/extensions/core/load3d/Load3dAnimation.ts
@@ -1,6 +1,7 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import * as THREE from 'three'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
+
 import { AnimationManager } from './AnimationManager'
 import Load3d from './Load3d'
 import { Load3DOptions } from './interfaces'
diff --git a/src/extensions/core/load3d/NodeStorage.ts b/src/extensions/core/load3d/NodeStorage.ts
index 7791a1fe2..44417fbc6 100644
--- a/src/extensions/core/load3d/NodeStorage.ts
+++ b/src/extensions/core/load3d/NodeStorage.ts
@@ -1,4 +1,4 @@
-import { LGraphNode } from '@comfyorg/litegraph'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 import { NodeStorageInterface } from './interfaces'
 
diff --git a/src/extensions/core/load3d/interfaces.ts b/src/extensions/core/load3d/interfaces.ts
index 603740518..0142fda91 100644
--- a/src/extensions/core/load3d/interfaces.ts
+++ b/src/extensions/core/load3d/interfaces.ts
@@ -1,4 +1,3 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import * as THREE from 'three'
 import { OrbitControls } from 'three/examples/jsm/controls/OrbitControls'
 import { ViewHelper } from 'three/examples/jsm/helpers/ViewHelper'
@@ -8,6 +7,7 @@ import { MTLLoader } from 'three/examples/jsm/loaders/MTLLoader'
 import { OBJLoader } from 'three/examples/jsm/loaders/OBJLoader'
 import { STLLoader } from 'three/examples/jsm/loaders/STLLoader'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { CustomInputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 
 export type Load3DNodeType = 'Load3D' | 'Preview3D'
diff --git a/src/extensions/core/nodeTemplates.ts b/src/extensions/core/nodeTemplates.ts
index 05c569a22..5ca26a934 100644
--- a/src/extensions/core/nodeTemplates.ts
+++ b/src/extensions/core/nodeTemplates.ts
@@ -1,6 +1,5 @@
-import { LGraphCanvas } from '@comfyorg/litegraph'
-
 import { t } from '@/i18n'
+import { LGraphCanvas } from '@/lib/litegraph/src/litegraph'
 import { useDialogService } from '@/services/dialogService'
 import { useToastStore } from '@/stores/toastStore'
 import { deserialiseAndCreate } from '@/utils/vintageClipboard'
diff --git a/src/extensions/core/noteNode.ts b/src/extensions/core/noteNode.ts
index d5cbf0dee..8b2d9ed99 100644
--- a/src/extensions/core/noteNode.ts
+++ b/src/extensions/core/noteNode.ts
@@ -1,5 +1,5 @@
-import { LGraphCanvas, LiteGraph } from '@comfyorg/litegraph'
-import { LGraphNode } from '@comfyorg/litegraph'
+import { LGraphCanvas, LiteGraph } from '@/lib/litegraph/src/litegraph'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 import { app } from '../../scripts/app'
 import { ComfyWidgets } from '../../scripts/widgets'
diff --git a/src/extensions/core/rerouteNode.ts b/src/extensions/core/rerouteNode.ts
index 0df8263ab..64b152e19 100644
--- a/src/extensions/core/rerouteNode.ts
+++ b/src/extensions/core/rerouteNode.ts
@@ -1,5 +1,9 @@
-import type { IContextMenuValue } from '@comfyorg/litegraph'
-import { LGraphCanvas, LGraphNode, LiteGraph } from '@comfyorg/litegraph'
+import type { IContextMenuValue } from '@/lib/litegraph/src/litegraph'
+import {
+  LGraphCanvas,
+  LGraphNode,
+  LiteGraph
+} from '@/lib/litegraph/src/litegraph'
 
 import { app } from '../../scripts/app'
 import { getWidgetConfig, mergeIfValid, setWidgetConfig } from './widgetInputs'
diff --git a/src/extensions/core/simpleTouchSupport.ts b/src/extensions/core/simpleTouchSupport.ts
index 5734ada72..c129711ec 100644
--- a/src/extensions/core/simpleTouchSupport.ts
+++ b/src/extensions/core/simpleTouchSupport.ts
@@ -1,4 +1,4 @@
-import { LGraphCanvas, LiteGraph } from '@comfyorg/litegraph'
+import { LGraphCanvas, LiteGraph } from '@/lib/litegraph/src/litegraph'
 
 import { app } from '../../scripts/app'
 
diff --git a/src/extensions/core/slotDefaults.ts b/src/extensions/core/slotDefaults.ts
index c3bfd657e..f7ca293a4 100644
--- a/src/extensions/core/slotDefaults.ts
+++ b/src/extensions/core/slotDefaults.ts
@@ -1,4 +1,4 @@
-import { LiteGraph } from '@comfyorg/litegraph'
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
 
 import { app } from '../../scripts/app'
 import { ComfyWidgets } from '../../scripts/widgets'
diff --git a/src/extensions/core/uploadAudio.ts b/src/extensions/core/uploadAudio.ts
index 701fc0b5d..dd6351244 100644
--- a/src/extensions/core/uploadAudio.ts
+++ b/src/extensions/core/uploadAudio.ts
@@ -1,8 +1,3 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type {
-  IBaseWidget,
-  IStringWidget
-} from '@comfyorg/litegraph/dist/types/widgets'
 import { MediaRecorder as ExtendableMediaRecorder } from 'extendable-media-recorder'
 
 import { useChainCallback } from '@/composables/functional/useChainCallback'
@@ -10,6 +5,11 @@ import { useNodeDragAndDrop } from '@/composables/node/useNodeDragAndDrop'
 import { useNodeFileInput } from '@/composables/node/useNodeFileInput'
 import { useNodePaste } from '@/composables/node/useNodePaste'
 import { t } from '@/i18n'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type {
+  IBaseWidget,
+  IStringWidget
+} from '@/lib/litegraph/src/types/widgets'
 import type { ResultItemType } from '@/schemas/apiSchema'
 import type { ComfyNodeDef } from '@/schemas/nodeDefSchema'
 import type { DOMWidget } from '@/scripts/domWidget'
diff --git a/src/extensions/core/widgetInputs.ts b/src/extensions/core/widgetInputs.ts
index fbc23b231..bc0725627 100644
--- a/src/extensions/core/widgetInputs.ts
+++ b/src/extensions/core/widgetInputs.ts
@@ -1,18 +1,17 @@
-import { LGraphNode, LiteGraph } from '@comfyorg/litegraph'
+import {
+  type CallbackParams,
+  useChainCallback
+} from '@/composables/functional/useChainCallback'
+import { LGraphNode, LiteGraph } from '@/lib/litegraph/src/litegraph'
 import type {
   INodeInputSlot,
   INodeOutputSlot,
   ISlotType,
   LLink,
   Vector2
-} from '@comfyorg/litegraph'
-import type { CanvasPointerEvent } from '@comfyorg/litegraph/dist/types/events'
-import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
-import {
-  type CallbackParams,
-  useChainCallback
-} from '@/composables/functional/useChainCallback'
+} from '@/lib/litegraph/src/litegraph'
+import type { CanvasPointerEvent } from '@/lib/litegraph/src/types/events'
+import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
 import type { InputSpec } from '@/schemas/nodeDefSchema'
 import { app } from '@/scripts/app'
 import { ComfyWidgets, addValueControlWidgets } from '@/scripts/widgets'
diff --git a/src/lib/litegraph/lint-staged.config.js b/src/lib/litegraph/lint-staged.config.js
index 8dc7f8a74..49c761f0c 100644
--- a/src/lib/litegraph/lint-staged.config.js
+++ b/src/lib/litegraph/lint-staged.config.js
@@ -3,10 +3,7 @@ export default {
 
   "*.js": stagedFiles => prettierAndEslint(stagedFiles),
 
-  "*.{ts,mts}": stagedFiles => [
-    ...prettierAndEslint(stagedFiles),
-    "tsc",
-  ],
+  "*.{ts,mts}": stagedFiles => [...prettierAndEslint(stagedFiles), "tsc"],
 }
 
 function prettierAndEslint(fileNames) {
diff --git a/src/main.ts b/src/main.ts
index ddace3ee1..d196d2303 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -1,4 +1,4 @@
-import '@comfyorg/litegraph/style.css'
+import '@/lib/litegraph/src/style.css'
 import { definePreset } from '@primevue/themes'
 import Aura from '@primevue/themes/aura'
 import * as Sentry from '@sentry/vue'
diff --git a/src/schemas/apiSchema.ts b/src/schemas/apiSchema.ts
index f5b62b3b8..00e51c7a2 100644
--- a/src/schemas/apiSchema.ts
+++ b/src/schemas/apiSchema.ts
@@ -1,7 +1,7 @@
-import { LinkMarkerShape } from '@comfyorg/litegraph'
 import { z } from 'zod'
 import { fromZodError } from 'zod-validation-error'
 
+import { LinkMarkerShape } from '@/lib/litegraph/src/litegraph'
 import { colorPalettesSchema } from '@/schemas/colorPaletteSchema'
 import { zComfyWorkflow, zNodeId } from '@/schemas/comfyWorkflowSchema'
 import { zKeybinding } from '@/schemas/keyBindingSchema'
diff --git a/src/schemas/colorPaletteSchema.ts b/src/schemas/colorPaletteSchema.ts
index 9001eff2a..1a29c3940 100644
--- a/src/schemas/colorPaletteSchema.ts
+++ b/src/schemas/colorPaletteSchema.ts
@@ -1,6 +1,7 @@
-import { LiteGraph } from '@comfyorg/litegraph'
 import { z } from 'zod'
 
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
+
 const nodeSlotSchema = z.object({
   CLIP: z.string(),
   CLIP_VISION: z.string(),
diff --git a/src/scripts/app.ts b/src/scripts/app.ts
index 67240b3af..2baed3618 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -1,12 +1,3 @@
-import {
-  LGraph,
-  LGraphCanvas,
-  LGraphEventMode,
-  LGraphNode,
-  LiteGraph
-} from '@comfyorg/litegraph'
-import type { Vector2 } from '@comfyorg/litegraph'
-import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import _ from 'lodash'
 import type { ToastMessageOptions } from 'primevue/toast'
 import { reactive } from 'vue'
@@ -14,6 +5,15 @@ import { reactive } from 'vue'
 import { useCanvasPositionConversion } from '@/composables/element/useCanvasPositionConversion'
 import { useWorkflowValidation } from '@/composables/useWorkflowValidation'
 import { st, t } from '@/i18n'
+import {
+  LGraph,
+  LGraphCanvas,
+  LGraphEventMode,
+  LGraphNode,
+  LiteGraph
+} from '@/lib/litegraph/src/litegraph'
+import type { Vector2 } from '@/lib/litegraph/src/litegraph'
+import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
 import type {
   ExecutionErrorWsMessage,
   NodeError,
diff --git a/src/scripts/changeTracker.ts b/src/scripts/changeTracker.ts
index ee7ccae0c..230095e5b 100644
--- a/src/scripts/changeTracker.ts
+++ b/src/scripts/changeTracker.ts
@@ -1,8 +1,8 @@
-import { LGraphCanvas, LiteGraph } from '@comfyorg/litegraph'
 import * as jsondiffpatch from 'jsondiffpatch'
 import _ from 'lodash'
 import log from 'loglevel'
 
+import { LGraphCanvas, LiteGraph } from '@/lib/litegraph/src/litegraph'
 import type { ExecutedWsMessage } from '@/schemas/apiSchema'
 import type { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import { useExecutionStore } from '@/stores/executionStore'
diff --git a/src/scripts/domWidget.ts b/src/scripts/domWidget.ts
index a24b2619d..39bbfcf0b 100644
--- a/src/scripts/domWidget.ts
+++ b/src/scripts/domWidget.ts
@@ -1,12 +1,16 @@
-import { LGraphNode, LegacyWidget, LiteGraph } from '@comfyorg/litegraph'
-import type {
-  IBaseWidget,
-  IWidgetOptions
-} from '@comfyorg/litegraph/dist/types/widgets'
 import _ from 'lodash'
 import { type Component, toRaw } from 'vue'
 
 import { useChainCallback } from '@/composables/functional/useChainCallback'
+import {
+  LGraphNode,
+  LegacyWidget,
+  LiteGraph
+} from '@/lib/litegraph/src/litegraph'
+import type {
+  IBaseWidget,
+  IWidgetOptions
+} from '@/lib/litegraph/src/types/widgets'
 import type { InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import { useDomWidgetStore } from '@/stores/domWidgetStore'
 import { generateUUID } from '@/utils/formatUtil'
diff --git a/src/scripts/errorNodeWidgets.ts b/src/scripts/errorNodeWidgets.ts
index 48a195b28..71a7ddf1b 100644
--- a/src/scripts/errorNodeWidgets.ts
+++ b/src/scripts/errorNodeWidgets.ts
@@ -1,10 +1,9 @@
-import { LGraphNode } from '@comfyorg/litegraph'
-import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
-
 import { useChainCallback } from '@/composables/functional/useChainCallback'
 import { useBooleanWidget } from '@/composables/widgets/useBooleanWidget'
 import { useFloatWidget } from '@/composables/widgets/useFloatWidget'
 import { useStringWidget } from '@/composables/widgets/useStringWidget'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
 
 const StringWidget = useStringWidget()
 const FloatWidget = useFloatWidget()
diff --git a/src/scripts/fluxKontextEditNode.ts b/src/scripts/fluxKontextEditNode.ts
index e684e6d6f..a2e367f0f 100644
--- a/src/scripts/fluxKontextEditNode.ts
+++ b/src/scripts/fluxKontextEditNode.ts
@@ -1,3 +1,5 @@
+import _ from 'lodash'
+
 import {
   type INodeOutputSlot,
   type LGraph,
@@ -5,10 +7,8 @@ import {
   LLink,
   LiteGraph,
   type Point
-} from '@comfyorg/litegraph'
-import type { IBaseWidget } from '@comfyorg/litegraph/dist/types/widgets'
-import _ from 'lodash'
-
+} from '@/lib/litegraph/src/litegraph'
+import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
 import { parseFilePath } from '@/utils/formatUtil'
 
 import { app } from './app'
diff --git a/src/scripts/pnginfo.ts b/src/scripts/pnginfo.ts
index 689537f7e..7d6eea535 100644
--- a/src/scripts/pnginfo.ts
+++ b/src/scripts/pnginfo.ts
@@ -1,4 +1,4 @@
-import { LiteGraph } from '@comfyorg/litegraph'
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
 
 import { api } from './api'
 import { getFromAvifFile } from './metadata/avif'
diff --git a/src/scripts/widgets.ts b/src/scripts/widgets.ts
index 90f5e54f7..b0b607917 100644
--- a/src/scripts/widgets.ts
+++ b/src/scripts/widgets.ts
@@ -1,10 +1,3 @@
-import { type LGraphNode, isComboWidget } from '@comfyorg/litegraph'
-import type {
-  IBaseWidget,
-  IComboWidget,
-  IStringWidget
-} from '@comfyorg/litegraph/dist/types/widgets'
-
 import { useBooleanWidget } from '@/composables/widgets/useBooleanWidget'
 import { useComboWidget } from '@/composables/widgets/useComboWidget'
 import { useFloatWidget } from '@/composables/widgets/useFloatWidget'
@@ -13,6 +6,12 @@ import { useIntWidget } from '@/composables/widgets/useIntWidget'
 import { useMarkdownWidget } from '@/composables/widgets/useMarkdownWidget'
 import { useStringWidget } from '@/composables/widgets/useStringWidget'
 import { t } from '@/i18n'
+import { type LGraphNode, isComboWidget } from '@/lib/litegraph/src/litegraph'
+import type {
+  IBaseWidget,
+  IComboWidget,
+  IStringWidget
+} from '@/lib/litegraph/src/types/widgets'
 import { transformInputSpecV1ToV2 } from '@/schemas/nodeDef/migration'
 import type { InputSpec as InputSpecV2 } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import type { InputSpec } from '@/schemas/nodeDefSchema'
diff --git a/src/services/colorPaletteService.ts b/src/services/colorPaletteService.ts
index 75249082f..549b99bde 100644
--- a/src/services/colorPaletteService.ts
+++ b/src/services/colorPaletteService.ts
@@ -1,9 +1,9 @@
-import { LGraphCanvas } from '@comfyorg/litegraph'
-import { LiteGraph } from '@comfyorg/litegraph'
 import { toRaw } from 'vue'
 import { fromZodError } from 'zod-validation-error'
 
 import { useErrorHandling } from '@/composables/useErrorHandling'
+import { LGraphCanvas } from '@/lib/litegraph/src/litegraph'
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
 import {
   Colors,
   type Palette,
diff --git a/src/services/litegraphService.ts b/src/services/litegraphService.ts
index 4a53ed448..3570d5985 100644
--- a/src/services/litegraphService.ts
+++ b/src/services/litegraphService.ts
@@ -1,3 +1,9 @@
+import _ from 'lodash'
+
+import { useNodeAnimatedImage } from '@/composables/node/useNodeAnimatedImage'
+import { useNodeCanvasImagePreview } from '@/composables/node/useNodeCanvasImagePreview'
+import { useNodeImage, useNodeVideo } from '@/composables/node/useNodeImage'
+import { st, t } from '@/i18n'
 import {
   type IContextMenuValue,
   LGraphBadge,
@@ -10,19 +16,13 @@ import {
   SubgraphNode,
   type Vector2,
   createBounds
-} from '@comfyorg/litegraph'
+} from '@/lib/litegraph/src/litegraph'
 import type {
   ExportedSubgraphInstance,
   ISerialisableNodeInput,
   ISerialisableNodeOutput,
   ISerialisedNode
-} from '@comfyorg/litegraph/dist/types/serialisation'
-import _ from 'lodash'
-
-import { useNodeAnimatedImage } from '@/composables/node/useNodeAnimatedImage'
-import { useNodeCanvasImagePreview } from '@/composables/node/useNodeCanvasImagePreview'
-import { useNodeImage, useNodeVideo } from '@/composables/node/useNodeImage'
-import { st, t } from '@/i18n'
+} from '@/lib/litegraph/src/types/serialisation'
 import type { NodeId } from '@/schemas/comfyWorkflowSchema'
 import { transformInputSpecV2ToV1 } from '@/schemas/nodeDef/migration'
 import type {
diff --git a/src/services/load3dService.ts b/src/services/load3dService.ts
index 515579ef9..f32685fb5 100644
--- a/src/services/load3dService.ts
+++ b/src/services/load3dService.ts
@@ -1,8 +1,8 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { toRaw } from 'vue'
 
 import Load3d from '@/extensions/core/load3d/Load3d'
 import Load3dAnimation from '@/extensions/core/load3d/Load3dAnimation'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { CustomInputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 
 type Load3dReadyCallback = (load3d: Load3d | Load3dAnimation) => void
diff --git a/src/services/subgraphService.ts b/src/services/subgraphService.ts
index ac1cb3d0a..865016e4d 100644
--- a/src/services/subgraphService.ts
+++ b/src/services/subgraphService.ts
@@ -2,8 +2,7 @@ import {
   type ExportedSubgraph,
   type ExportedSubgraphInstance,
   type Subgraph
-} from '@comfyorg/litegraph'
-
+} from '@/lib/litegraph/src/litegraph'
 import type { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import type { ComfyNodeDef as ComfyNodeDefV1 } from '@/schemas/nodeDefSchema'
 import { app as comfyApp } from '@/scripts/app'
diff --git a/src/services/workflowService.ts b/src/services/workflowService.ts
index 4fcbd44b5..78c50d892 100644
--- a/src/services/workflowService.ts
+++ b/src/services/workflowService.ts
@@ -1,8 +1,8 @@
-import { LGraph, LGraphCanvas } from '@comfyorg/litegraph'
-import type { SerialisableGraph, Vector2 } from '@comfyorg/litegraph'
 import { toRaw } from 'vue'
 
 import { t } from '@/i18n'
+import { LGraph, LGraphCanvas } from '@/lib/litegraph/src/litegraph'
+import type { SerialisableGraph, Vector2 } from '@/lib/litegraph/src/litegraph'
 import { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import { app } from '@/scripts/app'
 import { blankGraph, defaultGraph } from '@/scripts/defaultGraph'
diff --git a/src/stores/executionStore.ts b/src/stores/executionStore.ts
index bc33b75cd..6bc5db077 100644
--- a/src/stores/executionStore.ts
+++ b/src/stores/executionStore.ts
@@ -1,10 +1,10 @@
-import type { LGraph, Subgraph } from '@comfyorg/litegraph'
 import { defineStore } from 'pinia'
 import { computed, ref } from 'vue'
 
 import type ChatHistoryWidget from '@/components/graph/widgets/ChatHistoryWidget.vue'
 import { useNodeChatHistory } from '@/composables/node/useNodeChatHistory'
 import { useNodeProgressText } from '@/composables/node/useNodeProgressText'
+import type { LGraph, Subgraph } from '@/lib/litegraph/src/litegraph'
 import type {
   DisplayComponentWsMessage,
   ExecutedWsMessage,
diff --git a/src/stores/graphStore.ts b/src/stores/graphStore.ts
index a1e0d0043..b76c25139 100644
--- a/src/stores/graphStore.ts
+++ b/src/stores/graphStore.ts
@@ -1,8 +1,12 @@
-import type { LGraphCanvas, LGraphGroup, LGraphNode } from '@comfyorg/litegraph'
-import type { Positionable } from '@comfyorg/litegraph/dist/interfaces'
 import { defineStore } from 'pinia'
 import { type Raw, computed, markRaw, ref, shallowRef } from 'vue'
 
+import type { Positionable } from '@/lib/litegraph/src/interfaces'
+import type {
+  LGraphCanvas,
+  LGraphGroup,
+  LGraphNode
+} from '@/lib/litegraph/src/litegraph'
 import { isLGraphGroup, isLGraphNode, isReroute } from '@/utils/litegraphUtil'
 
 export const useTitleEditorStore = defineStore('titleEditor', () => {
diff --git a/src/stores/imagePreviewStore.ts b/src/stores/imagePreviewStore.ts
index b2c19b8f9..2af468d6f 100644
--- a/src/stores/imagePreviewStore.ts
+++ b/src/stores/imagePreviewStore.ts
@@ -1,6 +1,6 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import { defineStore } from 'pinia'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import {
   ExecutedWsMessage,
   ResultItem,
diff --git a/src/stores/nodeDefStore.ts b/src/stores/nodeDefStore.ts
index 8a5a46ade..9d32d0c4e 100644
--- a/src/stores/nodeDefStore.ts
+++ b/src/stores/nodeDefStore.ts
@@ -1,9 +1,9 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import axios from 'axios'
 import _ from 'lodash'
 import { defineStore } from 'pinia'
 import { computed, ref } from 'vue'
 
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { transformNodeDefV1ToV2 } from '@/schemas/nodeDef/migration'
 import type {
   ComfyNodeDef as ComfyNodeDefV2,
diff --git a/src/stores/subgraphNavigationStore.ts b/src/stores/subgraphNavigationStore.ts
index 01b2265cb..0ff72f205 100644
--- a/src/stores/subgraphNavigationStore.ts
+++ b/src/stores/subgraphNavigationStore.ts
@@ -1,9 +1,9 @@
 import QuickLRU from '@alloc/quick-lru'
-import type { Subgraph } from '@comfyorg/litegraph'
-import type { DragAndScaleState } from '@comfyorg/litegraph/dist/DragAndScale'
 import { defineStore } from 'pinia'
 import { computed, ref, shallowRef, watch } from 'vue'
 
+import type { DragAndScaleState } from '@/lib/litegraph/src/DragAndScale'
+import type { Subgraph } from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { findSubgraphPathById } from '@/utils/graphTraversalUtil'
 import { isNonNullish } from '@/utils/typeGuardUtil'
diff --git a/src/stores/workflowStore.ts b/src/stores/workflowStore.ts
index f11068852..7de048646 100644
--- a/src/stores/workflowStore.ts
+++ b/src/stores/workflowStore.ts
@@ -1,8 +1,8 @@
-import type { LGraph, Subgraph } from '@comfyorg/litegraph'
 import _ from 'lodash'
 import { defineStore } from 'pinia'
 import { type Raw, computed, markRaw, ref, shallowRef, watch } from 'vue'
 
+import type { LGraph, Subgraph } from '@/lib/litegraph/src/litegraph'
 import { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import type { NodeId } from '@/schemas/comfyWorkflowSchema'
 import { api } from '@/scripts/api'
diff --git a/src/types/comfy.ts b/src/types/comfy.ts
index f78fccaeb..622753357 100644
--- a/src/types/comfy.ts
+++ b/src/types/comfy.ts
@@ -1,6 +1,5 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import { Positionable } from '@comfyorg/litegraph/dist/interfaces'
-
+import { Positionable } from '@/lib/litegraph/src/interfaces'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import type { ComfyWorkflowJSON } from '@/schemas/comfyWorkflowSchema'
 import type { Keybinding } from '@/schemas/keyBindingSchema'
 import type { ComfyNodeDef } from '@/schemas/nodeDefSchema'
diff --git a/src/types/index.ts b/src/types/index.ts
index 377581465..b8d6a81ae 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -1,5 +1,4 @@
-import type { LGraph } from '@comfyorg/litegraph'
-
+import type { LGraph } from '@/lib/litegraph/src/litegraph'
 import type {
   DeviceStats,
   EmbeddingsResponse,
diff --git a/src/types/litegraph-augmentation.d.ts b/src/types/litegraph-augmentation.d.ts
index 68e2ac8f4..2e04dc501 100644
--- a/src/types/litegraph-augmentation.d.ts
+++ b/src/types/litegraph-augmentation.d.ts
@@ -1,6 +1,5 @@
-import '@comfyorg/litegraph'
-import type { LLink, Size } from '@comfyorg/litegraph'
-
+import '@/lib/litegraph/src/litegraph'
+import type { LLink, Size } from '@/lib/litegraph/src/litegraph'
 import type { ComfyNodeDef as ComfyNodeDefV2 } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import type { ComfyNodeDef as ComfyNodeDefV1 } from '@/schemas/nodeDefSchema'
 import type { DOMWidget, DOMWidgetOptions } from '@/scripts/domWidget'
@@ -8,7 +7,7 @@ import type { DOMWidget, DOMWidgetOptions } from '@/scripts/domWidget'
 import type { NodeId } from '../schemas/comfyWorkflowSchema'
 
 /** ComfyUI extensions of litegraph */
-declare module '@comfyorg/litegraph/dist/types/widgets' {
+declare module '@/lib/litegraph/src/types/widgets' {
   interface IWidgetOptions {
     /** Currently used by DOM widgets only.  Declaring here reduces complexity. */
     onHide?: (widget: DOMWidget) => void
@@ -59,7 +58,7 @@ declare module '@comfyorg/litegraph/dist/types/widgets' {
 /**
  * ComfyUI extensions of litegraph interfaces
  */
-declare module '@comfyorg/litegraph/dist/interfaces' {
+declare module '@/lib/litegraph/src/interfaces' {
   interface IWidgetLocator {
     [key: symbol]: unknown
   }
@@ -68,9 +67,12 @@ 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'
+declare module '@/lib/litegraph/src/litegraph' {
+  import type {
+    ExecutableLGraphNode,
+    ExecutionId
+  } from '@/lib/litegraph/src/litegraph'
+  import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
 
   interface LGraphNodeConstructor<T extends LGraphNode = LGraphNode> {
     type?: string
@@ -197,7 +199,7 @@ declare module '@comfyorg/litegraph' {
 /**
  * Extended types for litegraph, to be merged upstream once it has stabilized.
  */
-declare module '@comfyorg/litegraph' {
+declare module '@/lib/litegraph/src/litegraph' {
   /**
    * widgets_values is set to LGraphNode by `LGraphNode.configure`, but it is not
    * used by litegraph internally. We should remove the dependency on it later.
diff --git a/src/utils/errorReportUtil.ts b/src/utils/errorReportUtil.ts
index 643ec6d0f..65f5aa830 100644
--- a/src/utils/errorReportUtil.ts
+++ b/src/utils/errorReportUtil.ts
@@ -1,5 +1,4 @@
-import type { ISerialisedGraph } from '@comfyorg/litegraph'
-
+import type { ISerialisedGraph } from '@/lib/litegraph/src/litegraph'
 import type { SystemStats } from '@/schemas/apiSchema'
 import type { NodeId } from '@/schemas/comfyWorkflowSchema'
 
diff --git a/src/utils/executableGroupNodeChildDTO.ts b/src/utils/executableGroupNodeChildDTO.ts
index 1250874ca..8d2d573e6 100644
--- a/src/utils/executableGroupNodeChildDTO.ts
+++ b/src/utils/executableGroupNodeChildDTO.ts
@@ -1,3 +1,4 @@
+import type { GroupNodeHandler } from '@/extensions/core/groupNode'
 import {
   type ExecutableLGraphNode,
   ExecutableNodeDTO,
@@ -5,9 +6,7 @@ import {
   type LGraphNode,
   type NodeId,
   type SubgraphNode
-} from '@comfyorg/litegraph'
-
-import type { GroupNodeHandler } from '@/extensions/core/groupNode'
+} from '@/lib/litegraph/src/litegraph'
 
 export class ExecutableGroupNodeChildDTO extends ExecutableNodeDTO {
   groupNodeHandler?: GroupNodeHandler
diff --git a/src/utils/executableGroupNodeDto.ts b/src/utils/executableGroupNodeDto.ts
index 09044dcce..89570ae9e 100644
--- a/src/utils/executableGroupNodeDto.ts
+++ b/src/utils/executableGroupNodeDto.ts
@@ -4,7 +4,7 @@ import {
   type ISlotType,
   LGraphEventMode,
   type LGraphNode
-} from '@comfyorg/litegraph'
+} from '@/lib/litegraph/src/litegraph'
 
 export const GROUP = Symbol()
 
diff --git a/src/utils/executionUtil.ts b/src/utils/executionUtil.ts
index 727a631ca..f00ca0920 100644
--- a/src/utils/executionUtil.ts
+++ b/src/utils/executionUtil.ts
@@ -2,13 +2,12 @@ import type {
   ExecutableLGraphNode,
   ExecutionId,
   LGraph
-} from '@comfyorg/litegraph'
+} from '@/lib/litegraph/src/litegraph'
 import {
   ExecutableNodeDTO,
   LGraphEventMode,
   SubgraphNode
-} from '@comfyorg/litegraph'
-
+} from '@/lib/litegraph/src/litegraph'
 import type {
   ComfyApiWorkflow,
   ComfyWorkflowJSON
diff --git a/src/utils/graphTraversalUtil.ts b/src/utils/graphTraversalUtil.ts
index db8c57607..fa9e436ec 100644
--- a/src/utils/graphTraversalUtil.ts
+++ b/src/utils/graphTraversalUtil.ts
@@ -1,5 +1,8 @@
-import type { LGraph, LGraphNode, Subgraph } from '@comfyorg/litegraph'
-
+import type {
+  LGraph,
+  LGraphNode,
+  Subgraph
+} from '@/lib/litegraph/src/litegraph'
 import type { NodeExecutionId, NodeLocatorId } from '@/types/nodeIdentification'
 import { parseNodeLocatorId } from '@/types/nodeIdentification'
 
diff --git a/src/utils/linkFixer.ts b/src/utils/linkFixer.ts
index 218b1c962..125e450ee 100644
--- a/src/utils/linkFixer.ts
+++ b/src/utils/linkFixer.ts
@@ -24,13 +24,13 @@
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  * SOFTWARE.
  */
-import type { LGraph, LGraphNode, LLink } from '@comfyorg/litegraph'
-import type { NodeId } from '@comfyorg/litegraph/dist/LGraphNode'
-import type { SerialisedLLinkArray } from '@comfyorg/litegraph/dist/LLink'
+import type { NodeId } from '@/lib/litegraph/src/LGraphNode'
+import type { SerialisedLLinkArray } from '@/lib/litegraph/src/LLink'
+import type { LGraph, LGraphNode, LLink } from '@/lib/litegraph/src/litegraph'
 import type {
   ISerialisedGraph,
   ISerialisedNode
-} from '@comfyorg/litegraph/dist/types/serialisation'
+} from '@/lib/litegraph/src/types/serialisation'
 
 export interface BadLinksData<T = ISerialisedGraph | LGraph> {
   hasBadLinks: boolean
diff --git a/src/utils/litegraphUtil.ts b/src/utils/litegraphUtil.ts
index 5f147b99b..bb905fffd 100644
--- a/src/utils/litegraphUtil.ts
+++ b/src/utils/litegraphUtil.ts
@@ -1,16 +1,20 @@
-import { ColorOption, LGraph, Reroute } from '@comfyorg/litegraph'
-import { LGraphGroup, LGraphNode, isColorable } from '@comfyorg/litegraph'
+import _ from 'lodash'
+
+import { ColorOption, LGraph, Reroute } from '@/lib/litegraph/src/litegraph'
+import {
+  LGraphGroup,
+  LGraphNode,
+  isColorable
+} from '@/lib/litegraph/src/litegraph'
 import type {
   ExportedSubgraph,
   ISerialisableNodeInput,
   ISerialisedGraph
-} from '@comfyorg/litegraph/dist/types/serialisation'
+} from '@/lib/litegraph/src/types/serialisation'
 import type {
   IBaseWidget,
   IComboWidget
-} from '@comfyorg/litegraph/dist/types/widgets'
-import _ from 'lodash'
-
+} from '@/lib/litegraph/src/types/widgets'
 import type { InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 
 type ImageNode = LGraphNode & { imgs: HTMLImageElement[] | undefined }
diff --git a/src/utils/nodeFilterUtil.ts b/src/utils/nodeFilterUtil.ts
index 2456edd50..b2fab641b 100644
--- a/src/utils/nodeFilterUtil.ts
+++ b/src/utils/nodeFilterUtil.ts
@@ -1,4 +1,4 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 /**
  * Checks if a node is an output node.
diff --git a/src/utils/searchAndReplace.ts b/src/utils/searchAndReplace.ts
index 395d96406..1020d6fbe 100644
--- a/src/utils/searchAndReplace.ts
+++ b/src/utils/searchAndReplace.ts
@@ -1,5 +1,4 @@
-import type { LGraph, Subgraph } from '@comfyorg/litegraph'
-
+import type { LGraph, Subgraph } from '@/lib/litegraph/src/litegraph'
 import { formatDate } from '@/utils/formatUtil'
 import { collectAllNodes } from '@/utils/graphTraversalUtil'
 
diff --git a/src/utils/typeGuardUtil.ts b/src/utils/typeGuardUtil.ts
index a3bcb5f3d..af88ec55e 100644
--- a/src/utils/typeGuardUtil.ts
+++ b/src/utils/typeGuardUtil.ts
@@ -1,7 +1,6 @@
-import { LGraph, LGraphNode } from '@comfyorg/litegraph'
-import { Subgraph } from '@comfyorg/litegraph'
-
 import type { PrimitiveNode } from '@/extensions/core/widgetInputs'
+import { LGraph, LGraphNode } from '@/lib/litegraph/src/litegraph'
+import { Subgraph } from '@/lib/litegraph/src/litegraph'
 
 export function isPrimitiveNode(
   node: LGraphNode
diff --git a/src/utils/vintageClipboard.ts b/src/utils/vintageClipboard.ts
index b2d13e2ad..c83354364 100644
--- a/src/utils/vintageClipboard.ts
+++ b/src/utils/vintageClipboard.ts
@@ -1,5 +1,9 @@
-import type { LGraph, LGraphCanvas, LGraphNode } from '@comfyorg/litegraph'
-import { LiteGraph } from '@comfyorg/litegraph'
+import type {
+  LGraph,
+  LGraphCanvas,
+  LGraphNode
+} from '@/lib/litegraph/src/litegraph'
+import { LiteGraph } from '@/lib/litegraph/src/litegraph'
 
 /**
  * Serialises an array of nodes using a modified version of the old Litegraph copy (& paste) function
diff --git a/tests-ui/tests/components/dialog/content/MissingCoreNodesMessage.spec.ts b/tests-ui/tests/components/dialog/content/MissingCoreNodesMessage.spec.ts
index 5e9fdec3f..eeecae0a7 100644
--- a/tests-ui/tests/components/dialog/content/MissingCoreNodesMessage.spec.ts
+++ b/tests-ui/tests/components/dialog/content/MissingCoreNodesMessage.spec.ts
@@ -1,4 +1,3 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { mount } from '@vue/test-utils'
 import PrimeVue from 'primevue/config'
 import Message from 'primevue/message'
@@ -6,6 +5,7 @@ import { beforeEach, describe, expect, it, vi } from 'vitest'
 import { nextTick } from 'vue'
 
 import MissingCoreNodesMessage from '@/components/dialog/content/MissingCoreNodesMessage.vue'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { useSystemStatsStore } from '@/stores/systemStatsStore'
 
 // Mock the stores
diff --git a/tests-ui/tests/composables/canvas/useSelectedLiteGraphItems.test.ts b/tests-ui/tests/composables/canvas/useSelectedLiteGraphItems.test.ts
index f42c9c8ed..1774af294 100644
--- a/tests-ui/tests/composables/canvas/useSelectedLiteGraphItems.test.ts
+++ b/tests-ui/tests/composables/canvas/useSelectedLiteGraphItems.test.ts
@@ -1,13 +1,13 @@
+import { createPinia, setActivePinia } from 'pinia'
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+import { useSelectedLiteGraphItems } from '@/composables/canvas/useSelectedLiteGraphItems'
 import {
   LGraphEventMode,
   LGraphNode,
   Positionable,
   Reroute
-} from '@comfyorg/litegraph'
-import { createPinia, setActivePinia } from 'pinia'
-import { beforeEach, describe, expect, it, vi } from 'vitest'
-
-import { useSelectedLiteGraphItems } from '@/composables/canvas/useSelectedLiteGraphItems'
+} from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { useCanvasStore } from '@/stores/graphStore'
 
@@ -21,7 +21,7 @@ vi.mock('@/scripts/app', () => ({
 }))
 
 // Mock the litegraph module
-vi.mock('@comfyorg/litegraph', () => ({
+vi.mock('@/lib/litegraph/src/litegraph', () => ({
   Reroute: class Reroute {
     constructor() {}
   },
diff --git a/tests-ui/tests/composables/node/useNodePricing.test.ts b/tests-ui/tests/composables/node/useNodePricing.test.ts
index 67f78d7d3..6ca4ef918 100644
--- a/tests-ui/tests/composables/node/useNodePricing.test.ts
+++ b/tests-ui/tests/composables/node/useNodePricing.test.ts
@@ -1,8 +1,8 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
-import type { IComboWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import { describe, expect, it } from 'vitest'
 
 import { useNodePricing } from '@/composables/node/useNodePricing'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
+import type { IComboWidget } from '@/lib/litegraph/src/types/widgets'
 
 // Helper function to create a mock node
 function createMockNode(
diff --git a/tests-ui/tests/composables/useMissingNodes.test.ts b/tests-ui/tests/composables/useMissingNodes.test.ts
index 9dc0b08e5..e83bc7ff0 100644
--- a/tests-ui/tests/composables/useMissingNodes.test.ts
+++ b/tests-ui/tests/composables/useMissingNodes.test.ts
@@ -1,9 +1,9 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 import { nextTick, ref } from 'vue'
 
 import { useMissingNodes } from '@/composables/nodePack/useMissingNodes'
 import { useWorkflowPacks } from '@/composables/nodePack/useWorkflowPacks'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { app } from '@/scripts/app'
 import { useComfyManagerStore } from '@/stores/comfyManagerStore'
 import { useNodeDefStore } from '@/stores/nodeDefStore'
diff --git a/tests-ui/tests/composables/useNodeChatHistory.test.ts b/tests-ui/tests/composables/useNodeChatHistory.test.ts
index f1f66f1b5..6a6ecc6c5 100644
--- a/tests-ui/tests/composables/useNodeChatHistory.test.ts
+++ b/tests-ui/tests/composables/useNodeChatHistory.test.ts
@@ -1,7 +1,7 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
 import { useNodeChatHistory } from '@/composables/node/useNodeChatHistory'
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 vi.mock('@/composables/widgets/useChatHistoryWidget', () => ({
   useChatHistoryWidget: () => {
diff --git a/tests-ui/tests/composables/useWatchWidget.test.ts b/tests-ui/tests/composables/useWatchWidget.test.ts
index b31aaf625..61363a3f1 100644
--- a/tests-ui/tests/composables/useWatchWidget.test.ts
+++ b/tests-ui/tests/composables/useWatchWidget.test.ts
@@ -1,8 +1,8 @@
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { describe, expect, it, vi } from 'vitest'
 import { nextTick } from 'vue'
 
 import { useComputedWithWidgetWatch } from '@/composables/node/useWatchWidget'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 
 // Mock useChainCallback
 vi.mock('@/composables/functional/useChainCallback', () => ({
diff --git a/tests-ui/tests/litegraph.test.ts b/tests-ui/tests/litegraph.test.ts
index 88531e4a1..068da2fb5 100644
--- a/tests-ui/tests/litegraph.test.ts
+++ b/tests-ui/tests/litegraph.test.ts
@@ -1,6 +1,7 @@
-import { LGraph, LGraphNode, LiteGraph } from '@comfyorg/litegraph'
 import { describe, expect, it } from 'vitest'
 
+import { LGraph, LGraphNode, LiteGraph } from '@/lib/litegraph/src/litegraph'
+
 function swapNodes(nodes: LGraphNode[]) {
   const firstNode = nodes[0]
   const lastNode = nodes[nodes.length - 1]
diff --git a/tests-ui/tests/store/imagePreviewStore.test.ts b/tests-ui/tests/store/imagePreviewStore.test.ts
index 0a464ba81..d0df957dc 100644
--- a/tests-ui/tests/store/imagePreviewStore.test.ts
+++ b/tests-ui/tests/store/imagePreviewStore.test.ts
@@ -1,7 +1,7 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import { createPinia, setActivePinia } from 'pinia'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { ExecutedWsMessage } from '@/schemas/apiSchema'
 import { app } from '@/scripts/app'
 import { useNodeOutputStore } from '@/stores/imagePreviewStore'
diff --git a/tests-ui/tests/store/workflowStore.test.ts b/tests-ui/tests/store/workflowStore.test.ts
index a82e1adfc..d2cdc2f1f 100644
--- a/tests-ui/tests/store/workflowStore.test.ts
+++ b/tests-ui/tests/store/workflowStore.test.ts
@@ -1,8 +1,8 @@
-import type { Subgraph } from '@comfyorg/litegraph'
 import { createPinia, setActivePinia } from 'pinia'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 import { nextTick } from 'vue'
 
+import type { Subgraph } from '@/lib/litegraph/src/litegraph'
 import { api } from '@/scripts/api'
 import { app as comfyApp } from '@/scripts/app'
 import { defaultGraph, defaultGraphJSON } from '@/scripts/defaultGraph'
diff --git a/tests-ui/tests/utils/graphTraversalUtil.test.ts b/tests-ui/tests/utils/graphTraversalUtil.test.ts
index 185f7454c..f6cb74804 100644
--- a/tests-ui/tests/utils/graphTraversalUtil.test.ts
+++ b/tests-ui/tests/utils/graphTraversalUtil.test.ts
@@ -1,6 +1,10 @@
-import type { LGraph, LGraphNode, Subgraph } from '@comfyorg/litegraph'
 import { describe, expect, it, vi } from 'vitest'
 
+import type {
+  LGraph,
+  LGraphNode,
+  Subgraph
+} from '@/lib/litegraph/src/litegraph'
 import {
   collectAllNodes,
   collectFromNodes,
diff --git a/tests-ui/tests/utils/litegraphUtil.test.ts b/tests-ui/tests/utils/litegraphUtil.test.ts
index cc482588b..b4301941f 100644
--- a/tests-ui/tests/utils/litegraphUtil.test.ts
+++ b/tests-ui/tests/utils/litegraphUtil.test.ts
@@ -1,7 +1,7 @@
-import { ISerialisedGraph } from '@comfyorg/litegraph/dist/types/serialisation'
-import type { IWidget } from '@comfyorg/litegraph/dist/types/widgets'
 import { describe, expect, it } from 'vitest'
 
+import { ISerialisedGraph } from '@/lib/litegraph/src/types/serialisation'
+import type { IWidget } from '@/lib/litegraph/src/types/widgets'
 import type { InputSpec } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import {
   compressWidgetInputSlots,
diff --git a/tests-ui/tests/utils/nodeFilterUtil.test.ts b/tests-ui/tests/utils/nodeFilterUtil.test.ts
index 94f62b6f2..871f66109 100644
--- a/tests-ui/tests/utils/nodeFilterUtil.test.ts
+++ b/tests-ui/tests/utils/nodeFilterUtil.test.ts
@@ -1,6 +1,6 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import { describe, expect, it } from 'vitest'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { filterOutputNodes, isOutputNode } from '@/utils/nodeFilterUtil'
 
 describe('nodeFilterUtil', () => {
diff --git a/tests-ui/tests/utils/serachAndReplace.test.ts b/tests-ui/tests/utils/serachAndReplace.test.ts
index 2a8515369..2846ab13e 100644
--- a/tests-ui/tests/utils/serachAndReplace.test.ts
+++ b/tests-ui/tests/utils/serachAndReplace.test.ts
@@ -1,7 +1,7 @@
-import { LGraph } from '@comfyorg/litegraph'
-import type { LGraphNode } from '@comfyorg/litegraph'
 import { describe, expect, it } from 'vitest'
 
+import { LGraph } from '@/lib/litegraph/src/litegraph'
+import type { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { applyTextReplacements } from '@/utils/searchAndReplace'
 
 describe('applyTextReplacements', () => {
diff --git a/tests-ui/tests/widgets/domWidget.test.ts b/tests-ui/tests/widgets/domWidget.test.ts
index 8c5695f33..59d1d95c5 100644
--- a/tests-ui/tests/widgets/domWidget.test.ts
+++ b/tests-ui/tests/widgets/domWidget.test.ts
@@ -1,6 +1,6 @@
-import { LGraphNode } from '@comfyorg/litegraph'
 import { describe, expect, test, vi } from 'vitest'
 
+import { LGraphNode } from '@/lib/litegraph/src/litegraph'
 import { ComponentWidgetImpl, DOMWidgetImpl } from '@/scripts/domWidget'
 
 // Mock dependencies