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

fix(ui): invocation parsing for optional enum fields

Browse Source 
For example:
```py
my_field: Literal["foo", "bar"] | None = InputField(default=None)
```

Previously, this would cause a field parsing error and prevent the app from loading.

Two fixes:
- This type annotation and resultant schema are now parsed correctly
- Error handling added to template building logic to prevent the hang at startup when an error does occur
This commit is contained in:
psychedelicious
2025-05-20 08:24:45 +10:00
parent 53ffa98662
commit 15880e6ea7
2 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/frontend/web/src/features/nodes/util/schema/buildFieldInputTemplate.ts
View File

@@ -682,7 +682,7 @@ const buildEnumFieldInputTemplate: FieldInputTemplateBuilder<EnumFieldInputTempl
if (filteredAnyOf.length !== 1 || !isSchemaObject(firstAnyOf)) {
options = [];
} else {
options = firstAnyOf.enum ?? [];
options = firstAnyOf.const ? [firstAnyOf.const] : (firstAnyOf.enum ?? []);
}
} else if (schemaObject.const) {
options = [schemaObject.const];

22
invokeai/frontend/web/src/features/nodes/util/schema/parseSchema.ts
View File

@@ -161,8 +161,15 @@ export const parseSchema = (
fieldType.batch = true;
}
const fieldInputTemplate = buildFieldInputTemplate(property, propertyName, fieldType);
inputsAccumulator[propertyName] = fieldInputTemplate;
try {
const fieldInputTemplate = buildFieldInputTemplate(property, propertyName, fieldType);
inputsAccumulator[propertyName] = fieldInputTemplate;
} catch {
log.error(
{ node: type, field: propertyName, schema: parseify(property) },
'Problem building input field template'
);
}
return inputsAccumulator;
},
@@ -226,9 +233,16 @@ export const parseSchema = (
fieldType.batch = true;
}
const fieldOutputTemplate = buildFieldOutputTemplate(property, propertyName, fieldType);
try {
const fieldOutputTemplate = buildFieldOutputTemplate(property, propertyName, fieldType);
outputsAccumulator[propertyName] = fieldOutputTemplate;
} catch {
log.error(
{ node: type, field: propertyName, schema: parseify(property) },
'Problem building output field template'
);
}
outputsAccumulator[propertyName] = fieldOutputTemplate;
return outputsAccumulator;
},
{} as Record<string, FieldOutputTemplate>