mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-31 16:50:17 +08:00
68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ImageGenerationRequest(BaseModel):
|
|
model: str = Field(...)
|
|
prompt: str = Field(...)
|
|
aspect_ratio: str = Field(...)
|
|
n: int = Field(...)
|
|
seed: int = Field(...)
|
|
response_for: str = Field("url")
|
|
|
|
|
|
class InputUrlObject(BaseModel):
|
|
url: str = Field(...)
|
|
|
|
|
|
class ImageEditRequest(BaseModel):
|
|
model: str = Field(...)
|
|
image: InputUrlObject = Field(...)
|
|
prompt: str = Field(...)
|
|
resolution: str = Field(...)
|
|
n: int = Field(...)
|
|
seed: int = Field(...)
|
|
response_for: str = Field("url")
|
|
|
|
|
|
class VideoGenerationRequest(BaseModel):
|
|
model: str = Field(...)
|
|
prompt: str = Field(...)
|
|
image: InputUrlObject | None = Field(...)
|
|
duration: int = Field(...)
|
|
aspect_ratio: str | None = Field(...)
|
|
resolution: str = Field(...)
|
|
seed: int = Field(...)
|
|
|
|
|
|
class VideoEditRequest(BaseModel):
|
|
model: str = Field(...)
|
|
prompt: str = Field(...)
|
|
video: InputUrlObject = Field(...)
|
|
seed: int = Field(...)
|
|
|
|
|
|
class ImageResponseObject(BaseModel):
|
|
url: str | None = Field(None)
|
|
b64_json: str | None = Field(None)
|
|
revised_prompt: str | None = Field(None)
|
|
|
|
|
|
class ImageGenerationResponse(BaseModel):
|
|
data: list[ImageResponseObject] = Field(...)
|
|
|
|
|
|
class VideoGenerationResponse(BaseModel):
|
|
request_id: str = Field(...)
|
|
|
|
|
|
class VideoResponseObject(BaseModel):
|
|
url: str = Field(...)
|
|
upsampled_prompt: str | None = Field(None)
|
|
duration: int = Field(...)
|
|
|
|
|
|
class VideoStatusResponse(BaseModel):
|
|
status: str | None = Field(None)
|
|
video: VideoResponseObject | None = Field(None)
|
|
model: str | None = Field(None)
|