feat(api): api update (#94)

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-15f705a5789a4671a5cba160123f7325eff333b93dab4292e25ee92e2ef15a68.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/luma-ai-karanganesan%2Fluma_ai-7514d44d25b8dd931de982d593ba0d03467acdda95fc8367ef2732e7ad2e8086.yml

src/lumaai/resources/generations/generations.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import Union
 from typing_extensions import Literal
 
 import httpx
@@ -88,10 +89,13 @@ def create(
         *,
         aspect_ratio: Literal["1:1", "16:9", "9:16", "4:3", "3:4", "21:9", "9:21"] | NotGiven = NOT_GIVEN,
         callback_url: str | NotGiven = NOT_GIVEN,
+        duration: Union[Literal["5s", "9s"], str] | NotGiven = NOT_GIVEN,
         generation_type: Literal["video"] | NotGiven = NOT_GIVEN,
         keyframes: generation_create_params.Keyframes | NotGiven = NOT_GIVEN,
         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,
         # 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,
@@ -109,12 +113,18 @@ def create(
               be sent to the callback URL when the generation is dreaming, completed, or
               failed
 
+          duration: The duration of the generation
+
           keyframes: The keyframes of the generation
 
           loop: Whether to loop the video
 
+          model: The model used for the generation
+
           prompt: The prompt of the generation
 
+          resolution: The resolution of the generation
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -129,10 +139,13 @@ def create(
                 {
                     "aspect_ratio": aspect_ratio,
                     "callback_url": callback_url,
+                    "duration": duration,
                     "generation_type": generation_type,
                     "keyframes": keyframes,
                     "loop": loop,
+                    "model": model,
                     "prompt": prompt,
+                    "resolution": resolution,
                 },
                 generation_create_params.GenerationCreateParams,
             ),
@@ -289,10 +302,13 @@ async def create(
         *,
         aspect_ratio: Literal["1:1", "16:9", "9:16", "4:3", "3:4", "21:9", "9:21"] | NotGiven = NOT_GIVEN,
         callback_url: str | NotGiven = NOT_GIVEN,
+        duration: Union[Literal["5s", "9s"], str] | NotGiven = NOT_GIVEN,
         generation_type: Literal["video"] | NotGiven = NOT_GIVEN,
         keyframes: generation_create_params.Keyframes | NotGiven = NOT_GIVEN,
         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,
         # 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,
@@ -310,12 +326,18 @@ async def create(
               be sent to the callback URL when the generation is dreaming, completed, or
               failed
 
+          duration: The duration of the generation
+
           keyframes: The keyframes of the generation
 
           loop: Whether to loop the video
 
+          model: The model used for the generation
+
           prompt: The prompt of the generation
 
+          resolution: The resolution of the generation
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -330,10 +352,13 @@ async def create(
                 {
                     "aspect_ratio": aspect_ratio,
                     "callback_url": callback_url,
+                    "duration": duration,
                     "generation_type": generation_type,
                     "keyframes": keyframes,
                     "loop": loop,
+                    "model": model,
                     "prompt": prompt,
+                    "resolution": resolution,
                 },
                 generation_create_params.GenerationCreateParams,
             ),

src/lumaai/resources/generations/video.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import Union
 from typing_extensions import Literal
 
 import httpx
@@ -51,10 +52,13 @@ def create(
         *,
         aspect_ratio: Literal["1:1", "16:9", "9:16", "4:3", "3:4", "21:9", "9:21"] | NotGiven = NOT_GIVEN,
         callback_url: str | NotGiven = NOT_GIVEN,
+        duration: Union[Literal["5s", "9s"], str] | NotGiven = NOT_GIVEN,
         generation_type: Literal["video"] | NotGiven = NOT_GIVEN,
         keyframes: video_create_params.Keyframes | NotGiven = NOT_GIVEN,
         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,
         # 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,
@@ -72,12 +76,18 @@ def create(
               be sent to the callback URL when the generation is dreaming, completed, or
               failed
 
+          duration: The duration of the generation
+
           keyframes: The keyframes of the generation
 
           loop: Whether to loop the video
 
+          model: The model used for the generation
+
           prompt: The prompt of the generation
 
+          resolution: The resolution of the generation
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -92,10 +102,13 @@ def create(
                 {
                     "aspect_ratio": aspect_ratio,
                     "callback_url": callback_url,
+                    "duration": duration,
                     "generation_type": generation_type,
                     "keyframes": keyframes,
                     "loop": loop,
+                    "model": model,
                     "prompt": prompt,
+                    "resolution": resolution,
                 },
                 video_create_params.VideoCreateParams,
             ),
@@ -131,10 +144,13 @@ async def create(
         *,
         aspect_ratio: Literal["1:1", "16:9", "9:16", "4:3", "3:4", "21:9", "9:21"] | NotGiven = NOT_GIVEN,
         callback_url: str | NotGiven = NOT_GIVEN,
+        duration: Union[Literal["5s", "9s"], str] | NotGiven = NOT_GIVEN,
         generation_type: Literal["video"] | NotGiven = NOT_GIVEN,
         keyframes: video_create_params.Keyframes | NotGiven = NOT_GIVEN,
         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,
         # 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,
@@ -152,12 +168,18 @@ async def create(
               be sent to the callback URL when the generation is dreaming, completed, or
               failed
 
+          duration: The duration of the generation
+
           keyframes: The keyframes of the generation
 
           loop: Whether to loop the video
 
+          model: The model used for the generation
+
           prompt: The prompt of the generation
 
+          resolution: The resolution of the generation
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -172,10 +194,13 @@ async def create(
                 {
                     "aspect_ratio": aspect_ratio,
                     "callback_url": callback_url,
+                    "duration": duration,
                     "generation_type": generation_type,
                     "keyframes": keyframes,
                     "loop": loop,
+                    "model": model,
                     "prompt": prompt,
+                    "resolution": resolution,
                 },
                 video_create_params.VideoCreateParams,
             ),

src/lumaai/types/generation.py

@@ -101,6 +101,9 @@ class RequestGenerationRequest(BaseModel):
     failed
     """
 
+    duration: Union[Literal["5s", "9s"], str, None] = None
+    """The duration of the generation"""
+
     generation_type: Optional[Literal["video"]] = None
 
     keyframes: Optional[RequestGenerationRequestKeyframes] = None
@@ -109,9 +112,15 @@ class RequestGenerationRequest(BaseModel):
     loop: Optional[bool] = None
     """Whether to loop the video"""
 
+    model: Optional[Literal["ray-1-6", "ray-2"]] = None
+    """The model used for the generation"""
+
     prompt: Optional[str] = None
     """The prompt of the generation"""
 
+    resolution: Union[Literal["540p", "720p"], str, None] = None
+    """The resolution of the generation"""
+
 
 class RequestImageGenerationRequestCharacterRefIdentity0(BaseModel):
     images: Optional[List[str]] = None

src/lumaai/types/generation_create_params.py

@@ -28,6 +28,9 @@ class GenerationCreateParams(TypedDict, total=False):
     failed
     """
 
+    duration: Union[Literal["5s", "9s"], str]
+    """The duration of the generation"""
+
     generation_type: Literal["video"]
 
     keyframes: Keyframes
@@ -36,9 +39,15 @@ class GenerationCreateParams(TypedDict, total=False):
     loop: bool
     """Whether to loop the video"""
 
+    model: Literal["ray-1-6", "ray-2"]
+    """The model used for the generation"""
+
     prompt: str
     """The prompt of the generation"""
 
+    resolution: Union[Literal["540p", "720p"], str]
+    """The resolution of the generation"""
+
 
 class KeyframesFrame0GenerationReference(TypedDict, total=False):
     id: Required[str]

src/lumaai/types/generations/video_create_params.py

@@ -28,6 +28,9 @@ class VideoCreateParams(TypedDict, total=False):
     failed
     """
 
+    duration: Union[Literal["5s", "9s"], str]
+    """The duration of the generation"""
+
     generation_type: Literal["video"]
 
     keyframes: Keyframes
@@ -36,9 +39,15 @@ class VideoCreateParams(TypedDict, total=False):
     loop: bool
     """Whether to loop the video"""
 
+    model: Literal["ray-1-6", "ray-2"]
+    """The model used for the generation"""
+
     prompt: str
     """The prompt of the generation"""
 
+    resolution: Union[Literal["540p", "720p"], str]
+    """The resolution of the generation"""
+
 
 class KeyframesFrame0GenerationReference(TypedDict, total=False):
     id: Required[str]

tests/api_resources/generations/test_video.py

@@ -27,6 +27,7 @@ def test_method_create_with_all_params(self, client: LumaAI) -> None:
         video = client.generations.video.create(
             aspect_ratio="1:1",
             callback_url="https://example.com",
+            duration="5s",
             generation_type="video",
             keyframes={
                 "frame0": {
@@ -39,7 +40,9 @@ def test_method_create_with_all_params(self, client: LumaAI) -> None:
                 },
             },
             loop=True,
+            model="ray-1-6",
             prompt="A serene lake surrounded by mountains at sunset",
+            resolution="540p",
         )
         assert_matches_type(Generation, video, path=["response"])
 
@@ -77,6 +80,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncLumaAI) ->
         video = await async_client.generations.video.create(
             aspect_ratio="1:1",
             callback_url="https://example.com",
+            duration="5s",
             generation_type="video",
             keyframes={
                 "frame0": {
@@ -89,7 +93,9 @@ async def test_method_create_with_all_params(self, async_client: AsyncLumaAI) ->
                 },
             },
             loop=True,
+            model="ray-1-6",
             prompt="A serene lake surrounded by mountains at sunset",
+            resolution="540p",
         )
         assert_matches_type(Generation, video, path=["response"])
 

tests/api_resources/test_generations.py

@@ -27,6 +27,7 @@ def test_method_create_with_all_params(self, client: LumaAI) -> None:
         generation = client.generations.create(
             aspect_ratio="1:1",
             callback_url="https://example.com",
+            duration="5s",
             generation_type="video",
             keyframes={
                 "frame0": {
@@ -39,7 +40,9 @@ def test_method_create_with_all_params(self, client: LumaAI) -> None:
                 },
             },
             loop=True,
+            model="ray-1-6",
             prompt="A serene lake surrounded by mountains at sunset",
+            resolution="540p",
         )
         assert_matches_type(Generation, generation, path=["response"])
 
@@ -186,6 +189,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncLumaAI) ->
         generation = await async_client.generations.create(
             aspect_ratio="1:1",
             callback_url="https://example.com",
+            duration="5s",
             generation_type="video",
             keyframes={
                 "frame0": {
@@ -198,7 +202,9 @@ async def test_method_create_with_all_params(self, async_client: AsyncLumaAI) ->
                 },
             },
             loop=True,
+            model="ray-1-6",
             prompt="A serene lake surrounded by mountains at sunset",
+            resolution="540p",
         )
         assert_matches_type(Generation, generation, path=["response"])