mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-16 16:32:34 +08:00
e2c71ceb00
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
* feat(api-nodes-Tencent): add ModelTo3DUV, 3DTextureEdit, 3DParts nodes * add image output to TencentModelTo3DUV node * commented out two nodes * added rate_limit check to other hunyuan3d nodes
87 lines
2.2 KiB
Python
87 lines
2.2 KiB
Python
from typing import TypedDict
|
|
|
|
from pydantic import BaseModel, Field, model_validator
|
|
|
|
|
|
class InputGenerateType(TypedDict):
|
|
generate_type: str
|
|
polygon_type: str
|
|
pbr: bool
|
|
|
|
|
|
class Hunyuan3DViewImage(BaseModel):
|
|
ViewType: str = Field(..., description="Valid values: back, left, right.")
|
|
ViewImageUrl: str = Field(...)
|
|
|
|
|
|
class To3DProTaskRequest(BaseModel):
|
|
Model: str = Field(...)
|
|
Prompt: str | None = Field(None)
|
|
ImageUrl: str | None = Field(None)
|
|
MultiViewImages: list[Hunyuan3DViewImage] | None = Field(None)
|
|
EnablePBR: bool | None = Field(...)
|
|
FaceCount: int | None = Field(...)
|
|
GenerateType: str | None = Field(...)
|
|
PolygonType: str | None = Field(...)
|
|
|
|
|
|
class RequestError(BaseModel):
|
|
Code: str = Field("")
|
|
Message: str = Field("")
|
|
|
|
|
|
class To3DProTaskCreateResponse(BaseModel):
|
|
JobId: str | None = Field(None)
|
|
Error: RequestError | None = Field(None)
|
|
|
|
@model_validator(mode="before")
|
|
@classmethod
|
|
def unwrap_data(cls, values: dict) -> dict:
|
|
if "Response" in values and isinstance(values["Response"], dict):
|
|
return values["Response"]
|
|
return values
|
|
|
|
|
|
class ResultFile3D(BaseModel):
|
|
Type: str = Field(...)
|
|
Url: str = Field(...)
|
|
PreviewImageUrl: str = Field("")
|
|
|
|
|
|
class To3DProTaskResultResponse(BaseModel):
|
|
ErrorCode: str = Field("")
|
|
ErrorMessage: str = Field("")
|
|
ResultFile3Ds: list[ResultFile3D] = Field([])
|
|
Status: str = Field(...)
|
|
|
|
@model_validator(mode="before")
|
|
@classmethod
|
|
def unwrap_data(cls, values: dict) -> dict:
|
|
if "Response" in values and isinstance(values["Response"], dict):
|
|
return values["Response"]
|
|
return values
|
|
|
|
|
|
class To3DProTaskQueryRequest(BaseModel):
|
|
JobId: str = Field(...)
|
|
|
|
|
|
class To3DUVFileInput(BaseModel):
|
|
Type: str = Field(..., description="File type: GLB, OBJ, or FBX")
|
|
Url: str = Field(...)
|
|
|
|
|
|
class To3DUVTaskRequest(BaseModel):
|
|
File: To3DUVFileInput = Field(...)
|
|
|
|
|
|
class TextureEditImageInfo(BaseModel):
|
|
Url: str = Field(...)
|
|
|
|
|
|
class TextureEditTaskRequest(BaseModel):
|
|
File3D: To3DUVFileInput = Field(...)
|
|
Image: TextureEditImageInfo | None = Field(None)
|
|
Prompt: str | None = Field(None)
|
|
EnablePBR: bool | None = Field(None)
|