mirror of
https://github.com/Comfy-Org/ComfyUI.git
synced 2026-08-15 14:14:28 +08:00
f4da7a827b
Amp-Thread-ID: https://ampcode.com/threads/T-019fd06e-b484-753d-a063-f3c79085e235 Co-authored-by: Amp <amp@ampcode.com> # Conflicts: # comfy_api_nodes/apis/comfy_cloud.py # comfy_api_nodes/nodes_comfy_cloud.py # tests-unit/comfy_api_nodes_test/comfy_cloud_test.py
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
ComfyCloudWorkflow = Literal[
|
|
"text-to-image",
|
|
"text-to-video",
|
|
"image-to-video",
|
|
"image-edit",
|
|
"image.ideogram-4-design.v1",
|
|
"image.krea-2-creative-image.v1",
|
|
"image.mage-flow-image.v1",
|
|
"image.flux-2-reference-edit.v1",
|
|
"image.qwen-image-edit-2511.v1",
|
|
"image.seedvr2-image-upscale.v1",
|
|
"video.minimax-h3-text-sound.v1",
|
|
"video.minimax-h3-image-sound.v1",
|
|
"video.ltx-2-3-image-audio-performance.v1",
|
|
"video.ltx-2-3-first-last-frame.v1",
|
|
"video.wan-2-2-14b-first-last-frame.v1",
|
|
"video.scail-2-character-replacement.v1",
|
|
]
|
|
|
|
|
|
class ComfyCloudWorkflowInputs(BaseModel):
|
|
prompt: str | None = Field(None)
|
|
image_url: str | None = Field(None)
|
|
assets: dict[str, "ComfyCloudAssetInput"] | None = Field(None)
|
|
audio_url: str | None = Field(None)
|
|
first_frame_url: str | None = Field(None)
|
|
last_frame_url: str | None = Field(None)
|
|
reference_character_url: str | None = Field(None)
|
|
driving_video_url: str | None = Field(None)
|
|
instruction: str | None = Field(None)
|
|
prompt_enhance: bool | None = Field(None)
|
|
enhance_prompt: bool | None = Field(None)
|
|
negative_prompt: str | None = Field(None)
|
|
aspect_ratio: str | None = Field(None)
|
|
duration_seconds: float | None = Field(None)
|
|
guidance: float | None = Field(None)
|
|
quality_mode: str | None = Field(None)
|
|
seed: int | None = Field(None, ge=0, le=0xFFFFFFFFFFFFFFFF)
|
|
scale: str | None = Field(None)
|
|
scene_prompt: str | None = Field(None)
|
|
driving_subject: str | None = Field(None)
|
|
reference_subject: str | None = Field(None)
|
|
|
|
|
|
class ComfyCloudAssetInput(BaseModel):
|
|
type: Literal["IMAGE", "VIDEO", "AUDIO"] = Field(...)
|
|
url: str = Field(...)
|
|
|
|
|
|
class ComfyCloudGenerateRequest(BaseModel):
|
|
workflow: ComfyCloudWorkflow = Field(...)
|
|
inputs: ComfyCloudWorkflowInputs = Field(...)
|
|
|
|
|
|
class ComfyCloudGenerateResponse(BaseModel):
|
|
task_id: str = Field(...)
|
|
status: str = Field(...)
|
|
polling_url: str = Field(...)
|
|
cancel_url: str = Field(...)
|
|
|
|
|
|
class ComfyCloudStatusResponse(BaseModel):
|
|
task_id: str = Field(...)
|
|
status: str = Field(...)
|
|
progress: float | None = Field(None)
|
|
output_url: str | None = Field(None)
|
|
error: str | None = Field(None)
|