Merge 'feat/support-draft' into 'integration_2026-01-09_1102561052674'

feat: [development task] ark runtime (2013661)

See merge request: !1019

Author: BitsAdmin

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/resources/images/images.py

@@ -27,7 +27,7 @@
 )
 from ...types.images.image_gen_stream_event import ImageGenStreamEvent
 from ..._types import Body, Query, Headers
-from ..._streaming import Stream
+from ..._streaming import Stream, AsyncStream
 from ...types.images.images import OptimizePromptOptions
 
 
@@ -222,7 +222,7 @@ async def generate(
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
         stream: Literal[True],
-    ) -> Stream[ImageGenStreamEvent]: ...
+    ) -> AsyncStream[ImageGenStreamEvent]: ...
 
     @overload
     async def generate(
@@ -246,7 +246,7 @@ async def generate(
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
         stream: bool,
-    ) -> ImagesResponse | Stream[ImageGenStreamEvent]: ...
+    ) -> ImagesResponse | AsyncStream[ImageGenStreamEvent]: ...
 
     @async_apikey_required
     async def generate(
@@ -270,7 +270,7 @@ async def generate(
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
         stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
-    ) -> ImagesResponse | Stream[ImageGenStreamEvent]:
+    ) -> ImagesResponse | AsyncStream[ImageGenStreamEvent]:
         return await self._post(
             "/images/generations",
             body={
@@ -304,5 +304,5 @@ async def generate(
             ),
             cast_to=ImagesResponse,
             stream=stream or False,
-            stream_cls=Stream[ImageGenStreamEvent],
+            stream_cls=AsyncStream[ImageGenStreamEvent],
         )

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]