fix(nodes): fix image_type in ImageField

This commit is contained in:
psychedelicious 2023-04-04 15:46:34 +10:00
parent daf1bc6b67
commit 406039426a

View File

@ -12,7 +12,7 @@ class ImageType(str, Enum):
class ImageField(BaseModel):
"""An image field used for passing image objects between invocations"""
image_type: str = Field(
image_type: ImageType = Field(
default=ImageType.RESULT, description="The type of the image"
)
image_name: Optional[str] = Field(default=None, description="The name of the image")