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): auto-update nodes on loading workflow

Browse Source
This commit is contained in:
psychedelicious
2025-02-07 14:50:09 +11:00
parent ef0ef875dd
commit 211bb8a204
2 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/frontend/web/public/locales/en.json
View File

@@ -934,6 +934,7 @@
"noWorkflows": "No Workflows",
"noMatchingWorkflows": "No Matching Workflows",
"noWorkflow": "No Workflow",
"unableToUpdateNode": "Node update failed: node {{node}} of type {{type}} (may require deleting and recreating)",
"mismatchedVersion": "Invalid node: node {{node}} of type {{type}} has mismatched version (try updating?)",
"missingTemplate": "Invalid node: node {{node}} of type {{type}} missing template (not installed?)",
"sourceNodeDoesNotExist": "Invalid edge: source/output node {{node}} does not exist",

44
invokeai/frontend/web/src/features/nodes/util/workflow/validateWorkflow.ts
View File

@@ -7,9 +7,8 @@ import {
} from 'features/nodes/types/field';
import type { WorkflowV3 } from 'features/nodes/types/workflow';
import { buildNodeFieldElement, isWorkflowInvocationNode } from 'features/nodes/types/workflow';
import { getNeedsUpdate } from 'features/nodes/util/node/nodeUpdate';
import { getNeedsUpdate, updateNode } from 'features/nodes/util/node/nodeUpdate';
import { t } from 'i18next';
import { keyBy } from 'lodash-es';
import type { JsonObject } from 'type-fest';
import { parseAndMigrateWorkflow } from './migrations';
@@ -72,11 +71,11 @@ export const validateWorkflow = async (
const { nodes, edges } = _workflow;
const warnings: WorkflowWarning[] = [];
// We don't need to validate Note nodes or CurrentImage nodes - only Invocation nodes
const invocationNodes = nodes.filter(isWorkflowInvocationNode);
const keyedNodes = keyBy(invocationNodes, 'id');
for (const node of Object.values(invocationNodes)) {
for (const [i, node] of nodes.entries()) {
if (!isWorkflowInvocationNode(node)) {
// We don't need to validate Note nodes or CurrentImage nodes - only Invocation nodes
continue;
}
const template = templates[node.data.type];
if (!template) {
// This node's type template does not exist
@@ -91,17 +90,22 @@ export const validateWorkflow = async (
continue;
}
// This node needs to be updated, based on comparison of its version to the template version
if (getNeedsUpdate(node.data, template)) {
// This node needs to be updated, based on comparison of its version to the template version
const message = t('nodes.mismatchedVersion', {
node: node.id,
type: node.data.type,
});
warnings.push({
message,
data: parseify({ node, nodeTemplate: template }),
});
continue;
try {
const updatedNode = updateNode(node, template);
nodes[i] = updatedNode;
} catch (e) {
const message = t('nodes.unableToUpdateNode', {
node: node.id,
type: node.data.type,
});
warnings.push({
message,
data: parseify({ node, nodeTemplate: template }),
});
continue;
}
}
for (const input of Object.values(node.data.inputs)) {
@@ -150,8 +154,8 @@ export const validateWorkflow = async (
}
edges.forEach((edge, i) => {
// Validate each edge. If the edge is invalid, we must remove it to prevent runtime errors with reactflow.
const sourceNode = keyedNodes[edge.source];
const targetNode = keyedNodes[edge.target];
const sourceNode = nodes.find(({ id }) => id === edge.source);
const targetNode = nodes.find(({ id }) => id === edge.target);
const sourceTemplate = sourceNode ? templates[sourceNode.data.type] : undefined;
const targetTemplate = targetNode ? templates[targetNode.data.type] : undefined;
const issues: string[] = [];
@@ -230,7 +234,7 @@ export const validateWorkflow = async (
if (_workflow.exposedFields.length > 0 && Object.values(_workflow.form.elements).length === 0) {
// Migrated exposed fields to form elements
for (const { nodeId, fieldName } of _workflow.exposedFields) {
const node = keyedNodes[nodeId];
const node = nodes.find(({ id }) => id === nodeId);
if (!node) {
continue;
}