sarvamai 0.1.7__py3-none-any.whl → 0.1.8a0__py3-none-any.whl

sarvamai/__init__.py

@@ -5,6 +5,8 @@
 from .types import (
     AudioData,
     AudioMessage,
+    AudioOutput,
+    AudioOutputData,
     ChatCompletionRequestAssistantMessage,
     ChatCompletionRequestMessage,
     ChatCompletionRequestMessage_Assistant,
@@ -14,6 +16,7 @@ from .types import (
     ChatCompletionRequestUserMessage,
     ChatCompletionResponseMessage,
     Choice,
+    CloseConnection,
     CompletionUsage,
     ConfigMessage,
     CreateChatCompletionResponse,
@@ -23,15 +26,26 @@ from .types import (
     ErrorData,
     ErrorDetails,
     ErrorMessage,
+    ErrorResponse,
+    ErrorResponseData,
     EventsData,
     FinishReason,
     Format,
+    InitializeConnection,
+    InitializeConnectionData,
+    InitializeConnectionDataModel,
+    InitializeConnectionDataOutputAudioBitrate,
+    InitializeConnectionDataSpeaker,
+    InitializeConnectionDataTargetLanguageCode,
     LanguageIdentificationResponse,
     NumeralsFormat,
+    PingSignal,
     ReasoningEffort,
     ResponseType,
     Role,
     SarvamModelIds,
+    SendText,
+    SendTextData,
     SpeechSampleRate,
     SpeechToTextLanguage,
     SpeechToTextModel,
@@ -72,12 +86,22 @@ from .errors import (
     TooManyRequestsError,
     UnprocessableEntityError,
 )
-from . import chat, speech_to_text, speech_to_text_streaming, speech_to_text_translate_streaming, text, text_to_speech
+from . import (
+    chat,
+    speech_to_text,
+    speech_to_text_streaming,
+    speech_to_text_translate_streaming,
+    text,
+    text_to_speech,
+    text_to_speech_streaming,
+)
 from .client import AsyncSarvamAI, SarvamAI
 from .environment import SarvamAIEnvironment
 from .requests import (
     AudioDataParams,
     AudioMessageParams,
+    AudioOutputDataParams,
+    AudioOutputParams,
     ChatCompletionRequestAssistantMessageParams,
     ChatCompletionRequestMessageParams,
     ChatCompletionRequestMessage_AssistantParams,
@@ -87,6 +111,7 @@ from .requests import (
     ChatCompletionRequestUserMessageParams,
     ChatCompletionResponseMessageParams,
     ChoiceParams,
+    CloseConnectionParams,
     CompletionUsageParams,
     ConfigMessageParams,
     CreateChatCompletionResponseParams,
@@ -95,8 +120,15 @@ from .requests import (
     ErrorDataParams,
     ErrorDetailsParams,
     ErrorMessageParams,
+    ErrorResponseDataParams,
+    ErrorResponseParams,
     EventsDataParams,
+    InitializeConnectionDataParams,
+    InitializeConnectionParams,
     LanguageIdentificationResponseParams,
+    PingSignalParams,
+    SendTextDataParams,
+    SendTextParams,
     SpeechToTextResponseDataParams,
     SpeechToTextResponseParams,
     SpeechToTextStreamingResponseParams,
@@ -131,6 +163,10 @@ __all__ = [
     "AudioDataParams",
     "AudioMessage",
     "AudioMessageParams",
+    "AudioOutput",
+    "AudioOutputData",
+    "AudioOutputDataParams",
+    "AudioOutputParams",
     "BadRequestError",
     "ChatCompletionRequestAssistantMessage",
     "ChatCompletionRequestAssistantMessageParams",
@@ -150,6 +186,8 @@ __all__ = [
     "ChatCompletionResponseMessageParams",
     "Choice",
     "ChoiceParams",
+    "CloseConnection",
+    "CloseConnectionParams",
     "CompletionUsage",
     "CompletionUsageParams",
     "ConfigMessage",
@@ -167,21 +205,39 @@ __all__ = [
     "ErrorDetailsParams",
     "ErrorMessage",
     "ErrorMessageParams",
+    "ErrorResponse",
+    "ErrorResponseData",
+    "ErrorResponseDataParams",
+    "ErrorResponseParams",
     "EventsData",
     "EventsDataParams",
     "FinishReason",
     "ForbiddenError",
     "Format",
+    "InitializeConnection",
+    "InitializeConnectionData",
+    "InitializeConnectionDataModel",
+    "InitializeConnectionDataOutputAudioBitrate",
+    "InitializeConnectionDataParams",
+    "InitializeConnectionDataSpeaker",
+    "InitializeConnectionDataTargetLanguageCode",
+    "InitializeConnectionParams",
     "InternalServerError",
     "LanguageIdentificationResponse",
     "LanguageIdentificationResponseParams",
     "NumeralsFormat",
+    "PingSignal",
+    "PingSignalParams",
     "ReasoningEffort",
     "ResponseType",
     "Role",
     "SarvamAI",
     "SarvamAIEnvironment",
     "SarvamModelIds",
+    "SendText",
+    "SendTextData",
+    "SendTextDataParams",
+    "SendTextParams",
     "ServiceUnavailableError",
     "SpeechSampleRate",
     "SpeechToTextLanguage",
@@ -244,4 +300,5 @@ __all__ = [
     "speech_to_text_translate_streaming",
     "text",
     "text_to_speech",
+    "text_to_speech_streaming",
 ]

sarvamai/client.py

@@ -16,6 +16,7 @@ from .speech_to_text_translate_streaming.client import (
 )
 from .text.client import AsyncTextClient, TextClient
 from .text_to_speech.client import AsyncTextToSpeechClient, TextToSpeechClient
+from .text_to_speech_streaming.client import AsyncTextToSpeechStreamingClient, TextToSpeechStreamingClient
 
 
 class SarvamAI:
@@ -86,6 +87,7 @@ class SarvamAI:
         self.speech_to_text_translate_streaming = SpeechToTextTranslateStreamingClient(
             client_wrapper=self._client_wrapper
         )
+        self.text_to_speech_streaming = TextToSpeechStreamingClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncSarvamAI:
@@ -156,3 +158,4 @@ class AsyncSarvamAI:
         self.speech_to_text_translate_streaming = AsyncSpeechToTextTranslateStreamingClient(
             client_wrapper=self._client_wrapper
         )
+        self.text_to_speech_streaming = AsyncTextToSpeechStreamingClient(client_wrapper=self._client_wrapper)

sarvamai/core/client_wrapper.py

@@ -17,10 +17,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "sarvamai/0.1.7",
+            "User-Agent": "sarvamai/0.1.8a0",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.7",
+            "X-Fern-SDK-Version": "0.1.8a0",
         }
         headers["api-subscription-key"] = self.api_subscription_key
         return headers

sarvamai/requests/__init__.py

@@ -4,6 +4,8 @@
 
 from .audio_data import AudioDataParams
 from .audio_message import AudioMessageParams
+from .audio_output import AudioOutputParams
+from .audio_output_data import AudioOutputDataParams
 from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessageParams
 from .chat_completion_request_message import (
     ChatCompletionRequestMessageParams,
@@ -15,6 +17,7 @@ from .chat_completion_request_system_message import ChatCompletionRequestSystemM
 from .chat_completion_request_user_message import ChatCompletionRequestUserMessageParams
 from .chat_completion_response_message import ChatCompletionResponseMessageParams
 from .choice import ChoiceParams
+from .close_connection import CloseConnectionParams
 from .completion_usage import CompletionUsageParams
 from .config_message import ConfigMessageParams
 from .create_chat_completion_response import CreateChatCompletionResponseParams
@@ -23,8 +26,15 @@ from .diarized_transcript import DiarizedTranscriptParams
 from .error_data import ErrorDataParams
 from .error_details import ErrorDetailsParams
 from .error_message import ErrorMessageParams
+from .error_response import ErrorResponseParams
+from .error_response_data import ErrorResponseDataParams
 from .events_data import EventsDataParams
+from .initialize_connection import InitializeConnectionParams
+from .initialize_connection_data import InitializeConnectionDataParams
 from .language_identification_response import LanguageIdentificationResponseParams
+from .ping_signal import PingSignalParams
+from .send_text import SendTextParams
+from .send_text_data import SendTextDataParams
 from .speech_to_text_response import SpeechToTextResponseParams
 from .speech_to_text_response_data import SpeechToTextResponseDataParams
 from .speech_to_text_streaming_response import SpeechToTextStreamingResponseParams
@@ -43,6 +53,8 @@ from .transliteration_response import TransliterationResponseParams
 __all__ = [
     "AudioDataParams",
     "AudioMessageParams",
+    "AudioOutputDataParams",
+    "AudioOutputParams",
     "ChatCompletionRequestAssistantMessageParams",
     "ChatCompletionRequestMessageParams",
     "ChatCompletionRequestMessage_AssistantParams",
@@ -52,6 +64,7 @@ __all__ = [
     "ChatCompletionRequestUserMessageParams",
     "ChatCompletionResponseMessageParams",
     "ChoiceParams",
+    "CloseConnectionParams",
     "CompletionUsageParams",
     "ConfigMessageParams",
     "CreateChatCompletionResponseParams",
@@ -60,8 +73,15 @@ __all__ = [
     "ErrorDataParams",
     "ErrorDetailsParams",
     "ErrorMessageParams",
+    "ErrorResponseDataParams",
+    "ErrorResponseParams",
     "EventsDataParams",
+    "InitializeConnectionDataParams",
+    "InitializeConnectionParams",
     "LanguageIdentificationResponseParams",
+    "PingSignalParams",
+    "SendTextDataParams",
+    "SendTextParams",
     "SpeechToTextResponseDataParams",
     "SpeechToTextResponseParams",
     "SpeechToTextStreamingResponseParams",

sarvamai/requests/audio_output.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+from .audio_output_data import AudioOutputDataParams
+
+
+class AudioOutputParams(typing_extensions.TypedDict):
+    type: typing.Literal["audio"]
+    data: AudioOutputDataParams

sarvamai/requests/audio_output_data.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing_extensions
+
+
+class AudioOutputDataParams(typing_extensions.TypedDict):
+    content_type: str
+    """
+    MIME type of the audio content (e.g., 'audio/mp3', 'audio/wav')
+    """
+
+    audio: str
+    """
+    Base64-encoded audio data ready for playback or download
+    """

sarvamai/requests/close_connection.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+
+
+class CloseConnectionParams(typing_extensions.TypedDict):
+    type: typing.Literal["flush"]

sarvamai/requests/error_response.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+from .error_response_data import ErrorResponseDataParams
+
+
+class ErrorResponseParams(typing_extensions.TypedDict):
+    type: typing.Literal["error"]
+    data: ErrorResponseDataParams

sarvamai/requests/error_response_data.py

@@ -0,0 +1,18 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+
+
+class ErrorResponseDataParams(typing_extensions.TypedDict):
+    message: str
+    code: typing_extensions.NotRequired[int]
+    """
+    Optional error code for programmatic error handling
+    """
+
+    details: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    """
+    Additional error details and context information
+    """

sarvamai/requests/initialize_connection.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+from .initialize_connection_data import InitializeConnectionDataParams
+
+
+class InitializeConnectionParams(typing_extensions.TypedDict):
+    type: typing.Literal["config"]
+    data: InitializeConnectionDataParams

sarvamai/requests/initialize_connection_data.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+from ..types.initialize_connection_data_model import InitializeConnectionDataModel
+from ..types.initialize_connection_data_output_audio_bitrate import InitializeConnectionDataOutputAudioBitrate
+from ..types.initialize_connection_data_speaker import InitializeConnectionDataSpeaker
+from ..types.initialize_connection_data_target_language_code import InitializeConnectionDataTargetLanguageCode
+
+
+class InitializeConnectionDataParams(typing_extensions.TypedDict):
+    target_language_code: InitializeConnectionDataTargetLanguageCode
+    speaker: InitializeConnectionDataSpeaker
+    pitch: typing_extensions.NotRequired[float]
+    pace: typing_extensions.NotRequired[float]
+    loudness: typing_extensions.NotRequired[float]
+    speech_sample_rate: typing_extensions.NotRequired[int]
+    enable_preprocessing: typing_extensions.NotRequired[bool]
+    model: InitializeConnectionDataModel
+    output_audio_codec: typing_extensions.NotRequired[typing.Literal["mp3"]]
+    output_audio_bitrate: typing_extensions.NotRequired[InitializeConnectionDataOutputAudioBitrate]
+    min_buffer_size: typing_extensions.NotRequired[int]
+    max_chunk_length: typing_extensions.NotRequired[int]

sarvamai/requests/ping_signal.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+
+
+class PingSignalParams(typing_extensions.TypedDict):
+    type: typing.Literal["ping"]

sarvamai/requests/send_text.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import typing_extensions
+from .send_text_data import SendTextDataParams
+
+
+class SendTextParams(typing_extensions.TypedDict):
+    type: typing.Literal["text"]
+    data: SendTextDataParams

sarvamai/requests/send_text_data.py

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

sarvamai/text_to_speech_streaming/__init__.py

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

sarvamai/text_to_speech_streaming/client.py

@@ -0,0 +1,136 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from contextlib import asynccontextmanager, contextmanager
+
+import websockets
+import websockets.sync.client as websockets_sync_client
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from .raw_client import AsyncRawTextToSpeechStreamingClient, RawTextToSpeechStreamingClient
+from .socket_client import AsyncTextToSpeechStreamingSocketClient, TextToSpeechStreamingSocketClient
+
+
+class TextToSpeechStreamingClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawTextToSpeechStreamingClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawTextToSpeechStreamingClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawTextToSpeechStreamingClient
+        """
+        return self._raw_client
+
+    @contextmanager
+    def connect(
+        self,
+        *,
+        api_subscription_key: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Iterator[TextToSpeechStreamingSocketClient]:
+        """
+        Bidirectional WebSocket channel for real-time TTS synthesis.
+        Supports streaming, flushing, config updates, and audio playback.
+
+        Parameters
+        ----------
+        api_subscription_key : typing.Optional[str]
+            API subscription key for authentication
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TextToSpeechStreamingSocketClient
+        """
+        ws_url = self._raw_client._client_wrapper.get_environment().production + "/text-to-speech/ws"
+        headers = self._raw_client._client_wrapper.get_headers()
+        if api_subscription_key is not None:
+            headers["Api-Subscription-Key"] = str(api_subscription_key)
+        if request_options and "additional_headers" in request_options:
+            headers.update(request_options["additional_headers"])
+        try:
+            with websockets_sync_client.connect(ws_url, additional_headers=headers) as protocol:
+                yield TextToSpeechStreamingSocketClient(websocket=protocol)
+        except websockets.exceptions.InvalidStatusCode as exc:
+            status_code: int = exc.status_code
+            if status_code == 401:
+                raise ApiError(
+                    status_code=status_code,
+                    headers=dict(headers),
+                    body="Websocket initialized with invalid credentials.",
+                )
+            raise ApiError(
+                status_code=status_code,
+                headers=dict(headers),
+                body="Unexpected error when initializing websocket connection.",
+            )
+
+
+class AsyncTextToSpeechStreamingClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawTextToSpeechStreamingClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawTextToSpeechStreamingClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawTextToSpeechStreamingClient
+        """
+        return self._raw_client
+
+    @asynccontextmanager
+    async def connect(
+        self,
+        *,
+        api_subscription_key: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.AsyncIterator[AsyncTextToSpeechStreamingSocketClient]:
+        """
+        Bidirectional WebSocket channel for real-time TTS synthesis.
+        Supports streaming, flushing, config updates, and audio playback.
+
+        Parameters
+        ----------
+        api_subscription_key : typing.Optional[str]
+            API subscription key for authentication
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncTextToSpeechStreamingSocketClient
+        """
+        ws_url = self._raw_client._client_wrapper.get_environment().production + "/text-to-speech/ws"
+        headers = self._raw_client._client_wrapper.get_headers()
+        if api_subscription_key is not None:
+            headers["Api-Subscription-Key"] = str(api_subscription_key)
+        if request_options and "additional_headers" in request_options:
+            headers.update(request_options["additional_headers"])
+        try:
+            async with websockets.connect(ws_url, extra_headers=headers) as protocol:
+                yield AsyncTextToSpeechStreamingSocketClient(websocket=protocol)
+        except websockets.exceptions.InvalidStatusCode as exc:
+            status_code: int = exc.status_code
+            if status_code == 401:
+                raise ApiError(
+                    status_code=status_code,
+                    headers=dict(headers),
+                    body="Websocket initialized with invalid credentials.",
+                )
+            raise ApiError(
+                status_code=status_code,
+                headers=dict(headers),
+                body="Unexpected error when initializing websocket connection.",
+            )

sarvamai/text_to_speech_streaming/raw_client.py

@@ -0,0 +1,113 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from contextlib import asynccontextmanager, contextmanager
+
+import websockets
+import websockets.sync.client as websockets_sync_client
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from .socket_client import AsyncTextToSpeechStreamingSocketClient, TextToSpeechStreamingSocketClient
+
+
+class RawTextToSpeechStreamingClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    @contextmanager
+    def connect(
+        self,
+        *,
+        api_subscription_key: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Iterator[TextToSpeechStreamingSocketClient]:
+        """
+        Bidirectional WebSocket channel for real-time TTS synthesis.
+        Supports streaming, flushing, config updates, and audio playback.
+
+        Parameters
+        ----------
+        api_subscription_key : typing.Optional[str]
+            API subscription key for authentication
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TextToSpeechStreamingSocketClient
+        """
+        ws_url = self._client_wrapper.get_environment().production + "/text-to-speech/ws"
+        headers = self._client_wrapper.get_headers()
+        if api_subscription_key is not None:
+            headers["Api-Subscription-Key"] = str(api_subscription_key)
+        if request_options and "additional_headers" in request_options:
+            headers.update(request_options["additional_headers"])
+        try:
+            with websockets_sync_client.connect(ws_url, additional_headers=headers) as protocol:
+                yield TextToSpeechStreamingSocketClient(websocket=protocol)
+        except websockets.exceptions.InvalidStatusCode as exc:
+            status_code: int = exc.status_code
+            if status_code == 401:
+                raise ApiError(
+                    status_code=status_code,
+                    headers=dict(headers),
+                    body="Websocket initialized with invalid credentials.",
+                )
+            raise ApiError(
+                status_code=status_code,
+                headers=dict(headers),
+                body="Unexpected error when initializing websocket connection.",
+            )
+
+
+class AsyncRawTextToSpeechStreamingClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    @asynccontextmanager
+    async def connect(
+        self,
+        *,
+        api_subscription_key: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.AsyncIterator[AsyncTextToSpeechStreamingSocketClient]:
+        """
+        Bidirectional WebSocket channel for real-time TTS synthesis.
+        Supports streaming, flushing, config updates, and audio playback.
+
+        Parameters
+        ----------
+        api_subscription_key : typing.Optional[str]
+            API subscription key for authentication
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncTextToSpeechStreamingSocketClient
+        """
+        ws_url = self._client_wrapper.get_environment().production + "/text-to-speech/ws"
+        headers = self._client_wrapper.get_headers()
+        if api_subscription_key is not None:
+            headers["Api-Subscription-Key"] = str(api_subscription_key)
+        if request_options and "additional_headers" in request_options:
+            headers.update(request_options["additional_headers"])
+        try:
+            async with websockets.connect(ws_url, extra_headers=headers) as protocol:
+                yield AsyncTextToSpeechStreamingSocketClient(websocket=protocol)
+        except websockets.exceptions.InvalidStatusCode as exc:
+            status_code: int = exc.status_code
+            if status_code == 401:
+                raise ApiError(
+                    status_code=status_code,
+                    headers=dict(headers),
+                    body="Websocket initialized with invalid credentials.",
+                )
+            raise ApiError(
+                status_code=status_code,
+                headers=dict(headers),
+                body="Unexpected error when initializing websocket connection.",
+            )