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): add string generator

Browse Source
This commit is contained in:
psychedelicious
2025-01-18 22:26:37 +11:00
parent 526d64a5e2
commit 43c98fd99e
10 changed files with 264 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
invokeai/frontend/web/src/features/queue/store/readiness.ts
View File

@@ -30,8 +30,10 @@ import {
isIntegerGeneratorFieldInputInstance,
isStringFieldCollectionInputInstance,
isStringFieldCollectionInputTemplate,
isStringGeneratorFieldInputInstance,
resolveFloatGeneratorField,
resolveIntegerGeneratorField,
resolveStringGeneratorField,
} from 'features/nodes/types/field';
import {
validateImageFieldCollectionValue,
@@ -76,9 +78,21 @@ export const resolveBatchValue = (batchNode: InvocationNode, nodes: InvocationNo
return ownValue;
} else if (batchNode.data.type === 'string_batch') {
assert(isStringFieldCollectionInputInstance(batchNode.data.inputs.strings));
const ownValue = batchNode.data.inputs.strings.value ?? [];
// no generators for strings yet
return ownValue;
const ownValue = batchNode.data.inputs.strings.value;
const edgeToStrings = edges.find((edge) => edge.target === batchNode.id && edge.targetHandle === 'strings');
if (!edgeToStrings) {
return ownValue ?? [];
}
const generatorNode = nodes.find((node) => node.id === edgeToStrings.source);
assert(generatorNode, 'Missing edge from string generator to string batch');
const generatorField = generatorNode.data.inputs['generator'];
assert(isStringGeneratorFieldInputInstance(generatorField), 'Invalid string generator');
const generatorValue = resolveStringGeneratorField(generatorField);
return generatorValue;
} else if (batchNode.data.type === 'float_batch') {
assert(isFloatFieldCollectionInputInstance(batchNode.data.inputs.floats));
const ownValue = batchNode.data.inputs.floats.value;