feat(*): support draft

Author: jizhongBao

volcenginesdkarkruntime/resources/content_generation/tasks.py

@@ -49,6 +49,14 @@ def create(
         service_tier: Optional[str] = None,
         execution_expires_after: Optional[int] = None,
         generate_audio: Optional[bool] = None,
+        draft: Optional[bool] = None,
+        camera_fixed: Optional[bool] = None,
+        watermark: Optional[bool] = None,
+        seed: Optional[int] = None,
+        resolution: Optional[str] = None,
+        ratio: Optional[str] = None,
+        duration: Optional[int] = None,
+        frames: Optional[int] = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -64,6 +72,14 @@ def create(
                 "service_tier": service_tier,
                 "execution_expires_after": execution_expires_after,
                 "generate_audio": generate_audio,
+                "draft": draft,
+                "camera_fixed": camera_fixed,
+                "watermark": watermark,
+                "seed": seed,
+                "resolution": resolution,
+                "ratio": ratio,
+                "duration": duration,
+                "frames": frames,
             },
             options=make_request_options(
                 extra_headers=extra_headers,
@@ -179,6 +195,14 @@ async def create(
         service_tier: Optional[str] = None,
         execution_expires_after: Optional[int] = None,
         generate_audio: Optional[bool] = None,
+        draft: Optional[bool] = None,
+        camera_fixed: Optional[bool] = None,
+        watermark: Optional[bool] = None,
+        seed: Optional[int] = None,
+        resolution: Optional[str] = None,
+        ratio: Optional[str] = None,
+        duration: Optional[int] = None,
+        frames: Optional[int] = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -194,6 +218,14 @@ async def create(
                 "service_tier": service_tier,
                 "execution_expires_after": execution_expires_after,
                 "generate_audio": generate_audio,
+                "draft": draft,
+                "camera_fixed": camera_fixed,
+                "watermark": watermark,
+                "seed": seed,
+                "resolution": resolution,
+                "ratio": ratio,
+                "duration": duration,
+                "frames": frames,
             },
             options=make_request_options(
                 extra_headers=extra_headers,

volcenginesdkarkruntime/types/content_generation/content_generation_task.py

@@ -89,3 +89,18 @@ class ContentGenerationTask(BaseModel):
 
     generate_audio: bool
     """Whether to generate audio."""
+
+    duration: int
+    """The duration of the generated content in seconds."""
+
+    ratio: str
+    """The aspect ratio of the generated content."""
+
+    resolution: str
+    """The resolution of the generated content."""
+
+    draft: bool
+    """Whether to generate a draft version of the content."""
+
+    draft_task_id: str
+    """The ID of the draft task, if applicable."""

volcenginesdkarkruntime/types/content_generation/create_task_content_param.py

@@ -41,4 +41,13 @@ class CreateTaskContentImageParam(TypedDict):
     """Image data object containing image URL."""
 
 
-CreateTaskContentParam = Union[CreateTaskContentTextParam, CreateTaskContentImageParam]
+class CreateTaskContentDraftTaskParam(TypedDict):
+    type: Literal["draft_task"]
+    draft_task: CreateTaskContentDraftTaskDataParam
+    """ID of the draft task to be used for content generation."""
+
+class CreateTaskContentDraftTaskDataParam(TypedDict):
+    id: str
+    """ID of the draft task to be used for content generation."""
+
+CreateTaskContentParam = Union[CreateTaskContentTextParam, CreateTaskContentImageParam, CreateTaskContentDraftTaskParam]