runwayml 3.4.0__tar.gz → 3.6.0__tar.gz

runwayml-3.6.0/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "3.6.0"
+}

{runwayml-3.4.0 → runwayml-3.6.0}/CHANGELOG.md

@@ -1,5 +1,50 @@
 # Changelog
 
+## 3.6.0 (2025-06-23)
+
+Full Changelog: [v3.5.0...v3.6.0](https://github.com/runwayml/sdk-python/compare/v3.5.0...v3.6.0)
+
+### Features
+
+* **client:** add support for aiohttp ([7d5490c](https://github.com/runwayml/sdk-python/commit/7d5490ca015dd0ff55789889036cc1bd32a58c43))
+
+
+### Bug Fixes
+
+* **tests:** fix: tests which call HTTP endpoints directly with the example parameters ([33301dd](https://github.com/runwayml/sdk-python/commit/33301ddf84cc64581d715d6b8ce1eda48ff66ed5))
+
+
+### Chores
+
+* **readme:** update badges ([147dc95](https://github.com/runwayml/sdk-python/commit/147dc95f9ab390065ab96403a88ca5a87e59f597))
+
+
+### Documentation
+
+* **client:** fix httpx.Timeout documentation reference ([af9fecb](https://github.com/runwayml/sdk-python/commit/af9fecb38a4075fcc8f7687f2ef130ad8c813389))
+
+## 3.5.0 (2025-06-17)
+
+Full Changelog: [v3.4.0...v3.5.0](https://github.com/runwayml/sdk-python/compare/v3.4.0...v3.5.0)
+
+### Features
+
+* **api:** SDK updates, contentModeration for i2v ([1286e32](https://github.com/runwayml/sdk-python/commit/1286e3239f388143404c4b0a61816f5f6282fe1c))
+* **api:** SDK updates, contentModeration for i2v ([8ff202e](https://github.com/runwayml/sdk-python/commit/8ff202ee39dc5856118bf1f3f5537e459a120967))
+
+
+### Bug Fixes
+
+* **client:** correctly parse binary response | stream ([fc9bedb](https://github.com/runwayml/sdk-python/commit/fc9bedb3cbb47849e83a5982ed92a694796e80c7))
+
+
+### Chores
+
+* **ci:** enable for pull requests ([4b439c9](https://github.com/runwayml/sdk-python/commit/4b439c99012918849be4fa64b1ab8d1802cee6e6))
+* **internal:** update conftest.py ([0ff71fd](https://github.com/runwayml/sdk-python/commit/0ff71fd67a421dbff729e98418b6cd72ed0e57b3))
+* **tests:** add tests for httpx client instantiation & proxies ([89e5fbe](https://github.com/runwayml/sdk-python/commit/89e5fbef9e8a64d050252f489b2d76e4c510c663))
+* **tests:** run tests in parallel ([7edbcd1](https://github.com/runwayml/sdk-python/commit/7edbcd1db04ed139e026701fe24501199b4eaffd))
+
 ## 3.4.0 (2025-06-04)
 
 Full Changelog: [v3.3.0...v3.4.0](https://github.com/runwayml/sdk-python/compare/v3.3.0...v3.4.0)

{runwayml-3.4.0 → runwayml-3.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: runwayml
-Version: 3.4.0
+Version: 3.6.0
 Summary: The official Python library for the runwayml API
 Project-URL: Homepage, https://github.com/runwayml/sdk-python
 Project-URL: Repository, https://github.com/runwayml/sdk-python
@@ -27,11 +27,14 @@ Requires-Dist: httpx<1,>=0.23.0
 Requires-Dist: pydantic<3,>=1.9.0
 Requires-Dist: sniffio
 Requires-Dist: typing-extensions<5,>=4.10
+Provides-Extra: aiohttp
+Requires-Dist: aiohttp; extra == 'aiohttp'
+Requires-Dist: httpx-aiohttp>=0.1.6; extra == 'aiohttp'
 Description-Content-Type: text/markdown
 
 # RunwayML Python API library
 
-[![PyPI version](https://img.shields.io/pypi/v/runwayml.svg)](https://pypi.org/project/runwayml/)
+[![PyPI version](https://github.com/runwayml/sdk-python/tree/main/<https://img.shields.io/pypi/v/runwayml.svg?label=pypi%20(stable)>)](https://pypi.org/project/runwayml/)
 
 The RunwayML Python library provides convenient access to the RunwayML REST API from any Python 3.8+
 application. The library includes type definitions for all request params and response fields,
@@ -105,6 +108,43 @@ asyncio.run(main())
 
 Functionality between the synchronous and asynchronous clients is otherwise identical.
 
+### With aiohttp
+
+By default, the async client uses `httpx` for HTTP requests. However, for improved concurrency performance you may also use `aiohttp` as the HTTP backend.
+
+You can enable this by installing `aiohttp`:
+
+```sh
+# install from PyPI
+pip install runwayml[aiohttp]
+```
+
+Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:
+
+```python
+import os
+import asyncio
+from runwayml import DefaultAioHttpClient
+from runwayml import AsyncRunwayML
+
+
+async def main() -> None:
+    async with AsyncRunwayML(
+        api_key=os.environ.get("RUNWAYML_API_SECRET"),  # This is the default and can be omitted
+        http_client=DefaultAioHttpClient(),
+    ) as client:
+        image_to_video = await client.image_to_video.create(
+            model="gen4_turbo",
+            prompt_image="https://example.com/assets/bunny.jpg",
+            ratio="1280:720",
+            prompt_text="The bunny is eating a carrot",
+        )
+        print(image_to_video.id)
+
+
+asyncio.run(main())
+```
+
 ## Using types
 
 Nested request parameters are [TypedDicts](https://docs.python.org/3/library/typing.html#typing.TypedDict). Responses are [Pydantic models](https://docs.pydantic.dev) which also provide helper methods for things like:
@@ -206,7 +246,7 @@ client.with_options(max_retries=5).image_to_video.create(
 ### Timeouts
 
 By default requests time out after 1 minute. You can configure this with a `timeout` option,
-which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/#fine-tuning-the-configuration) object:
+which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/timeouts/#fine-tuning-the-configuration) object:
 
 ```python
 from runwayml import RunwayML

{runwayml-3.4.0 → runwayml-3.6.0}/README.md

@@ -1,6 +1,6 @@
 # RunwayML Python API library
 
-[![PyPI version](https://img.shields.io/pypi/v/runwayml.svg)](https://pypi.org/project/runwayml/)
+[![PyPI version](<https://img.shields.io/pypi/v/runwayml.svg?label=pypi%20(stable)>)](https://pypi.org/project/runwayml/)
 
 The RunwayML Python library provides convenient access to the RunwayML REST API from any Python 3.8+
 application. The library includes type definitions for all request params and response fields,
@@ -74,6 +74,43 @@ asyncio.run(main())
 
 Functionality between the synchronous and asynchronous clients is otherwise identical.
 
+### With aiohttp
+
+By default, the async client uses `httpx` for HTTP requests. However, for improved concurrency performance you may also use `aiohttp` as the HTTP backend.
+
+You can enable this by installing `aiohttp`:
+
+```sh
+# install from PyPI
+pip install runwayml[aiohttp]
+```
+
+Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:
+
+```python
+import os
+import asyncio
+from runwayml import DefaultAioHttpClient
+from runwayml import AsyncRunwayML
+
+
+async def main() -> None:
+    async with AsyncRunwayML(
+        api_key=os.environ.get("RUNWAYML_API_SECRET"),  # This is the default and can be omitted
+        http_client=DefaultAioHttpClient(),
+    ) as client:
+        image_to_video = await client.image_to_video.create(
+            model="gen4_turbo",
+            prompt_image="https://example.com/assets/bunny.jpg",
+            ratio="1280:720",
+            prompt_text="The bunny is eating a carrot",
+        )
+        print(image_to_video.id)
+
+
+asyncio.run(main())
+```
+
 ## Using types
 
 Nested request parameters are [TypedDicts](https://docs.python.org/3/library/typing.html#typing.TypedDict). Responses are [Pydantic models](https://docs.pydantic.dev) which also provide helper methods for things like:
@@ -175,7 +212,7 @@ client.with_options(max_retries=5).image_to_video.create(
 ### Timeouts
 
 By default requests time out after 1 minute. You can configure this with a `timeout` option,
-which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/#fine-tuning-the-configuration) object:
+which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/timeouts/#fine-tuning-the-configuration) object:
 
 ```python
 from runwayml import RunwayML

runwayml-3.6.0/examples/generate_image.py

@@ -0,0 +1,16 @@
+from runwayml import RunwayML, TaskFailedError
+
+client = RunwayML()
+
+image_task = client.text_to_image.create(
+    model="gen4_image",
+    prompt_text="A beautiful sunset over a calm ocean",
+    ratio="1920:1080",
+)
+try:
+    output = image_task.wait_for_task_output()
+except TaskFailedError as e:
+    print('The image failed to generate.')
+    print(e.task_details)
+else:
+    print(output.output[0])  # type: ignore

{runwayml-3.4.0 → runwayml-3.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runwayml"
-version = "3.4.0"
+version = "3.6.0"
 description = "The official Python library for the runwayml API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -37,6 +37,8 @@ classifiers = [
 Homepage = "https://github.com/runwayml/sdk-python"
 Repository = "https://github.com/runwayml/sdk-python"
 
+[project.optional-dependencies]
+aiohttp = ["aiohttp", "httpx_aiohttp>=0.1.6"]
 
 [tool.rye]
 managed = true
@@ -54,6 +56,7 @@ dev-dependencies = [
     "importlib-metadata>=6.7.0",
     "rich>=13.7.1",
     "nest_asyncio==1.6.0",
+    "pytest-xdist>=3.6.1",
 ]
 
 [tool.rye.scripts]
@@ -125,7 +128,7 @@ replacement = '[\1](https://github.com/runwayml/sdk-python/tree/main/\g<2>)'
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
-addopts = "--tb=short"
+addopts = "--tb=short -n auto"
 xfail_strict = true
 asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "session"

{runwayml-3.4.0 → runwayml-3.6.0}/requirements-dev.lock

@@ -10,6 +10,13 @@
 #   universal: false
 
 -e file:.
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.12.8
+    # via httpx-aiohttp
+    # via runwayml
+aiosignal==1.3.2
+    # via aiohttp
 annotated-types==0.6.0
     # via pydantic
 anyio==4.4.0
@@ -17,6 +24,10 @@ anyio==4.4.0
     # via runwayml
 argcomplete==3.1.2
     # via nox
+async-timeout==5.0.1
+    # via aiohttp
+attrs==25.3.0
+    # via aiohttp
 certifi==2023.7.22
     # via httpcore
     # via httpx
@@ -30,18 +41,27 @@ distro==1.8.0
 exceptiongroup==1.2.2
     # via anyio
     # via pytest
+execnet==2.1.1
+    # via pytest-xdist
 filelock==3.12.4
     # via virtualenv
+frozenlist==1.6.2
+    # via aiohttp
+    # via aiosignal
 h11==0.14.0
     # via httpcore
 httpcore==1.0.2
     # via httpx
 httpx==0.28.1
+    # via httpx-aiohttp
     # via respx
     # via runwayml
+httpx-aiohttp==0.1.6
+    # via runwayml
 idna==3.4
     # via anyio
     # via httpx
+    # via yarl
 importlib-metadata==7.0.0
 iniconfig==2.0.0
     # via pytest
@@ -49,6 +69,9 @@ markdown-it-py==3.0.0
     # via rich
 mdurl==0.1.2
     # via markdown-it-py
+multidict==6.4.4
+    # via aiohttp
+    # via yarl
 mypy==1.14.1
 mypy-extensions==1.0.0
     # via mypy
@@ -63,6 +86,9 @@ platformdirs==3.11.0
     # via virtualenv
 pluggy==1.5.0
     # via pytest
+propcache==0.3.1
+    # via aiohttp
+    # via yarl
 pydantic==2.10.3
     # via runwayml
 pydantic-core==2.27.1
@@ -72,7 +98,9 @@ pygments==2.18.0
 pyright==1.1.399
 pytest==8.3.3
     # via pytest-asyncio
+    # via pytest-xdist
 pytest-asyncio==0.24.0
+pytest-xdist==3.7.0
 python-dateutil==2.8.2
     # via time-machine
 pytz==2023.3.post1
@@ -93,6 +121,7 @@ tomli==2.0.2
     # via pytest
 typing-extensions==4.12.2
     # via anyio
+    # via multidict
     # via mypy
     # via pydantic
     # via pydantic-core
@@ -100,5 +129,7 @@ typing-extensions==4.12.2
     # via runwayml
 virtualenv==20.24.5
     # via nox
+yarl==1.20.0
+    # via aiohttp
 zipp==3.17.0
     # via importlib-metadata

{runwayml-3.4.0 → runwayml-3.6.0}/requirements.lock

@@ -10,11 +10,22 @@
 #   universal: false
 
 -e file:.
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.12.8
+    # via httpx-aiohttp
+    # via runwayml
+aiosignal==1.3.2
+    # via aiohttp
 annotated-types==0.6.0
     # via pydantic
 anyio==4.4.0
     # via httpx
     # via runwayml
+async-timeout==5.0.1
+    # via aiohttp
+attrs==25.3.0
+    # via aiohttp
 certifi==2023.7.22
     # via httpcore
     # via httpx
@@ -22,15 +33,28 @@ distro==1.8.0
     # via runwayml
 exceptiongroup==1.2.2
     # via anyio
+frozenlist==1.6.2
+    # via aiohttp
+    # via aiosignal
 h11==0.14.0
     # via httpcore
 httpcore==1.0.2
     # via httpx
 httpx==0.28.1
+    # via httpx-aiohttp
+    # via runwayml
+httpx-aiohttp==0.1.6
     # via runwayml
 idna==3.4
     # via anyio
     # via httpx
+    # via yarl
+multidict==6.4.4
+    # via aiohttp
+    # via yarl
+propcache==0.3.1
+    # via aiohttp
+    # via yarl
 pydantic==2.10.3
     # via runwayml
 pydantic-core==2.27.1
@@ -40,6 +64,9 @@ sniffio==1.3.0
     # via runwayml
 typing-extensions==4.12.2
     # via anyio
+    # via multidict
     # via pydantic
     # via pydantic-core
     # via runwayml
+yarl==1.20.0
+    # via aiohttp

{runwayml-3.4.0 → runwayml-3.6.0}/src/runwayml/__init__.py

@@ -36,7 +36,8 @@ from ._exceptions import (
     UnprocessableEntityError,
     APIResponseValidationError,
 )
-from ._base_client import DefaultHttpxClient, DefaultAsyncHttpxClient
+from .lib.polling import TaskFailedError, TaskTimeoutError
+from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
 from ._utils._logs import setup_logging as _setup_logging
 
 __all__ = [
@@ -78,6 +79,9 @@ __all__ = [
     "DEFAULT_CONNECTION_LIMITS",
     "DefaultHttpxClient",
     "DefaultAsyncHttpxClient",
+    "DefaultAioHttpClient",
+    "TaskFailedError",
+    "TaskTimeoutError",
 ]
 
 if not _t.TYPE_CHECKING:

{runwayml-3.4.0 → runwayml-3.6.0}/src/runwayml/_base_client.py

@@ -1071,7 +1071,14 @@ class SyncAPIClient(BaseClient[httpx.Client, Stream[Any]]):
     ) -> ResponseT:
         origin = get_origin(cast_to) or cast_to
 
-        if inspect.isclass(origin) and issubclass(origin, BaseAPIResponse):
+        if (
+            inspect.isclass(origin)
+            and issubclass(origin, BaseAPIResponse)
+            # we only want to actually return the custom BaseAPIResponse class if we're
+            # returning the raw response, or if we're not streaming SSE, as if we're streaming
+            # SSE then `cast_to` doesn't actively reflect the type we need to parse into
+            and (not stream or bool(response.request.headers.get(RAW_RESPONSE_HEADER)))
+        ):
             if not issubclass(origin, APIResponse):
                 raise TypeError(f"API Response types must subclass {APIResponse}; Received {origin}")
 
@@ -1282,6 +1289,24 @@ class _DefaultAsyncHttpxClient(httpx.AsyncClient):
         super().__init__(**kwargs)
 
 
+try:
+    import httpx_aiohttp
+except ImportError:
+
+    class _DefaultAioHttpClient(httpx.AsyncClient):
+        def __init__(self, **_kwargs: Any) -> None:
+            raise RuntimeError("To use the aiohttp client you must have installed the package with the `aiohttp` extra")
+else:
+
+    class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: Any) -> None:
+            kwargs.setdefault("timeout", DEFAULT_TIMEOUT)
+            kwargs.setdefault("limits", DEFAULT_CONNECTION_LIMITS)
+            kwargs.setdefault("follow_redirects", True)
+
+            super().__init__(**kwargs)
+
+
 if TYPE_CHECKING:
     DefaultAsyncHttpxClient = httpx.AsyncClient
     """An alias to `httpx.AsyncClient` that provides the same defaults that this SDK
@@ -1290,8 +1315,12 @@ if TYPE_CHECKING:
     This is useful because overriding the `http_client` with your own instance of
     `httpx.AsyncClient` will result in httpx's defaults being used, not ours.
     """
+
+    DefaultAioHttpClient = httpx.AsyncClient
+    """An alias to `httpx.AsyncClient` that changes the default HTTP transport to `aiohttp`."""
 else:
     DefaultAsyncHttpxClient = _DefaultAsyncHttpxClient
+    DefaultAioHttpClient = _DefaultAioHttpClient
 
 
 class AsyncHttpxClientWrapper(DefaultAsyncHttpxClient):
@@ -1574,7 +1603,14 @@ class AsyncAPIClient(BaseClient[httpx.AsyncClient, AsyncStream[Any]]):
     ) -> ResponseT:
         origin = get_origin(cast_to) or cast_to
 
-        if inspect.isclass(origin) and issubclass(origin, BaseAPIResponse):
+        if (
+            inspect.isclass(origin)
+            and issubclass(origin, BaseAPIResponse)
+            # we only want to actually return the custom BaseAPIResponse class if we're
+            # returning the raw response, or if we're not streaming SSE, as if we're streaming
+            # SSE then `cast_to` doesn't actively reflect the type we need to parse into
+            and (not stream or bool(response.request.headers.get(RAW_RESPONSE_HEADER)))
+        ):
             if not issubclass(origin, AsyncAPIResponse):
                 raise TypeError(f"API Response types must subclass {AsyncAPIResponse}; Received {origin}")
 

{runwayml-3.4.0 → runwayml-3.6.0}/src/runwayml/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "runwayml"
-__version__ = "3.4.0"  # x-release-please-version
+__version__ = "3.6.0"  # x-release-please-version

runwayml-3.6.0/src/runwayml/lib/polling.py

@@ -0,0 +1,127 @@
+import time
+import random
+from typing import TYPE_CHECKING, Type, Union, TypeVar, cast
+from typing_extensions import ParamSpec
+
+import anyio
+
+from .._models import BaseModel
+from ..types.task_retrieve_response import TaskRetrieveResponse
+
+if TYPE_CHECKING:
+    from .._client import RunwayML, AsyncRunwayML
+
+P = ParamSpec("P")
+T = TypeVar("T")
+
+POLL_TIME = 6
+POLL_JITTER = 3
+
+
+class AwaitableTaskResponseMixin:
+    def wait_for_task_output(self, timeout: Union[float, None] = 60 * 10) -> TaskRetrieveResponse:  # type: ignore[empty-body]
+        """
+        When called, this will block until the task is complete.
+
+        If the task fails or is cancelled, a `TaskFailedError` will be raised.
+
+        Args:
+          timeout: The maximum amount of time to wait for the task to complete in seconds. If not
+              specified, the default timeout is 10 minutes. Will raise a `TaskTimeoutError` if the
+              task does not complete within the timeout.
+
+        Returns:
+          The task details, equivalent to calling `client.tasks.retrieve(task_id)`.
+        """
+        ...
+
+
+class NewTaskCreatedResponse(AwaitableTaskResponseMixin, BaseModel):
+    id: str
+
+
+class AwaitableTaskRetrieveResponse(AwaitableTaskResponseMixin, TaskRetrieveResponse):
+    pass
+
+
+class AsyncAwaitableTaskResponseMixin:
+    async def wait_for_task_output(self, timeout: Union[float, None] = 60 * 10) -> TaskRetrieveResponse:  # type: ignore[empty-body]
+        """
+        When called, this will wait until the task is complete.
+
+        If the task fails or is cancelled, a `TaskFailedError` will be raised.
+
+        Args:
+          timeout: The maximum amount of time to wait for the task to complete in seconds. If not
+              specified, the default timeout is 10 minutes. Will raise a `TaskTimeoutError` if the
+              task does not complete within the timeout. Setting this to `None` will wait
+              indefinitely (disabling the timeout).
+
+        Returns:
+          The task details, equivalent to awaiting `client.tasks.retrieve(task_id)`.
+        """
+        ...
+
+
+class AsyncNewTaskCreatedResponse(AsyncAwaitableTaskResponseMixin, BaseModel):
+    id: str
+
+
+class AsyncAwaitableTaskRetrieveResponse(AsyncAwaitableTaskResponseMixin, TaskRetrieveResponse):
+    pass
+
+
+def create_waitable_resource(base_class: Type[T], client: "RunwayML") -> Type[NewTaskCreatedResponse]:
+    class WithClient(base_class):  # type: ignore[valid-type,misc]
+        id: str
+
+        def wait_for_task_output(self, timeout: Union[float, None] = 60 * 10) -> TaskRetrieveResponse:
+            start_time = time.time()
+            while True:
+                time.sleep(POLL_TIME + random.random() * POLL_JITTER - POLL_JITTER / 2)
+                task_details = client.tasks.retrieve(self.id)
+                if task_details.status == "SUCCEEDED":
+                    return task_details
+                if task_details.status == "FAILED":
+                    raise TaskFailedError(task_details)
+                if timeout is not None and time.time() - start_time > timeout:
+                    raise TaskTimeoutError(task_details)
+
+    WithClient.__name__ = base_class.__name__
+    WithClient.__qualname__ = base_class.__qualname__
+
+    return cast(Type[NewTaskCreatedResponse], WithClient)
+
+
+def create_async_waitable_resource(base_class: Type[T], client: "AsyncRunwayML") -> Type[AsyncNewTaskCreatedResponse]:
+    class WithClient(base_class):  # type: ignore[valid-type,misc]
+        id: str
+
+        async def wait_for_task_output(self, timeout: Union[float, None] = 60 * 10) -> TaskRetrieveResponse:
+            start_time = anyio.current_time()
+            while True:
+                await anyio.sleep(POLL_TIME + random.random() * POLL_JITTER - POLL_JITTER / 2)
+                task_details = await client.tasks.retrieve(self.id)
+                if task_details.status == "SUCCEEDED":
+                    return task_details
+                if task_details.status == "FAILED" or task_details.status == "CANCELLED":
+                    raise TaskFailedError(task_details)
+                if timeout is not None and anyio.current_time() - start_time > timeout:
+                    raise TaskTimeoutError(task_details)
+
+    WithClient.__name__ = base_class.__name__
+    WithClient.__qualname__ = base_class.__qualname__
+
+    return cast(Type[AsyncNewTaskCreatedResponse], WithClient)
+
+
+class TaskFailedError(Exception):
+    def __init__(self, task_details: TaskRetrieveResponse):
+        self.task_details = task_details
+        super().__init__(f"Task failed")
+
+
+class TaskTimeoutError(Exception):
+    def __init__(self, task_details: TaskRetrieveResponse):
+        self.task_details = task_details
+        super().__init__(f"Task timed out")