mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-05 05:46:12 +02:00
bd21363563
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Build package / Build Test (3.10) (push) Has been cancelled
Build package / Build Test (3.11) (push) Has been cancelled
Build package / Build Test (3.12) (push) Has been cancelled
Build package / Build Test (3.13) (push) Has been cancelled
Build package / Build Test (3.14) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
76 lines
1.9 KiB
Python
76 lines
1.9 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_format: str = Field("url")
|
|
resolution: str = Field(...)
|
|
|
|
|
|
class InputUrlObject(BaseModel):
|
|
url: str = Field(...)
|
|
|
|
|
|
class ImageEditRequest(BaseModel):
|
|
model: str = Field(...)
|
|
images: list[InputUrlObject] = Field(...)
|
|
prompt: str = Field(...)
|
|
resolution: str = Field(...)
|
|
n: int = Field(...)
|
|
seed: int = Field(...)
|
|
response_format: str = Field("url")
|
|
aspect_ratio: str | None = Field(...)
|
|
|
|
|
|
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 UsageObject(BaseModel):
|
|
cost_in_usd_ticks: int | None = Field(None)
|
|
|
|
|
|
class ImageGenerationResponse(BaseModel):
|
|
data: list[ImageResponseObject] = Field(...)
|
|
usage: UsageObject | None = Field(None)
|
|
|
|
|
|
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)
|
|
usage: UsageObject | None = Field(None)
|