sarvamai 0.1.11a6__tar.gz → 0.1.12a1__tar.gz

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarvamai
-Version: 0.1.11a6
+Version: 0.1.12a1
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/pyproject.toml

@@ -3,7 +3,7 @@ name = "sarvamai"
 
 [tool.poetry]
 name = "sarvamai"
-version = "0.1.11a6"
+version = "0.1.12a1"
 description = ""
 readme = "README.md"
 authors = []

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/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.11a6",
+            "User-Agent": "sarvamai/0.1.12a1",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.11a6",
+            "X-Fern-SDK-Version": "0.1.12a1",
             **(self.get_custom_headers() or {}),
         }
         headers["api-subscription-key"] = self.api_subscription_key

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/core/events.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import inspect
 import typing
 from enum import Enum
 
@@ -19,7 +20,7 @@ class EventEmitterMixin:
     def __init__(self) -> None:
         self._callbacks: typing.Dict[EventType, typing.List[typing.Callable]] = {}
 
-    def on(self, event_name: EventType, callback: typing.Callable[[typing.Any], None]) -> None:
+    def on(self, event_name: EventType, callback: typing.Callable[[typing.Any], typing.Any]) -> None:
         if event_name not in self._callbacks:
             self._callbacks[event_name] = []
         self._callbacks[event_name].append(callback)
@@ -28,3 +29,10 @@ class EventEmitterMixin:
         if event_name in self._callbacks:
             for cb in self._callbacks[event_name]:
                 cb(data)
+
+    async def _emit_async(self, event_name: EventType, data: typing.Any) -> None:
+        if event_name in self._callbacks:
+            for cb in self._callbacks[event_name]:
+                res = cb(data)
+                if inspect.isawaitable(res):
+                    await res

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/speech_to_text_streaming/client.py

@@ -4,7 +4,7 @@ import typing
 from contextlib import asynccontextmanager, contextmanager
 
 import httpx
-import websockets
+import websockets.exceptions
 import websockets.sync.client as websockets_sync_client
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -16,6 +16,11 @@ from .types.speech_to_text_streaming_language_code import SpeechToTextStreamingL
 from .types.speech_to_text_streaming_model import SpeechToTextStreamingModel
 from .types.speech_to_text_streaming_vad_signals import SpeechToTextStreamingVadSignals
 
+try:
+    from websockets.legacy.client import connect as websockets_client_connect  # type: ignore
+except ImportError:
+    from websockets import connect as websockets_client_connect  # type: ignore
+
 
 class SpeechToTextStreamingClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -174,7 +179,7 @@ class AsyncSpeechToTextStreamingClient:
         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:
+            async with websockets_client_connect(ws_url, extra_headers=headers) as protocol:
                 yield AsyncSpeechToTextStreamingSocketClient(websocket=protocol)
         except websockets.exceptions.InvalidStatusCode as exc:
             status_code: int = exc.status_code

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/speech_to_text_streaming/raw_client.py

@@ -4,7 +4,7 @@ import typing
 from contextlib import asynccontextmanager, contextmanager
 
 import httpx
-import websockets
+import websockets.exceptions
 import websockets.sync.client as websockets_sync_client
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -15,6 +15,11 @@ from .types.speech_to_text_streaming_language_code import SpeechToTextStreamingL
 from .types.speech_to_text_streaming_model import SpeechToTextStreamingModel
 from .types.speech_to_text_streaming_vad_signals import SpeechToTextStreamingVadSignals
 
+try:
+    from websockets.legacy.client import connect as websockets_client_connect  # type: ignore
+except ImportError:
+    from websockets import connect as websockets_client_connect  # type: ignore
+
 
 class RawSpeechToTextStreamingClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -151,7 +156,7 @@ class AsyncRawSpeechToTextStreamingClient:
         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:
+            async with websockets_client_connect(ws_url, extra_headers=headers) as protocol:
                 yield AsyncSpeechToTextStreamingSocketClient(websocket=protocol)
         except websockets.exceptions.InvalidStatusCode as exc:
             status_code: int = exc.status_code

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/speech_to_text_translate_streaming/client.py

@@ -4,7 +4,7 @@ import typing
 from contextlib import asynccontextmanager, contextmanager
 
 import httpx
-import websockets
+import websockets.exceptions
 import websockets.sync.client as websockets_sync_client
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -17,6 +17,11 @@ from .types.speech_to_text_translate_streaming_high_vad_sensitivity import (
 from .types.speech_to_text_translate_streaming_model import SpeechToTextTranslateStreamingModel
 from .types.speech_to_text_translate_streaming_vad_signals import SpeechToTextTranslateStreamingVadSignals
 
+try:
+    from websockets.legacy.client import connect as websockets_client_connect  # type: ignore
+except ImportError:
+    from websockets import connect as websockets_client_connect  # type: ignore
+
 
 class SpeechToTextTranslateStreamingClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -163,7 +168,7 @@ class AsyncSpeechToTextTranslateStreamingClient:
         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:
+            async with websockets_client_connect(ws_url, extra_headers=headers) as protocol:
                 yield AsyncSpeechToTextTranslateStreamingSocketClient(websocket=protocol)
         except websockets.exceptions.InvalidStatusCode as exc:
             status_code: int = exc.status_code

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/speech_to_text_translate_streaming/raw_client.py

@@ -4,7 +4,7 @@ import typing
 from contextlib import asynccontextmanager, contextmanager
 
 import httpx
-import websockets
+import websockets.exceptions
 import websockets.sync.client as websockets_sync_client
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -16,6 +16,11 @@ from .types.speech_to_text_translate_streaming_high_vad_sensitivity import (
 from .types.speech_to_text_translate_streaming_model import SpeechToTextTranslateStreamingModel
 from .types.speech_to_text_translate_streaming_vad_signals import SpeechToTextTranslateStreamingVadSignals
 
+try:
+    from websockets.legacy.client import connect as websockets_client_connect  # type: ignore
+except ImportError:
+    from websockets import connect as websockets_client_connect  # type: ignore
+
 
 class RawSpeechToTextTranslateStreamingClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -140,7 +145,7 @@ class AsyncRawSpeechToTextTranslateStreamingClient:
         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:
+            async with websockets_client_connect(ws_url, extra_headers=headers) as protocol:
                 yield AsyncSpeechToTextTranslateStreamingSocketClient(websocket=protocol)
         except websockets.exceptions.InvalidStatusCode as exc:
             status_code: int = exc.status_code

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/text_to_speech_streaming/client.py

@@ -4,7 +4,7 @@ import typing
 from contextlib import asynccontextmanager, contextmanager
 
 import httpx
-import websockets
+import websockets.exceptions
 import websockets.sync.client as websockets_sync_client
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -12,6 +12,11 @@ from ..core.request_options import RequestOptions
 from .raw_client import AsyncRawTextToSpeechStreamingClient, RawTextToSpeechStreamingClient
 from .socket_client import AsyncTextToSpeechStreamingSocketClient, TextToSpeechStreamingSocketClient
 
+try:
+    from websockets.legacy.client import connect as websockets_client_connect  # type: ignore
+except ImportError:
+    from websockets import connect as websockets_client_connect  # type: ignore
+
 
 class TextToSpeechStreamingClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -136,7 +141,7 @@ class AsyncTextToSpeechStreamingClient:
         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:
+            async with websockets_client_connect(ws_url, extra_headers=headers) as protocol:
                 yield AsyncTextToSpeechStreamingSocketClient(websocket=protocol)
         except websockets.exceptions.InvalidStatusCode as exc:
             status_code: int = exc.status_code

{sarvamai-0.1.11a6 → sarvamai-0.1.12a1}/src/sarvamai/text_to_speech_streaming/raw_client.py

@@ -4,13 +4,18 @@ import typing
 from contextlib import asynccontextmanager, contextmanager
 
 import httpx
-import websockets
+import websockets.exceptions
 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
 
+try:
+    from websockets.legacy.client import connect as websockets_client_connect  # type: ignore
+except ImportError:
+    from websockets import connect as websockets_client_connect  # type: ignore
+
 
 class RawTextToSpeechStreamingClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -113,7 +118,7 @@ class AsyncRawTextToSpeechStreamingClient:
         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:
+            async with websockets_client_connect(ws_url, extra_headers=headers) as protocol:
                 yield AsyncTextToSpeechStreamingSocketClient(websocket=protocol)
         except websockets.exceptions.InvalidStatusCode as exc:
             status_code: int = exc.status_code