runapi-topaz 0.1.0__tar.gz

runapi_topaz-0.1.0/.gitignore

@@ -0,0 +1,29 @@
+# Build artifacts
+dist/
+build/
+*.egg-info/
+*.egg
+
+# Bytecode
+__pycache__/
+*.py[cod]
+
+# Virtual environments
+.venv/
+venv/
+
+# uv
+uv.lock
+
+# Test / type caches
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+.coverage
+htmlcov/
+
+# IDE / OS
+.idea/
+.vscode/
+*.swp
+.DS_Store

runapi_topaz-0.1.0/PKG-INFO

@@ -0,0 +1,86 @@
+Metadata-Version: 2.4
+Name: runapi-topaz
+Version: 0.1.0
+Summary: Topaz image and video upscale client for RunAPI
+Project-URL: Homepage, https://runapi.ai/models/topaz
+Project-URL: Documentation, https://runapi.ai/docs#sdk-topaz
+Author-email: RunAPI <contact@runapi.ai>
+License-Expression: Apache-2.0
+Keywords: ai,runapi,sdk,topaz,upscale-image,upscale-video
+Requires-Python: >=3.9
+Requires-Dist: runapi-core
+Description-Content-Type: text/markdown
+
+# Topaz Python SDK for RunAPI
+
+The Topaz Python SDK is the language-specific package for Topaz on RunAPI. Use it
+for image upscale and video upscale flows when your application needs JSON request
+bodies, task status lookup, and consistent RunAPI errors in Python.
+
+For model details, use https://runapi.ai/models/topaz; for API reference, use
+https://runapi.ai/docs#topaz; for SDK docs, use https://runapi.ai/docs#sdk-topaz.
+
+## Install
+
+```bash
+pip install runapi-topaz
+```
+
+## Quick start
+
+```python
+from runapi.topaz import TopazClient
+
+client = TopazClient()  # reads RUNAPI_API_KEY, or pass api_key="sk-..."
+
+task = client.upscale_image.create(
+    model="topaz-upscale-image",
+    source_image_url="https://example.com/in.jpg",
+    upscale_factor=4,
+)
+status = client.upscale_image.get(task.id)
+
+video = client.upscale_video.create(
+    model="topaz-upscale-video",
+    source_video_url="https://example.com/in.mp4",
+    upscale_factor=2,
+)
+```
+
+Use `create` to submit a task and return quickly, `get` to fetch the latest task
+state, and `run` to create and poll until completion:
+
+```python
+result = client.upscale_image.run(
+    model="topaz-upscale-image",
+    source_image_url="https://example.com/in.jpg",
+    upscale_factor=4,
+)
+print(result.images[0].url)
+```
+
+In web request handlers, prefer `create` plus webhook or later `get` polling so a
+worker is not held open.
+
+RunAPI-generated file URLs are temporary. Download and store generated images and
+videos in your own durable storage within 7 days; do not treat returned URLs as
+long-term assets.
+
+## Language notes
+
+Pass parameters as keyword arguments and catch the `runapi.topaz` error classes
+when building upscaling jobs or scripts. The available resources are
+`upscale_image` and `upscale_video`. Keep `RUNAPI_API_KEY` in the environment or
+your secret manager; never commit API keys or callback secrets.
+
+## Links
+
+- Model page: https://runapi.ai/models/topaz
+- SDK docs: https://runapi.ai/docs#sdk-topaz
+- Product docs: https://runapi.ai/docs#topaz
+- Pricing and rate limits: https://runapi.ai/models/topaz
+- Full catalog: https://runapi.ai/models
+
+## License
+
+Licensed under the Apache License, Version 2.0.

runapi_topaz-0.1.0/README.md

@@ -0,0 +1,73 @@
+# Topaz Python SDK for RunAPI
+
+The Topaz Python SDK is the language-specific package for Topaz on RunAPI. Use it
+for image upscale and video upscale flows when your application needs JSON request
+bodies, task status lookup, and consistent RunAPI errors in Python.
+
+For model details, use https://runapi.ai/models/topaz; for API reference, use
+https://runapi.ai/docs#topaz; for SDK docs, use https://runapi.ai/docs#sdk-topaz.
+
+## Install
+
+```bash
+pip install runapi-topaz
+```
+
+## Quick start
+
+```python
+from runapi.topaz import TopazClient
+
+client = TopazClient()  # reads RUNAPI_API_KEY, or pass api_key="sk-..."
+
+task = client.upscale_image.create(
+    model="topaz-upscale-image",
+    source_image_url="https://example.com/in.jpg",
+    upscale_factor=4,
+)
+status = client.upscale_image.get(task.id)
+
+video = client.upscale_video.create(
+    model="topaz-upscale-video",
+    source_video_url="https://example.com/in.mp4",
+    upscale_factor=2,
+)
+```
+
+Use `create` to submit a task and return quickly, `get` to fetch the latest task
+state, and `run` to create and poll until completion:
+
+```python
+result = client.upscale_image.run(
+    model="topaz-upscale-image",
+    source_image_url="https://example.com/in.jpg",
+    upscale_factor=4,
+)
+print(result.images[0].url)
+```
+
+In web request handlers, prefer `create` plus webhook or later `get` polling so a
+worker is not held open.
+
+RunAPI-generated file URLs are temporary. Download and store generated images and
+videos in your own durable storage within 7 days; do not treat returned URLs as
+long-term assets.
+
+## Language notes
+
+Pass parameters as keyword arguments and catch the `runapi.topaz` error classes
+when building upscaling jobs or scripts. The available resources are
+`upscale_image` and `upscale_video`. Keep `RUNAPI_API_KEY` in the environment or
+your secret manager; never commit API keys or callback secrets.
+
+## Links
+
+- Model page: https://runapi.ai/models/topaz
+- SDK docs: https://runapi.ai/docs#sdk-topaz
+- Product docs: https://runapi.ai/docs#topaz
+- Pricing and rate limits: https://runapi.ai/models/topaz
+- Full catalog: https://runapi.ai/models
+
+## License
+
+Licensed under the Apache License, Version 2.0.

runapi_topaz-0.1.0/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "runapi-topaz"
+version = "0.1.0"
+description = "Topaz image and video upscale client for RunAPI"
+readme = "README.md"
+requires-python = ">=3.9"
+license = "Apache-2.0"
+authors = [{ name = "RunAPI", email = "contact@runapi.ai" }]
+keywords = ["runapi", "topaz", "upscale-image", "upscale-video", "ai", "sdk"]
+dependencies = ["runapi-core"]
+
+[project.urls]
+Homepage = "https://runapi.ai/models/topaz"
+Documentation = "https://runapi.ai/docs#sdk-topaz"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/runapi"]
+
+[tool.uv]
+package = true
+
+[tool.uv.sources]
+runapi-core = { workspace = true }
+
+[dependency-groups]
+dev = ["pytest>=8"]
+
+[tool.runapi]
+slug = "topaz"

runapi_topaz-0.1.0/src/runapi/topaz/__init__.py

@@ -0,0 +1,24 @@
+"""Topaz client for RunAPI."""
+
+from runapi.core import (
+    AuthenticationError,
+    InsufficientCreditsError,
+    NotFoundError,
+    RateLimitError,
+    TaskFailedError,
+    TaskTimeoutError,
+    ValidationError,
+)
+
+from .client import TopazClient
+
+__all__ = [
+    "TopazClient",
+    "AuthenticationError",
+    "RateLimitError",
+    "InsufficientCreditsError",
+    "NotFoundError",
+    "ValidationError",
+    "TaskFailedError",
+    "TaskTimeoutError",
+]

runapi_topaz-0.1.0/src/runapi/topaz/client.py

@@ -0,0 +1,31 @@
+"""Topaz client."""
+
+from __future__ import annotations
+
+from typing import Any, Optional
+
+from runapi.core import ClientOptions, HttpClient, resolve_api_key
+
+from .resources.upscale_image import UpscaleImage
+from .resources.upscale_video import UpscaleVideo
+
+
+class TopazClient:
+    """Topaz image and video upscale client.
+
+    Example::
+
+        client = TopazClient(api_key="sk-...")
+        result = client.upscale_image.run(
+            model="topaz-upscale-image",
+            source_image_url="https://example.com/in.jpg",
+            upscale_factor=4,
+        )
+    """
+
+    def __init__(self, api_key: Optional[str] = None, **options: Any) -> None:
+        resolved_api_key = resolve_api_key(api_key)
+        client_options = ClientOptions(api_key=resolved_api_key, **options)
+        http = client_options.http_client or HttpClient(client_options)
+        self.upscale_image = UpscaleImage(http)
+        self.upscale_video = UpscaleVideo(http)

runapi_topaz-0.1.0/src/runapi/topaz/contract_gen.py

@@ -0,0 +1,31 @@
+CONTRACT = {
+    "upscale-image": {
+        "models": ["topaz-upscale-image"],
+        "fields_by_model": {
+            "topaz-upscale-image": {
+                "source_image_url": {
+                    "required": True
+                },
+                "upscale_factor": {
+                    "enum": [1, 2, 4, 8],
+                    "required": True,
+                    "type": "integer"
+                }
+            }
+        }
+    },
+    "upscale-video": {
+        "models": ["topaz-upscale-video"],
+        "fields_by_model": {
+            "topaz-upscale-video": {
+                "source_video_url": {
+                    "required": True
+                },
+                "upscale_factor": {
+                    "enum": [1, 2, 4],
+                    "type": "integer"
+                }
+            }
+        }
+    }
+}

runapi_topaz-0.1.0/src/runapi/topaz/resources/__init__.py

@@ -0,0 +1,4 @@
+from .upscale_image import UpscaleImage
+from .upscale_video import UpscaleVideo
+
+__all__ = ["UpscaleImage", "UpscaleVideo"]

runapi_topaz-0.1.0/src/runapi/topaz/resources/upscale_image.py

@@ -0,0 +1,58 @@
+"""Topaz upscale-image resource."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from runapi.core import Resource
+
+from ..contract_gen import CONTRACT
+from ..types import (
+    CompletedUpscaleImageResponse,
+    UpscaleImageResponse,
+)
+
+
+class UpscaleImage(Resource):
+    """Upscale images with Topaz."""
+
+    ENDPOINT = "/api/v1/topaz/upscale_image"
+
+    RESPONSE_CLASS = UpscaleImageResponse
+    COMPLETED_RESPONSE_CLASS = CompletedUpscaleImageResponse
+
+    def run(self, **params: Any) -> Any:
+        """Upscale an image and poll until it completes.
+
+        Args:
+            **params: image upscale parameters (model, ...).
+
+        Returns:
+            The completed (narrowed) image upscale response.
+        """
+        task = self.create(**params)
+        return self._poll_until_complete(lambda: self.get(task.id))
+
+    def create(self, **params: Any) -> Any:
+        """Create an image upscale task and return immediately with an id.
+
+        Args:
+            **params: image upscale parameters (model, ...).
+
+        Returns:
+            The task creation result with an id.
+        """
+        compacted = self._compact_params(params)
+        self._validate_contract(CONTRACT["upscale-image"], compacted)
+        return self._request("post", self.ENDPOINT, body=compacted)
+
+    def get(self, id: str) -> Any:
+        """Fetch the current status of an image upscale task.
+
+        Args:
+            id: The task id returned by ``create``.
+
+        Returns:
+            The current task status.
+        """
+        return self._request("get", f"{self.ENDPOINT}/{id}")

runapi_topaz-0.1.0/src/runapi/topaz/resources/upscale_video.py

@@ -0,0 +1,58 @@
+"""Topaz upscale-video resource."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from runapi.core import Resource
+
+from ..contract_gen import CONTRACT
+from ..types import (
+    CompletedUpscaleVideoResponse,
+    UpscaleVideoResponse,
+)
+
+
+class UpscaleVideo(Resource):
+    """Upscale videos with Topaz."""
+
+    ENDPOINT = "/api/v1/topaz/upscale_video"
+
+    RESPONSE_CLASS = UpscaleVideoResponse
+    COMPLETED_RESPONSE_CLASS = CompletedUpscaleVideoResponse
+
+    def run(self, **params: Any) -> Any:
+        """Upscale a video and poll until it completes.
+
+        Args:
+            **params: video upscale parameters (model, ...).
+
+        Returns:
+            The completed (narrowed) video upscale response.
+        """
+        task = self.create(**params)
+        return self._poll_until_complete(lambda: self.get(task.id))
+
+    def create(self, **params: Any) -> Any:
+        """Create a video upscale task and return immediately with an id.
+
+        Args:
+            **params: video upscale parameters (model, ...).
+
+        Returns:
+            The task creation result with an id.
+        """
+        compacted = self._compact_params(params)
+        self._validate_contract(CONTRACT["upscale-video"], compacted)
+        return self._request("post", self.ENDPOINT, body=compacted)
+
+    def get(self, id: str) -> Any:
+        """Fetch the current status of a video upscale task.
+
+        Args:
+            id: The task id returned by ``create``.
+
+        Returns:
+            The current task status.
+        """
+        return self._request("get", f"{self.ENDPOINT}/{id}")

runapi_topaz-0.1.0/src/runapi/topaz/types.py

@@ -0,0 +1,43 @@
+"""Topaz response models."""
+
+from __future__ import annotations
+
+from runapi.core import BaseModel, TaskResponse, optional, required
+
+
+class Image(BaseModel):
+    url = optional(str)
+
+
+class Video(BaseModel):
+    url = optional(str)
+
+
+class UpscaleImageResponse(TaskResponse):
+    """Topaz image upscale task status response."""
+
+    id = required(str)
+    status = optional(str, enum=lambda: TaskResponse.Status.ALL)
+    images = optional([lambda: Image])
+    error = optional(str)
+
+
+class UpscaleVideoResponse(TaskResponse):
+    """Topaz video upscale task status response."""
+
+    id = required(str)
+    status = optional(str, enum=lambda: TaskResponse.Status.ALL)
+    videos = optional([lambda: Video])
+    error = optional(str)
+
+
+class CompletedUpscaleImageResponse(UpscaleImageResponse):
+    """Narrowed response from ``run()`` once polling observes completion."""
+
+    images = required([lambda: Image])
+
+
+class CompletedUpscaleVideoResponse(UpscaleVideoResponse):
+    """Narrowed response from ``run()`` once polling observes completion."""
+
+    videos = required([lambda: Video])

runapi_topaz-0.1.0/tests/test_client.py

@@ -0,0 +1,202 @@
+import pytest
+
+from runapi.core import config
+from runapi.core.errors import AuthenticationError, ValidationError
+from runapi.topaz import TopazClient
+from runapi.topaz.resources.upscale_image import UpscaleImage
+from runapi.topaz.resources.upscale_video import UpscaleVideo
+from runapi.topaz.types import (
+    CompletedUpscaleImageResponse,
+    CompletedUpscaleVideoResponse,
+    UpscaleImageResponse,
+    UpscaleVideoResponse,
+)
+
+
+class FakeHttp:
+    def __init__(self, *responses):
+        self._responses = list(responses)
+        self.calls = []
+
+    def request(self, method, path, body=None, options=None):
+        self.calls.append((method, path, body))
+        if self._responses:
+            return self._responses.pop(0)
+        return {"id": "task_1", "status": "pending"}
+
+
+@pytest.fixture(autouse=True)
+def reset_config(monkeypatch):
+    monkeypatch.delenv("RUNAPI_API_KEY", raising=False)
+    monkeypatch.setattr(config, "api_key", None)
+    yield
+
+
+# --- auth -----------------------------------------------------------------
+
+
+def test_accepts_api_key_parameter():
+    assert isinstance(TopazClient(api_key="k", http_client=FakeHttp()), TopazClient)
+
+
+def test_falls_back_to_global(monkeypatch):
+    monkeypatch.setattr(config, "api_key", "global-key")
+    assert isinstance(TopazClient(http_client=FakeHttp()), TopazClient)
+
+
+def test_falls_back_to_env(monkeypatch):
+    monkeypatch.setenv("RUNAPI_API_KEY", "env-key")
+    assert isinstance(TopazClient(http_client=FakeHttp()), TopazClient)
+
+
+def test_raises_without_api_key():
+    with pytest.raises(AuthenticationError, match="API key is required"):
+        TopazClient()
+
+
+# --- injection / accessors ------------------------------------------------
+
+
+def test_uses_injected_http_client():
+    fake = FakeHttp()
+    client = TopazClient(api_key="k", http_client=fake)
+    assert client.upscale_image._http is fake
+    assert client.upscale_video._http is fake
+
+
+def test_exposes_resource_accessors():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    assert isinstance(client.upscale_image, UpscaleImage)
+    assert isinstance(client.upscale_video, UpscaleVideo)
+
+
+# --- request shapes -------------------------------------------------------
+
+
+def test_create_posts_compacted_body():
+    fake = FakeHttp({"id": "t1", "status": "pending"})
+    client = TopazClient(api_key="k", http_client=fake)
+    result = client.upscale_image.create(
+        model="topaz-upscale-image",
+        source_image_url="https://x/y.jpg",
+        upscale_factor=4,
+        note=None,
+    )
+    assert fake.calls == [
+        (
+            "post",
+            "/api/v1/topaz/upscale_image",
+            {"model": "topaz-upscale-image", "source_image_url": "https://x/y.jpg", "upscale_factor": 4},
+        ),
+    ]
+    assert isinstance(result, UpscaleImageResponse)
+
+
+def test_get_fetches_by_id():
+    fake = FakeHttp({"id": "t1", "status": "processing"})
+    client = TopazClient(api_key="k", http_client=fake)
+    client.upscale_image.get("t1")
+    assert fake.calls == [("get", "/api/v1/topaz/upscale_image/t1", None)]
+
+
+def test_video_create_posts_compacted_body():
+    fake = FakeHttp({"id": "t1", "status": "pending"})
+    client = TopazClient(api_key="k", http_client=fake)
+    result = client.upscale_video.create(
+        model="topaz-upscale-video",
+        source_video_url="https://x/y.mp4",
+        upscale_factor=2,
+    )
+    assert fake.calls == [
+        (
+            "post",
+            "/api/v1/topaz/upscale_video",
+            {"model": "topaz-upscale-video", "source_video_url": "https://x/y.mp4", "upscale_factor": 2},
+        ),
+    ]
+    assert isinstance(result, UpscaleVideoResponse)
+
+
+def test_run_narrows_completed_image_type():
+    fake = FakeHttp(
+        {"id": "t1", "status": "pending"},
+        {"id": "t1", "status": "completed", "images": [{"url": "https://x/y.png"}]},
+    )
+    client = TopazClient(api_key="k", http_client=fake)
+    result = client.upscale_image.run(
+        model="topaz-upscale-image", source_image_url="https://x/y.jpg", upscale_factor=4
+    )
+    assert isinstance(result, CompletedUpscaleImageResponse)
+    assert result.images[0].url == "https://x/y.png"
+
+
+def test_run_narrows_completed_video_type():
+    fake = FakeHttp(
+        {"id": "t1", "status": "pending"},
+        {"id": "t1", "status": "completed", "videos": [{"url": "https://x/y.mp4"}]},
+    )
+    client = TopazClient(api_key="k", http_client=fake)
+    result = client.upscale_video.run(
+        model="topaz-upscale-video", source_video_url="https://x/y.mp4", upscale_factor=2
+    )
+    assert isinstance(result, CompletedUpscaleVideoResponse)
+    assert result.videos[0].url == "https://x/y.mp4"
+
+
+# --- validation -----------------------------------------------------------
+
+
+def test_image_requires_model():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="model must be one of: topaz-upscale-image"):
+        client.upscale_image.create(source_image_url="https://x/y.jpg", upscale_factor=4)
+
+
+def test_image_requires_source_image_url():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="source_image_url is required"):
+        client.upscale_image.create(model="topaz-upscale-image", upscale_factor=4)
+
+
+def test_image_requires_upscale_factor():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="upscale_factor is required"):
+        client.upscale_image.create(model="topaz-upscale-image", source_image_url="https://x/y.jpg")
+
+
+def test_image_rejects_bad_factor():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="upscale_factor must be one of: 1, 2, 4, 8"):
+        client.upscale_image.create(
+            model="topaz-upscale-image", source_image_url="https://x/y.jpg", upscale_factor=3
+        )
+
+
+def test_video_requires_model():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="model must be one of: topaz-upscale-video"):
+        client.upscale_video.create(source_video_url="https://x/y.mp4")
+
+
+def test_video_requires_source_video_url():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="source_video_url is required"):
+        client.upscale_video.create(model="topaz-upscale-video")
+
+
+def test_video_allows_default_upscale_factor():
+    fake = FakeHttp({"id": "t1", "status": "pending"})
+    client = TopazClient(api_key="k", http_client=fake)
+    client.upscale_video.create(
+        model="topaz-upscale-video", source_video_url="https://x/y.mp4"
+    )
+    _, _, body = fake.calls[0]
+    assert "upscale_factor" not in body
+
+
+def test_video_rejects_bad_factor():
+    client = TopazClient(api_key="k", http_client=FakeHttp())
+    with pytest.raises(ValidationError, match="upscale_factor must be one of: 1, 2, 4"):
+        client.upscale_video.create(
+            model="topaz-upscale-video", source_video_url="https://x/y.mp4", upscale_factor=8
+        )