runwayml 3.3.0__py3-none-any.whl → 3.5.0__py3-none-any.whl

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/_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/_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/_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/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/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/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/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/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),
         )
 
 

runwayml/resources/video_upscale.py

@@ -0,0 +1,201 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal
+
+import httpx
+
+from ..types import video_upscale_create_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import maybe_transform, async_maybe_transform
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    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.video_upscale_create_response import VideoUpscaleCreateResponse
+
+__all__ = ["VideoUpscaleResource", "AsyncVideoUpscaleResource"]
+
+
+class VideoUpscaleResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> VideoUpscaleResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/runwayml/sdk-python#accessing-raw-response-data-eg-headers
+        """
+        return VideoUpscaleResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> VideoUpscaleResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/runwayml/sdk-python#with_streaming_response
+        """
+        return VideoUpscaleResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        model: Literal["upscale_v1"],
+        video_uri: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> NewTaskCreatedResponse:
+        """This endpoint will start a new task to upscale a video.
+
+        Videos will be upscaled
+        by a factor of 4X, capped at a maximum of 4096px along each side.
+
+        Args:
+          model: The model variant to use.
+
+          video_uri: A HTTPS URL pointing to a video or a data URI containing a video. The video must
+              be less than 4096px on each side. The video duration may not exceed 40 seconds.
+              See [our docs](/assets/inputs#videos) on video inputs for more information.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/v1/video_upscale",
+            body=maybe_transform(
+                {
+                    "model": model,
+                    "video_uri": video_uri,
+                },
+                video_upscale_create_params.VideoUpscaleCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=create_waitable_resource(VideoUpscaleCreateResponse, self._client),
+        )
+
+
+class AsyncVideoUpscaleResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncVideoUpscaleResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/runwayml/sdk-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncVideoUpscaleResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncVideoUpscaleResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/runwayml/sdk-python#with_streaming_response
+        """
+        return AsyncVideoUpscaleResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        model: Literal["upscale_v1"],
+        video_uri: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> AsyncNewTaskCreatedResponse:
+        """This endpoint will start a new task to upscale a video.
+
+        Videos will be upscaled
+        by a factor of 4X, capped at a maximum of 4096px along each side.
+
+        Args:
+          model: The model variant to use.
+
+          video_uri: A HTTPS URL pointing to a video or a data URI containing a video. The video must
+              be less than 4096px on each side. The video duration may not exceed 40 seconds.
+              See [our docs](/assets/inputs#videos) on video inputs for more information.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/v1/video_upscale",
+            body=await async_maybe_transform(
+                {
+                    "model": model,
+                    "video_uri": video_uri,
+                },
+                video_upscale_create_params.VideoUpscaleCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=create_async_waitable_resource(VideoUpscaleCreateResponse, self._client),
+        )
+
+
+class VideoUpscaleResourceWithRawResponse:
+    def __init__(self, video_upscale: VideoUpscaleResource) -> None:
+        self._video_upscale = video_upscale
+
+        self.create = to_raw_response_wrapper(
+            video_upscale.create,
+        )
+
+
+class AsyncVideoUpscaleResourceWithRawResponse:
+    def __init__(self, video_upscale: AsyncVideoUpscaleResource) -> None:
+        self._video_upscale = video_upscale
+
+        self.create = async_to_raw_response_wrapper(
+            video_upscale.create,
+        )
+
+
+class VideoUpscaleResourceWithStreamingResponse:
+    def __init__(self, video_upscale: VideoUpscaleResource) -> None:
+        self._video_upscale = video_upscale
+
+        self.create = to_streamed_response_wrapper(
+            video_upscale.create,
+        )
+
+
+class AsyncVideoUpscaleResourceWithStreamingResponse:
+    def __init__(self, video_upscale: AsyncVideoUpscaleResource) -> None:
+        self._video_upscale = video_upscale
+
+        self.create = async_to_streamed_response_wrapper(
+            video_upscale.create,
+        )

runwayml/types/__init__.py

@@ -4,7 +4,9 @@ from __future__ import annotations
 
 from .task_retrieve_response import TaskRetrieveResponse as TaskRetrieveResponse
 from .text_to_image_create_params import TextToImageCreateParams as TextToImageCreateParams
+from .video_upscale_create_params import VideoUpscaleCreateParams as VideoUpscaleCreateParams
 from .image_to_video_create_params import ImageToVideoCreateParams as ImageToVideoCreateParams
 from .text_to_image_create_response import TextToImageCreateResponse as TextToImageCreateResponse
+from .video_upscale_create_response import VideoUpscaleCreateResponse as VideoUpscaleCreateResponse
 from .image_to_video_create_response import ImageToVideoCreateResponse as ImageToVideoCreateResponse
 from .organization_retrieve_response import OrganizationRetrieveResponse as OrganizationRetrieveResponse

runwayml/types/image_to_video_create_params.py

@@ -7,11 +7,11 @@ from typing_extensions import Literal, Required, Annotated, TypedDict
 
 from .._utils import PropertyInfo
 
-__all__ = ["ImageToVideoCreateParams", "PromptImagePromptImage"]
+__all__ = ["ImageToVideoCreateParams", "PromptImagePromptImage", "ContentModeration"]
 
 
 class ImageToVideoCreateParams(TypedDict, total=False):
-    model: Required[Literal["gen4_turbo", "gen3a_turbo"]]
+    model: Required[Literal["gen3a_turbo", "gen4_turbo"]]
     """The model variant to use."""
 
     prompt_image: Required[Annotated[Union[str, Iterable[PromptImagePromptImage]], PropertyInfo(alias="promptImage")]]
@@ -41,6 +41,9 @@ class ImageToVideoCreateParams(TypedDict, total=False):
     - `768:1280`
     """
 
+    content_moderation: Annotated[ContentModeration, PropertyInfo(alias="contentModeration")]
+    """Settings that affect the behavior of the content moderation system."""
+
     duration: Literal[5, 10]
     """The number of seconds of duration for the output video."""
 
@@ -74,3 +77,11 @@ class PromptImagePromptImage(TypedDict, total=False):
 
     See [our docs](/assets/inputs#images) on image inputs for more information.
     """
+
+
+class ContentModeration(TypedDict, total=False):
+    public_figure_threshold: Annotated[Literal["auto", "low"], PropertyInfo(alias="publicFigureThreshold")]
+    """
+    When set to `low`, the content moderation system will be less strict about
+    preventing generations that include recognizable public figures.
+    """

runwayml/types/organization_retrieve_response.py

@@ -13,11 +13,13 @@ __all__ = [
     "TierModelsGen3aTurbo",
     "TierModelsGen4Image",
     "TierModelsGen4Turbo",
+    "TierModelsUpscaleV1",
     "Usage",
     "UsageModels",
     "UsageModelsGen3aTurbo",
     "UsageModelsGen4Image",
     "UsageModelsGen4Turbo",
+    "UsageModelsUpscaleV1",
 ]
 
 
@@ -45,6 +47,14 @@ class TierModelsGen4Turbo(BaseModel):
     """The maximum number of generations that can be created each day for this model."""
 
 
+class TierModelsUpscaleV1(BaseModel):
+    max_concurrent_generations: int = FieldInfo(alias="maxConcurrentGenerations")
+    """The maximum number of generations that can be run concurrently for this model."""
+
+    max_daily_generations: int = FieldInfo(alias="maxDailyGenerations")
+    """The maximum number of generations that can be created each day for this model."""
+
+
 class TierModels(BaseModel):
     gen3a_turbo: Optional[TierModelsGen3aTurbo] = None
     """Limits associated with the gen3a_turbo model."""
@@ -55,6 +65,9 @@ class TierModels(BaseModel):
     gen4_turbo: Optional[TierModelsGen4Turbo] = None
     """Limits associated with the gen4_turbo model."""
 
+    upscale_v1: Optional[TierModelsUpscaleV1] = None
+    """Limits associated with the upscale_v1 model."""
+
 
 class Tier(BaseModel):
     max_monthly_credit_spend: int = FieldInfo(alias="maxMonthlyCreditSpend")
@@ -79,6 +92,11 @@ class UsageModelsGen4Turbo(BaseModel):
     """The number of generations that have been run for this model in the past day."""
 
 
+class UsageModelsUpscaleV1(BaseModel):
+    daily_generations: int = FieldInfo(alias="dailyGenerations")
+    """The number of generations that have been run for this model in the past day."""
+
+
 class UsageModels(BaseModel):
     gen3a_turbo: Optional[UsageModelsGen3aTurbo] = None
     """Usage data for the gen3a_turbo model."""
@@ -89,6 +107,9 @@ class UsageModels(BaseModel):
     gen4_turbo: Optional[UsageModelsGen4Turbo] = None
     """Usage data for the gen4_turbo model."""
 
+    upscale_v1: Optional[UsageModelsUpscaleV1] = None
+    """Usage data for the upscale_v1 model."""
+
 
 class Usage(BaseModel):
     models: UsageModels

runwayml/types/video_upscale_create_params.py

@@ -0,0 +1,22 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["VideoUpscaleCreateParams"]
+
+
+class VideoUpscaleCreateParams(TypedDict, total=False):
+    model: Required[Literal["upscale_v1"]]
+    """The model variant to use."""
+
+    video_uri: Required[Annotated[str, PropertyInfo(alias="videoUri")]]
+    """A HTTPS URL pointing to a video or a data URI containing a video.
+
+    The video must be less than 4096px on each side. The video duration may not
+    exceed 40 seconds. See [our docs](/assets/inputs#videos) on video inputs for
+    more information.
+    """

runwayml/types/video_upscale_create_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .._models import BaseModel
+
+__all__ = ["VideoUpscaleCreateResponse"]
+
+
+class VideoUpscaleCreateResponse(BaseModel):
+    id: str
+    """The ID of the newly created task."""

{runwayml-3.3.0.dist-info → runwayml-3.5.0.dist-info}/METADATA

@@ -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.dist-info → runwayml-3.5.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-runwayml/__init__.py,sha256=w7Il_Fc71HY9Ebs62Q5VLcsCQGraefXwcOmtzeg4r5I,2587
-runwayml/_base_client.py,sha256=C7nXQufaETXu-KXQeQtb6h6lEMCA-xWQUKWlIh9x0Ro,65094
-runwayml/_client.py,sha256=e2fMzTRERF3s2UDqSfaK7wXl_E9FJy8KjBxUsFvLRBs,17814
+runwayml/__init__.py,sha256=y_OizTn3A4yOML16rBpWDwOU4caMAtiu6Iz02DHETgA,2693
+runwayml/_base_client.py,sha256=Ix0P9AvG5RoBrE4TOAyaAmMipKftcg4eZTPbuI4Ugio,65886
+runwayml/_client.py,sha256=tEooz-jH6JGowMQwrTZE7CKcyqpILEHKfSFf7OevoIU,18513
 runwayml/_compat.py,sha256=VWemUKbj6DDkQ-O4baSpHVLJafotzeXmCQGJugfVTIw,6580
 runwayml/_constants.py,sha256=S14PFzyN9-I31wiV7SmIlL5Ga0MLHxdvegInGdXH7tM,462
 runwayml/_exceptions.py,sha256=p2Q8kywHCVQzArLQL4Ht-HetTBhAvevU6yDvEq7PpIE,3224
@@ -11,7 +11,7 @@ runwayml/_resource.py,sha256=BF-j3xY5eRTKmuTxg8eDhLtLP4MLB1phDh_B6BKipKA,1112
 runwayml/_response.py,sha256=WxjSEXX-j01ZhlSxYyMCVSEKxo20pgy40RA7iyski8M,28800
 runwayml/_streaming.py,sha256=NSVuAgknVQWU1cgZEjQn01IdZKKynb5rOeYp5Lo-OEQ,10108
 runwayml/_types.py,sha256=YL6SdhLq5SHlT644GjzDwOJ_Slyr8QDRCoacOp4trhI,6199
-runwayml/_version.py,sha256=iyQEcKcorFeYuq-S88vke94gaOp6VY42fw2ih8_8G8k,160
+runwayml/_version.py,sha256=TTPBS3eF1AlhzcYEdnHx7EWU5BZA4Oj9bDtQQ7AX6Wo,160
 runwayml/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 runwayml/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 runwayml/_utils/_logs.py,sha256=ZfS5W59hdqEBVV86lNrk28PhvUxtHOzs9JqiLhSu0pI,780
@@ -24,19 +24,23 @@ runwayml/_utils/_transform.py,sha256=n7kskEWz6o__aoNvhFoGVyDoalNe6mJwp-g7BWkdj88
 runwayml/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,4602
 runwayml/_utils/_utils.py,sha256=ts4CiiuNpFiGB6YMdkQRh2SZvYvsl7mAF-JWHCcLDf4,12312
 runwayml/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
-runwayml/resources/__init__.py,sha256=ucZzkzpJ_Wn8QymaDzZyb1MO9YJPGCp39yYRd7tOAok,2114
-runwayml/resources/image_to_video.py,sha256=dS2yTuOx10xSUAhAZ0FmuAEDXqo-WLyWe41aVovR5yo,10120
+runwayml/lib/polling.py,sha256=4fF0gP-h4iR0jvxWifsqtR1iH9vRoKEgGykkaZNT9Ek,4743
+runwayml/resources/__init__.py,sha256=5XJIeBB9D2GqQU-9Ub5AuzRVgLYp1QdLr8Fjh6DCDxM,2643
+runwayml/resources/image_to_video.py,sha256=O6YbnD7QEE_YK5UeRDq8RzWwiuS5nTQSAkCl9RvZe74,10880
 runwayml/resources/organization.py,sha256=XBg5nhkycPU3rllRvf9aaeHuZNtzGDKHlLPrPqDCAsw,5419
-runwayml/resources/tasks.py,sha256=-VT3qetYcaqn4FskekxhN_fCTozMl1GqxGpGwxV8M60,9673
-runwayml/resources/text_to_image.py,sha256=E-Fbu2fL96QFbp_j_hz7fw9iQTQwlqjpzH32K4HwoRg,9958
-runwayml/types/__init__.py,sha256=AHeIAsK4Yw29OpmnNVTsQZkwHszr5ZJq6fYus_RbTis,695
-runwayml/types/image_to_video_create_params.py,sha256=TgoNwKkz_PK_sFBeATqMFkXpBhdx8DSZTLA6hLbMhFM,2352
+runwayml/resources/tasks.py,sha256=mjdBqB1G4u9v3xB_9yn6aIdvsDmawxSNcTENkMpKSms,10146
+runwayml/resources/text_to_image.py,sha256=OJ9oD1Fc5NBGqci6Ox_-M9lFvsYG31fSwSzsr0pwPak,10200
+runwayml/resources/video_upscale.py,sha256=8Mz_g5Swxmgp14jfcfexurUYpPi73q_iU-9D1jOddt0,7691
+runwayml/types/__init__.py,sha256=zXevA88F25dbdsjgpvdlKlvELiv9aamz1hS3xcLbe40,889
+runwayml/types/image_to_video_create_params.py,sha256=6M_xJRx0ws8nQ0a3k3jEICDm-WXJUG9j-j1UIxAAg-s,2869
 runwayml/types/image_to_video_create_response.py,sha256=WvZHbZxxJz8KerRNogzb1RYBrxa1x0iCPDi9-LCpHyE,345
-runwayml/types/organization_retrieve_response.py,sha256=SQxoIUxCnDDBv83A3K7TmNjcAdz_SpHB4MGA9WOJ00w,3553
+runwayml/types/organization_retrieve_response.py,sha256=_7vny0YNmNmREe8jp58ldJbXDdzNAGzDT8V0kK9pbsA,4385
 runwayml/types/task_retrieve_response.py,sha256=v8y2bLxsW6srzScW-B3Akv72q_PI_NQmduGrGRQMHds,2139
 runwayml/types/text_to_image_create_params.py,sha256=I8Dr4UG6VnciQ87zN6qp03FKwlQNnrwyn6cacHLqw20,2794
 runwayml/types/text_to_image_create_response.py,sha256=koMzUg82dYFQPp77wln3UR1z8WO2sHCNMWGgoQ9Id8M,262
-runwayml-3.3.0.dist-info/METADATA,sha256=w0IyAYnBwQCYp_3EaV3HIDkcPbRX84VALDHZOG535L4,13986
-runwayml-3.3.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-runwayml-3.3.0.dist-info/licenses/LICENSE,sha256=baeFj6izBWIm6A5_7N3-WAsy_VYpDF05Dd4zS1zsfZI,11338
-runwayml-3.3.0.dist-info/RECORD,,
+runwayml/types/video_upscale_create_params.py,sha256=Ta3BNQy9aeTUBU5Ui-CMJtF32HeNRqbNpqjAAOKXyks,743
+runwayml/types/video_upscale_create_response.py,sha256=zf-79HbJa68dUHltBiZjVtnW_U6HUI-htmkTm5URBSU,264
+runwayml-3.5.0.dist-info/METADATA,sha256=XVkA49TYYef7Lo64G3DhqxZ_TnLk6sAD-syfEscSpE0,13986
+runwayml-3.5.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+runwayml-3.5.0.dist-info/licenses/LICENSE,sha256=baeFj6izBWIm6A5_7N3-WAsy_VYpDF05Dd4zS1zsfZI,11338
+runwayml-3.5.0.dist-info/RECORD,,