runapi-seedream 0.3.0__tar.gz → 0.5.0__tar.gz

{runapi_seedream-0.3.0 → runapi_seedream-0.5.0}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.5
 Name: runapi-seedream
-Version: 0.3.0
+Version: 0.5.0
 Summary: RunAPI Seedream SDK for text-to-image and edit-image workflows in JavaScript, Python, Ruby, Go, Java, and PHP
 Project-URL: Homepage, https://runapi.ai/models/seedream
 Project-URL: Documentation, https://runapi.ai/docs/resources/sdks
 Project-URL: Source, https://github.com/runapi-ai/seedream-sdk
 Project-URL: Issues, https://github.com/runapi-ai/seedream-sdk/issues
 Project-URL: Changelog, https://github.com/runapi-ai/seedream-sdk/blob/main/CHANGELOG.md
-Project-URL: Release Notes, https://github.com/runapi-ai/seedream-sdk/releases/tag/python%2Fv0.3.0
+Project-URL: Release Notes, https://github.com/runapi-ai/seedream-sdk/releases/tag/python%2Fv0.5.0
 Author-email: RunAPI <contact@runapi.ai>
 License-Expression: Apache-2.0
 Keywords: api,golang,gradle,image-api,image-generation,java,maven,python,ruby,runapi,runapi-ai,sdk,seedream,seedream-api,text-to-image,typescript
 Requires-Python: >=3.9
-Requires-Dist: runapi-core>=0.8.0
+Requires-Dist: runapi-core~=0.10.1
 Description-Content-Type: text/markdown
 
 # Seedream Python SDK for RunAPI

{runapi_seedream-0.3.0 → runapi_seedream-0.5.0}/pyproject.toml

@@ -4,14 +4,14 @@ build-backend = "hatchling.build"
 
 [project]
 name = "runapi-seedream"
-version = "0.3.0"
+version = "0.5.0"
 description = "RunAPI Seedream SDK for text-to-image and edit-image workflows in JavaScript, Python, Ruby, Go, Java, and PHP"
 readme = "README.md"
 requires-python = ">=3.9"
 license = "Apache-2.0"
 authors = [{ name = "RunAPI", email = "contact@runapi.ai" }]
 keywords = ["runapi", "runapi-ai", "seedream", "api", "sdk", "typescript", "python", "ruby", "golang", "java", "maven", "gradle", "image-generation", "text-to-image", "image-api", "seedream-api"]
-dependencies = ["runapi-core>=0.8.0"]
+dependencies = ["runapi-core~=0.10.1"]
 
 [project.urls]
 Homepage = "https://runapi.ai/models/seedream"
@@ -19,7 +19,7 @@ Documentation = "https://runapi.ai/docs/resources/sdks"
 Source = "https://github.com/runapi-ai/seedream-sdk"
 Issues = "https://github.com/runapi-ai/seedream-sdk/issues"
 Changelog = "https://github.com/runapi-ai/seedream-sdk/blob/main/CHANGELOG.md"
-"Release Notes" = "https://github.com/runapi-ai/seedream-sdk/releases/tag/python%2Fv0.3.0"
+"Release Notes" = "https://github.com/runapi-ai/seedream-sdk/releases/tag/python%2Fv0.5.0"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/runapi"]

{runapi_seedream-0.3.0 → runapi_seedream-0.5.0}/src/runapi/seedream/resources/decompose_layers.py

@@ -6,7 +6,6 @@ from typing import Any, Optional
 
 from runapi.core import RequestOptions, Resource
 
-from ..contract_gen import CONTRACT
 from ..types import CompletedDecomposeLayersResponse, DecomposeLayersResponse
 
 
@@ -23,7 +22,6 @@ class DecomposeLayers(Resource):
 
     def create(self, options: Optional[RequestOptions] = None, **params: Any) -> Any:
         compacted = self._compact_params(params)
-        self._validate_contract(CONTRACT["decompose-layers"], compacted)
         return self._request("post", self.ENDPOINT, body=compacted, options=options)
 
     def get(self, id: str, options: Optional[RequestOptions] = None) -> Any:

{runapi_seedream-0.3.0 → runapi_seedream-0.5.0}/src/runapi/seedream/resources/edit_image.py

@@ -2,15 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
-from runapi.core import Resource, ValidationError, RequestOptions
+from runapi.core import Resource, RequestOptions
 
-from ..contract_gen import CONTRACT
 from ..types import (
-    LONG_PROMPT_MODELS,
-    MINIMUM_THREE_PROMPT_MODELS,
-    V4_MODELS,
     CompletedEditImageResponse,
     EditImageResponse,
 )
@@ -24,10 +20,6 @@ class EditImage(Resource):
     RESPONSE_CLASS = EditImageResponse
     COMPLETED_RESPONSE_CLASS = CompletedEditImageResponse
 
-    PROMPT_MAX_LENGTH = 3000
-    V4_PROMPT_MAX_LENGTH = 5000
-    PROMPT_MIN_LENGTH_LITE = 3
-
     def run(self, options: Optional[RequestOptions] = None, **params: Any) -> Any:
         """Edit an image and poll until it completes.
 
@@ -50,7 +42,6 @@ class EditImage(Resource):
             The task creation result with an id.
         """
         compacted = self._compact_params(params)
-        self._validate_params(compacted)
         return self._request("post", self.ENDPOINT, body=compacted, options=options)
 
     def get(self, id: str, options: Optional[RequestOptions] = None) -> Any:
@@ -63,31 +54,3 @@ class EditImage(Resource):
             The current task status.
         """
         return self._request("get", f"{self.ENDPOINT}/{id}", options=options)
-
-    def _validate_params(self, params: Dict[str, Any]) -> None:
-        self._validate_contract(CONTRACT["edit-image"], params)
-
-        model = params.get("model")
-
-        prompt = params.get("prompt")
-        if not (isinstance(prompt, str) and prompt):
-            raise ValidationError("prompt is required")
-        max_length = self.V4_PROMPT_MAX_LENGTH if model in LONG_PROMPT_MODELS else self.PROMPT_MAX_LENGTH
-        if len(prompt) > max_length:
-            raise ValidationError(f"prompt must be at most {max_length} characters")
-        if model in MINIMUM_THREE_PROMPT_MODELS and len(prompt) < self.PROMPT_MIN_LENGTH_LITE:
-            raise ValidationError(
-                f"prompt must be between {self.PROMPT_MIN_LENGTH_LITE} and {max_length} characters"
-            )
-
-        if model in V4_MODELS:
-            self._validate_integer(params, "seed")
-
-    @staticmethod
-    def _validate_integer(params: Dict[str, Any], key: str) -> None:
-        value = params.get(key)
-        if value is None:
-            return
-        if isinstance(value, int) and not isinstance(value, bool):
-            return
-        raise ValidationError(f"{key} must be an integer")

runapi_seedream-0.5.0/src/runapi/seedream/resources/text_to_image.py

@@ -0,0 +1,56 @@
+"""Seedream text-to-image resource."""
+
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from runapi.core import Resource, RequestOptions
+
+from ..types import (
+    CompletedTextToImageResponse,
+    TextToImageResponse,
+)
+
+
+class TextToImage(Resource):
+    """Generate images from text prompts with Seedream models."""
+
+    ENDPOINT = "/api/v1/seedream/text_to_image"
+
+    RESPONSE_CLASS = TextToImageResponse
+    COMPLETED_RESPONSE_CLASS = CompletedTextToImageResponse
+
+    def run(self, options: Optional[RequestOptions] = None, **params: Any) -> Any:
+        """Create a text-to-image task and poll until it completes.
+
+        Args:
+            **params: text-to-image parameters (model, ...).
+
+        Returns:
+            The completed (narrowed) text-to-image response.
+        """
+        task = self.create(options=options, **params)
+        return self._poll_until_complete(lambda: self.get(task.id, options=options))
+
+    def create(self, options: Optional[RequestOptions] = None, **params: Any) -> Any:
+        """Create a text-to-image task and return immediately with an id.
+
+        Args:
+            **params: text-to-image parameters (model, ...).
+
+        Returns:
+            The task creation result with an id.
+        """
+        compacted = self._compact_params(params)
+        return self._request("post", self.ENDPOINT, body=compacted, options=options)
+
+    def get(self, id: str, options: Optional[RequestOptions] = None) -> Any:
+        """Fetch the current status of a text-to-image task.
+
+        Args:
+            id: The task id returned by ``create``.
+
+        Returns:
+            The current task status.
+        """
+        return self._request("get", f"{self.ENDPOINT}/{id}", options=options)

{runapi_seedream-0.3.0 → runapi_seedream-0.5.0}/src/runapi/seedream/types.py

@@ -1,18 +1,9 @@
-"""Seedream model groups and responses; request enums come from CONTRACT."""
+"""Seedream response models."""
 
 from __future__ import annotations
 
 from runapi.core import BaseModel, TaskResponse, optional, required
 
-# Model groupings used by bespoke prompt-length selection (per-model rule the
-# contract cannot express). Model membership and field enums are validated by
-# the generated CONTRACT.
-LITE_MODELS = ["seedream-5-lite-text-to-image", "seedream-5-lite-edit"]
-PRO_MODELS = ["seedream-5-pro-text-to-image", "seedream-5-pro-edit"]
-V4_MODELS = ["seedream-v4-text-to-image", "seedream-v4-edit"]
-LONG_PROMPT_MODELS = [*PRO_MODELS, *V4_MODELS]
-MINIMUM_THREE_PROMPT_MODELS = [*LITE_MODELS, *PRO_MODELS]
-
 
 class Image(BaseModel):
     url = optional(str)

{runapi_seedream-0.3.0 → runapi_seedream-0.5.0}/tests/test_client.py

@@ -1,7 +1,12 @@
+import json
+
+import httpx
 import pytest
 
 from runapi.core import config
 from runapi.core.errors import AuthenticationError, ValidationError
+from runapi.core.http_client import HttpClient
+from runapi.core.options import ClientOptions
 from runapi.seedream import SeedreamClient
 from runapi.seedream.resources.edit_image import EditImage
 from runapi.seedream.resources.decompose_layers import DecomposeLayers
@@ -73,10 +78,9 @@ def test_decompose_layers_create_get_and_completed_response():
         {"id": "layers-1", "status": "processing"},
         {
             "id": "layers-1",
-            "status": "completed",
+            "status": "completed", "usage": {"cost": 0.05, "generated_images": 2},
             "base_image": {"url": "https://file.runapi.ai/base.jpeg"},
-            "layers": [{"url": "https://file.runapi.ai/layer.png", "z_index": 1}],
-        },
+            "layers": [{"url": "https://file.runapi.ai/layer.png", "z_index": 1}]},
     )
     client = SeedreamClient(api_key="k", http_client=fake)
     created = client.decompose_layers.create(
@@ -92,11 +96,11 @@ def test_decompose_layers_create_get_and_completed_response():
         {
             "model": "seedream-5-pro-layer-decomposition",
             "image_url": "https://cdn.runapi.ai/public/samples/image.jpg",
-            "output_format": "jpeg",
-        },
+            "output_format": "jpeg"},
     )
     assert isinstance(result, DecomposeLayersResponse)
     assert result.base_image.url == "https://file.runapi.ai/base.jpeg"
+    assert result.usage.generated_images == 2
 
 
 # --- request shapes -------------------------------------------------------
@@ -109,8 +113,7 @@ def test_create_posts_compacted_body():
         model="seedream-v4-text-to-image", prompt="hello world", aspect_ratio="1:1", seed=None
     )
     assert fake.calls == [
-        ("post", "/api/v1/seedream/text_to_image", {"model": "seedream-v4-text-to-image", "prompt": "hello world", "aspect_ratio": "1:1"}),
-    ]
+        ("post", "/api/v1/seedream/text_to_image", {"model": "seedream-v4-text-to-image", "prompt": "hello world", "aspect_ratio": "1:1"})]
     assert isinstance(result, TextToImageResponse)
 
 
@@ -135,10 +138,8 @@ def test_lite_output_format_is_forwarded():
                 "source_image_urls": ["https://cdn.runapi.ai/public/samples/image.jpg"],
                 "aspect_ratio": "1:1",
                 "output_quality": "high",
-                "output_format": "jpeg",
-            },
-        ),
-    ]
+                "output_format": "jpeg"},
+        )]
 
 
 def test_pro_edit_is_forwarded():
@@ -151,9 +152,11 @@ def test_pro_edit_is_forwarded():
         aspect_ratio="3:2",
         output_quality="basic",
         output_format="png",
+        background="transparent",
     )
     assert fake.calls[0][2]["model"] == "seedream-5-pro-edit"
     assert fake.calls[0][2]["output_format"] == "png"
+    assert fake.calls[0][2]["background"] == "transparent"
 
 
 def test_get_fetches_by_id():
@@ -166,7 +169,7 @@ def test_get_fetches_by_id():
 def test_run_narrows_completed_type():
     fake = FakeHttp(
         {"id": "t1", "status": "pending"},
-        {"id": "t1", "status": "completed", "images": [{"url": "https://x/y.png"}]},
+        {"id": "t1", "status": "completed", "usage": {"cost": 0.05}, "images": [{"url": "https://x/y.png"}]},
     )
     client = SeedreamClient(api_key="k", http_client=fake)
     result = client.text_to_image.run(model="seedream-v4-text-to-image", prompt="a serene lake")
@@ -174,82 +177,27 @@ def test_run_narrows_completed_type():
     assert result.images[0].url == "https://x/y.png"
 
 
-# --- validation -----------------------------------------------------------
-
-
-def test_rejects_unknown_model():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="model must be one of:"):
-        client.text_to_image.create(model="nope", prompt="hi there")
-
-
-def test_requires_prompt():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="prompt is required"):
-        client.text_to_image.create(model="seedream-v4-text-to-image")
-
+def test_server_decides_unknown_models_and_params():
+    sent = []
 
-def test_v4_output_count_range():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="output_count must be one of: 1, 2, 3, 4, 5, 6"):
-        client.text_to_image.create(model="seedream-v4-text-to-image", prompt="hi there", output_count=9)
-
-
-def test_non_v4_requires_aspect_ratio_and_quality():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="aspect_ratio is required"):
-        client.text_to_image.create(model="seedream-4.5-text-to-image", prompt="hi there")
-
-
-def test_text_to_image_rejects_source_image_urls():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="source_image_urls is not supported"):
-        client.text_to_image.create(
-            model="seedream-v4-text-to-image", prompt="hi there", source_image_urls=["https://x/a.png"]
-        )
+    def handler(request):
+        body = json.loads(request.content)
+        sent.append(body)
+        if body["output_quality"] == "bad":
+            return httpx.Response(400, json={"error": "output_quality is not supported"})
+        return httpx.Response(200, json={"id": "task_1", "status": "processing"})
 
-
-def test_edit_requires_source_image_urls():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="source_image_urls is required"):
-        client.edit_image.create(model="seedream-v4-edit", prompt="make it pop")
-
-
-def test_lite_prompt_min_length():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="between 3 and 3000"):
-        client.text_to_image.create(model="seedream-5-lite-text-to-image", prompt="hi", aspect_ratio="1:1", output_quality="high")
-
-
-def test_pro_prompt_allows_5000_characters():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    client.text_to_image.create(
-        model="seedream-5-pro-text-to-image",
-        prompt="x" * 5000,
-        aspect_ratio="1:1",
-        output_quality="high",
+    http = HttpClient(
+        ClientOptions(api_key="k", base_url="https://runapi.ai", max_retries=0),
+        transport=httpx.MockTransport(handler),
     )
+    client = SeedreamClient(api_key="k", http_client=http)
+    params = {"model": "seedream-future-text-to-image", "prompt": "a lake", "future_setting": "on"}
 
+    result = client.text_to_image.create(**params, output_quality="ultra")
+    with pytest.raises(ValidationError) as error:
+        client.text_to_image.create(**params, output_quality="bad")
 
-def test_pro_edit_rejects_more_than_ten_source_images():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="source_image_urls must contain between 1 and 10 items"):
-        client.edit_image.create(
-            model="seedream-5-pro-edit",
-            prompt="restyle this image",
-            source_image_urls=["https://cdn.runapi.ai/public/samples/image.jpg"] * 11,
-            aspect_ratio="1:1",
-            output_quality="high",
-        )
-
-
-def test_lite_output_format_enum():
-    client = SeedreamClient(api_key="k", http_client=FakeHttp())
-    with pytest.raises(ValidationError, match="output_format must be one of: png, jpeg"):
-        client.text_to_image.create(
-            model="seedream-5-lite-text-to-image",
-            prompt="hi there",
-            aspect_ratio="1:1",
-            output_quality="high",
-            output_format="webp",
-        )
+    assert sent[0] == {**params, "output_quality": "ultra"}
+    assert result.id == "task_1"
+    assert error.value.message == "output_quality is not supported"

runapi_seedream-0.3.0/src/runapi/seedream/contract_gen.py

@@ -1,222 +0,0 @@
-CONTRACT = {
-    "decompose-layers": {
-        "models": ["seedream-5-pro-layer-decomposition"],
-        "fields_by_model": {
-            "seedream-5-pro-layer-decomposition": {
-                "image_url": {
-                    "required": True
-                },
-                "output_format": {
-                    "enum": ["png", "jpeg"]
-                },
-                "prompt": {
-                    "max": 5000,
-                    "length": True
-                },
-                "size": {
-                    "enum": ["auto", "1K", "1.5K", "2K"]
-                }
-            }
-        }
-    },
-    "edit-image": {
-        "models": ["seedream-4.5-edit", "seedream-5-lite-edit", "seedream-5-pro-edit", "seedream-v4-edit"],
-        "fields_by_model": {
-            "seedream-4.5-edit": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "16:9", "9:16", "2:3", "3:2", "21:9"],
-                    "required": True
-                },
-                "output_count": {
-                    "type": "integer"
-                },
-                "output_quality": {
-                    "enum": ["basic", "high"],
-                    "required": True
-                },
-                "seed": {
-                    "type": "integer"
-                },
-                "source_image_urls": {
-                    "required": True,
-                    "min_items": 1,
-                    "max_items": 14
-                }
-            },
-            "seedream-5-lite-edit": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "16:9", "9:16", "2:3", "3:2", "21:9"],
-                    "required": True
-                },
-                "output_count": {
-                    "type": "integer"
-                },
-                "output_format": {
-                    "enum": ["png", "jpeg"]
-                },
-                "output_quality": {
-                    "enum": ["basic", "high", "ultra"],
-                    "required": True
-                },
-                "seed": {
-                    "type": "integer"
-                },
-                "source_image_urls": {
-                    "required": True,
-                    "min_items": 1,
-                    "max_items": 14
-                }
-            },
-            "seedream-5-pro-edit": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "16:9", "9:16", "2:3", "3:2", "21:9"],
-                    "required": True
-                },
-                "output_format": {
-                    "enum": ["png", "jpeg"]
-                },
-                "output_quality": {
-                    "enum": ["basic", "high"],
-                    "required": True
-                },
-                "prompt": {
-                    "required": True,
-                    "min": 3,
-                    "max": 5000,
-                    "length": True
-                },
-                "source_image_urls": {
-                    "required": True,
-                    "min_items": 1,
-                    "max_items": 10
-                }
-            },
-            "seedream-v4-edit": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "3:2", "2:3", "16:9", "9:16", "21:9"]
-                },
-                "output_count": {
-                    "enum": [1, 2, 3, 4, 5, 6],
-                    "type": "integer"
-                },
-                "output_resolution": {
-                    "enum": ["1k", "2k", "4k"]
-                },
-                "seed": {
-                    "type": "integer"
-                },
-                "source_image_urls": {
-                    "required": True,
-                    "min_items": 1,
-                    "max_items": 10
-                }
-            }
-        },
-        "rules": [{
-            "when": {
-                "model": "seedream-4.5-edit"
-            },
-            "forbidden": ["output_format"]
-        }, {
-            "when": {
-                "model": "seedream-5-pro-edit"
-            },
-            "forbidden": ["output_resolution", "output_count", "seed"]
-        }, {
-            "when": {
-                "model": "seedream-v4-edit"
-            },
-            "forbidden": ["output_format"]
-        }]
-    },
-    "text-to-image": {
-        "models": ["seedream-4.5-text-to-image", "seedream-5-lite-text-to-image", "seedream-5-pro-text-to-image", "seedream-v4-text-to-image"],
-        "fields_by_model": {
-            "seedream-4.5-text-to-image": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "16:9", "9:16", "2:3", "3:2", "21:9"],
-                    "required": True
-                },
-                "output_count": {
-                    "type": "integer"
-                },
-                "output_quality": {
-                    "enum": ["basic", "high"],
-                    "required": True
-                },
-                "seed": {
-                    "type": "integer"
-                }
-            },
-            "seedream-5-lite-text-to-image": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "16:9", "9:16", "2:3", "3:2", "21:9"],
-                    "required": True
-                },
-                "output_count": {
-                    "type": "integer"
-                },
-                "output_format": {
-                    "enum": ["png", "jpeg"]
-                },
-                "output_quality": {
-                    "enum": ["basic", "high", "ultra"],
-                    "required": True
-                },
-                "seed": {
-                    "type": "integer"
-                }
-            },
-            "seedream-5-pro-text-to-image": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "16:9", "9:16", "2:3", "3:2", "21:9"],
-                    "required": True
-                },
-                "output_format": {
-                    "enum": ["png", "jpeg"]
-                },
-                "output_quality": {
-                    "enum": ["basic", "high"],
-                    "required": True
-                },
-                "prompt": {
-                    "required": True,
-                    "min": 3,
-                    "max": 5000,
-                    "length": True
-                }
-            },
-            "seedream-v4-text-to-image": {
-                "aspect_ratio": {
-                    "enum": ["1:1", "4:3", "3:4", "3:2", "2:3", "16:9", "9:16", "21:9"]
-                },
-                "output_count": {
-                    "enum": [1, 2, 3, 4, 5, 6],
-                    "type": "integer"
-                },
-                "output_resolution": {
-                    "enum": ["1k", "2k", "4k"]
-                },
-                "seed": {
-                    "type": "integer"
-                }
-            }
-        },
-        "rules": [{
-            "when": {
-                "model": "seedream-4.5-text-to-image"
-            },
-            "forbidden": ["output_format"]
-        }, {
-            "when": {
-                "model": "seedream-5-pro-text-to-image"
-            },
-            "forbidden": ["output_resolution", "output_count", "seed"]
-        }, {
-            "when": {
-                "model": "seedream-v4-text-to-image"
-            },
-            "forbidden": ["output_format"]
-        }]
-    }
-}

runapi_seedream-0.3.0/src/runapi/seedream/resources/text_to_image.py

@@ -1,105 +0,0 @@
-"""Seedream text-to-image resource."""
-
-from __future__ import annotations
-
-from typing import Any, Dict, Optional
-
-from runapi.core import Resource, ValidationError, RequestOptions
-
-from ..contract_gen import CONTRACT
-from ..types import (
-    LONG_PROMPT_MODELS,
-    MINIMUM_THREE_PROMPT_MODELS,
-    V4_MODELS,
-    CompletedTextToImageResponse,
-    TextToImageResponse,
-)
-
-
-class TextToImage(Resource):
-    """Generate images from text prompts with Seedream models."""
-
-    ENDPOINT = "/api/v1/seedream/text_to_image"
-
-    RESPONSE_CLASS = TextToImageResponse
-    COMPLETED_RESPONSE_CLASS = CompletedTextToImageResponse
-
-    PROMPT_MAX_LENGTH = 3000
-    V4_PROMPT_MAX_LENGTH = 5000
-    PROMPT_MIN_LENGTH_LITE = 3
-
-    def run(self, options: Optional[RequestOptions] = None, **params: Any) -> Any:
-        """Create a text-to-image task and poll until it completes.
-
-        Args:
-            **params: text-to-image parameters (model, ...).
-
-        Returns:
-            The completed (narrowed) text-to-image response.
-        """
-        task = self.create(options=options, **params)
-        return self._poll_until_complete(lambda: self.get(task.id, options=options))
-
-    def create(self, options: Optional[RequestOptions] = None, **params: Any) -> Any:
-        """Create a text-to-image task and return immediately with an id.
-
-        Args:
-            **params: text-to-image parameters (model, ...).
-
-        Returns:
-            The task creation result with an id.
-        """
-        compacted = self._compact_params(params)
-        self._validate_params(compacted)
-        return self._request("post", self.ENDPOINT, body=compacted, options=options)
-
-    def get(self, id: str, options: Optional[RequestOptions] = None) -> Any:
-        """Fetch the current status of a text-to-image task.
-
-        Args:
-            id: The task id returned by ``create``.
-
-        Returns:
-            The current task status.
-        """
-        return self._request("get", f"{self.ENDPOINT}/{id}", options=options)
-
-    def _validate_params(self, params: Dict[str, Any]) -> None:
-        self._validate_contract(CONTRACT["text-to-image"], params)
-
-        model = params.get("model")
-
-        prompt = params.get("prompt")
-        if not (isinstance(prompt, str) and prompt):
-            raise ValidationError("prompt is required")
-        max_length = self.V4_PROMPT_MAX_LENGTH if model in LONG_PROMPT_MODELS else self.PROMPT_MAX_LENGTH
-        if len(prompt) > max_length:
-            raise ValidationError(f"prompt must be at most {max_length} characters")
-        if model in MINIMUM_THREE_PROMPT_MODELS and len(prompt) < self.PROMPT_MIN_LENGTH_LITE:
-            raise ValidationError(
-                f"prompt must be between {self.PROMPT_MIN_LENGTH_LITE} and {max_length} characters"
-            )
-
-        if model in V4_MODELS:
-            self._validate_integer(params, "seed")
-
-        if self._field_present(params, "source_image_urls"):
-            raise ValidationError(f"source_image_urls is not supported for {model}")
-
-    @staticmethod
-    def _field_present(params: Dict[str, Any], key: str) -> bool:
-        value = params.get(key)
-        if value is None:
-            return False
-        if hasattr(value, "__len__"):
-            return len(value) > 0
-        return True
-
-    @staticmethod
-    def _validate_integer(params: Dict[str, Any], key: str) -> None:
-        value = params.get(key)
-        if value is None:
-            return
-        if isinstance(value, int) and not isinstance(value, bool):
-            return
-        raise ValidationError(f"{key} must be an integer")