sarvamai 0.1.8rc7__tar.gz → 0.1.9a2__tar.gz

{sarvamai-0.1.8rc7 → sarvamai-0.1.9a2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarvamai
-Version: 0.1.8rc7
+Version: 0.1.9a2
 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.8rc7 → sarvamai-0.1.9a2}/pyproject.toml

@@ -3,7 +3,7 @@ name = "sarvamai"
 
 [tool.poetry]
 name = "sarvamai"
-version = "0.1.8rc7"
+version = "0.1.9a2"
 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.8rc7 → sarvamai-0.1.9a2}/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.8rc7 → sarvamai-0.1.9a2}/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.8rc7",
+            "User-Agent": "sarvamai/0.1.9a2",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.8rc7",
+            "X-Fern-SDK-Version": "0.1.9a2",
+            **(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.8rc7 → sarvamai-0.1.9a2}/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.8rc7 → sarvamai-0.1.9a2}/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.8rc7 → sarvamai-0.1.9a2}/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.8rc7 → sarvamai-0.1.9a2}/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.8rc7 → sarvamai-0.1.9a2}/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.

sarvamai-0.1.9a2/src/sarvamai/types/speech_to_text_model.py

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

sarvamai-0.1.9a2/src/sarvamai/types/speech_to_text_translate_model.py

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

sarvamai-0.1.8rc7/src/sarvamai/types/speech_to_text_model.py

@@ -1,7 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-SpeechToTextModel = typing.Union[
-    typing.Literal["saarika:v1", "saarika:v2", "saarika:v2.5", "saarika:flash"], typing.Any
-]

sarvamai-0.1.8rc7/src/sarvamai/types/speech_to_text_translate_model.py

@@ -1,7 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-SpeechToTextTranslateModel = typing.Union[
-    typing.Literal["saaras:v1", "saaras:v2", "saaras:v2.5", "saaras:flash"], typing.Any
-]