sarvamai 0.1.19a0__py3-none-any.whl → 0.1.19a2__py3-none-any.whl

sarvamai/core/client_wrapper.py

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

sarvamai/speech_to_text_streaming/socket_client.py

@@ -9,7 +9,10 @@ from ..core.events import EventEmitterMixin, EventType
 from ..core.pydantic_utilities import parse_obj_as
 from ..types.audio_data import AudioData
 from ..types.audio_message import AudioMessage
-from ..types.speech_to_text_streaming_response import SpeechToTextStreamingResponse
+from ..types.speech_to_text_streaming_response import (
+    SpeechToTextStreamingResponse,
+)
+from ..types.stt_flush_signal import SttFlushSignal
 
 SpeechToTextStreamingSocketClientResponse = typing.Union[SpeechToTextStreamingResponse]
 
@@ -22,7 +25,9 @@ class AsyncSpeechToTextStreamingSocketClient(EventEmitterMixin):
     async def __aiter__(self):
         async for message in self._websocket:
             message = json.loads(message) if isinstance(message, str) else message
-            yield parse_obj_as(SpeechToTextStreamingSocketClientResponse, message)  # type: ignore
+            yield parse_obj_as(
+                SpeechToTextStreamingSocketClientResponse, message
+            )  # type: ignore
 
     async def start_listening(self):
         """
@@ -37,8 +42,14 @@ class AsyncSpeechToTextStreamingSocketClient(EventEmitterMixin):
         self._emit(EventType.OPEN, None)
         try:
             async for raw_message in self._websocket:
-                raw_message = json.loads(raw_message) if isinstance(raw_message, str) else raw_message
-                parsed = parse_obj_as(SpeechToTextStreamingSocketClientResponse, raw_message)  # type: ignore
+                raw_message = (
+                    json.loads(raw_message)
+                    if isinstance(raw_message, str)
+                    else raw_message
+                )
+                parsed = parse_obj_as(
+                    SpeechToTextStreamingSocketClientResponse, raw_message
+                )  # type: ignore
                 self._emit(EventType.MESSAGE, parsed)
         except websockets.WebSocketException as exc:
             self._emit(EventType.ERROR, exc)
@@ -54,10 +65,26 @@ class AsyncSpeechToTextStreamingSocketClient(EventEmitterMixin):
         """
 
         return await self._send_speech_to_text_streaming_audio_message(
-            message=AudioMessage(audio=AudioData(data=audio, sample_rate=sample_rate, encoding=encoding))
+            message=AudioMessage(
+                audio=AudioData(data=audio, sample_rate=sample_rate, encoding=encoding)
+            )
         )
 
-    async def _send_speech_to_text_streaming_audio_message(self, message: AudioMessage) -> None:
+    async def flush(self, reason: typing.Optional[str] = None) -> None:
+        """
+        Signal to flush the audio buffer and force finalize partial transcriptions.
+        Use this to force processing of any remaining audio that hasn't been
+        transcribed yet.
+
+        :param reason: Optional reason for flushing (e.g., "end_of_segment",
+                      "manual_flush")
+        """
+        message = SttFlushSignal(reason=reason)
+        await self._send_model(message)
+
+    async def _send_speech_to_text_streaming_audio_message(
+        self, message: AudioMessage
+    ) -> None:
         """
         Send a message to the websocket connection.
         The message will be sent as a AudioMessage.
@@ -70,7 +97,9 @@ class AsyncSpeechToTextStreamingSocketClient(EventEmitterMixin):
         """
         data = await self._websocket.recv()
         data = json.loads(data) if isinstance(data, str) else data
-        return parse_obj_as(SpeechToTextStreamingSocketClientResponse, data)  # type: ignore
+        return parse_obj_as(
+            SpeechToTextStreamingSocketClientResponse, data
+        )  # type: ignore
 
     async def _send(self, data: typing.Any) -> None:
         """
@@ -95,7 +124,9 @@ class SpeechToTextStreamingSocketClient(EventEmitterMixin):
     def __iter__(self):
         for message in self._websocket:
             message = json.loads(message) if isinstance(message, str) else message
-            yield parse_obj_as(SpeechToTextStreamingSocketClientResponse, message)  # type: ignore
+            yield parse_obj_as(
+                SpeechToTextStreamingSocketClientResponse, message
+            )  # type: ignore
 
     def start_listening(self):
         """
@@ -110,8 +141,14 @@ class SpeechToTextStreamingSocketClient(EventEmitterMixin):
         self._emit(EventType.OPEN, None)
         try:
             for raw_message in self._websocket:
-                raw_message = json.loads(raw_message) if isinstance(raw_message, str) else raw_message
-                parsed = parse_obj_as(SpeechToTextStreamingSocketClientResponse, raw_message)  # type: ignore
+                raw_message = (
+                    json.loads(raw_message)
+                    if isinstance(raw_message, str)
+                    else raw_message
+                )
+                parsed = parse_obj_as(
+                    SpeechToTextStreamingSocketClientResponse, raw_message
+                )  # type: ignore
                 self._emit(EventType.MESSAGE, parsed)
         except websockets.WebSocketException as exc:
             self._emit(EventType.ERROR, exc)
@@ -126,18 +163,36 @@ class SpeechToTextStreamingSocketClient(EventEmitterMixin):
         :param sample_rate (Optional): Audio sample rate in Hz (default is 16000)
         """
         return self._send_speech_to_text_streaming_audio_message(
-            message=AudioMessage(audio=AudioData(data=audio, sample_rate=sample_rate, encoding=encoding))
+            message=AudioMessage(
+                audio=AudioData(data=audio, sample_rate=sample_rate, encoding=encoding)
+            )
         )
 
+    def flush(self, reason: typing.Optional[str] = None) -> None:
+        """
+        Signal to flush the audio buffer and force finalize partial transcriptions.
+        Use this to force processing of any remaining audio that hasn't been
+        transcribed yet.
+
+        :param reason: Optional reason for flushing (e.g., "end_of_segment",
+                      "manual_flush")
+        """
+        message = SttFlushSignal(reason=reason)
+        self._send_model(message)
+
     def recv(self) -> SpeechToTextStreamingSocketClientResponse:
         """
         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(SpeechToTextStreamingSocketClientResponse, data)  # type: ignore
+        return parse_obj_as(
+            SpeechToTextStreamingSocketClientResponse, data
+        )  # type: ignore
 
-    def _send_speech_to_text_streaming_audio_message(self, message: AudioMessage) -> None:
+    def _send_speech_to_text_streaming_audio_message(
+        self, message: AudioMessage
+    ) -> None:
         """
         Send a message to the websocket connection.
         The message will be sent as a AudioMessage.

{sarvamai-0.1.19a0.dist-info → sarvamai-0.1.19a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarvamai
-Version: 0.1.19a0
+Version: 0.1.19a2
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{sarvamai-0.1.19a0.dist-info → sarvamai-0.1.19a2.dist-info}/RECORD

@@ -5,7 +5,7 @@ sarvamai/chat/raw_client.py,sha256=A2kRuZcVWlJhyYCD7YKgqNkZEp3cYa1731KhRkhirU0,1
 sarvamai/client.py,sha256=J30X_os1lPf8Wml0KDFEf6p8VGHhgF_lf3nw1T2D3qo,8207
 sarvamai/core/__init__.py,sha256=YE2CtXeASe1RAbaI39twKWYKCuT4tW5is9HWHhJjR_g,1653
 sarvamai/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-sarvamai/core/client_wrapper.py,sha256=hO-qp8Qjk4fEujFbYae3L32zLbnMl_vjtZIWPhZixUo,2570
+sarvamai/core/client_wrapper.py,sha256=wE2VNWPeyHeA2m4ZKv3TvURhnIPSxCz9G48V5lZPbX4,2570
 sarvamai/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 sarvamai/core/events.py,sha256=HvKBdSoYcFetk7cgNXb7FxuY-FtY8NtUhZIN7mGVx8U,1159
 sarvamai/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
@@ -94,7 +94,7 @@ sarvamai/speech_to_text_job/raw_client.py,sha256=OZTPzMhAn-ckE_xKzfZ9QLsEX5EZVOJ
 sarvamai/speech_to_text_streaming/__init__.py,sha256=aVcI8kSlUpwLrPtA-4oUkuJ_D9Onha_qU49lA2vtEqU,515
 sarvamai/speech_to_text_streaming/client.py,sha256=15k-UH2ROia2EymXia0BxVKbzBz7k1Rh9k7L9hgO5Fg,9206
 sarvamai/speech_to_text_streaming/raw_client.py,sha256=69hAAD9mcpplo3gR7B5JXbo6Yys8bPT3DfSk1FjyI8Q,8387
-sarvamai/speech_to_text_streaming/socket_client.py,sha256=QI0vEjDvNTG9-EH99NolIuARYwD-P8Fms8rqKOD9HJQ,6419
+sarvamai/speech_to_text_streaming/socket_client.py,sha256=N0wjuQ37WtTggQysWfB8pOwLIwmIkGRglZO9N2y5yHI,7870
 sarvamai/speech_to_text_streaming/types/__init__.py,sha256=A99VJwG5fkik5SFTMDMKPZCjvKSaU2aQ02EbA5rrDlM,723
 sarvamai/speech_to_text_streaming/types/speech_to_text_streaming_flush_signal.py,sha256=dDJOBlzAjhuiSVqW2RHHY1f6xy0DU_Yoo9UV8-7MjnA,173
 sarvamai/speech_to_text_streaming/types/speech_to_text_streaming_high_vad_sensitivity.py,sha256=OwPwffa8TkLPGMnOTn5S7d-HmV8QmN3B7fHz8I1-VT8,180
@@ -215,6 +215,6 @@ sarvamai/types/transliterate_mode.py,sha256=1jSEMlGcoLkWuk12TgoOpSgwifa4rThGKZ1h
 sarvamai/types/transliterate_source_language.py,sha256=bSY9wJszF0sg-Cgg6F-YcWC8ly1mIlj9rqa15-jBtx8,283
 sarvamai/types/transliteration_response.py,sha256=yt-lzTbDeJ_ZL4I8kQa6oESxA9ebeJJY7LfFHpdEsmM,815
 sarvamai/version.py,sha256=Qkp3Ee9YH-O9RTix90e0i7iNrFAGN-QDt2AFwGA4n8k,75
-sarvamai-0.1.19a0.dist-info/METADATA,sha256=huhmFTOKlTUOLAlRGDWJn0Ah_qw2zNq01USmrljNH3w,26753
-sarvamai-0.1.19a0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-sarvamai-0.1.19a0.dist-info/RECORD,,
+sarvamai-0.1.19a2.dist-info/METADATA,sha256=YaTRDFMvJFy1UF8DtXSN2itvtgQqEcrKns2JglAUHn8,26753
+sarvamai-0.1.19a2.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+sarvamai-0.1.19a2.dist-info/RECORD,,