From 81a97f379636827ee4fbc26dc1facea600ba18d4 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 12 Mar 2025 19:59:31 +1000
Subject: [PATCH] fix(ui): load workflow from object

---
 .../components/LoadWorkflowConfirmationAlertDialog.tsx    | 6 +++---
 .../LoadWorkflowFromGraphModal.tsx                        | 8 ++++----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowConfirmationAlertDialog.tsx b/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowConfirmationAlertDialog.tsx
index 5376b208db..f4048be9e6 100644
--- a/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowConfirmationAlertDialog.tsx
+++ b/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowConfirmationAlertDialog.tsx
@@ -25,8 +25,8 @@ type LoadLibraryWorkflowData = Callbacks & {
 };
 
 type LoadWorkflowFromObjectData = Callbacks & {
-  type: 'direct';
-  data: WorkflowV3;
+  type: 'object';
+  data: unknown;
 };
 
 type LoadWorkflowFromFileData = Callbacks & {
@@ -70,7 +70,7 @@ const useLoadImmediate = () => {
       onError,
       onCompleted,
     };
-    if (type === 'direct') {
+    if (type === 'object') {
       await loadWorkflowFromObject(data, options);
     } else if (type === 'file') {
       await loadWorkflowFromFile(data, options);
diff --git a/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowFromGraphModal/LoadWorkflowFromGraphModal.tsx b/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowFromGraphModal/LoadWorkflowFromGraphModal.tsx
index e4b9d2a2cd..131409e7cf 100644
--- a/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowFromGraphModal/LoadWorkflowFromGraphModal.tsx
+++ b/invokeai/frontend/web/src/features/workflowLibrary/components/LoadWorkflowFromGraphModal/LoadWorkflowFromGraphModal.tsx
@@ -15,7 +15,7 @@ import {
 } from '@invoke-ai/ui-library';
 import { useStore } from '@nanostores/react';
 import { graphToWorkflow } from 'features/nodes/util/workflow/graphToWorkflow';
-import { useLoadWorkflowFromObject } from 'features/workflowLibrary/hooks/useLoadWorkflowFromObject';
+import { useLoadWorkflowWithDialog } from 'features/workflowLibrary/components/LoadWorkflowConfirmationAlertDialog';
 import { atom } from 'nanostores';
 import type { ChangeEvent } from 'react';
 import { useCallback, useState } from 'react';
@@ -38,7 +38,7 @@ export const useLoadWorkflowFromGraphModal = () => {
 export const LoadWorkflowFromGraphModal = () => {
   const { t } = useTranslation();
   const { isOpen, onClose } = useLoadWorkflowFromGraphModal();
-  const loadWorkflowFromObject = useLoadWorkflowFromObject();
+  const loadWorkflowWithDialog = useLoadWorkflowWithDialog();
   const [graphRaw, setGraphRaw] = useState<string>('');
   const [unvalidatedWorkflow, setUnvalidatedWorkflow] = useState<unknown>();
   const [unvalidatedWorkflowAsString, setUnvalidatedWorkflowAsString] = useState<string>('');
@@ -59,9 +59,9 @@ export const LoadWorkflowFromGraphModal = () => {
     setUnvalidatedWorkflowAsString(JSON.stringify(workflow, null, 2));
   }, [graphRaw, shouldAutoLayout]);
   const loadWorkflow = useCallback(async () => {
-    await loadWorkflowFromObject(unvalidatedWorkflow);
+    await loadWorkflowWithDialog({ type: 'object', data: unvalidatedWorkflow });
     onClose();
-  }, [loadWorkflowFromObject, unvalidatedWorkflow, onClose]);
+  }, [loadWorkflowWithDialog, unvalidatedWorkflow, onClose]);
   return (
     <Modal isOpen={isOpen} onClose={onClose} isCentered useInert={false}>
       <ModalOverlay />