feat: seededit

Author: Lin Fangyuan

volcenginesdkarkruntime/resources/images/images.py

@@ -16,11 +16,13 @@ def generate(
         *,
         model: str,
         prompt: str,
+        image: 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,
@@ -31,11 +33,13 @@ def generate(
             body={
                 "model": model,
                 "prompt": prompt,
+                "image": image,
                 "response_format": response_format,
                 "size": size,
                 "seed": seed,
                 "guidance_scale": guidance_scale,
                 "watermark": watermark,
+                "optimize_prompt": optimize_prompt,
             },
             options=make_request_options(
                 extra_headers=extra_headers,
@@ -56,11 +60,13 @@ async def generate(
         *,
         model: str,
         prompt: str,
+        image: 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,
@@ -71,11 +77,13 @@ async def generate(
             body={
                 "model": model,
                 "prompt": prompt,
+                "image": image,
                 "response_format": response_format,
                 "size": size,
                 "seed": seed,
                 "guidance_scale": guidance_scale,
                 "watermark": watermark,
+                "optimize_prompt": optimize_prompt,
             },
             options=make_request_options(
                 extra_headers=extra_headers,

volcenginesdkexamples/volcenginesdkarkruntime/async_image_generations.py

@@ -11,10 +11,10 @@
 
 
 async def main():
-    print("----- async generate images -----")
+    print("----- [Seedream] async generate images -----")
 
     result = client.images.generate(
-        model="${YOUR_ENDPOINT_ID}",
+        model="${YOUR_SEEDREAM_ENDPOINT_ID}",
         prompt="龙与地下城女骑士背景是起伏的平原，目光从镜头转向平原",
         seed=1234567890,
         watermark=True,
@@ -24,5 +24,19 @@ async def main():
 
     print(await result)
 
+    print("----- [Seededit] async generate images async -----")
+
+    result = client.images.generate(
+        model="${YOUR_SEEDEDIT_ENDPOINT_ID}",
+        prompt="龙与地下城女骑士背景是起伏的平原，目光从镜头转向平原",
+        image="https://an-test-imgs.tos-cn-beijing.volces.com/avi/9m_001.jpg",
+        seed=1234567890,
+        watermark=True,
+        size="adaptive",
+        guidance_scale=2.5,
+    )
+
+    print(await result)
+
 if __name__ == "__main__":
     asyncio.run(main())