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(nodes): improved error messages for invalid defaults

Browse Source
This commit is contained in:
psychedelicious
2025-05-02 10:45:04 +10:00
parent 758f94d3c6
commit b6d58774f4
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/app/invocations/baseinvocation.py
View File

@@ -495,7 +495,9 @@ class NoDefaultSentinel:
pass
def validate_field_default(field_name: str, invocation_type: str, annotation: Any, field_info: FieldInfo) -> None:
def validate_field_default(
cls_name: str, field_name: str, invocation_type: str, annotation: Any, field_info: FieldInfo
) -> None:
"""Validates the default value of a field against its pydantic field definition."""
assert isinstance(field_info.json_schema_extra, dict), "json_schema_extra is not a dict"
@@ -507,13 +509,15 @@ def validate_field_default(field_name: str, invocation_type: str, annotation: An
if orig_default is NoDefaultSentinel:
return
TempDefaultValidator = create_model("TempDefaultValidator", field_to_validate=(annotation, field_info))
TempDefaultValidator = create_model(cls_name, **{field_name: (annotation, field_info)})
# Validate the default value against the annotation
try:
TempDefaultValidator(field_to_validate=orig_default)
TempDefaultValidator.model_validate({field_name: orig_default})
except Exception as e:
raise ValueError(f"Default value for field {field_name} on invocation {invocation_type} is invalid, {e}") from e
raise InvalidFieldError(
f'Default value for field "{field_name}" on invocation "{invocation_type}" is invalid, {e}'
) from e
def is_optional(annotation: Any) -> bool:
@@ -571,7 +575,7 @@ def invocation(
f"{field_name} on invocation {invocation_type} has a non-dict json_schema_extra, did you forget to use InputField?"
)
validate_field_default(field_name, invocation_type, annotation, field_info)
validate_field_default(cls.__name__, field_name, invocation_type, annotation, field_info)
if field_info.default is None and not is_optional(annotation):
annotation = annotation | None