feat: add subdivisionlevel and fileformat in resp

Author: yiguang

volcenginesdkarkruntime/resources/images/images.py

@@ -21,7 +21,7 @@
 from ..._base_client import make_request_options
 from ..._utils._utils import apikey_required, async_apikey_required
 from ..._resource import SyncAPIResource, AsyncAPIResource
-from ...types.images import SequentialImageGenerationOptions, ImagesResponse
+from ...types.images import OptimizePromptOptions, SequentialImageGenerationOptions, ImagesResponse
 from ...types.images.image_gen_stream_event import ImageGenStreamEvent
 from ..._types import Body, Query, Headers
 from ..._streaming import Stream
@@ -41,6 +41,7 @@ def generate(
         guidance_scale: float | None = None,
         watermark: bool | None = None,
         optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -72,6 +73,7 @@ def generate(
         guidance_scale: float | None = None,
         watermark: bool | None = None,
         optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -95,6 +97,7 @@ def generate(
         guidance_scale: float | None = None,
         watermark: bool | None = None,
         optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -118,6 +121,7 @@ def generate(
         guidance_scale: float | None = None,
         watermark: bool | None = None,
         optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -138,6 +142,10 @@ def generate(
                 "guidance_scale": guidance_scale,
                 "watermark": watermark,
                 "optimize_prompt": optimize_prompt,
+                "optimize_prompt_options": (
+                    optimize_prompt_options.model_dump(mode="json")
+                    if optimize_prompt_options is not None else None
+                ),
                 "sequential_image_generation": sequential_image_generation,
                 "sequential_image_generation_options": (
                     sequential_image_generation_options.model_dump(mode="json")
@@ -162,70 +170,73 @@ def generate(
 class AsyncImages(AsyncAPIResource):
     @overload
     async def generate(
-            self,
-            *,
-            model: str,
-            prompt: str,
-            image: str | list[str] | None = None,
-            response_format: str | None = None,
-            size: str | None = None,
-            seed: int | None = None,
-            guidance_scale: float | None = None,
-            watermark: bool | None = None,
-            optimize_prompt: bool | None = None,
-            extra_headers: Headers | None = None,
-            extra_query: Query | None = None,
-            extra_body: Body | None = None,
-            timeout: float | httpx.Timeout | None = None,
-            sequential_image_generation: str | None = None,
-            sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
-            stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
+        self,
+        *,
+        model: str,
+        prompt: str,
+        image: str | list[str] | None = None,
+        response_format: str | None = None,
+        size: str | None = None,
+        seed: int | None = None,
+        guidance_scale: float | None = None,
+        watermark: bool | None = None,
+        optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None = None,
+        sequential_image_generation: str | None = None,
+        sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
+        stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
     ) -> ImagesResponse:
         ...
 
     @overload
     async def generate(
-            self,
-            *,
-            model: str,
-            prompt: str,
-            image: str | list[str] | None = None,
-            response_format: str | None = None,
-            size: str | None = None,
-            seed: int | None = None,
-            guidance_scale: float | None = None,
-            watermark: bool | None = None,
-            optimize_prompt: bool | None = None,
-            extra_headers: Headers | None = None,
-            extra_query: Query | None = None,
-            extra_body: Body | None = None,
-            timeout: float | httpx.Timeout | None = None,
-            sequential_image_generation: str | None = None,
-            sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
-            stream: Literal[True],
+        self,
+        *,
+        model: str,
+        prompt: str,
+        image: str | list[str] | None = None,
+        response_format: str | None = None,
+        size: str | None = None,
+        seed: int | None = None,
+        guidance_scale: float | None = None,
+        watermark: bool | None = None,
+        optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None = None,
+        sequential_image_generation: str | None = None,
+        sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
+        stream: Literal[True],
     ) -> Stream[ImageGenStreamEvent]:
         ...
 
     @overload
     async def generate(
-            self,
-            *,
-            model: str,
-            prompt: str,
-            image: str | list[str] | None = None,
-            response_format: str | None = None,
-            size: str | None = None,
-            seed: int | None = None,
-            guidance_scale: float | None = None,
-            watermark: bool | None = None,
-            optimize_prompt: bool | None = None,
-            extra_headers: Headers | None = None,
-            extra_query: Query | None = None,
-            extra_body: Body | None = None,
-            timeout: float | httpx.Timeout | None = None,
-            sequential_image_generation: str | None = None,
-            sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
-            stream: bool,
+        self,
+        *,
+        model: str,
+        prompt: str,
+        image: str | list[str] | None = None,
+        response_format: str | None = None,
+        size: str | None = None,
+        seed: int | None = None,
+        guidance_scale: float | None = None,
+        watermark: bool | None = None,
+        optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None = None,
+        sequential_image_generation: str | None = None,
+        sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
+        stream: bool,
     ) -> ImagesResponse | Stream[ImageGenStreamEvent]:
         ...
 
@@ -242,6 +253,7 @@ async def generate(
         guidance_scale: float | None = None,
         watermark: bool | None = None,
         optimize_prompt: bool | None = None,
+        optimize_prompt_options: OptimizePromptOptions | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -262,6 +274,10 @@ async def generate(
                 "guidance_scale": guidance_scale,
                 "watermark": watermark,
                 "optimize_prompt": optimize_prompt,
+                "optimize_prompt_options": (
+                    optimize_prompt_options.model_dump(mode="json")
+                    if optimize_prompt_options is not None else None
+                ),
                 "sequential_image_generation": sequential_image_generation,
                 "sequential_image_generation_options": (
                     sequential_image_generation_options.model_dump(mode="json")

volcenginesdkarkruntime/types/content_generation/content_generation_task.py

@@ -27,6 +27,9 @@ class Content(BaseModel):
     last_frame_url: str
     """The URL of the last frame of the generated video, if any."""
 
+    file_url: str
+    """The URL of the generated file, if any."""
+
 
 class ContentGenerationError(BaseModel):
     message: str
@@ -55,6 +58,18 @@ class ContentGenerationTask(BaseModel):
     usage: Usage
     """The usage information for the task."""
 
+    subdivisionlevel: str
+    """The subdivision level of the task."""
+
+    fileformat: str
+    """The file format of the generated content."""
+
+    frames: int
+    """The number of frames in the generated content."""
+
+    framespersecond: int
+    """The number of frames per second in the generated content."""
+
     created_at: int
     """The Unix timestamp when the task was created."""
 

volcenginesdkarkruntime/types/images/__init__.py

@@ -10,7 +10,7 @@
 #
 # This modified file is released under the same license.
 
-from .images import SequentialImageGenerationOptions, ImagesResponse
+from .images import OptimizePromptOptions, SequentialImageGenerationOptions, ImagesResponse
 
 
-__all__ = ["SequentialImageGenerationOptions", "ImagesResponse"]
+__all__ = ["OptimizePromptOptions", "SequentialImageGenerationOptions", "ImagesResponse"]

volcenginesdkarkruntime/types/images/images.py

@@ -14,13 +14,19 @@
 
 from volcenginesdkarkruntime._models import BaseModel
 
-__all__ = ["SequentialImageGenerationOptions", "ImagesResponse"]
+__all__ = ["OptimizePromptOptions", "SequentialImageGenerationOptions", "ImagesResponse"]
+
+
+class OptimizePromptOptions(BaseModel):
+    thinking: Optional[str] = None
+    mode: Optional[str] = None
 
 
 class SequentialImageGenerationOptions(BaseModel):
     max_images: Optional[int] = None
     """ Maximum number of images to generate in this request; effective only when the multi-image feature is enabled """
 
+
 class Usage(BaseModel):
     generated_images: int
     """The number of images generated."""