mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-15 01:07:03 +08:00
908fd7d749
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 (macos-latest) (push) Waiting to run
Unit Tests / test (ubuntu-latest) (push) Waiting to run
Unit Tests / test (windows-2022) (push) Waiting to run
105 lines
3.9 KiB
Python
105 lines
3.9 KiB
Python
from pydantic import BaseModel, Field
|
||
|
||
|
||
class OmniProText2VideoRequest(BaseModel):
|
||
model_name: str = Field(..., description="kling-video-o1")
|
||
aspect_ratio: str = Field(..., description="'16:9', '9:16' or '1:1'")
|
||
duration: str = Field(..., description="'5' or '10'")
|
||
prompt: str = Field(...)
|
||
mode: str = Field("pro")
|
||
|
||
|
||
class OmniParamImage(BaseModel):
|
||
image_url: str = Field(...)
|
||
type: str | None = Field(None, description="Can be 'first_frame' or 'end_frame'")
|
||
|
||
|
||
class OmniParamVideo(BaseModel):
|
||
video_url: str = Field(...)
|
||
refer_type: str | None = Field(..., description="Can be 'base' or 'feature'")
|
||
keep_original_sound: str = Field(..., description="'yes' or 'no'")
|
||
|
||
|
||
class OmniProFirstLastFrameRequest(BaseModel):
|
||
model_name: str = Field(..., description="kling-video-o1")
|
||
image_list: list[OmniParamImage] = Field(..., min_length=1, max_length=7)
|
||
duration: str = Field(..., description="'5' or '10'")
|
||
prompt: str = Field(...)
|
||
mode: str = Field("pro")
|
||
|
||
|
||
class OmniProReferences2VideoRequest(BaseModel):
|
||
model_name: str = Field(..., description="kling-video-o1")
|
||
aspect_ratio: str | None = Field(..., description="'16:9', '9:16' or '1:1'")
|
||
image_list: list[OmniParamImage] | None = Field(
|
||
None, max_length=7, description="Max length 4 when video is present."
|
||
)
|
||
video_list: list[OmniParamVideo] | None = Field(None, max_length=1)
|
||
duration: str | None = Field(..., description="From 3 to 10.")
|
||
prompt: str = Field(...)
|
||
mode: str = Field("pro")
|
||
|
||
|
||
class TaskStatusVideoResult(BaseModel):
|
||
duration: str | None = Field(None, description="Total video duration")
|
||
id: str | None = Field(None, description="Generated video ID")
|
||
url: str | None = Field(None, description="URL for generated video")
|
||
|
||
|
||
class TaskStatusImageResult(BaseModel):
|
||
index: int = Field(..., description="Image Number，0-9")
|
||
url: str = Field(..., description="URL for generated image")
|
||
|
||
|
||
class TaskStatusResults(BaseModel):
|
||
videos: list[TaskStatusVideoResult] | None = Field(None)
|
||
images: list[TaskStatusImageResult] | None = Field(None)
|
||
|
||
|
||
class TaskStatusResponseData(BaseModel):
|
||
created_at: int | None = Field(None, description="Task creation time")
|
||
updated_at: int | None = Field(None, description="Task update time")
|
||
task_status: str | None = None
|
||
task_status_msg: str | None = Field(None, description="Additional failure reason. Only for polling endpoint.")
|
||
task_id: str | None = Field(None, description="Task ID")
|
||
task_result: TaskStatusResults | None = Field(None)
|
||
|
||
|
||
class TaskStatusResponse(BaseModel):
|
||
code: int | None = Field(None, description="Error code")
|
||
message: str | None = Field(None, description="Error message")
|
||
request_id: str | None = Field(None, description="Request ID")
|
||
data: TaskStatusResponseData | None = Field(None)
|
||
|
||
|
||
class OmniImageParamImage(BaseModel):
|
||
image: str = Field(...)
|
||
|
||
|
||
class OmniProImageRequest(BaseModel):
|
||
model_name: str = Field(..., description="kling-image-o1")
|
||
resolution: str = Field(..., description="'1k' or '2k'")
|
||
aspect_ratio: str | None = Field(...)
|
||
prompt: str = Field(...)
|
||
mode: str = Field("pro")
|
||
n: int | None = Field(1, le=9)
|
||
image_list: list[OmniImageParamImage] | None = Field(..., max_length=10)
|
||
|
||
|
||
class TextToVideoWithAudioRequest(BaseModel):
|
||
model_name: str = Field(..., description="kling-v2-6")
|
||
aspect_ratio: str = Field(..., description="'16:9', '9:16' or '1:1'")
|
||
duration: str = Field(..., description="'5' or '10'")
|
||
prompt: str = Field(...)
|
||
mode: str = Field("pro")
|
||
sound: str = Field(..., description="'on' or 'off'")
|
||
|
||
|
||
class ImageToVideoWithAudioRequest(BaseModel):
|
||
model_name: str = Field(..., description="kling-v2-6")
|
||
image: str = Field(...)
|
||
duration: str = Field(..., description="'5' or '10'")
|
||
prompt: str = Field(...)
|
||
mode: str = Field("pro")
|
||
sound: str = Field(..., description="'on' or 'off'")
|