fix(arkruntime): add image size for seedream api

Author: yiguang

volcenginesdkarkruntime/resources/images/images.py

@@ -125,7 +125,7 @@ def generate(
         sequential_image_generation: str | None = None,
         sequential_image_generation_options: SequentialImageGenerationOptions | None = None,
         stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
-    ) -> ImagesResponse:
+    ) -> ImagesResponse | Stream[ImageGenStreamEvent]:
         resp = self._post(
             "/images/generations",
             body={

volcenginesdkarkruntime/types/images/images.py

@@ -33,6 +33,9 @@ class Image(BaseModel):
     b64_json: str
     """The Base 64 encoded string of the generated image, if any."""
 
+    size: str
+    """The size of the generated image, if any."""
+
 
 class Error(BaseModel):
     message: str

volcenginesdkexamples/volcenginesdkarkruntime/image_generations.py

@@ -24,12 +24,12 @@
     print("----- [Seededit] generate images -----")
 
     result = client.images.generate(
-        model="${YOUR_SEEDEDIT_ENDPOINT_ID}",
+        model="${YOUR_SEEDREAM_ENDPOINT_ID}",
         prompt="龙与地下城女骑士背景是起伏的平原，目光从镜头转向平原",
-        image="${YOUR_IMAGE_URL}",
+        # image="${YOUR_IMAGE_URL}",
         seed=1234567890,
         watermark=True,
-        size="adaptive",
+        size="1024x1024",
     )
 
     print(result)
@@ -40,7 +40,7 @@
         model="${YOUR_SEEDREAM_ENDPOINT_ID}",
         prompt="星球大战, 需要三幅图片描绘不同的战斗场景",
         response_format="url",
-        image=["YOUR_IMAGE_URL_HERE", "YOUR_IMAGE_URL_HERE"],
+        # image=["YOUR_IMAGE_URL_HERE", "YOUR_IMAGE_URL_HERE"],
         seed=1234567890,
         watermark=True,
         size="1024x1024",
@@ -66,6 +66,3 @@
             if event.error is None:
                 print("Final completed event:")
                 print("recv.Usage:", event.usage)
-
-        elif event.type == "image_generation.partial_image":
-            print(f"Partial image index={event.partial_image_index}, size={len(event.b64_json)}")