feat(api): Fix audio transcription response formats

groq · Jun 11, 2024 · 88e3d61 · 88e3d61
1 parent c081730
commit 88e3d61
Show file tree

Hide file tree

Showing 17 changed files with 75 additions and 259 deletions.
diff --git a/.stats.yml b/.stats.yml
@@ -1,2 +1,2 @@
 configured_endpoints: 7
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/groqcloud%2Fgroqcloud-862b73eb4a57968b79e739f345b7f4523b8273477544fb03056c9e7d4230f42d.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/groqcloud%2Fgroqcloud-a4c1214ecaa24ad37fbb3c12b8392787ebe0fd51c5e7e08bdf25d7608dc7900b.yml
diff --git a/api.md b/api.md
@@ -65,24 +65,24 @@ Methods:
 Types:
 
 ```python
-from groq.types.audio import Transcription, TranscriptionCreateResponse
+from groq.types.audio import Transcription
 ```
 
 Methods:
 
-- <code title="post /openai/v1/audio/transcriptions">client.audio.transcriptions.<a href="./src/groq/resources/audio/transcriptions.py">create</a>(\*\*<a href="src/groq/types/audio/transcription_create_params.py">params</a>) -> <a href="./src/groq/types/audio/transcription_create_response.py">TranscriptionCreateResponse</a></code>
+- <code title="post /openai/v1/audio/transcriptions">client.audio.transcriptions.<a href="./src/groq/resources/audio/transcriptions.py">create</a>(\*\*<a href="src/groq/types/audio/transcription_create_params.py">params</a>) -> <a href="./src/groq/types/audio/transcription.py">Transcription</a></code>
 
 ## Translations
 
 Types:
 
 ```python
-from groq.types.audio import Translation, TranslationCreateResponse
+from groq.types.audio import Translation
 ```
 
 Methods:
 
-- <code title="post /openai/v1/audio/translations">client.audio.translations.<a href="./src/groq/resources/audio/translations.py">create</a>(\*\*<a href="src/groq/types/audio/translation_create_params.py">params</a>) -> <a href="./src/groq/types/audio/translation_create_response.py">TranslationCreateResponse</a></code>
+- <code title="post /openai/v1/audio/translations">client.audio.translations.<a href="./src/groq/resources/audio/translations.py">create</a>(\*\*<a href="src/groq/types/audio/translation_create_params.py">params</a>) -> <a href="./src/groq/types/audio/translation.py">Translation</a></code>
 
 # Models
 

diff --git a/src/groq/resources/audio/transcriptions.py b/src/groq/resources/audio/transcriptions.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Any, List, Union, Mapping, cast
+from typing import List, Union, Mapping, cast
 from typing_extensions import Literal
 
 import httpx
@@ -26,7 +26,7 @@
 from ..._base_client import (
     make_request_options,
 )
-from ...types.audio.transcription_create_response import TranscriptionCreateResponse
+from ...types.audio.transcription import Transcription
 
 __all__ = ["Transcriptions", "AsyncTranscriptions"]
 
@@ -56,7 +56,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TranscriptionCreateResponse:
+    ) -> Transcription:
         """
         Transcribes audio into the input language.
 
@@ -115,19 +115,14 @@ def create(
             # sent to the server will contain a `boundary` parameter, e.g.
             # multipart/form-data; boundary=---abc--
             extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
-        return cast(
-            TranscriptionCreateResponse,
-            self._post(
-                "/openai/v1/audio/transcriptions",
-                body=maybe_transform(body, transcription_create_params.TranscriptionCreateParams),
-                files=files,
-                options=make_request_options(
-                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-                ),
-                cast_to=cast(
-                    Any, TranscriptionCreateResponse
-                ),  # Union types cannot be passed in as arguments in the type system
+        return self._post(
+            "/openai/v1/audio/transcriptions",
+            body=maybe_transform(body, transcription_create_params.TranscriptionCreateParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
+            cast_to=Transcription,
         )
 
 
@@ -156,7 +151,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TranscriptionCreateResponse:
+    ) -> Transcription:
         """
         Transcribes audio into the input language.
 
@@ -215,19 +210,14 @@ async def create(
             # sent to the server will contain a `boundary` parameter, e.g.
             # multipart/form-data; boundary=---abc--
             extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
-        return cast(
-            TranscriptionCreateResponse,
-            await self._post(
-                "/openai/v1/audio/transcriptions",
-                body=await async_maybe_transform(body, transcription_create_params.TranscriptionCreateParams),
-                files=files,
-                options=make_request_options(
-                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-                ),
-                cast_to=cast(
-                    Any, TranscriptionCreateResponse
-                ),  # Union types cannot be passed in as arguments in the type system
+        return await self._post(
+            "/openai/v1/audio/transcriptions",
+            body=await async_maybe_transform(body, transcription_create_params.TranscriptionCreateParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
+            cast_to=Transcription,
         )
 
 

diff --git a/src/groq/resources/audio/translations.py b/src/groq/resources/audio/translations.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Union, Mapping, cast
+from typing import Union, Mapping, cast
 from typing_extensions import Literal
 
 import httpx
@@ -26,7 +26,7 @@
 from ..._base_client import (
     make_request_options,
 )
-from ...types.audio.translation_create_response import TranslationCreateResponse
+from ...types.audio.translation import Translation
 
 __all__ = ["Translations", "AsyncTranslations"]
 
@@ -54,7 +54,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TranslationCreateResponse:
+    ) -> Translation:
         """
         Translates audio into English.
 
@@ -100,19 +100,14 @@ def create(
             # sent to the server will contain a `boundary` parameter, e.g.
             # multipart/form-data; boundary=---abc--
             extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
-        return cast(
-            TranslationCreateResponse,
-            self._post(
-                "/openai/v1/audio/translations",
-                body=maybe_transform(body, translation_create_params.TranslationCreateParams),
-                files=files,
-                options=make_request_options(
-                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-                ),
-                cast_to=cast(
-                    Any, TranslationCreateResponse
-                ),  # Union types cannot be passed in as arguments in the type system
+        return self._post(
+            "/openai/v1/audio/translations",
+            body=maybe_transform(body, translation_create_params.TranslationCreateParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
+            cast_to=Translation,
         )
 
 
@@ -139,7 +134,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> TranslationCreateResponse:
+    ) -> Translation:
         """
         Translates audio into English.
 
@@ -185,19 +180,14 @@ async def create(
             # sent to the server will contain a `boundary` parameter, e.g.
             # multipart/form-data; boundary=---abc--
             extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
-        return cast(
-            TranslationCreateResponse,
-            await self._post(
-                "/openai/v1/audio/translations",
-                body=await async_maybe_transform(body, translation_create_params.TranslationCreateParams),
-                files=files,
-                options=make_request_options(
-                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-                ),
-                cast_to=cast(
-                    Any, TranslationCreateResponse
-                ),  # Union types cannot be passed in as arguments in the type system
+        return await self._post(
+            "/openai/v1/audio/translations",
+            body=await async_maybe_transform(body, translation_create_params.TranslationCreateParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
+            cast_to=Translation,
         )
 
 

diff --git a/src/groq/resources/chat/completions.py b/src/groq/resources/chat/completions.py
@@ -157,6 +157,7 @@ def create(
         logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
         max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
         n: Optional[int] | NotGiven = NOT_GIVEN,
+        parallel_tool_calls: Optional[bool] | NotGiven = NOT_GIVEN,
         presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
         response_format: Optional[completion_create_params.ResponseFormat] | NotGiven = NOT_GIVEN,
         seed: Optional[int] | NotGiven = NOT_GIVEN,
@@ -218,6 +219,8 @@ def create(
               the current moment, only n=1 is supported. Other values will result in a 400
               response.
 
+          parallel_tool_calls: Whether to enable parallel function calling during tool use.
+
           presence_penalty: Number between -2.0 and 2.0. Positive values penalize new tokens based on
               whether they appear in the text so far, increasing the model's likelihood to
               talk about new topics.
@@ -296,6 +299,7 @@ def create(
                     "logprobs": logprobs,
                     "max_tokens": max_tokens,
                     "n": n,
+                    "parallel_tool_calls": parallel_tool_calls,
                     "presence_penalty": presence_penalty,
                     "response_format": response_format,
                     "seed": seed,
@@ -442,6 +446,7 @@ async def create(
         logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
         max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
         n: Optional[int] | NotGiven = NOT_GIVEN,
+        parallel_tool_calls: Optional[bool] | NotGiven = NOT_GIVEN,
         presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
         response_format: Optional[completion_create_params.ResponseFormat] | NotGiven = NOT_GIVEN,
         seed: Optional[int] | NotGiven = NOT_GIVEN,
@@ -503,6 +508,8 @@ async def create(
               the current moment, only n=1 is supported. Other values will result in a 400
               response.
 
+          parallel_tool_calls: Whether to enable parallel function calling during tool use.
+
           presence_penalty: Number between -2.0 and 2.0. Positive values penalize new tokens based on
               whether they appear in the text so far, increasing the model's likelihood to
               talk about new topics.
@@ -581,6 +588,7 @@ async def create(
                     "logprobs": logprobs,
                     "max_tokens": max_tokens,
                     "n": n,
+                    "parallel_tool_calls": parallel_tool_calls,
                     "presence_penalty": presence_penalty,
                     "response_format": response_format,
                     "seed": seed,

diff --git a/src/groq/types/audio/__init__.py b/src/groq/types/audio/__init__.py
@@ -6,5 +6,3 @@
 from .transcription import Transcription as Transcription
 from .translation_create_params import TranslationCreateParams as TranslationCreateParams
 from .transcription_create_params import TranscriptionCreateParams as TranscriptionCreateParams
-from .translation_create_response import TranslationCreateResponse as TranslationCreateResponse
-from .transcription_create_response import TranscriptionCreateResponse as TranscriptionCreateResponse
diff --git a/src/groq/types/audio/transcription_create_response.py b/src/groq/types/audio/transcription_create_response.py