Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): allow collect node inputs to connect to multiple fields when using lazy connect

Browse Source
This commit is contained in:
psychedelicious 2024-05-16 23:00:12 +10:00
parent 78cb4d75ad
commit 76825f4261
1 changed files with 7 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
invokeai/frontend/web/src/features/nodes/store/util/findConnectionToValidHandle.ts
View File

@ -169,8 +169,9 @@ export const getFirstValidConnection = (
}
} else {
// Connecting from a target to a source
// Ensure we there is not already an edge to the target
// Ensure we there is not already an edge to the target, except for collect nodes
if (
pendingConnection.node.data.type !== 'collect' &&
edges.some(
(e) => e.target === pendingConnection.node.id && e.targetHandle === pendingConnection.fieldTemplate.name
)
@ -182,20 +183,13 @@ export const getFirstValidConnection = (
return null;
}
if (candidateNode.data.type === 'collect') {
// Special handling for collect node - connect to the `collection` field
return {
source: candidateNode.id,
sourceHandle: 'collection',
target: pendingConnection.node.id,
targetHandle: pendingConnection.fieldTemplate.name,
};
}
// Sources/outputs can have any number of edges, we can take the first matching output field
const candidateFields = map(candidateTemplate.outputs);
const candidateField = candidateFields.find((field) =>
validateSourceAndTargetTypes(field.type, pendingConnection.fieldTemplate.type)
);
const candidateField = candidateFields.find((field) => {
const isValid = validateSourceAndTargetTypes(field.type, pendingConnection.fieldTemplate.type);
const isAlreadyConnected = edges.some((e) => e.source === candidateNode.id && e.sourceHandle === field.name);
return isValid && !isAlreadyConnected;
});
if (candidateField) {
return {
source: candidateNode.id,