sarvamai 0.1.5a6__py3-none-any.whl → 0.1.5a7__py3-none-any.whl

sarvamai/text_to_speech/raw_client.py

@@ -1,24 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
-from ..types.text_to_speech_language import TextToSpeechLanguage
-from ..types.text_to_speech_speaker import TextToSpeechSpeaker
-from ..types.speech_sample_rate import SpeechSampleRate
-from ..types.text_to_speech_model import TextToSpeechModel
-from ..core.request_options import RequestOptions
-from ..core.http_response import HttpResponse
-from ..types.text_to_speech_response import TextToSpeechResponse
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
 from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
 from ..errors.bad_request_error import BadRequestError
 from ..errors.forbidden_error import ForbiddenError
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..errors.too_many_requests_error import TooManyRequestsError
 from ..errors.internal_server_error import InternalServerError
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper
-from ..core.http_response import AsyncHttpResponse
+from ..errors.too_many_requests_error import TooManyRequestsError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.speech_sample_rate import SpeechSampleRate
+from ..types.text_to_speech_language import TextToSpeechLanguage
+from ..types.text_to_speech_model import TextToSpeechModel
+from ..types.text_to_speech_response import TextToSpeechResponse
+from ..types.text_to_speech_speaker import TextToSpeechSpeaker
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -97,6 +96,7 @@ class RawTextToSpeechClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "text-to-speech",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "text": text,
@@ -127,58 +127,63 @@ class RawTextToSpeechClient:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 422:
                 raise UnprocessableEntityError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 429:
                 raise TooManyRequestsError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
 class AsyncRawTextToSpeechClient:
@@ -254,6 +259,7 @@ class AsyncRawTextToSpeechClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "text-to-speech",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "text": text,
@@ -284,55 +290,60 @@ class AsyncRawTextToSpeechClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 422:
                 raise UnprocessableEntityError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 429:
                 raise TooManyRequestsError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

sarvamai/types/__init__.py

@@ -1,5 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
+from .audio_data import AudioData
+from .audio_data_encoding import AudioDataEncoding
+from .audio_message import AudioMessage
 from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessage
 from .chat_completion_request_message import (
     ChatCompletionRequestMessage,
@@ -16,9 +21,12 @@ from .create_chat_completion_response import CreateChatCompletionResponse
 from .diarized_entry import DiarizedEntry
 from .diarized_transcript import DiarizedTranscript
 from .error_code import ErrorCode
+from .error_data import ErrorData
 from .error_details import ErrorDetails
 from .error_message import ErrorMessage
+from .events_data import EventsData
 from .finish_reason import FinishReason
+from .format import Format
 from .language_identification_response import LanguageIdentificationResponse
 from .numerals_format import NumeralsFormat
 from .reasoning_effort import ReasoningEffort
@@ -28,6 +36,9 @@ from .speech_sample_rate import SpeechSampleRate
 from .speech_to_text_language import SpeechToTextLanguage
 from .speech_to_text_model import SpeechToTextModel
 from .speech_to_text_response import SpeechToTextResponse
+from .speech_to_text_streaming_response import SpeechToTextStreamingResponse
+from .speech_to_text_streaming_response_data import SpeechToTextStreamingResponseData
+from .speech_to_text_streaming_response_type import SpeechToTextStreamingResponseType
 from .speech_to_text_translate_language import SpeechToTextTranslateLanguage
 from .speech_to_text_translate_model import SpeechToTextTranslateModel
 from .speech_to_text_translate_response import SpeechToTextTranslateResponse
@@ -38,6 +49,8 @@ from .text_to_speech_model import TextToSpeechModel
 from .text_to_speech_response import TextToSpeechResponse
 from .text_to_speech_speaker import TextToSpeechSpeaker
 from .timestamps_model import TimestampsModel
+from .transcription_data import TranscriptionData
+from .transcription_metrics import TranscriptionMetrics
 from .translate_mode import TranslateMode
 from .translate_model import TranslateModel
 from .translate_source_language import TranslateSourceLanguage
@@ -50,6 +63,9 @@ from .transliterate_source_language import TransliterateSourceLanguage
 from .transliteration_response import TransliterationResponse
 
 __all__ = [
+    "AudioData",
+    "AudioDataEncoding",
+    "AudioMessage",
     "ChatCompletionRequestAssistantMessage",
     "ChatCompletionRequestMessage",
     "ChatCompletionRequestMessage_Assistant",
@@ -64,9 +80,12 @@ __all__ = [
     "DiarizedEntry",
     "DiarizedTranscript",
     "ErrorCode",
+    "ErrorData",
     "ErrorDetails",
     "ErrorMessage",
+    "EventsData",
     "FinishReason",
+    "Format",
     "LanguageIdentificationResponse",
     "NumeralsFormat",
     "ReasoningEffort",
@@ -76,6 +95,9 @@ __all__ = [
     "SpeechToTextLanguage",
     "SpeechToTextModel",
     "SpeechToTextResponse",
+    "SpeechToTextStreamingResponse",
+    "SpeechToTextStreamingResponseData",
+    "SpeechToTextStreamingResponseType",
     "SpeechToTextTranslateLanguage",
     "SpeechToTextTranslateModel",
     "SpeechToTextTranslateResponse",
@@ -86,6 +108,8 @@ __all__ = [
     "TextToSpeechResponse",
     "TextToSpeechSpeaker",
     "TimestampsModel",
+    "TranscriptionData",
+    "TranscriptionMetrics",
     "TranslateMode",
     "TranslateModel",
     "TranslateSourceLanguage",

sarvamai/types/audio_data.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .audio_data_encoding import AudioDataEncoding
+
+
+class AudioData(UniversalBaseModel):
+    data: str = pydantic.Field()
+    """
+    Base64 encoded audio data
+    """
+
+    sample_rate: int = pydantic.Field()
+    """
+    Audio sample rate in Hz (16kHz preferred, 8kHz least preferred)
+    """
+
+    encoding: AudioDataEncoding = pydantic.Field()
+    """
+    Audio encoding format
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/audio_data_encoding.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+AudioDataEncoding = typing.Union[typing.Literal["audio/wav", "audio/pcm", "audio/flac", "audio/mp3"], typing.Any]

sarvamai/types/audio_message.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .audio_data import AudioData
+
+
+class AudioMessage(UniversalBaseModel):
+    audio: AudioData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/chat_completion_request_assistant_message.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class ChatCompletionRequestAssistantMessage(UniversalBaseModel):
     content: str = pydantic.Field()

sarvamai/types/chat_completion_request_message.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class ChatCompletionRequestMessage_Assistant(UniversalBaseModel):

sarvamai/types/chat_completion_request_system_message.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class ChatCompletionRequestSystemMessage(UniversalBaseModel):
     content: str = pydantic.Field()

sarvamai/types/chat_completion_request_user_message.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class ChatCompletionRequestUserMessage(UniversalBaseModel):
     content: str = pydantic.Field()

sarvamai/types/chat_completion_response_message.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .role import Role
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
 
 
 class ChatCompletionResponseMessage(UniversalBaseModel):

sarvamai/types/choice.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .finish_reason import FinishReason
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .chat_completion_response_message import ChatCompletionResponseMessage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
+from .finish_reason import FinishReason
 
 
 class Choice(UniversalBaseModel):

sarvamai/types/completion_usage.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class CompletionUsage(UniversalBaseModel):
     completion_tokens: int = pydantic.Field()

sarvamai/types/create_chat_completion_response.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .choice import Choice
 from .completion_usage import CompletionUsage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class CreateChatCompletionResponse(UniversalBaseModel):

sarvamai/types/diarized_entry.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class DiarizedEntry(UniversalBaseModel):
     transcript: str = pydantic.Field()

sarvamai/types/diarized_transcript.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .diarized_entry import DiarizedEntry
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .diarized_entry import DiarizedEntry
 
 
 class DiarizedTranscript(UniversalBaseModel):

sarvamai/types/error_data.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ErrorData(UniversalBaseModel):
+    error: str = pydantic.Field()
+    """
+    Error message
+    """
+
+    code: str = pydantic.Field()
+    """
+    Error code
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/error_details.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .error_code import ErrorCode
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class ErrorDetails(UniversalBaseModel):

sarvamai/types/error_message.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .error_details import ErrorDetails
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .error_details import ErrorDetails
+
 
 class ErrorMessage(UniversalBaseModel):
     error: ErrorDetails = pydantic.Field()

sarvamai/types/events_data.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class EventsData(UniversalBaseModel):
+    event_type: str = pydantic.Field()
+    """
+    Type of event
+    """
+
+    timestamp: dt.datetime = pydantic.Field()
+    """
+    Event timestamp
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/format.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+Format = typing.Union[typing.Literal["wav", "mp3"], typing.Any]

sarvamai/types/language_identification_response.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class LanguageIdentificationResponse(UniversalBaseModel):

sarvamai/types/speech_to_text_response.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
 import pydantic
-from .timestamps_model import TimestampsModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .diarized_transcript import DiarizedTranscript
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .timestamps_model import TimestampsModel
 
 
 class SpeechToTextResponse(UniversalBaseModel):

sarvamai/types/speech_to_text_streaming_response.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .speech_to_text_streaming_response_data import SpeechToTextStreamingResponseData
+from .speech_to_text_streaming_response_type import SpeechToTextStreamingResponseType
+
+
+class SpeechToTextStreamingResponse(UniversalBaseModel):
+    type: SpeechToTextStreamingResponseType
+    data: SpeechToTextStreamingResponseData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/speech_to_text_streaming_response_data.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .error_data import ErrorData
+from .events_data import EventsData
+from .transcription_data import TranscriptionData
+
+SpeechToTextStreamingResponseData = typing.Union[TranscriptionData, ErrorData, EventsData]

sarvamai/types/speech_to_text_streaming_response_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+SpeechToTextStreamingResponseType = typing.Union[typing.Literal["data", "error", "events"], typing.Any]