github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): be double extra sure migrated workflows are parsed before loading

Browse Source
This commit is contained in:
psychedelicious
2025-02-07 14:03:28 +11:00
parent 2abe2f52f7
commit 4ca45f7651
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
invokeai/frontend/web/src/features/nodes/util/workflow/migrations.ts
View File

@@ -12,7 +12,7 @@ import { zWorkflowV2 } from 'features/nodes/types/v2/workflow';
import type { WorkflowV3 } from 'features/nodes/types/workflow';
import { zWorkflowV3 } from 'features/nodes/types/workflow';
import { t } from 'i18next';
import { forEach } from 'lodash-es';
import { forEach, get } from 'lodash-es';
import { z } from 'zod';
/**
@@ -86,17 +86,20 @@ export const parseAndMigrateWorkflow = (data: unknown): WorkflowV3 => {
throw new WorkflowVersionError(t('nodes.unableToGetWorkflowVersion'));
}
let workflow = deepClone(data) as WorkflowV1 | WorkflowV2 | WorkflowV3;
let workflow = deepClone(data);
if (workflow.meta.version === '1.0.0') {
if (get(workflow, 'meta.version') === '1.0.0') {
const v1 = zWorkflowV1.parse(workflow);
workflow = migrateV1toV2(v1);
}
if (workflow.meta.version === '2.0.0') {
if (get(workflow, 'meta.version') === '2.0.0') {
const v2 = zWorkflowV2.parse(workflow);
workflow = migrateV2toV3(v2);
}
return workflow as WorkflowV3;
// We should now have a V3 workflow
const migratedWorkflow = zWorkflowV3.parse(workflow);
return migratedWorkflow;
};