mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
27 lines
661 B
Python
27 lines
661 B
Python
from enum import Enum
|
|
from typing import Optional
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ImageType(str, Enum):
|
|
RESULT = "results"
|
|
INTERMEDIATE = "intermediates"
|
|
UPLOAD = "uploads"
|
|
|
|
|
|
class ImageField(BaseModel):
|
|
"""An image field used for passing image objects between invocations"""
|
|
|
|
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")
|
|
|
|
class Config:
|
|
schema_extra = {
|
|
"required": [
|
|
"image_type",
|
|
"image_name",
|
|
]
|
|
}
|