runwayml 3.3.0__tar.gz → 3.5.0__tar.gz

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

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

{runwayml-3.3.0 → runwayml-3.5.0}/CHANGELOG.md

@@ -1,5 +1,35 @@
 # Changelog
 
+## 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)
+
+### Features
+
+* **api:** Add video upscale endpoint ([aac6a70](https://github.com/runwayml/sdk-python/commit/aac6a707ca2925034fa141ceac8a9515958a77c1))
+
 ## 3.3.0 (2025-06-04)
 
 Full Changelog: [v3.2.0...v3.3.0](https://github.com/runwayml/sdk-python/compare/v3.2.0...v3.3.0)

{runwayml-3.3.0 → runwayml-3.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: runwayml
-Version: 3.3.0
+Version: 3.5.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

{runwayml-3.3.0 → runwayml-3.5.0}/api.md

@@ -35,6 +35,18 @@ Methods:
 
 - <code title="post /v1/text_to_image">client.text_to_image.<a href="./src/runwayml/resources/text_to_image.py">create</a>(\*\*<a href="src/runwayml/types/text_to_image_create_params.py">params</a>) -> <a href="./src/runwayml/types/text_to_image_create_response.py">TextToImageCreateResponse</a></code>
 
+# VideoUpscale
+
+Types:
+
+```python
+from runwayml.types import VideoUpscaleCreateResponse
+```
+
+Methods:
+
+- <code title="post /v1/video_upscale">client.video_upscale.<a href="./src/runwayml/resources/video_upscale.py">create</a>(\*\*<a href="src/runwayml/types/video_upscale_create_params.py">params</a>) -> <a href="./src/runwayml/types/video_upscale_create_response.py">VideoUpscaleCreateResponse</a></code>
+
 # Organization
 
 Types:

runwayml-3.5.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.3.0 → runwayml-3.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runwayml"
-version = "3.3.0"
+version = "3.5.0"
 description = "The official Python library for the runwayml API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -54,6 +54,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 +126,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.3.0 → runwayml-3.5.0}/requirements-dev.lock

@@ -30,6 +30,8 @@ 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
 h11==0.14.0
@@ -72,7 +74,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

{runwayml-3.3.0 → runwayml-3.5.0}/src/runwayml/__init__.py

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

{runwayml-3.3.0 → runwayml-3.5.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}")
 
@@ -1574,7 +1581,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.3.0 → runwayml-3.5.0}/src/runwayml/_client.py

@@ -21,7 +21,7 @@ from ._types import (
 )
 from ._utils import is_given, get_async_library
 from ._version import __version__
-from .resources import tasks, organization, text_to_image, image_to_video
+from .resources import tasks, organization, text_to_image, video_upscale, image_to_video
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import RunwayMLError, APIStatusError
 from ._base_client import (
@@ -46,6 +46,7 @@ class RunwayML(SyncAPIClient):
     tasks: tasks.TasksResource
     image_to_video: image_to_video.ImageToVideoResource
     text_to_image: text_to_image.TextToImageResource
+    video_upscale: video_upscale.VideoUpscaleResource
     organization: organization.OrganizationResource
     with_raw_response: RunwayMLWithRawResponse
     with_streaming_response: RunwayMLWithStreamedResponse
@@ -113,6 +114,7 @@ class RunwayML(SyncAPIClient):
         self.tasks = tasks.TasksResource(self)
         self.image_to_video = image_to_video.ImageToVideoResource(self)
         self.text_to_image = text_to_image.TextToImageResource(self)
+        self.video_upscale = video_upscale.VideoUpscaleResource(self)
         self.organization = organization.OrganizationResource(self)
         self.with_raw_response = RunwayMLWithRawResponse(self)
         self.with_streaming_response = RunwayMLWithStreamedResponse(self)
@@ -229,6 +231,7 @@ class AsyncRunwayML(AsyncAPIClient):
     tasks: tasks.AsyncTasksResource
     image_to_video: image_to_video.AsyncImageToVideoResource
     text_to_image: text_to_image.AsyncTextToImageResource
+    video_upscale: video_upscale.AsyncVideoUpscaleResource
     organization: organization.AsyncOrganizationResource
     with_raw_response: AsyncRunwayMLWithRawResponse
     with_streaming_response: AsyncRunwayMLWithStreamedResponse
@@ -296,6 +299,7 @@ class AsyncRunwayML(AsyncAPIClient):
         self.tasks = tasks.AsyncTasksResource(self)
         self.image_to_video = image_to_video.AsyncImageToVideoResource(self)
         self.text_to_image = text_to_image.AsyncTextToImageResource(self)
+        self.video_upscale = video_upscale.AsyncVideoUpscaleResource(self)
         self.organization = organization.AsyncOrganizationResource(self)
         self.with_raw_response = AsyncRunwayMLWithRawResponse(self)
         self.with_streaming_response = AsyncRunwayMLWithStreamedResponse(self)
@@ -413,6 +417,7 @@ class RunwayMLWithRawResponse:
         self.tasks = tasks.TasksResourceWithRawResponse(client.tasks)
         self.image_to_video = image_to_video.ImageToVideoResourceWithRawResponse(client.image_to_video)
         self.text_to_image = text_to_image.TextToImageResourceWithRawResponse(client.text_to_image)
+        self.video_upscale = video_upscale.VideoUpscaleResourceWithRawResponse(client.video_upscale)
         self.organization = organization.OrganizationResourceWithRawResponse(client.organization)
 
 
@@ -421,6 +426,7 @@ class AsyncRunwayMLWithRawResponse:
         self.tasks = tasks.AsyncTasksResourceWithRawResponse(client.tasks)
         self.image_to_video = image_to_video.AsyncImageToVideoResourceWithRawResponse(client.image_to_video)
         self.text_to_image = text_to_image.AsyncTextToImageResourceWithRawResponse(client.text_to_image)
+        self.video_upscale = video_upscale.AsyncVideoUpscaleResourceWithRawResponse(client.video_upscale)
         self.organization = organization.AsyncOrganizationResourceWithRawResponse(client.organization)
 
 
@@ -429,6 +435,7 @@ class RunwayMLWithStreamedResponse:
         self.tasks = tasks.TasksResourceWithStreamingResponse(client.tasks)
         self.image_to_video = image_to_video.ImageToVideoResourceWithStreamingResponse(client.image_to_video)
         self.text_to_image = text_to_image.TextToImageResourceWithStreamingResponse(client.text_to_image)
+        self.video_upscale = video_upscale.VideoUpscaleResourceWithStreamingResponse(client.video_upscale)
         self.organization = organization.OrganizationResourceWithStreamingResponse(client.organization)
 
 
@@ -437,6 +444,7 @@ class AsyncRunwayMLWithStreamedResponse:
         self.tasks = tasks.AsyncTasksResourceWithStreamingResponse(client.tasks)
         self.image_to_video = image_to_video.AsyncImageToVideoResourceWithStreamingResponse(client.image_to_video)
         self.text_to_image = text_to_image.AsyncTextToImageResourceWithStreamingResponse(client.text_to_image)
+        self.video_upscale = video_upscale.AsyncVideoUpscaleResourceWithStreamingResponse(client.video_upscale)
         self.organization = organization.AsyncOrganizationResourceWithStreamingResponse(client.organization)
 
 

{runwayml-3.3.0 → runwayml-3.5.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.3.0"  # x-release-please-version
+__version__ = "3.5.0"  # x-release-please-version

runwayml-3.5.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")

{runwayml-3.3.0 → runwayml-3.5.0}/src/runwayml/resources/__init__.py

@@ -24,6 +24,14 @@ from .text_to_image import (
     TextToImageResourceWithStreamingResponse,
     AsyncTextToImageResourceWithStreamingResponse,
 )
+from .video_upscale import (
+    VideoUpscaleResource,
+    AsyncVideoUpscaleResource,
+    VideoUpscaleResourceWithRawResponse,
+    AsyncVideoUpscaleResourceWithRawResponse,
+    VideoUpscaleResourceWithStreamingResponse,
+    AsyncVideoUpscaleResourceWithStreamingResponse,
+)
 from .image_to_video import (
     ImageToVideoResource,
     AsyncImageToVideoResource,
@@ -52,6 +60,12 @@ __all__ = [
     "AsyncTextToImageResourceWithRawResponse",
     "TextToImageResourceWithStreamingResponse",
     "AsyncTextToImageResourceWithStreamingResponse",
+    "VideoUpscaleResource",
+    "AsyncVideoUpscaleResource",
+    "VideoUpscaleResourceWithRawResponse",
+    "AsyncVideoUpscaleResourceWithRawResponse",
+    "VideoUpscaleResourceWithStreamingResponse",
+    "AsyncVideoUpscaleResourceWithStreamingResponse",
     "OrganizationResource",
     "AsyncOrganizationResource",
     "OrganizationResourceWithRawResponse",

{runwayml-3.3.0 → runwayml-3.5.0}/src/runwayml/resources/image_to_video.py

@@ -18,6 +18,12 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ..lib.polling import (
+    NewTaskCreatedResponse,
+    AsyncNewTaskCreatedResponse,
+    create_waitable_resource,
+    create_async_waitable_resource,
+)
 from .._base_client import make_request_options
 from ..types.image_to_video_create_response import ImageToVideoCreateResponse
 
@@ -47,9 +53,10 @@ class ImageToVideoResource(SyncAPIResource):
     def create(
         self,
         *,
-        model: Literal["gen4_turbo", "gen3a_turbo"],
+        model: Literal["gen3a_turbo", "gen4_turbo"],
         prompt_image: Union[str, Iterable[image_to_video_create_params.PromptImagePromptImage]],
         ratio: Literal["1280:720", "720:1280", "1104:832", "832:1104", "960:960", "1584:672", "1280:768", "768:1280"],
+        content_moderation: image_to_video_create_params.ContentModeration | NotGiven = NOT_GIVEN,
         duration: Literal[5, 10] | NotGiven = NOT_GIVEN,
         prompt_text: str | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
@@ -59,7 +66,7 @@ class ImageToVideoResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ImageToVideoCreateResponse:
+    ) -> NewTaskCreatedResponse:
         """
         This endpoint will start a new task to generate a video from an image prompt.
 
@@ -86,6 +93,8 @@ class ImageToVideoResource(SyncAPIResource):
               - `1280:768`
               - `768:1280`
 
+          content_moderation: Settings that affect the behavior of the content moderation system.
+
           duration: The number of seconds of duration for the output video.
 
           prompt_text: A non-empty string up to 1000 characters (measured in UTF-16 code units). This
@@ -110,6 +119,7 @@ class ImageToVideoResource(SyncAPIResource):
                     "model": model,
                     "prompt_image": prompt_image,
                     "ratio": ratio,
+                    "content_moderation": content_moderation,
                     "duration": duration,
                     "prompt_text": prompt_text,
                     "seed": seed,
@@ -119,7 +129,7 @@ class ImageToVideoResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ImageToVideoCreateResponse,
+            cast_to=create_waitable_resource(ImageToVideoCreateResponse, self._client),
         )
 
 
@@ -146,9 +156,10 @@ class AsyncImageToVideoResource(AsyncAPIResource):
     async def create(
         self,
         *,
-        model: Literal["gen4_turbo", "gen3a_turbo"],
+        model: Literal["gen3a_turbo", "gen4_turbo"],
         prompt_image: Union[str, Iterable[image_to_video_create_params.PromptImagePromptImage]],
         ratio: Literal["1280:720", "720:1280", "1104:832", "832:1104", "960:960", "1584:672", "1280:768", "768:1280"],
+        content_moderation: image_to_video_create_params.ContentModeration | NotGiven = NOT_GIVEN,
         duration: Literal[5, 10] | NotGiven = NOT_GIVEN,
         prompt_text: str | NotGiven = NOT_GIVEN,
         seed: int | NotGiven = NOT_GIVEN,
@@ -158,7 +169,7 @@ class AsyncImageToVideoResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> ImageToVideoCreateResponse:
+    ) -> AsyncNewTaskCreatedResponse:
         """
         This endpoint will start a new task to generate a video from an image prompt.
 
@@ -185,6 +196,8 @@ class AsyncImageToVideoResource(AsyncAPIResource):
               - `1280:768`
               - `768:1280`
 
+          content_moderation: Settings that affect the behavior of the content moderation system.
+
           duration: The number of seconds of duration for the output video.
 
           prompt_text: A non-empty string up to 1000 characters (measured in UTF-16 code units). This
@@ -209,6 +222,7 @@ class AsyncImageToVideoResource(AsyncAPIResource):
                     "model": model,
                     "prompt_image": prompt_image,
                     "ratio": ratio,
+                    "content_moderation": content_moderation,
                     "duration": duration,
                     "prompt_text": prompt_text,
                     "seed": seed,
@@ -218,7 +232,7 @@ class AsyncImageToVideoResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ImageToVideoCreateResponse,
+            cast_to=create_async_waitable_resource(ImageToVideoCreateResponse, self._client),
         )
 
 

{runwayml-3.3.0 → runwayml-3.5.0}/src/runwayml/resources/tasks.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+from typing import cast
+
 import httpx
 
 from .._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
@@ -13,6 +15,12 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ..lib.polling import (
+    AwaitableTaskRetrieveResponse,
+    AsyncAwaitableTaskRetrieveResponse,
+    create_waitable_resource,
+    create_async_waitable_resource,
+)
 from .._base_client import make_request_options
 from ..types.task_retrieve_response import TaskRetrieveResponse
 
@@ -49,7 +57,7 @@ class TasksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TaskRetrieveResponse:
+    ) -> AwaitableTaskRetrieveResponse:
         """Return details about a task.
 
         Consumers of this API should not expect updates
@@ -71,7 +79,9 @@ class TasksResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TaskRetrieveResponse,
+            cast_to=cast(
+                type[AwaitableTaskRetrieveResponse], create_waitable_resource(TaskRetrieveResponse, self._client)
+            ),
         )
 
     def delete(
@@ -144,7 +154,7 @@ class AsyncTasksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TaskRetrieveResponse:
+    ) -> AsyncAwaitableTaskRetrieveResponse:
         """Return details about a task.
 
         Consumers of this API should not expect updates
@@ -166,7 +176,10 @@ class AsyncTasksResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TaskRetrieveResponse,
+            cast_to=cast(
+                type[AsyncAwaitableTaskRetrieveResponse],
+                create_async_waitable_resource(TaskRetrieveResponse, self._client),
+            ),
         )
 
     async def delete(

{runwayml-3.3.0 → runwayml-3.5.0}/src/runwayml/resources/text_to_image.py

@@ -18,6 +18,12 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ..lib.polling import (
+    NewTaskCreatedResponse,
+    AsyncNewTaskCreatedResponse,
+    create_waitable_resource,
+    create_async_waitable_resource,
+)
 from .._base_client import make_request_options
 from ..types.text_to_image_create_response import TextToImageCreateResponse
 
@@ -76,7 +82,7 @@ class TextToImageResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TextToImageCreateResponse:
+    ) -> NewTaskCreatedResponse:
         """
         This endpoint will start a new task to generate images from text.
 
@@ -121,7 +127,7 @@ class TextToImageResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TextToImageCreateResponse,
+            cast_to=create_waitable_resource(TextToImageCreateResponse, self._client),
         )
 
 
@@ -177,7 +183,7 @@ class AsyncTextToImageResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TextToImageCreateResponse:
+    ) -> AsyncNewTaskCreatedResponse:
         """
         This endpoint will start a new task to generate images from text.
 
@@ -222,7 +228,7 @@ class AsyncTextToImageResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=TextToImageCreateResponse,
+            cast_to=create_async_waitable_resource(TextToImageCreateResponse, self._client),
         )