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

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.",
+            )

sarvamai/text_to_speech_streaming/socket_client.py

@@ -0,0 +1,177 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import json
+import typing
+
+import websockets
+import websockets.sync.connection as websockets_sync_connection
+from ..core.events import EventEmitterMixin, EventType
+from ..core.pydantic_utilities import parse_obj_as
+from ..types.audio_output import AudioOutput
+from ..types.close_connection import CloseConnection
+from ..types.error_response import ErrorResponse
+from ..types.initialize_connection import InitializeConnection
+from ..types.ping_signal import PingSignal
+from ..types.send_text import SendText
+
+TextToSpeechStreamingSocketClientResponse = typing.Union[AudioOutput, ErrorResponse]
+
+
+class AsyncTextToSpeechStreamingSocketClient(EventEmitterMixin):
+    def __init__(self, *, websocket: websockets.WebSocketClientProtocol):
+        super().__init__()
+        self._websocket = websocket
+
+    async def __aiter__(self):
+        async for message in self._websocket:
+            yield parse_obj_as(TextToSpeechStreamingSocketClientResponse, message)  # type: ignore
+
+    async def start_listening(self):
+        """
+        Start listening for messages on the websocket connection.
+
+        Emits events in the following order:
+        - EventType.OPEN when connection is established
+        - EventType.MESSAGE for each message received
+        - EventType.ERROR if an error occurs
+        - EventType.CLOSE when connection is closed
+        """
+        self._emit(EventType.OPEN, None)
+        try:
+            async for raw_message in self._websocket:
+                parsed = parse_obj_as(TextToSpeechStreamingSocketClientResponse, raw_message)  # type: ignore
+                self._emit(EventType.MESSAGE, parsed)
+        except websockets.WebSocketException as exc:
+            self._emit(EventType.ERROR, exc)
+        finally:
+            self._emit(EventType.CLOSE, None)
+
+    async def send_initialize_connection(self, message: InitializeConnection) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a InitializeConnection.
+        """
+        await self._send_model(message)
+
+    async def send_send_text(self, message: SendText) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a SendText.
+        """
+        await self._send_model(message)
+
+    async def send_close_connection(self, message: CloseConnection) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a CloseConnection.
+        """
+        await self._send_model(message)
+
+    async def send_ping_signal(self, message: PingSignal) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a PingSignal.
+        """
+        await self._send_model(message)
+
+    async def recv(self) -> TextToSpeechStreamingSocketClientResponse:
+        """
+        Receive a message from the websocket connection.
+        """
+        data = await self._websocket.recv()
+        data = json.loads(data) if isinstance(data, str) else data
+        return parse_obj_as(TextToSpeechStreamingSocketClientResponse, data)  # type: ignore
+
+    async def _send(self, data: typing.Any) -> None:
+        """
+        Send a message to the websocket connection.
+        """
+        if isinstance(data, dict):
+            data = json.dumps(data)
+        await self._websocket.send(data)
+
+    async def _send_model(self, data: typing.Any) -> None:
+        """
+        Send a Pydantic model to the websocket connection.
+        """
+        await self._send(data.dict())
+
+
+class TextToSpeechStreamingSocketClient(EventEmitterMixin):
+    def __init__(self, *, websocket: websockets_sync_connection.Connection):
+        super().__init__()
+        self._websocket = websocket
+
+    def __iter__(self):
+        for message in self._websocket:
+            yield parse_obj_as(TextToSpeechStreamingSocketClientResponse, message)  # type: ignore
+
+    def start_listening(self):
+        """
+        Start listening for messages on the websocket connection.
+
+        Emits events in the following order:
+        - EventType.OPEN when connection is established
+        - EventType.MESSAGE for each message received
+        - EventType.ERROR if an error occurs
+        - EventType.CLOSE when connection is closed
+        """
+        self._emit(EventType.OPEN, None)
+        try:
+            for raw_message in self._websocket:
+                parsed = parse_obj_as(TextToSpeechStreamingSocketClientResponse, raw_message)  # type: ignore
+                self._emit(EventType.MESSAGE, parsed)
+        except websockets.WebSocketException as exc:
+            self._emit(EventType.ERROR, exc)
+        finally:
+            self._emit(EventType.CLOSE, None)
+
+    def send_initialize_connection(self, message: InitializeConnection) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a InitializeConnection.
+        """
+        self._send_model(message)
+
+    def send_send_text(self, message: SendText) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a SendText.
+        """
+        self._send_model(message)
+
+    def send_close_connection(self, message: CloseConnection) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a CloseConnection.
+        """
+        self._send_model(message)
+
+    def send_ping_signal(self, message: PingSignal) -> None:
+        """
+        Send a message to the websocket connection.
+        The message will be sent as a PingSignal.
+        """
+        self._send_model(message)
+
+    def recv(self) -> TextToSpeechStreamingSocketClientResponse:
+        """
+        Receive a message from the websocket connection.
+        """
+        data = self._websocket.recv()
+        data = json.loads(data) if isinstance(data, str) else data
+        return parse_obj_as(TextToSpeechStreamingSocketClientResponse, data)  # type: ignore
+
+    def _send(self, data: typing.Any) -> None:
+        """
+        Send a message to the websocket connection.
+        """
+        if isinstance(data, dict):
+            data = json.dumps(data)
+        self._websocket.send(data)
+
+    def _send_model(self, data: typing.Any) -> None:
+        """
+        Send a Pydantic model to the websocket connection.
+        """
+        self._send(data.dict())

sarvamai/types/__init__.py

@@ -4,9 +4,8 @@
 
 from .audio_data import AudioData
 from .audio_message import AudioMessage
-from .base_job_parameters import BaseJobParameters
-from .bulk_job_callback import BulkJobCallback
-from .bulk_job_init_response_v_1 import BulkJobInitResponseV1
+from .audio_output import AudioOutput
+from .audio_output_data import AudioOutputData
 from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessage
 from .chat_completion_request_message import (
     ChatCompletionRequestMessage,
@@ -18,6 +17,7 @@ from .chat_completion_request_system_message import ChatCompletionRequestSystemM
 from .chat_completion_request_user_message import ChatCompletionRequestUserMessage
 from .chat_completion_response_message import ChatCompletionResponseMessage
 from .choice import Choice
+from .close_connection import CloseConnection
 from .completion_usage import CompletionUsage
 from .config_message import ConfigMessage
 from .create_chat_completion_response import CreateChatCompletionResponse
@@ -27,23 +27,27 @@ from .error_code import ErrorCode
 from .error_data import ErrorData
 from .error_details import ErrorDetails
 from .error_message import ErrorMessage
+from .error_response import ErrorResponse
+from .error_response_data import ErrorResponseData
 from .events_data import EventsData
-from .file_signed_url_details import FileSignedUrlDetails
-from .files_download_response import FilesDownloadResponse
-from .files_request import FilesRequest
-from .files_upload_response import FilesUploadResponse
 from .finish_reason import FinishReason
 from .format import Format
-from .job_state import JobState
-from .job_status_v_1 import JobStatusV1
+from .initialize_connection import InitializeConnection
+from .initialize_connection_data import InitializeConnectionData
+from .initialize_connection_data_model import InitializeConnectionDataModel
+from .initialize_connection_data_output_audio_bitrate import InitializeConnectionDataOutputAudioBitrate
+from .initialize_connection_data_speaker import InitializeConnectionDataSpeaker
+from .initialize_connection_data_target_language_code import InitializeConnectionDataTargetLanguageCode
 from .language_identification_response import LanguageIdentificationResponse
 from .numerals_format import NumeralsFormat
+from .ping_signal import PingSignal
 from .reasoning_effort import ReasoningEffort
 from .response_type import ResponseType
 from .role import Role
 from .sarvam_model_ids import SarvamModelIds
+from .send_text import SendText
+from .send_text_data import SendTextData
 from .speech_sample_rate import SpeechSampleRate
-from .speech_to_text_job_parameters import SpeechToTextJobParameters
 from .speech_to_text_language import SpeechToTextLanguage
 from .speech_to_text_model import SpeechToTextModel
 from .speech_to_text_response import SpeechToTextResponse
@@ -58,10 +62,6 @@ from .speech_to_text_translate_streaming_response import SpeechToTextTranslateSt
 from .speech_to_text_translate_transcription_data import SpeechToTextTranslateTranscriptionData
 from .spoken_form_numerals_format import SpokenFormNumeralsFormat
 from .stop_configuration import StopConfiguration
-from .storage_container_type import StorageContainerType
-from .task_detail_v_1 import TaskDetailV1
-from .task_file_details import TaskFileDetails
-from .task_state import TaskState
 from .text_to_speech_language import TextToSpeechLanguage
 from .text_to_speech_model import TextToSpeechModel
 from .text_to_speech_response import TextToSpeechResponse
@@ -82,9 +82,8 @@ from .transliteration_response import TransliterationResponse
 __all__ = [
     "AudioData",
     "AudioMessage",
-    "BaseJobParameters",
-    "BulkJobCallback",
-    "BulkJobInitResponseV1",
+    "AudioOutput",
+    "AudioOutputData",
     "ChatCompletionRequestAssistantMessage",
     "ChatCompletionRequestMessage",
     "ChatCompletionRequestMessage_Assistant",
@@ -94,6 +93,7 @@ __all__ = [
     "ChatCompletionRequestUserMessage",
     "ChatCompletionResponseMessage",
     "Choice",
+    "CloseConnection",
     "CompletionUsage",
     "ConfigMessage",
     "CreateChatCompletionResponse",
@@ -103,23 +103,27 @@ __all__ = [
     "ErrorData",
     "ErrorDetails",
     "ErrorMessage",
+    "ErrorResponse",
+    "ErrorResponseData",
     "EventsData",
-    "FileSignedUrlDetails",
-    "FilesDownloadResponse",
-    "FilesRequest",
-    "FilesUploadResponse",
     "FinishReason",
     "Format",
-    "JobState",
-    "JobStatusV1",
+    "InitializeConnection",
+    "InitializeConnectionData",
+    "InitializeConnectionDataModel",
+    "InitializeConnectionDataOutputAudioBitrate",
+    "InitializeConnectionDataSpeaker",
+    "InitializeConnectionDataTargetLanguageCode",
     "LanguageIdentificationResponse",
     "NumeralsFormat",
+    "PingSignal",
     "ReasoningEffort",
     "ResponseType",
     "Role",
     "SarvamModelIds",
+    "SendText",
+    "SendTextData",
     "SpeechSampleRate",
-    "SpeechToTextJobParameters",
     "SpeechToTextLanguage",
     "SpeechToTextModel",
     "SpeechToTextResponse",
@@ -134,10 +138,6 @@ __all__ = [
     "SpeechToTextTranslateTranscriptionData",
     "SpokenFormNumeralsFormat",
     "StopConfiguration",
-    "StorageContainerType",
-    "TaskDetailV1",
-    "TaskFileDetails",
-    "TaskState",
     "TextToSpeechLanguage",
     "TextToSpeechModel",
     "TextToSpeechResponse",

sarvamai/types/audio_output.py

@@ -0,0 +1,21 @@
+# 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_output_data import AudioOutputData
+
+
+class AudioOutput(UniversalBaseModel):
+    type: typing.Literal["audio"] = "audio"
+    data: AudioOutputData
+
+    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/{bulk_job_callback.py → audio_output_data.py}

@@ -6,15 +6,15 @@ import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class BulkJobCallback(UniversalBaseModel):
-    url: str = pydantic.Field()
+class AudioOutputData(UniversalBaseModel):
+    content_type: str = pydantic.Field()
     """
-    Webhook url to call upon job completion
+    MIME type of the audio content (e.g., 'audio/mp3', 'audio/wav')
     """
 
-    auth_token: typing.Optional[str] = pydantic.Field(default=None)
+    audio: str = pydantic.Field()
     """
-    Authorization token required for the callback Url
+    Base64-encoded audio data ready for playback or download
     """
 
     if IS_PYDANTIC_V2:

sarvamai/types/{files_request.py → close_connection.py}

@@ -6,9 +6,8 @@ import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class FilesRequest(UniversalBaseModel):
-    job_id: str
-    files: typing.List[str]
+class CloseConnection(UniversalBaseModel):
+    type: typing.Literal["flush"] = "flush"
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

sarvamai/types/error_response.py

@@ -0,0 +1,21 @@
+# 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 .error_response_data import ErrorResponseData
+
+
+class ErrorResponse(UniversalBaseModel):
+    type: typing.Literal["error"] = "error"
+    data: ErrorResponseData
+
+    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/{files_download_response.py → error_response_data.py}

@@ -4,16 +4,19 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from .file_signed_url_details import FileSignedUrlDetails
-from .job_state import JobState
-from .storage_container_type import StorageContainerType
 
 
-class FilesDownloadResponse(UniversalBaseModel):
-    job_id: str
-    job_state: JobState
-    download_urls: typing.Dict[str, FileSignedUrlDetails]
-    storage_container_type: StorageContainerType
+class ErrorResponseData(UniversalBaseModel):
+    message: str
+    code: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Optional error code for programmatic error handling
+    """
+
+    details: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Additional error details and context information
+    """
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

sarvamai/types/initialize_connection.py

@@ -0,0 +1,21 @@
+# 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 .initialize_connection_data import InitializeConnectionData
+
+
+class InitializeConnection(UniversalBaseModel):
+    type: typing.Literal["config"] = "config"
+    data: InitializeConnectionData
+
+    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