feat(api): api update (#140)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 9
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/luma-ai-karanganesan%2Fluma_ai-5acd7a1b7a6bd23759fc755404fa2bc8c214c3239058ffe2808b3f1c4a3c521d.yml
+configured_endpoints: 10
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/luma-ai-karanganesan%2Fluma_ai-40360b5fde0fe57b8078d35565a402c5dc9af5cafa897fba042fdceff9d32617.yml

api.md

@@ -11,6 +11,7 @@ Methods:
 - <code title="post /generations">client.generations.<a href="./src/lumaai/resources/generations/generations.py">create</a>(\*\*<a href="src/lumaai/types/generation_create_params.py">params</a>) -> <a href="./src/lumaai/types/generation.py">Generation</a></code>
 - <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="post /generations/{id}/audio">client.generations.<a href="./src/lumaai/resources/generations/generations.py">audio</a>(id, \*\*<a href="src/lumaai/types/generation_audio_params.py">params</a>) -> <a href="./src/lumaai/types/generation.py">Generation</a></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>
 

src/lumaai/resources/generations/generations.py

@@ -23,7 +23,12 @@
     VideoResourceWithStreamingResponse,
     AsyncVideoResourceWithStreamingResponse,
 )
-from ...types import generation_list_params, generation_create_params, generation_upscale_params
+from ...types import (
+    generation_list_params,
+    generation_audio_params,
+    generation_create_params,
+    generation_upscale_params,
+)
 from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
 from ..._utils import (
     maybe_transform,
@@ -231,6 +236,58 @@ def delete(
             cast_to=NoneType,
         )
 
+    def audio(
+        self,
+        id: str,
+        *,
+        callback_url: str | NotGiven = NOT_GIVEN,
+        generation_type: Literal["add_audio"] | NotGiven = NOT_GIVEN,
+        negative_prompt: str | NotGiven = NOT_GIVEN,
+        prompt: 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:
+        """
+        Add audio to a generation by its ID
+
+        Args:
+          callback_url: The callback URL for the audio
+
+          negative_prompt: The negative prompt of the audio
+
+          prompt: The prompt of the audio
+
+          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}/audio",
+            body=maybe_transform(
+                {
+                    "callback_url": callback_url,
+                    "generation_type": generation_type,
+                    "negative_prompt": negative_prompt,
+                    "prompt": prompt,
+                },
+                generation_audio_params.GenerationAudioParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Generation,
+        )
+
     def get(
         self,
         id: str,
@@ -492,6 +549,58 @@ async def delete(
             cast_to=NoneType,
         )
 
+    async def audio(
+        self,
+        id: str,
+        *,
+        callback_url: str | NotGiven = NOT_GIVEN,
+        generation_type: Literal["add_audio"] | NotGiven = NOT_GIVEN,
+        negative_prompt: str | NotGiven = NOT_GIVEN,
+        prompt: 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:
+        """
+        Add audio to a generation by its ID
+
+        Args:
+          callback_url: The callback URL for the audio
+
+          negative_prompt: The negative prompt of the audio
+
+          prompt: The prompt of the audio
+
+          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}/audio",
+            body=await async_maybe_transform(
+                {
+                    "callback_url": callback_url,
+                    "generation_type": generation_type,
+                    "negative_prompt": negative_prompt,
+                    "prompt": prompt,
+                },
+                generation_audio_params.GenerationAudioParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Generation,
+        )
+
     async def get(
         self,
         id: str,
@@ -587,6 +696,9 @@ def __init__(self, generations: GenerationsResource) -> None:
         self.delete = to_raw_response_wrapper(
             generations.delete,
         )
+        self.audio = to_raw_response_wrapper(
+            generations.audio,
+        )
         self.get = to_raw_response_wrapper(
             generations.get,
         )
@@ -620,6 +732,9 @@ def __init__(self, generations: AsyncGenerationsResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             generations.delete,
         )
+        self.audio = async_to_raw_response_wrapper(
+            generations.audio,
+        )
         self.get = async_to_raw_response_wrapper(
             generations.get,
         )
@@ -653,6 +768,9 @@ def __init__(self, generations: GenerationsResource) -> None:
         self.delete = to_streamed_response_wrapper(
             generations.delete,
         )
+        self.audio = to_streamed_response_wrapper(
+            generations.audio,
+        )
         self.get = to_streamed_response_wrapper(
             generations.get,
         )
@@ -686,6 +804,9 @@ def __init__(self, generations: AsyncGenerationsResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             generations.delete,
         )
+        self.audio = async_to_streamed_response_wrapper(
+            generations.audio,
+        )
         self.get = async_to_streamed_response_wrapper(
             generations.get,
         )

src/lumaai/types/__init__.py

@@ -6,6 +6,7 @@
 from .credit_get_response import CreditGetResponse as CreditGetResponse
 from .ping_check_response import PingCheckResponse as PingCheckResponse
 from .generation_list_params import GenerationListParams as GenerationListParams
+from .generation_audio_params import GenerationAudioParams as GenerationAudioParams
 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

@@ -26,6 +26,7 @@
     "RequestImageGenerationRequestModifyImageRef",
     "RequestImageGenerationRequestStyleRef",
     "RequestUpscaleVideoGenerationRequest",
+    "RequestAudioGenerationRequest",
 ]
 
 
@@ -195,8 +196,24 @@ class RequestUpscaleVideoGenerationRequest(BaseModel):
     """The resolution of the upscale"""
 
 
+class RequestAudioGenerationRequest(BaseModel):
+    callback_url: Optional[str] = None
+    """The callback URL for the audio"""
+
+    generation_type: Optional[Literal["add_audio"]] = None
+
+    negative_prompt: Optional[str] = None
+    """The negative prompt of the audio"""
+
+    prompt: Optional[str] = None
+    """The prompt of the audio"""
+
+
 Request: TypeAlias = Union[
-    RequestGenerationRequest, RequestImageGenerationRequest, RequestUpscaleVideoGenerationRequest
+    RequestGenerationRequest,
+    RequestImageGenerationRequest,
+    RequestUpscaleVideoGenerationRequest,
+    RequestAudioGenerationRequest,
 ]
 
 

src/lumaai/types/generation_audio_params.py

@@ -0,0 +1,20 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["GenerationAudioParams"]
+
+
+class GenerationAudioParams(TypedDict, total=False):
+    callback_url: str
+    """The callback URL for the audio"""
+
+    generation_type: Literal["add_audio"]
+
+    negative_prompt: str
+    """The negative prompt of the audio"""
+
+    prompt: str
+    """The prompt of the audio"""

tests/api_resources/test_generations.py

@@ -140,6 +140,55 @@ def test_path_params_delete(self, client: LumaAI) -> None:
                 "",
             )
 
+    @parametrize
+    def test_method_audio(self, client: LumaAI) -> None:
+        generation = client.generations.audio(
+            id="id",
+        )
+        assert_matches_type(Generation, generation, path=["response"])
+
+    @parametrize
+    def test_method_audio_with_all_params(self, client: LumaAI) -> None:
+        generation = client.generations.audio(
+            id="id",
+            callback_url="https://example.com",
+            generation_type="add_audio",
+            negative_prompt="negative_prompt",
+            prompt="prompt",
+        )
+        assert_matches_type(Generation, generation, path=["response"])
+
+    @parametrize
+    def test_raw_response_audio(self, client: LumaAI) -> None:
+        response = client.generations.with_raw_response.audio(
+            id="id",
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        generation = response.parse()
+        assert_matches_type(Generation, generation, path=["response"])
+
+    @parametrize
+    def test_streaming_response_audio(self, client: LumaAI) -> None:
+        with client.generations.with_streaming_response.audio(
+            id="id",
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            generation = response.parse()
+            assert_matches_type(Generation, generation, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @parametrize
+    def test_path_params_audio(self, client: LumaAI) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `id` but received ''"):
+            client.generations.with_raw_response.audio(
+                id="",
+            )
+
     @parametrize
     def test_method_get(self, client: LumaAI) -> None:
         generation = client.generations.get(
@@ -350,6 +399,55 @@ async def test_path_params_delete(self, async_client: AsyncLumaAI) -> None:
                 "",
             )
 
+    @parametrize
+    async def test_method_audio(self, async_client: AsyncLumaAI) -> None:
+        generation = await async_client.generations.audio(
+            id="id",
+        )
+        assert_matches_type(Generation, generation, path=["response"])
+
+    @parametrize
+    async def test_method_audio_with_all_params(self, async_client: AsyncLumaAI) -> None:
+        generation = await async_client.generations.audio(
+            id="id",
+            callback_url="https://example.com",
+            generation_type="add_audio",
+            negative_prompt="negative_prompt",
+            prompt="prompt",
+        )
+        assert_matches_type(Generation, generation, path=["response"])
+
+    @parametrize
+    async def test_raw_response_audio(self, async_client: AsyncLumaAI) -> None:
+        response = await async_client.generations.with_raw_response.audio(
+            id="id",
+        )
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        generation = await response.parse()
+        assert_matches_type(Generation, generation, path=["response"])
+
+    @parametrize
+    async def test_streaming_response_audio(self, async_client: AsyncLumaAI) -> None:
+        async with async_client.generations.with_streaming_response.audio(
+            id="id",
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            generation = await response.parse()
+            assert_matches_type(Generation, generation, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
+    @parametrize
+    async def test_path_params_audio(self, async_client: AsyncLumaAI) -> None:
+        with pytest.raises(ValueError, match=r"Expected a non-empty value for `id` but received ''"):
+            await async_client.generations.with_raw_response.audio(
+                id="",
+            )
+
     @parametrize
     async def test_method_get(self, async_client: AsyncLumaAI) -> None:
         generation = await async_client.generations.get(