correctly set is_published when loading a workflow

This commit is contained in:
Mary Hipp
2025-04-07 15:50:06 -04:00
committed by psychedelicious
parent ded080ae04
commit 1558fe9a37

View File

@@ -1,4 +1,6 @@
import { useToast } from '@invoke-ai/ui-library';
import { useAppDispatch } from 'app/store/storeHooks';
import { workflowIsPublishedChanged } from 'features/nodes/store/workflowSlice';
import type { WorkflowV3 } from 'features/nodes/types/workflow';
import { useValidateAndLoadWorkflow } from 'features/workflowLibrary/hooks/useValidateAndLoadWorkflow';
import { useCallback } from 'react';
@@ -17,6 +19,7 @@ export const useLoadWorkflowFromLibrary = () => {
const validateAndLoadWorkflow = useValidateAndLoadWorkflow();
const [getWorkflow] = useLazyGetWorkflowQuery();
const [updateOpenedAt] = useUpdateOpenedAtMutation();
const dispatch = useAppDispatch();
const loadWorkflowFromLibrary = useCallback(
async (
workflowId: string,
@@ -36,6 +39,7 @@ export const useLoadWorkflowFromLibrary = () => {
onError?.();
return;
}
dispatch(workflowIsPublishedChanged(res.is_published ?? false));
updateOpenedAt({ workflow_id: workflowId });
onSuccess?.(validatedWorkflow);
} catch {
@@ -50,7 +54,7 @@ export const useLoadWorkflowFromLibrary = () => {
onCompleted?.();
}
},
[getWorkflow, validateAndLoadWorkflow, updateOpenedAt, toast, t]
[getWorkflow, validateAndLoadWorkflow, updateOpenedAt, toast, t, dispatch]
);
return loadWorkflowFromLibrary;