sarvamai 0.1.8rc6__tar.gz → 0.1.9__tar.gz

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarvamai
-Version: 0.1.8rc6
+Version: 0.1.9
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers
@@ -20,9 +20,9 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
 Requires-Dist: httpx (>=0.21.2)
 Requires-Dist: pydantic (>=1.9.2)
-Requires-Dist: pydantic-core (>=2.18.2,<3.0.0)
+Requires-Dist: pydantic-core (>=2.18.2)
 Requires-Dist: typing_extensions (>=4.0.0)
-Requires-Dist: websockets (==12.0)
+Requires-Dist: websockets (>=12.0)
 Description-Content-Type: text/markdown
 
 # Sarvam Python Library

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/pyproject.toml

@@ -3,7 +3,7 @@ name = "sarvamai"
 
 [tool.poetry]
 name = "sarvamai"
-version = "0.1.8rc6"
+version = "0.1.9"
 description = ""
 readme = "README.md"
 authors = []
@@ -37,9 +37,9 @@ Repository = 'https://github.com/sarvamai/sarvam-python-sdk'
 python = "^3.8"
 httpx = ">=0.21.2"
 pydantic = ">= 1.9.2"
-pydantic-core = "^2.18.2"
+pydantic-core = ">=2.18.2"
 typing_extensions = ">= 4.0.0"
-websockets = "12.0"
+websockets = ">=12.0"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "==1.13.0"

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/__init__.py

@@ -18,6 +18,11 @@ from .types import (
     Choice,
     CompletionUsage,
     ConfigMessage,
+    ConfigureConnection,
+    ConfigureConnectionData,
+    ConfigureConnectionDataOutputAudioBitrate,
+    ConfigureConnectionDataSpeaker,
+    ConfigureConnectionDataTargetLanguageCode,
     CreateChatCompletionResponse,
     DiarizedEntry,
     DiarizedTranscript,
@@ -31,11 +36,6 @@ from .types import (
     FinishReason,
     FlushSignal,
     Format,
-    InitializeConnection,
-    InitializeConnectionData,
-    InitializeConnectionDataOutputAudioBitrate,
-    InitializeConnectionDataSpeaker,
-    InitializeConnectionDataTargetLanguageCode,
     LanguageIdentificationResponse,
     NumeralsFormat,
     PingSignal,
@@ -112,6 +112,8 @@ from .requests import (
     ChoiceParams,
     CompletionUsageParams,
     ConfigMessageParams,
+    ConfigureConnectionDataParams,
+    ConfigureConnectionParams,
     CreateChatCompletionResponseParams,
     DiarizedEntryParams,
     DiarizedTranscriptParams,
@@ -122,8 +124,6 @@ from .requests import (
     ErrorResponseParams,
     EventsDataParams,
     FlushSignalParams,
-    InitializeConnectionDataParams,
-    InitializeConnectionParams,
     LanguageIdentificationResponseParams,
     PingSignalParams,
     SendTextDataParams,
@@ -189,6 +189,13 @@ __all__ = [
     "CompletionUsageParams",
     "ConfigMessage",
     "ConfigMessageParams",
+    "ConfigureConnection",
+    "ConfigureConnectionData",
+    "ConfigureConnectionDataOutputAudioBitrate",
+    "ConfigureConnectionDataParams",
+    "ConfigureConnectionDataSpeaker",
+    "ConfigureConnectionDataTargetLanguageCode",
+    "ConfigureConnectionParams",
     "CreateChatCompletionResponse",
     "CreateChatCompletionResponseParams",
     "DiarizedEntry",
@@ -213,13 +220,6 @@ __all__ = [
     "FlushSignalParams",
     "ForbiddenError",
     "Format",
-    "InitializeConnection",
-    "InitializeConnectionData",
-    "InitializeConnectionDataOutputAudioBitrate",
-    "InitializeConnectionDataParams",
-    "InitializeConnectionDataSpeaker",
-    "InitializeConnectionDataTargetLanguageCode",
-    "InitializeConnectionParams",
     "InternalServerError",
     "LanguageIdentificationResponse",
     "LanguageIdentificationResponseParams",

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/client.py

@@ -35,6 +35,9 @@ class SarvamAI:
 
 
     api_subscription_key : typing.Optional[str]
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -58,6 +61,7 @@ class SarvamAI:
         *,
         environment: SarvamAIEnvironment = SarvamAIEnvironment.PRODUCTION,
         api_subscription_key: typing.Optional[str] = os.getenv("SARVAM_API_KEY"),
+        headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.Client] = None,
@@ -72,6 +76,7 @@ class SarvamAI:
         self._client_wrapper = SyncClientWrapper(
             environment=environment,
             api_subscription_key=api_subscription_key,
+            headers=headers,
             httpx_client=httpx_client
             if httpx_client is not None
             else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
@@ -106,6 +111,9 @@ class AsyncSarvamAI:
 
 
     api_subscription_key : typing.Optional[str]
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -129,6 +137,7 @@ class AsyncSarvamAI:
         *,
         environment: SarvamAIEnvironment = SarvamAIEnvironment.PRODUCTION,
         api_subscription_key: typing.Optional[str] = os.getenv("SARVAM_API_KEY"),
+        headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.AsyncClient] = None,
@@ -143,6 +152,7 @@ class AsyncSarvamAI:
         self._client_wrapper = AsyncClientWrapper(
             environment=environment,
             api_subscription_key=api_subscription_key,
+            headers=headers,
             httpx_client=httpx_client
             if httpx_client is not None
             else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/core/client_wrapper.py

@@ -9,22 +9,32 @@ from .http_client import AsyncHttpClient, HttpClient
 
 class BaseClientWrapper:
     def __init__(
-        self, *, api_subscription_key: str, environment: SarvamAIEnvironment, timeout: typing.Optional[float] = None
+        self,
+        *,
+        api_subscription_key: str,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        environment: SarvamAIEnvironment,
+        timeout: typing.Optional[float] = None,
     ):
         self.api_subscription_key = api_subscription_key
+        self._headers = headers
         self._environment = environment
         self._timeout = timeout
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "sarvamai/0.1.8rc6",
+            "User-Agent": "sarvamai/0.1.9",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.8rc6",
+            "X-Fern-SDK-Version": "0.1.9",
+            **(self.get_custom_headers() or {}),
         }
         headers["api-subscription-key"] = self.api_subscription_key
         return headers
 
+    def get_custom_headers(self) -> typing.Optional[typing.Dict[str, str]]:
+        return self._headers
+
     def get_environment(self) -> SarvamAIEnvironment:
         return self._environment
 
@@ -37,11 +47,14 @@ class SyncClientWrapper(BaseClientWrapper):
         self,
         *,
         api_subscription_key: str,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
         environment: SarvamAIEnvironment,
         timeout: typing.Optional[float] = None,
         httpx_client: httpx.Client,
     ):
-        super().__init__(api_subscription_key=api_subscription_key, environment=environment, timeout=timeout)
+        super().__init__(
+            api_subscription_key=api_subscription_key, headers=headers, environment=environment, timeout=timeout
+        )
         self.httpx_client = HttpClient(
             httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
         )
@@ -52,11 +65,14 @@ class AsyncClientWrapper(BaseClientWrapper):
         self,
         *,
         api_subscription_key: str,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
         environment: SarvamAIEnvironment,
         timeout: typing.Optional[float] = None,
         httpx_client: httpx.AsyncClient,
     ):
-        super().__init__(api_subscription_key=api_subscription_key, environment=environment, timeout=timeout)
+        super().__init__(
+            api_subscription_key=api_subscription_key, headers=headers, environment=environment, timeout=timeout
+        )
         self.httpx_client = AsyncHttpClient(
             httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
         )

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/core/pydantic_utilities.py

@@ -59,9 +59,9 @@ class UniversalBaseModel(pydantic.BaseModel):
             protected_namespaces=(),
         )
 
-        @pydantic.model_serializer(mode="wrap", when_used="json")  # type: ignore[attr-defined]
-        def serialize_model(self, handler: pydantic.SerializerFunctionWrapHandler) -> Any:  # type: ignore[name-defined]
-            serialized = handler(self)
+        @pydantic.model_serializer(mode="plain", when_used="json")  # type: ignore[attr-defined]
+        def serialize_model(self) -> Any:  # type: ignore[name-defined]
+            serialized = self.model_dump()
             data = {k: serialize_datetime(v) if isinstance(v, dt.datetime) else v for k, v in serialized.items()}
             return data
 

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/requests/__init__.py

@@ -19,6 +19,8 @@ from .chat_completion_response_message import ChatCompletionResponseMessageParam
 from .choice import ChoiceParams
 from .completion_usage import CompletionUsageParams
 from .config_message import ConfigMessageParams
+from .configure_connection import ConfigureConnectionParams
+from .configure_connection_data import ConfigureConnectionDataParams
 from .create_chat_completion_response import CreateChatCompletionResponseParams
 from .diarized_entry import DiarizedEntryParams
 from .diarized_transcript import DiarizedTranscriptParams
@@ -29,8 +31,6 @@ from .error_response import ErrorResponseParams
 from .error_response_data import ErrorResponseDataParams
 from .events_data import EventsDataParams
 from .flush_signal import FlushSignalParams
-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
@@ -66,6 +66,8 @@ __all__ = [
     "ChoiceParams",
     "CompletionUsageParams",
     "ConfigMessageParams",
+    "ConfigureConnectionDataParams",
+    "ConfigureConnectionParams",
     "CreateChatCompletionResponseParams",
     "DiarizedEntryParams",
     "DiarizedTranscriptParams",
@@ -76,8 +78,6 @@ __all__ = [
     "ErrorResponseParams",
     "EventsDataParams",
     "FlushSignalParams",
-    "InitializeConnectionDataParams",
-    "InitializeConnectionParams",
     "LanguageIdentificationResponseParams",
     "PingSignalParams",
     "SendTextDataParams",

sarvamai-0.1.8rc6/src/sarvamai/requests/initialize_connection.py → sarvamai-0.1.9/src/sarvamai/requests/configure_connection.py

@@ -3,10 +3,10 @@
 import typing
 
 import typing_extensions
-from .initialize_connection_data import InitializeConnectionDataParams
+from .configure_connection_data import ConfigureConnectionDataParams
 
 
-class InitializeConnectionParams(typing_extensions.TypedDict):
+class ConfigureConnectionParams(typing_extensions.TypedDict):
     """
     Configuration message required as the first message after establishing the WebSocket connection.
     This initializes TTS parameters and can be updated at any time during the WebSocket lifecycle
@@ -15,4 +15,4 @@ class InitializeConnectionParams(typing_extensions.TypedDict):
     """
 
     type: typing.Literal["config"]
-    data: InitializeConnectionDataParams
+    data: ConfigureConnectionDataParams

sarvamai-0.1.8rc6/src/sarvamai/requests/initialize_connection_data.py → sarvamai-0.1.9/src/sarvamai/requests/configure_connection_data.py

@@ -3,18 +3,18 @@
 import typing
 
 import typing_extensions
-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
+from ..types.configure_connection_data_output_audio_bitrate import ConfigureConnectionDataOutputAudioBitrate
+from ..types.configure_connection_data_speaker import ConfigureConnectionDataSpeaker
+from ..types.configure_connection_data_target_language_code import ConfigureConnectionDataTargetLanguageCode
 
 
-class InitializeConnectionDataParams(typing_extensions.TypedDict):
-    target_language_code: InitializeConnectionDataTargetLanguageCode
+class ConfigureConnectionDataParams(typing_extensions.TypedDict):
+    target_language_code: ConfigureConnectionDataTargetLanguageCode
     """
     The language of the text is BCP-47 format
     """
 
-    speaker: InitializeConnectionDataSpeaker
+    speaker: ConfigureConnectionDataSpeaker
     """
     The speaker voice to be used for the output audio.
     
@@ -67,7 +67,7 @@ class InitializeConnectionDataParams(typing_extensions.TypedDict):
     Audio codec (currently supports MP3 only, optimized for real-time playback)
     """
 
-    output_audio_bitrate: typing_extensions.NotRequired[InitializeConnectionDataOutputAudioBitrate]
+    output_audio_bitrate: typing_extensions.NotRequired[ConfigureConnectionDataOutputAudioBitrate]
     """
     Audio bitrate (choose from 5 supported bitrate options)
     """

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/speech_to_text/client.py

@@ -61,14 +61,12 @@ class SpeechToTextClient:
 
         model : typing.Optional[SpeechToTextModel]
             Specifies the model to use for speech-to-text conversion.
-            Note:- Default model is `saarika:v2`
+            Note:- Default model is `saarika:v2.5`
 
         language_code : typing.Optional[SpeechToTextLanguage]
-            Specifies the language of the input audio. This parameter is required to ensure accurate transcription.
-             For the `saarika:v1` model, this parameter is mandatory.
-             For the `saarika:v2` model, it is optional.
+            Specifies the language of the input audio.
+             For the `saarika:v2.5` model, it is optional.
             `unknown`: Use this when the language is not known; the API will detect it automatically.
-            Note:- that the `saarika:v1` model does not support `unknown` language code.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -192,14 +190,12 @@ class AsyncSpeechToTextClient:
 
         model : typing.Optional[SpeechToTextModel]
             Specifies the model to use for speech-to-text conversion.
-            Note:- Default model is `saarika:v2`
+            Note:- Default model is `saarika:v2.5`
 
         language_code : typing.Optional[SpeechToTextLanguage]
-            Specifies the language of the input audio. This parameter is required to ensure accurate transcription.
-             For the `saarika:v1` model, this parameter is mandatory.
-             For the `saarika:v2` model, it is optional.
+            Specifies the language of the input audio.
+             For the `saarika:v2.5` model, it is optional.
             `unknown`: Use this when the language is not known; the API will detect it automatically.
-            Note:- that the `saarika:v1` model does not support `unknown` language code.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/speech_to_text/raw_client.py

@@ -60,14 +60,12 @@ class RawSpeechToTextClient:
 
         model : typing.Optional[SpeechToTextModel]
             Specifies the model to use for speech-to-text conversion.
-            Note:- Default model is `saarika:v2`
+            Note:- Default model is `saarika:v2.5`
 
         language_code : typing.Optional[SpeechToTextLanguage]
-            Specifies the language of the input audio. This parameter is required to ensure accurate transcription.
-             For the `saarika:v1` model, this parameter is mandatory.
-             For the `saarika:v2` model, it is optional.
+            Specifies the language of the input audio.
+             For the `saarika:v2.5` model, it is optional.
             `unknown`: Use this when the language is not known; the API will detect it automatically.
-            Note:- that the `saarika:v1` model does not support `unknown` language code.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -346,14 +344,12 @@ class AsyncRawSpeechToTextClient:
 
         model : typing.Optional[SpeechToTextModel]
             Specifies the model to use for speech-to-text conversion.
-            Note:- Default model is `saarika:v2`
+            Note:- Default model is `saarika:v2.5`
 
         language_code : typing.Optional[SpeechToTextLanguage]
-            Specifies the language of the input audio. This parameter is required to ensure accurate transcription.
-             For the `saarika:v1` model, this parameter is mandatory.
-             For the `saarika:v2` model, it is optional.
+            Specifies the language of the input audio.
+             For the `saarika:v2.5` model, it is optional.
             `unknown`: Use this when the language is not known; the API will detect it automatically.
-            Note:- that the `saarika:v1` model does not support `unknown` language code.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/text/client.py

@@ -202,7 +202,7 @@ class TextClient:
         Parameters
         ----------
         input : str
-            The text input for language and script identification.
+            The text input for language and script identification. Max Input Limit is 1000 characters
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -534,7 +534,7 @@ class AsyncTextClient:
         Parameters
         ----------
         input : str
-            The text input for language and script identification.
+            The text input for language and script identification. Max Input Limit is 1000 characters
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

{sarvamai-0.1.8rc6 → sarvamai-0.1.9}/src/sarvamai/text/raw_client.py

@@ -263,7 +263,7 @@ class RawTextClient:
         Parameters
         ----------
         input : str
-            The text input for language and script identification.
+            The text input for language and script identification. Max Input Limit is 1000 characters
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -773,7 +773,7 @@ class AsyncRawTextClient:
         Parameters
         ----------
         input : str
-            The text input for language and script identification.
+            The text input for language and script identification. Max Input Limit is 1000 characters
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.