feat(nodes): make ImageField properties required

This commit is contained in:
psychedelicious 2023-04-03 18:56:48 +10:00
parent d1058adb59
commit 5a9157e628

View File

@ -20,7 +20,14 @@ class ImageField(BaseModel):
)
image_name: Optional[str] = Field(default=None, description="The name of the image")
class Config:
schema_extra = {
'required': [
'image_type',
'image_name',
]
}
class ImageOutput(BaseInvocationOutput):
"""Base class for invocations that output an image"""
#fmt: off