feat: support seedance and seedream param

Author: yiguang

volcenginesdkarkruntime/resources/images/images.py

@@ -24,6 +24,7 @@
     OptimizePromptOptions,
     SequentialImageGenerationOptions,
     ImagesResponse,
+    ContentGenerationTool
 )
 from ...types.images.image_gen_stream_event import ImageGenStreamEvent
 from ..._types import Body, Query, Headers
@@ -53,6 +54,8 @@ def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
     ) -> ImagesResponse:
         ...
@@ -86,6 +89,8 @@ def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: Literal[True],
     ) -> Stream[ImageGenStreamEvent]: ...
 
@@ -110,6 +115,8 @@ def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: bool,
     ) -> ImagesResponse | Stream[ImageGenStreamEvent]: ...
 
@@ -134,6 +141,8 @@ def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
     ) -> ImagesResponse | Stream[ImageGenStreamEvent]:
         resp = self._post(
@@ -159,6 +168,12 @@ def generate(
                     if sequential_image_generation_options is not None
                     else None
                 ),
+                "tools": (
+                    [t.model_dump(mode="json") for t in tools]
+                    if tools is not None
+                    else None
+                ),
+                "output_format": output_format,
                 "stream": stream,
             },
             options=make_request_options(
@@ -197,6 +212,8 @@ async def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
     ) -> ImagesResponse: ...
 
@@ -221,6 +238,8 @@ async def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: Literal[True],
     ) -> AsyncStream[ImageGenStreamEvent]: ...
 
@@ -245,6 +264,8 @@ async def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: bool,
     ) -> ImagesResponse | AsyncStream[ImageGenStreamEvent]: ...
 
@@ -269,6 +290,8 @@ async def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions
         | None = None,
+        tools: list[ContentGenerationTool] | None = None,
+        output_format: str | None = None,
         stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
     ) -> ImagesResponse | AsyncStream[ImageGenStreamEvent]:
         return await self._post(
@@ -294,6 +317,12 @@ async def generate(
                     if sequential_image_generation_options is not None
                     else None
                 ),
+                "tools": (
+                    [t.model_dump(mode="json") for t in tools]
+                    if tools is not None
+                    else None
+                ),
+                "output_format": output_format,
                 "stream": stream,
             },
             options=make_request_options(

volcenginesdkarkruntime/types/images/__init__.py

@@ -13,11 +13,13 @@
     OptimizePromptOptions,
     SequentialImageGenerationOptions,
     ImagesResponse,
+    ContentGenerationTool
 )
 
 
 __all__ = [
     "OptimizePromptOptions",
     "SequentialImageGenerationOptions",
     "ImagesResponse",
+    "ContentGenerationTool"
 ]

volcenginesdkarkruntime/types/images/image_gen_completed_event.py

@@ -10,18 +10,25 @@
 # This modified file is released under the same license.
 
 
+from typing import List, Optional
 from volcenginesdkarkruntime._models import BaseModel
 
 __all__ = ["ImageGenCompletedEvent"]
 
 
+class ToolUsage(BaseModel):
+    web_search: Optional[int] = None
+    """The number of web search."""
+
 class Usage(BaseModel):
     generated_images: int
     """The number of images generated."""
     output_tokens: int
     """The number of output tokens."""
     total_tokens: int
     """The total number of tokens."""
+    tool_usage: Optional[ToolUsage] = None
+    """The total number of images generated by this request."""
 
 
 class Error(BaseModel):

volcenginesdkarkruntime/types/images/images.py

@@ -17,6 +17,7 @@
     "OptimizePromptOptions",
     "SequentialImageGenerationOptions",
     "ImagesResponse",
+    "ContentGenerationTool"
 ]
 
 
@@ -30,9 +31,20 @@ class SequentialImageGenerationOptions(BaseModel):
     """ Maximum number of images to generate in this request; effective only when the multi-image feature is enabled """
 
 
+class ToolUsage(BaseModel):
+    web_search: Optional[int] = None
+    """The number of web search."""
+
+
 class Usage(BaseModel):
     generated_images: int
     """The number of images generated."""
+    output_tokens: Optional[int] = None
+    """The number of images generated by this request."""
+    total_tokens: Optional[int] = None
+    """The total number of images generated by this request."""
+    tool_usage: Optional[ToolUsage] = None
+    """The total number of images generated by this request."""
 
 
 class Image(BaseModel):
@@ -54,6 +66,11 @@ class Error(BaseModel):
     """The error code for failed image generation"""
 
 
+class ContentGenerationTool(BaseModel):
+    type: str
+    """The content generation tool used to generate the images."""
+
+
 class ImagesResponse(BaseModel):
     model: str
     """The model used to generated the images."""
@@ -69,3 +86,6 @@ class ImagesResponse(BaseModel):
 
     created_at: int
     """The Unix timestamp when the image was generated."""
+
+    tool: List[ContentGenerationTool]
+    """ The tool used to generate the image."""