sarvamai 0.1.4a0__tar.gz → 0.1.5a1__tar.gz

{sarvamai-0.1.4a0 → sarvamai-0.1.5a1}/PKG-INFO

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

{sarvamai-0.1.4a0 → sarvamai-0.1.5a1}/pyproject.toml

@@ -3,7 +3,7 @@ name = "sarvamai"
 
 [tool.poetry]
 name = "sarvamai"
-version = "0.1.4a0"
+version = "0.1.5a1"
 description = ""
 readme = "README.md"
 authors = []

{sarvamai-0.1.4a0 → sarvamai-0.1.5a1}/src/sarvamai/core/__init__.py

@@ -3,6 +3,7 @@
 from .api_error import ApiError
 from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
 from .datetime_utils import serialize_datetime
+from .events import EventEmitterMixin, EventType
 from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
 from .http_client import AsyncHttpClient, HttpClient
 from .http_response import AsyncHttpResponse, HttpResponse
@@ -27,6 +28,8 @@ __all__ = [
     "AsyncHttpClient",
     "AsyncHttpResponse",
     "BaseClientWrapper",
+    "EventEmitterMixin",
+    "EventType",
     "FieldMetadata",
     "File",
     "HttpClient",

{sarvamai-0.1.4a0 → sarvamai-0.1.5a1}/src/sarvamai/core/client_wrapper.py

@@ -17,10 +17,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "sarvamai/0.1.4a0",
+            "User-Agent": "sarvamai/0.1.5a1",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.4a0",
+            "X-Fern-SDK-Version": "0.1.5a1",
         }
         headers["api-subscription-key"] = self.api_subscription_key
         return headers

sarvamai-0.1.5a1/src/sarvamai/core/events.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from enum import Enum
+
+
+class EventType(str, Enum):
+    OPEN = "open"
+    MESSAGE = "message"
+    ERROR = "error"
+    CLOSE = "close"
+
+
+class EventEmitterMixin:
+    """
+    Simple mixin for registering and emitting events.
+    """
+
+    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:
+        if event_name not in self._callbacks:
+            self._callbacks[event_name] = []
+        self._callbacks[event_name].append(callback)
+
+    def _emit(self, event_name: EventType, data: typing.Any) -> None:
+        if event_name in self._callbacks:
+            for cb in self._callbacks[event_name]:
+                cb(data)