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

sarvamai/errors/service_unavailable_error.py

@@ -1,9 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import typing
+
 from ..core.api_error import ApiError
 from ..types.error_message import ErrorMessage
 
 
 class ServiceUnavailableError(ApiError):
-    def __init__(self, body: ErrorMessage):
-        super().__init__(status_code=503, body=body)
+    def __init__(self, body: ErrorMessage, headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=503, headers=headers, body=body)

sarvamai/errors/too_many_requests_error.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.api_error import ApiError
 import typing
 
+from ..core.api_error import ApiError
+
 
 class TooManyRequestsError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any]):
-        super().__init__(status_code=429, body=body)
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=429, headers=headers, body=body)

sarvamai/errors/unprocessable_entity_error.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.api_error import ApiError
 import typing
 
+from ..core.api_error import ApiError
+
 
 class UnprocessableEntityError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any]):
-        super().__init__(status_code=422, body=body)
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=422, headers=headers, body=body)

sarvamai/requests/__init__.py

@@ -1,5 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
+from .audio_data import AudioDataParams
+from .audio_message import AudioMessageParams
 from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessageParams
 from .chat_completion_request_message import (
     ChatCompletionRequestMessageParams,
@@ -15,18 +19,26 @@ from .completion_usage import CompletionUsageParams
 from .create_chat_completion_response import CreateChatCompletionResponseParams
 from .diarized_entry import DiarizedEntryParams
 from .diarized_transcript import DiarizedTranscriptParams
+from .error_data import ErrorDataParams
 from .error_details import ErrorDetailsParams
 from .error_message import ErrorMessageParams
+from .events_data import EventsDataParams
 from .language_identification_response import LanguageIdentificationResponseParams
 from .speech_to_text_response import SpeechToTextResponseParams
+from .speech_to_text_streaming_response import SpeechToTextStreamingResponseParams
+from .speech_to_text_streaming_response_data import SpeechToTextStreamingResponseDataParams
 from .speech_to_text_translate_response import SpeechToTextTranslateResponseParams
 from .stop_configuration import StopConfigurationParams
 from .text_to_speech_response import TextToSpeechResponseParams
 from .timestamps_model import TimestampsModelParams
+from .transcription_data import TranscriptionDataParams
+from .transcription_metrics import TranscriptionMetricsParams
 from .translation_response import TranslationResponseParams
 from .transliteration_response import TransliterationResponseParams
 
 __all__ = [
+    "AudioDataParams",
+    "AudioMessageParams",
     "ChatCompletionRequestAssistantMessageParams",
     "ChatCompletionRequestMessageParams",
     "ChatCompletionRequestMessage_AssistantParams",
@@ -40,14 +52,20 @@ __all__ = [
     "CreateChatCompletionResponseParams",
     "DiarizedEntryParams",
     "DiarizedTranscriptParams",
+    "ErrorDataParams",
     "ErrorDetailsParams",
     "ErrorMessageParams",
+    "EventsDataParams",
     "LanguageIdentificationResponseParams",
     "SpeechToTextResponseParams",
+    "SpeechToTextStreamingResponseDataParams",
+    "SpeechToTextStreamingResponseParams",
     "SpeechToTextTranslateResponseParams",
     "StopConfigurationParams",
     "TextToSpeechResponseParams",
     "TimestampsModelParams",
+    "TranscriptionDataParams",
+    "TranscriptionMetricsParams",
     "TranslationResponseParams",
     "TransliterationResponseParams",
 ]

sarvamai/requests/audio_data.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing_extensions
+from ..types.audio_data_encoding import AudioDataEncoding
+
+
+class AudioDataParams(typing_extensions.TypedDict):
+    data: str
+    """
+    Base64 encoded audio data
+    """
+
+    sample_rate: int
+    """
+    Audio sample rate in Hz (16kHz preferred, 8kHz least preferred)
+    """
+
+    encoding: AudioDataEncoding
+    """
+    Audio encoding format
+    """

sarvamai/requests/audio_message.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing_extensions
+from .audio_data import AudioDataParams
+
+
+class AudioMessageParams(typing_extensions.TypedDict):
+    audio: AudioDataParams

sarvamai/requests/chat_completion_request_message.py

@@ -1,9 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-import typing_extensions
+
 import typing
 
+import typing_extensions
+
 
 class ChatCompletionRequestMessage_AssistantParams(typing_extensions.TypedDict):
     role: typing.Literal["assistant"]

sarvamai/requests/create_chat_completion_response.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
 import typing
-from .choice import ChoiceParams
+
 import typing_extensions
+from .choice import ChoiceParams
 from .completion_usage import CompletionUsageParams
 
 

sarvamai/requests/diarized_transcript.py

@@ -1,7 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
 import typing
+
+import typing_extensions
 from .diarized_entry import DiarizedEntryParams
 
 

sarvamai/requests/error_data.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing_extensions
+
+
+class ErrorDataParams(typing_extensions.TypedDict):
+    error: str
+    """
+    Error message
+    """
+
+    code: str
+    """
+    Error code
+    """

sarvamai/requests/error_details.py

@@ -2,7 +2,6 @@
 
 import typing_extensions
 from ..types.error_code import ErrorCode
-import typing_extensions
 
 
 class ErrorDetailsParams(typing_extensions.TypedDict):

sarvamai/requests/events_data.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+
+import typing_extensions
+
+
+class EventsDataParams(typing_extensions.TypedDict):
+    event_type: str
+    """
+    Type of event
+    """
+
+    timestamp: dt.datetime
+    """
+    Event timestamp
+    """

sarvamai/requests/language_identification_response.py

@@ -1,7 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing_extensions
-import typing_extensions
 
 
 class LanguageIdentificationResponseParams(typing_extensions.TypedDict):

sarvamai/requests/speech_to_text_response.py

@@ -1,9 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing_extensions
-import typing_extensions
-from .timestamps_model import TimestampsModelParams
 from .diarized_transcript import DiarizedTranscriptParams
+from .timestamps_model import TimestampsModelParams
 
 
 class SpeechToTextResponseParams(typing_extensions.TypedDict):

sarvamai/requests/speech_to_text_streaming_response.py

@@ -0,0 +1,10 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing_extensions
+from ..types.speech_to_text_streaming_response_type import SpeechToTextStreamingResponseType
+from .speech_to_text_streaming_response_data import SpeechToTextStreamingResponseDataParams
+
+
+class SpeechToTextStreamingResponseParams(typing_extensions.TypedDict):
+    type: SpeechToTextStreamingResponseType
+    data: SpeechToTextStreamingResponseDataParams

sarvamai/requests/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 ErrorDataParams
+from .events_data import EventsDataParams
+from .transcription_data import TranscriptionDataParams
+
+SpeechToTextStreamingResponseDataParams = typing.Union[TranscriptionDataParams, ErrorDataParams, EventsDataParams]

sarvamai/requests/speech_to_text_translate_response.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
 import typing_extensions
 from ..types.speech_to_text_translate_language import SpeechToTextTranslateLanguage
 from .diarized_transcript import DiarizedTranscriptParams

sarvamai/requests/text_to_speech_response.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
-import typing_extensions
 import typing
 
+import typing_extensions
+
 
 class TextToSpeechResponseParams(typing_extensions.TypedDict):
     request_id: typing_extensions.NotRequired[str]

sarvamai/requests/timestamps_model.py

@@ -1,8 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing_extensions
 import typing
 
+import typing_extensions
+
 
 class TimestampsModelParams(typing_extensions.TypedDict):
     words: typing.Sequence[str]

sarvamai/requests/transcription_data.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+from .transcription_metrics import TranscriptionMetricsParams
+
+
+class TranscriptionDataParams(typing_extensions.TypedDict):
+    request_id: str
+    """
+    Unique identifier for the request
+    """
+
+    transcript: str
+    """
+    Transcript of the provided speech
+    """
+
+    timestamps: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    """
+    Timestamp information (if available)
+    """
+
+    diarized_transcript: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    """
+    Diarized transcript of the provided speech
+    """
+
+    language_code: typing_extensions.NotRequired[str]
+    """
+    BCP-47 code of detected language
+    """
+
+    metrics: TranscriptionMetricsParams

sarvamai/requests/transcription_metrics.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing_extensions
+
+
+class TranscriptionMetricsParams(typing_extensions.TypedDict):
+    audio_duration: float
+    """
+    Duration of processed audio in seconds
+    """
+
+    processing_latency: float
+    """
+    Processing latency in seconds
+    """

sarvamai/requests/translation_response.py

@@ -1,7 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing_extensions
-import typing_extensions
 
 
 class TranslationResponseParams(typing_extensions.TypedDict):

sarvamai/requests/transliteration_response.py

@@ -1,7 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing_extensions
-import typing_extensions
 
 
 class TransliterationResponseParams(typing_extensions.TypedDict):

sarvamai/speech_to_text/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

sarvamai/speech_to_text/client.py

@@ -1,17 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
-from .raw_client import RawSpeechToTextClient
+
 from .. import core
-from ..types.speech_to_text_model import SpeechToTextModel
-from ..types.speech_to_text_language import SpeechToTextLanguage
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from ..types.speech_to_text_language import SpeechToTextLanguage
+from ..types.speech_to_text_model import SpeechToTextModel
 from ..types.speech_to_text_response import SpeechToTextResponse
 from ..types.speech_to_text_translate_model import SpeechToTextTranslateModel
 from ..types.speech_to_text_translate_response import SpeechToTextTranslateResponse
-from ..core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawSpeechToTextClient
+from .raw_client import AsyncRawSpeechToTextClient, RawSpeechToTextClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -88,10 +87,10 @@ class SpeechToTextClient:
         )
         client.speech_to_text.transcribe()
         """
-        response = self._raw_client.transcribe(
+        _response = self._raw_client.transcribe(
             file=file, model=model, language_code=language_code, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     def translate(
         self,
@@ -144,8 +143,8 @@ class SpeechToTextClient:
         )
         client.speech_to_text.translate()
         """
-        response = self._raw_client.translate(file=file, prompt=prompt, model=model, request_options=request_options)
-        return response.data
+        _response = self._raw_client.translate(file=file, prompt=prompt, model=model, request_options=request_options)
+        return _response.data
 
 
 class AsyncSpeechToTextClient:
@@ -227,10 +226,10 @@ class AsyncSpeechToTextClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.transcribe(
+        _response = await self._raw_client.transcribe(
             file=file, model=model, language_code=language_code, request_options=request_options
         )
-        return response.data
+        return _response.data
 
     async def translate(
         self,
@@ -291,7 +290,7 @@ class AsyncSpeechToTextClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.translate(
+        _response = await self._raw_client.translate(
             file=file, prompt=prompt, model=model, request_options=request_options
         )
-        return response.data
+        return _response.data