feat(api): api update (#111)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 8
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/luma-ai-karanganesan%2Fluma_ai-7514d44d25b8dd931de982d593ba0d03467acdda95fc8367ef2732e7ad2e8086.yml
+configured_endpoints: 9
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/luma-ai-karanganesan%2Fluma_ai-5dbcb84a385ed55016c4d7ded7c0056e6e537b19fd3b626179a2c4c9bac8a90f.yml

api.md

@@ -12,6 +12,7 @@ Methods:
 - <code title="get /generations">client.generations.<a href="./src/lumaai/resources/generations/generations.py">list</a>(\*\*<a href="src/lumaai/types/generation_list_params.py">params</a>) -> <a href="./src/lumaai/types/generation_list_response.py">GenerationListResponse</a></code>
 - <code title="delete /generations/{id}">client.generations.<a href="./src/lumaai/resources/generations/generations.py">delete</a>(id) -> None</code>
 - <code title="get /generations/{id}">client.generations.<a href="./src/lumaai/resources/generations/generations.py">get</a>(id) -> <a href="./src/lumaai/types/generation.py">Generation</a></code>
+- <code title="post /generations/{id}/upscale">client.generations.<a href="./src/lumaai/resources/generations/generations.py">upscale</a>(id, \*\*<a href="src/lumaai/types/generation_upscale_params.py">params</a>) -> <a href="./src/lumaai/types/generation.py">Generation</a></code>
 
 ## CameraMotion
 

src/lumaai/resources/generations/generations.py

@@ -23,7 +23,7 @@
     VideoResourceWithStreamingResponse,
     AsyncVideoResourceWithStreamingResponse,
 )
-from ...types import generation_list_params, generation_create_params
+from ...types import generation_list_params, generation_create_params, generation_upscale_params
 from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
 from ..._utils import (
     maybe_transform,
@@ -95,7 +95,7 @@ def create(
         loop: bool | NotGiven = NOT_GIVEN,
         model: Literal["ray-1-6", "ray-2"] | NotGiven = NOT_GIVEN,
         prompt: str | NotGiven = NOT_GIVEN,
-        resolution: Union[Literal["540p", "720p"], str] | NotGiven = NOT_GIVEN,
+        resolution: Union[Literal["540p", "720p", "1080p", "4k"], str] | NotGiven = NOT_GIVEN,
         # 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,
@@ -264,6 +264,54 @@ def get(
             cast_to=Generation,
         )
 
+    def upscale(
+        self,
+        id: str,
+        *,
+        callback_url: str | NotGiven = NOT_GIVEN,
+        generation_type: Literal["upscale_video"] | NotGiven = NOT_GIVEN,
+        resolution: Union[Literal["540p", "720p", "1080p", "4k"], str] | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> Generation:
+        """
+        Upscale a generation by its ID
+
+        Args:
+          callback_url: The callback URL for the upscale
+
+          resolution: The resolution of the upscale
+
+          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
+        """
+        if not id:
+            raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
+        return self._post(
+            f"/generations/{id}/upscale",
+            body=maybe_transform(
+                {
+                    "callback_url": callback_url,
+                    "generation_type": generation_type,
+                    "resolution": resolution,
+                },
+                generation_upscale_params.GenerationUpscaleParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Generation,
+        )
+
 
 class AsyncGenerationsResource(AsyncAPIResource):
     @cached_property
@@ -308,7 +356,7 @@ async def create(
         loop: bool | NotGiven = NOT_GIVEN,
         model: Literal["ray-1-6", "ray-2"] | NotGiven = NOT_GIVEN,
         prompt: str | NotGiven = NOT_GIVEN,
-        resolution: Union[Literal["540p", "720p"], str] | NotGiven = NOT_GIVEN,
+        resolution: Union[Literal["540p", "720p", "1080p", "4k"], str] | NotGiven = NOT_GIVEN,
         # 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,
@@ -477,6 +525,54 @@ async def get(
             cast_to=Generation,
         )
 
+    async def upscale(
+        self,
+        id: str,
+        *,
+        callback_url: str | NotGiven = NOT_GIVEN,
+        generation_type: Literal["upscale_video"] | NotGiven = NOT_GIVEN,
+        resolution: Union[Literal["540p", "720p", "1080p", "4k"], str] | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> Generation:
+        """
+        Upscale a generation by its ID
+
+        Args:
+          callback_url: The callback URL for the upscale
+
+          resolution: The resolution of the upscale
+
+          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
+        """
+        if not id:
+            raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
+        return await self._post(
+            f"/generations/{id}/upscale",
+            body=await async_maybe_transform(
+                {
+                    "callback_url": callback_url,
+                    "generation_type": generation_type,
+                    "resolution": resolution,
+                },
+                generation_upscale_params.GenerationUpscaleParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Generation,
+        )
+
 
 class GenerationsResourceWithRawResponse:
     def __init__(self, generations: GenerationsResource) -> None:
@@ -494,6 +590,9 @@ def __init__(self, generations: GenerationsResource) -> None:
         self.get = to_raw_response_wrapper(
             generations.get,
         )
+        self.upscale = to_raw_response_wrapper(
+            generations.upscale,
+        )
 
     @cached_property
     def camera_motion(self) -> CameraMotionResourceWithRawResponse:
@@ -524,6 +623,9 @@ def __init__(self, generations: AsyncGenerationsResource) -> None:
         self.get = async_to_raw_response_wrapper(
             generations.get,
         )
+        self.upscale = async_to_raw_response_wrapper(
+            generations.upscale,
+        )
 
     @cached_property
     def camera_motion(self) -> AsyncCameraMotionResourceWithRawResponse:
@@ -554,6 +656,9 @@ def __init__(self, generations: GenerationsResource) -> None:
         self.get = to_streamed_response_wrapper(
             generations.get,
         )
+        self.upscale = to_streamed_response_wrapper(
+            generations.upscale,
+        )
 
     @cached_property
     def camera_motion(self) -> CameraMotionResourceWithStreamingResponse:
@@ -584,6 +689,9 @@ def __init__(self, generations: AsyncGenerationsResource) -> None:
         self.get = async_to_streamed_response_wrapper(
             generations.get,
         )
+        self.upscale = async_to_streamed_response_wrapper(
+            generations.upscale,
+        )
 
     @cached_property
     def camera_motion(self) -> AsyncCameraMotionResourceWithStreamingResponse:

src/lumaai/resources/generations/video.py

@@ -58,7 +58,7 @@ def create(
         loop: bool | NotGiven = NOT_GIVEN,
         model: Literal["ray-1-6", "ray-2"] | NotGiven = NOT_GIVEN,
         prompt: str | NotGiven = NOT_GIVEN,
-        resolution: Union[Literal["540p", "720p"], str] | NotGiven = NOT_GIVEN,
+        resolution: Union[Literal["540p", "720p", "1080p", "4k"], str] | NotGiven = NOT_GIVEN,
         # 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,
@@ -150,7 +150,7 @@ async def create(
         loop: bool | NotGiven = NOT_GIVEN,
         model: Literal["ray-1-6", "ray-2"] | NotGiven = NOT_GIVEN,
         prompt: str | NotGiven = NOT_GIVEN,
-        resolution: Union[Literal["540p", "720p"], str] | NotGiven = NOT_GIVEN,
+        resolution: Union[Literal["540p", "720p", "1080p", "4k"], str] | NotGiven = NOT_GIVEN,
         # 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,

src/lumaai/types/__init__.py

@@ -8,3 +8,4 @@
 from .generation_list_params import GenerationListParams as GenerationListParams
 from .generation_create_params import GenerationCreateParams as GenerationCreateParams
 from .generation_list_response import GenerationListResponse as GenerationListResponse
+from .generation_upscale_params import GenerationUpscaleParams as GenerationUpscaleParams

src/lumaai/types/generation.py

@@ -25,13 +25,17 @@
     "RequestImageGenerationRequestImageRef",
     "RequestImageGenerationRequestModifyImageRef",
     "RequestImageGenerationRequestStyleRef",
+    "RequestUpscaleVideoGenerationRequest",
 ]
 
 
 class Assets(BaseModel):
     image: Optional[str] = None
     """The URL of the image"""
 
+    progress_video: Optional[str] = None
+    """The URL of the progress video"""
+
     video: Optional[str] = None
     """The URL of the video"""
 
@@ -118,7 +122,7 @@ class RequestGenerationRequest(BaseModel):
     prompt: Optional[str] = None
     """The prompt of the generation"""
 
-    resolution: Union[Literal["540p", "720p"], str, None] = None
+    resolution: Union[Literal["540p", "720p", "1080p", "4k"], str, None] = None
     """The resolution of the generation"""
 
 
@@ -181,7 +185,19 @@ class RequestImageGenerationRequest(BaseModel):
     style_ref: Optional[List[RequestImageGenerationRequestStyleRef]] = None
 
 
-Request: TypeAlias = Union[RequestGenerationRequest, RequestImageGenerationRequest]
+class RequestUpscaleVideoGenerationRequest(BaseModel):
+    callback_url: Optional[str] = None
+    """The callback URL for the upscale"""
+
+    generation_type: Optional[Literal["upscale_video"]] = None
+
+    resolution: Union[Literal["540p", "720p", "1080p", "4k"], str, None] = None
+    """The resolution of the upscale"""
+
+
+Request: TypeAlias = Union[
+    RequestGenerationRequest, RequestImageGenerationRequest, RequestUpscaleVideoGenerationRequest
+]
 
 
 class Generation(BaseModel):

src/lumaai/types/generation_create_params.py

@@ -45,7 +45,7 @@ class GenerationCreateParams(TypedDict, total=False):
     prompt: str
     """The prompt of the generation"""
 
-    resolution: Union[Literal["540p", "720p"], str]
+    resolution: Union[Literal["540p", "720p", "1080p", "4k"], str]
     """The resolution of the generation"""
 
 

src/lumaai/types/generation_upscale_params.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Union
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["GenerationUpscaleParams"]
+
+
+class GenerationUpscaleParams(TypedDict, total=False):
+    callback_url: str
+    """The callback URL for the upscale"""
+
+    generation_type: Literal["upscale_video"]
+
+    resolution: Union[Literal["540p", "720p", "1080p", "4k"], str]
+    """The resolution of the upscale"""

src/lumaai/types/generations/video_create_params.py

@@ -45,7 +45,7 @@ class VideoCreateParams(TypedDict, total=False):
     prompt: str
     """The prompt of the generation"""
 
-    resolution: Union[Literal["540p", "720p"], str]
+    resolution: Union[Literal["540p", "720p", "1080p", "4k"], str]
     """The resolution of the generation"""