mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-13 07:40:50 +08:00
393d2880dd
Some checks are pending
Python Linting / Run Ruff (push) Waiting to run
Python Linting / Run Pylint (push) Waiting to run
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Waiting to run
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Waiting to run
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Waiting to run
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Waiting to run
Execution Tests / test (macos-latest) (push) Waiting to run
Execution Tests / test (ubuntu-latest) (push) Waiting to run
Execution Tests / test (windows-latest) (push) Waiting to run
Test server launches without errors / test (push) Waiting to run
Unit Tests / test (windows-2022) (push) Waiting to run
Unit Tests / test (macos-latest) (push) Waiting to run
Unit Tests / test (ubuntu-latest) (push) Waiting to run
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from pydantic import BaseModel, Field
|
|
|
|
|
|
class SubjectReference(BaseModel):
|
|
id: str = Field(...)
|
|
images: list[str] = Field(...)
|
|
|
|
|
|
class TaskCreationRequest(BaseModel):
|
|
model: str = Field(...)
|
|
prompt: str = Field(..., max_length=2000)
|
|
duration: int = Field(...)
|
|
seed: int = Field(..., ge=0, le=2147483647)
|
|
aspect_ratio: str | None = Field(None)
|
|
resolution: str | None = Field(None)
|
|
movement_amplitude: str | None = Field(None)
|
|
images: list[str] | None = Field(None, description="Base64 encoded string or image URL")
|
|
subjects: list[SubjectReference] | None = Field(None)
|
|
bgm: bool | None = Field(None)
|
|
audio: bool | None = Field(None)
|
|
|
|
|
|
class TaskCreationResponse(BaseModel):
|
|
task_id: str = Field(...)
|
|
state: str = Field(...)
|
|
created_at: str = Field(...)
|
|
code: int | None = Field(None, description="Error code")
|
|
|
|
|
|
class TaskResult(BaseModel):
|
|
id: str = Field(..., description="Creation id")
|
|
url: str = Field(..., description="The URL of the generated results, valid for one hour")
|
|
cover_url: str = Field(..., description="The cover URL of the generated results, valid for one hour")
|
|
|
|
|
|
class TaskStatusResponse(BaseModel):
|
|
state: str = Field(...)
|
|
err_code: str | None = Field(None)
|
|
progress: float | None = Field(None)
|
|
credits: int | None = Field(None)
|
|
creations: list[TaskResult] = Field(..., description="Generated results")
|