from modules.api.models import *

class InvocationsRequest(BaseModel):
    task: str
    username: Optional[str]
    checkpoint_info:Optional[dict]
    models: Optional[dict]
    txt2img_payload: Optional[StableDiffusionTxt2ImgProcessingAPI]
    img2img_payload: Optional[StableDiffusionImg2ImgProcessingAPI]
    extras_single_payload: Optional[ExtrasSingleImageRequest]
    extras_batch_payload: Optional[ExtrasBatchImagesRequest]
    interrogate_payload: Optional[InterrogateRequest]
    db_create_model_payload: Optional[str]
    merge_checkpoint_payload: Optional[dict]

class PingResponse(BaseModel):
    status: str