sarvamai 0.1.23a2__py3-none-any.whl → 0.1.23a4__py3-none-any.whl

sarvamai/speech_to_text_streaming/__init__.py

@@ -2,48 +2,20 @@
 
 # isort: skip_file
 
-import typing
-from importlib import import_module
-
-if typing.TYPE_CHECKING:
-    from .types import (
-        SpeechToTextStreamingFlushSignal,
-        SpeechToTextStreamingHighVadSensitivity,
-        SpeechToTextStreamingLanguageCode,
-        SpeechToTextStreamingVadSignals,
-    )
-_dynamic_imports: typing.Dict[str, str] = {
-    "SpeechToTextStreamingFlushSignal": ".types",
-    "SpeechToTextStreamingHighVadSensitivity": ".types",
-    "SpeechToTextStreamingLanguageCode": ".types",
-    "SpeechToTextStreamingVadSignals": ".types",
-}
-
-
-def __getattr__(attr_name: str) -> typing.Any:
-    module_name = _dynamic_imports.get(attr_name)
-    if module_name is None:
-        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
-    try:
-        module = import_module(module_name, __package__)
-        if module_name == f".{attr_name}":
-            return module
-        else:
-            return getattr(module, attr_name)
-    except ImportError as e:
-        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
-    except AttributeError as e:
-        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
-
-
-def __dir__():
-    lazy_attrs = list(_dynamic_imports.keys())
-    return sorted(lazy_attrs)
-
+from .types import (
+    SpeechToTextStreamingFlushSignal,
+    SpeechToTextStreamingHighVadSensitivity,
+    SpeechToTextStreamingLanguageCode,
+    SpeechToTextStreamingMode,
+    SpeechToTextStreamingModel,
+    SpeechToTextStreamingVadSignals,
+)
 
 __all__ = [
     "SpeechToTextStreamingFlushSignal",
     "SpeechToTextStreamingHighVadSensitivity",
     "SpeechToTextStreamingLanguageCode",
+    "SpeechToTextStreamingMode",
+    "SpeechToTextStreamingModel",
     "SpeechToTextStreamingVadSignals",
 ]

sarvamai/speech_to_text_streaming/client.py

@@ -14,6 +14,8 @@ from .socket_client import AsyncSpeechToTextStreamingSocketClient, SpeechToTextS
 from .types.speech_to_text_streaming_flush_signal import SpeechToTextStreamingFlushSignal
 from .types.speech_to_text_streaming_high_vad_sensitivity import SpeechToTextStreamingHighVadSensitivity
 from .types.speech_to_text_streaming_language_code import SpeechToTextStreamingLanguageCode
+from .types.speech_to_text_streaming_mode import SpeechToTextStreamingMode
+from .types.speech_to_text_streaming_model import SpeechToTextStreamingModel
 from .types.speech_to_text_streaming_vad_signals import SpeechToTextStreamingVadSignals
 
 try:
@@ -42,7 +44,8 @@ class SpeechToTextStreamingClient:
         self,
         *,
         language_code: SpeechToTextStreamingLanguageCode,
-        model: typing.Optional[typing.Literal["saarika:v2.5"]] = None,
+        model: typing.Optional[SpeechToTextStreamingModel] = None,
+        mode: typing.Optional[SpeechToTextStreamingMode] = None,
         sample_rate: typing.Optional[str] = None,
         high_vad_sensitivity: typing.Optional[SpeechToTextStreamingHighVadSensitivity] = None,
         vad_signals: typing.Optional[SpeechToTextStreamingVadSignals] = None,
@@ -60,11 +63,20 @@ class SpeechToTextStreamingClient:
         Parameters
         ----------
         language_code : SpeechToTextStreamingLanguageCode
-            Language code for speech recognition
+            Language code for speech recognition (BCP-47 format)
 
-        model : typing.Optional[typing.Literal["saarika:v2.5"]]
+        model : typing.Optional[SpeechToTextStreamingModel]
             Speech to text model to use
 
+        mode : typing.Optional[SpeechToTextStreamingMode]
+            Mode of operation for saaras:v3 model. Only applicable when model is 'saaras:v3'.
+            - transcribe: Standard Whisper transcription
+            - translate: Standard Whisper translation to English
+            - indic-en: Translate Indic languages to English
+            - verbatim: Exact transcription in original script
+            - translit: Transliteration to Latin script
+            - codemix: Code-mixed output (native + English)
+
         sample_rate : typing.Optional[str]
             Audio sample rate for the WebSocket connection. When specified as a connection parameter, only 16kHz and 8kHz are supported. 8kHz is only available via this connection parameter. If not specified, defaults to 16kHz.
 
@@ -93,6 +105,8 @@ class SpeechToTextStreamingClient:
             query_params = query_params.add("language-code", language_code)
         if model is not None:
             query_params = query_params.add("model", model)
+        if mode is not None:
+            query_params = query_params.add("mode", mode)
         if sample_rate is not None:
             query_params = query_params.add("sample_rate", sample_rate)
         if high_vad_sensitivity is not None:
@@ -145,7 +159,8 @@ class AsyncSpeechToTextStreamingClient:
         self,
         *,
         language_code: SpeechToTextStreamingLanguageCode,
-        model: typing.Optional[typing.Literal["saarika:v2.5"]] = None,
+        model: typing.Optional[SpeechToTextStreamingModel] = None,
+        mode: typing.Optional[SpeechToTextStreamingMode] = None,
         sample_rate: typing.Optional[str] = None,
         high_vad_sensitivity: typing.Optional[SpeechToTextStreamingHighVadSensitivity] = None,
         vad_signals: typing.Optional[SpeechToTextStreamingVadSignals] = None,
@@ -163,11 +178,20 @@ class AsyncSpeechToTextStreamingClient:
         Parameters
         ----------
         language_code : SpeechToTextStreamingLanguageCode
-            Language code for speech recognition
+            Language code for speech recognition (BCP-47 format)
 
-        model : typing.Optional[typing.Literal["saarika:v2.5"]]
+        model : typing.Optional[SpeechToTextStreamingModel]
             Speech to text model to use
 
+        mode : typing.Optional[SpeechToTextStreamingMode]
+            Mode of operation for saaras:v3 model. Only applicable when model is 'saaras:v3'.
+            - transcribe: Standard Whisper transcription
+            - translate: Standard Whisper translation to English
+            - indic-en: Translate Indic languages to English
+            - verbatim: Exact transcription in original script
+            - translit: Transliteration to Latin script
+            - codemix: Code-mixed output (native + English)
+
         sample_rate : typing.Optional[str]
             Audio sample rate for the WebSocket connection. When specified as a connection parameter, only 16kHz and 8kHz are supported. 8kHz is only available via this connection parameter. If not specified, defaults to 16kHz.
 
@@ -196,6 +220,8 @@ class AsyncSpeechToTextStreamingClient:
             query_params = query_params.add("language-code", language_code)
         if model is not None:
             query_params = query_params.add("model", model)
+        if mode is not None:
+            query_params = query_params.add("mode", mode)
         if sample_rate is not None:
             query_params = query_params.add("sample_rate", sample_rate)
         if high_vad_sensitivity is not None:

sarvamai/speech_to_text_streaming/raw_client.py

@@ -13,6 +13,8 @@ from .socket_client import AsyncSpeechToTextStreamingSocketClient, SpeechToTextS
 from .types.speech_to_text_streaming_flush_signal import SpeechToTextStreamingFlushSignal
 from .types.speech_to_text_streaming_high_vad_sensitivity import SpeechToTextStreamingHighVadSensitivity
 from .types.speech_to_text_streaming_language_code import SpeechToTextStreamingLanguageCode
+from .types.speech_to_text_streaming_mode import SpeechToTextStreamingMode
+from .types.speech_to_text_streaming_model import SpeechToTextStreamingModel
 from .types.speech_to_text_streaming_vad_signals import SpeechToTextStreamingVadSignals
 
 try:
@@ -30,7 +32,8 @@ class RawSpeechToTextStreamingClient:
         self,
         *,
         language_code: SpeechToTextStreamingLanguageCode,
-        model: typing.Optional[typing.Literal["saarika:v2.5"]] = None,
+        model: typing.Optional[SpeechToTextStreamingModel] = None,
+        mode: typing.Optional[SpeechToTextStreamingMode] = None,
         sample_rate: typing.Optional[str] = None,
         high_vad_sensitivity: typing.Optional[SpeechToTextStreamingHighVadSensitivity] = None,
         vad_signals: typing.Optional[SpeechToTextStreamingVadSignals] = None,
@@ -48,11 +51,20 @@ class RawSpeechToTextStreamingClient:
         Parameters
         ----------
         language_code : SpeechToTextStreamingLanguageCode
-            Language code for speech recognition
+            Language code for speech recognition (BCP-47 format)
 
-        model : typing.Optional[typing.Literal["saarika:v2.5"]]
+        model : typing.Optional[SpeechToTextStreamingModel]
             Speech to text model to use
 
+        mode : typing.Optional[SpeechToTextStreamingMode]
+            Mode of operation for saaras:v3 model. Only applicable when model is 'saaras:v3'.
+            - transcribe: Standard Whisper transcription
+            - translate: Standard Whisper translation to English
+            - indic-en: Translate Indic languages to English
+            - verbatim: Exact transcription in original script
+            - translit: Transliteration to Latin script
+            - codemix: Code-mixed output (native + English)
+
         sample_rate : typing.Optional[str]
             Audio sample rate for the WebSocket connection. When specified as a connection parameter, only 16kHz and 8kHz are supported. 8kHz is only available via this connection parameter. If not specified, defaults to 16kHz.
 
@@ -81,6 +93,8 @@ class RawSpeechToTextStreamingClient:
             query_params = query_params.add("language-code", language_code)
         if model is not None:
             query_params = query_params.add("model", model)
+        if mode is not None:
+            query_params = query_params.add("mode", mode)
         if sample_rate is not None:
             query_params = query_params.add("sample_rate", sample_rate)
         if high_vad_sensitivity is not None:
@@ -122,7 +136,8 @@ class AsyncRawSpeechToTextStreamingClient:
         self,
         *,
         language_code: SpeechToTextStreamingLanguageCode,
-        model: typing.Optional[typing.Literal["saarika:v2.5"]] = None,
+        model: typing.Optional[SpeechToTextStreamingModel] = None,
+        mode: typing.Optional[SpeechToTextStreamingMode] = None,
         sample_rate: typing.Optional[str] = None,
         high_vad_sensitivity: typing.Optional[SpeechToTextStreamingHighVadSensitivity] = None,
         vad_signals: typing.Optional[SpeechToTextStreamingVadSignals] = None,
@@ -140,11 +155,20 @@ class AsyncRawSpeechToTextStreamingClient:
         Parameters
         ----------
         language_code : SpeechToTextStreamingLanguageCode
-            Language code for speech recognition
+            Language code for speech recognition (BCP-47 format)
 
-        model : typing.Optional[typing.Literal["saarika:v2.5"]]
+        model : typing.Optional[SpeechToTextStreamingModel]
             Speech to text model to use
 
+        mode : typing.Optional[SpeechToTextStreamingMode]
+            Mode of operation for saaras:v3 model. Only applicable when model is 'saaras:v3'.
+            - transcribe: Standard Whisper transcription
+            - translate: Standard Whisper translation to English
+            - indic-en: Translate Indic languages to English
+            - verbatim: Exact transcription in original script
+            - translit: Transliteration to Latin script
+            - codemix: Code-mixed output (native + English)
+
         sample_rate : typing.Optional[str]
             Audio sample rate for the WebSocket connection. When specified as a connection parameter, only 16kHz and 8kHz are supported. 8kHz is only available via this connection parameter. If not specified, defaults to 16kHz.
 
@@ -173,6 +197,8 @@ class AsyncRawSpeechToTextStreamingClient:
             query_params = query_params.add("language-code", language_code)
         if model is not None:
             query_params = query_params.add("model", model)
+        if mode is not None:
+            query_params = query_params.add("mode", mode)
         if sample_rate is not None:
             query_params = query_params.add("sample_rate", sample_rate)
         if high_vad_sensitivity is not None:

sarvamai/speech_to_text_streaming/types/__init__.py

@@ -2,46 +2,18 @@
 
 # isort: skip_file
 
-import typing
-from importlib import import_module
-
-if typing.TYPE_CHECKING:
-    from .speech_to_text_streaming_flush_signal import SpeechToTextStreamingFlushSignal
-    from .speech_to_text_streaming_high_vad_sensitivity import SpeechToTextStreamingHighVadSensitivity
-    from .speech_to_text_streaming_language_code import SpeechToTextStreamingLanguageCode
-    from .speech_to_text_streaming_vad_signals import SpeechToTextStreamingVadSignals
-_dynamic_imports: typing.Dict[str, str] = {
-    "SpeechToTextStreamingFlushSignal": ".speech_to_text_streaming_flush_signal",
-    "SpeechToTextStreamingHighVadSensitivity": ".speech_to_text_streaming_high_vad_sensitivity",
-    "SpeechToTextStreamingLanguageCode": ".speech_to_text_streaming_language_code",
-    "SpeechToTextStreamingVadSignals": ".speech_to_text_streaming_vad_signals",
-}
-
-
-def __getattr__(attr_name: str) -> typing.Any:
-    module_name = _dynamic_imports.get(attr_name)
-    if module_name is None:
-        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
-    try:
-        module = import_module(module_name, __package__)
-        if module_name == f".{attr_name}":
-            return module
-        else:
-            return getattr(module, attr_name)
-    except ImportError as e:
-        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
-    except AttributeError as e:
-        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
-
-
-def __dir__():
-    lazy_attrs = list(_dynamic_imports.keys())
-    return sorted(lazy_attrs)
-
+from .speech_to_text_streaming_flush_signal import SpeechToTextStreamingFlushSignal
+from .speech_to_text_streaming_high_vad_sensitivity import SpeechToTextStreamingHighVadSensitivity
+from .speech_to_text_streaming_language_code import SpeechToTextStreamingLanguageCode
+from .speech_to_text_streaming_mode import SpeechToTextStreamingMode
+from .speech_to_text_streaming_model import SpeechToTextStreamingModel
+from .speech_to_text_streaming_vad_signals import SpeechToTextStreamingVadSignals
 
 __all__ = [
     "SpeechToTextStreamingFlushSignal",
     "SpeechToTextStreamingHighVadSensitivity",
     "SpeechToTextStreamingLanguageCode",
+    "SpeechToTextStreamingMode",
+    "SpeechToTextStreamingModel",
     "SpeechToTextStreamingVadSignals",
 ]

sarvamai/speech_to_text_streaming/types/speech_to_text_streaming_mode.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+SpeechToTextStreamingMode = typing.Union[
+    typing.Literal["transcribe", "translate", "indic-en", "verbatim", "translit", "codemix"], typing.Any
+]

sarvamai/speech_to_text_streaming/types/speech_to_text_streaming_model.py

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