sarvamai 0.1.5a0__py3-none-any.whl → 0.1.5a1__py3-none-any.whl

sarvamai/chat/raw_client.py

@@ -111,6 +111,7 @@ class RawChatClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "v1/chat/completions",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "messages": convert_and_respect_annotation_metadata(
@@ -291,6 +292,7 @@ class AsyncRawChatClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "v1/chat/completions",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "messages": convert_and_respect_annotation_metadata(

sarvamai/client.py

@@ -1,7 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing
 from .environment import SarvamAIEnvironment
+import typing
 import os
 import httpx
 from .core.api_error import ApiError
@@ -23,9 +23,6 @@ class SarvamAI:
 
     Parameters
     ----------
-    base_url : typing.Optional[str]
-        The base url to use for requests from the client.
-
     environment : SarvamAIEnvironment
         The environment to use for requests from the client. from .environment import SarvamAIEnvironment
 
@@ -57,7 +54,6 @@ class SarvamAI:
     def __init__(
         self,
         *,
-        base_url: typing.Optional[str] = None,
         environment: SarvamAIEnvironment = SarvamAIEnvironment.PRODUCTION,
         api_subscription_key: typing.Optional[str] = os.getenv("SARVAM_API_KEY"),
         timeout: typing.Optional[float] = None,
@@ -72,7 +68,7 @@ class SarvamAI:
                 body="The client must be instantiated be either passing in api_subscription_key or setting SARVAM_API_KEY"
             )
         self._client_wrapper = SyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
+            environment=environment,
             api_subscription_key=api_subscription_key,
             httpx_client=httpx_client
             if httpx_client is not None
@@ -93,9 +89,6 @@ class AsyncSarvamAI:
 
     Parameters
     ----------
-    base_url : typing.Optional[str]
-        The base url to use for requests from the client.
-
     environment : SarvamAIEnvironment
         The environment to use for requests from the client. from .environment import SarvamAIEnvironment
 
@@ -127,7 +120,6 @@ class AsyncSarvamAI:
     def __init__(
         self,
         *,
-        base_url: typing.Optional[str] = None,
         environment: SarvamAIEnvironment = SarvamAIEnvironment.PRODUCTION,
         api_subscription_key: typing.Optional[str] = os.getenv("SARVAM_API_KEY"),
         timeout: typing.Optional[float] = None,
@@ -142,7 +134,7 @@ class AsyncSarvamAI:
                 body="The client must be instantiated be either passing in api_subscription_key or setting SARVAM_API_KEY"
             )
         self._client_wrapper = AsyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
+            environment=environment,
             api_subscription_key=api_subscription_key,
             httpx_client=httpx_client
             if httpx_client is not None
@@ -155,12 +147,3 @@ class AsyncSarvamAI:
         self.speech_to_text = AsyncSpeechToTextClient(client_wrapper=self._client_wrapper)
         self.text_to_speech = AsyncTextToSpeechClient(client_wrapper=self._client_wrapper)
         self.chat = AsyncChatClient(client_wrapper=self._client_wrapper)
-
-
-def _get_base_url(*, base_url: typing.Optional[str] = None, environment: SarvamAIEnvironment) -> str:
-    if base_url is not None:
-        return base_url
-    elif environment is not None:
-        return environment.value
-    else:
-        raise Exception("Please pass in either base_url or environment to construct the client")

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/core/client_wrapper.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from ..environment import SarvamAIEnvironment
 import typing
 import httpx
 from .http_client import HttpClient
@@ -7,23 +8,25 @@ from .http_client import AsyncHttpClient
 
 
 class BaseClientWrapper:
-    def __init__(self, *, api_subscription_key: str, base_url: str, timeout: typing.Optional[float] = None):
+    def __init__(
+        self, *, api_subscription_key: str, environment: SarvamAIEnvironment, timeout: typing.Optional[float] = None
+    ):
         self.api_subscription_key = api_subscription_key
-        self._base_url = base_url
+        self._environment = environment
         self._timeout = timeout
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "sarvamai/0.1.5a0",
+            "User-Agent": "sarvamai/0.1.5a1",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.5a0",
+            "X-Fern-SDK-Version": "0.1.5a1",
         }
         headers["api-subscription-key"] = self.api_subscription_key
         return headers
 
-    def get_base_url(self) -> str:
-        return self._base_url
+    def get_environment(self) -> SarvamAIEnvironment:
+        return self._environment
 
     def get_timeout(self) -> typing.Optional[float]:
         return self._timeout
@@ -34,16 +37,13 @@ class SyncClientWrapper(BaseClientWrapper):
         self,
         *,
         api_subscription_key: str,
-        base_url: str,
+        environment: SarvamAIEnvironment,
         timeout: typing.Optional[float] = None,
         httpx_client: httpx.Client,
     ):
-        super().__init__(api_subscription_key=api_subscription_key, base_url=base_url, timeout=timeout)
+        super().__init__(api_subscription_key=api_subscription_key, environment=environment, timeout=timeout)
         self.httpx_client = HttpClient(
-            httpx_client=httpx_client,
-            base_headers=self.get_headers,
-            base_timeout=self.get_timeout,
-            base_url=self.get_base_url,
+            httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
         )
 
 
@@ -52,14 +52,11 @@ class AsyncClientWrapper(BaseClientWrapper):
         self,
         *,
         api_subscription_key: str,
-        base_url: str,
+        environment: SarvamAIEnvironment,
         timeout: typing.Optional[float] = None,
         httpx_client: httpx.AsyncClient,
     ):
-        super().__init__(api_subscription_key=api_subscription_key, base_url=base_url, timeout=timeout)
+        super().__init__(api_subscription_key=api_subscription_key, environment=environment, timeout=timeout)
         self.httpx_client = AsyncHttpClient(
-            httpx_client=httpx_client,
-            base_headers=self.get_headers,
-            base_timeout=self.get_timeout,
-            base_url=self.get_base_url,
+            httpx_client=httpx_client, base_headers=self.get_headers, base_timeout=self.get_timeout
         )

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)

sarvamai/environment.py

@@ -1,7 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
+from __future__ import annotations
 
 
-class SarvamAIEnvironment(enum.Enum):
-    PRODUCTION = "https://api.sarvam.ai"
+class SarvamAIEnvironment:
+    PRODUCTION: SarvamAIEnvironment
+
+    def __init__(self, *, base: str, production: str):
+        self.base = base
+        self.production = production
+
+
+SarvamAIEnvironment.PRODUCTION = SarvamAIEnvironment(base="https://api.sarvam.ai", production="wss://api.sarvam.ai")

sarvamai/speech_to_text/raw_client.py

@@ -80,6 +80,7 @@ class RawSpeechToTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "speech-to-text",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             data={
                 "model": model,
@@ -210,6 +211,7 @@ class RawSpeechToTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "speech-to-text-translate",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             data={
                 "prompt": prompt,
@@ -350,6 +352,7 @@ class AsyncRawSpeechToTextClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "speech-to-text",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             data={
                 "model": model,
@@ -480,6 +483,7 @@ class AsyncRawSpeechToTextClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "speech-to-text-translate",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             data={
                 "prompt": prompt,

sarvamai/text/raw_client.py

@@ -145,6 +145,7 @@ class RawTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "translate",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -249,6 +250,7 @@ class RawTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "text-lid",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -422,6 +424,7 @@ class RawTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "transliterate",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -616,6 +619,7 @@ class AsyncRawTextClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "translate",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -720,6 +724,7 @@ class AsyncRawTextClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "text-lid",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -893,6 +898,7 @@ class AsyncRawTextClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "transliterate",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,

sarvamai/text_to_speech/raw_client.py

@@ -97,6 +97,7 @@ class RawTextToSpeechClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "text-to-speech",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "text": text,
@@ -254,6 +255,7 @@ class AsyncRawTextToSpeechClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "text-to-speech",
+            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "text": text,

{sarvamai-0.1.5a0.dist-info → sarvamai-0.1.5a1.dist-info}/METADATA

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

{sarvamai-0.1.5a0.dist-info → sarvamai-0.1.5a1.dist-info}/RECORD

@@ -1,12 +1,13 @@
 sarvamai/__init__.py,sha256=j0eB2isvuZ1spRjp6coCb2mtTv45O636aYSoK79eddQ,5071
 sarvamai/chat/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 sarvamai/chat/client.py,sha256=4ADAA7HC-aoD-fO3ZLd0iodBOzBc_Xyib4E-7NbDkik,11791
-sarvamai/chat/raw_client.py,sha256=G7LVq-I0qZ6xsjoG_dJe2lAR_jAVU368eOn1MscYKbI,17825
-sarvamai/client.py,sha256=NrpjDx6AusBR2AfUWUrFSfteIlYkUhv5ahjdlCBNPeM,6936
-sarvamai/core/__init__.py,sha256=JZDozMOUlhvOLnOO6kGBxye5Ctfg_KDOjeUTSUoMmV8,1542
+sarvamai/chat/raw_client.py,sha256=GMvPA408o9eNJa-34apLRUpa53j8wkqR1dTj2vaIxFg,17957
+sarvamai/client.py,sha256=g9hNkKQeENNu7LoXXYZIjNJI_klN0ajb9kyRNHzeg4U,6235
+sarvamai/core/__init__.py,sha256=BCnS6NhY518p_t-WmBsa3qzwS-pmg3f8Thea-VnMaao,1633
 sarvamai/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-sarvamai/core/client_wrapper.py,sha256=EbW-_PB6M6SvRkHTxRiBdKcfjiVWejX_afBkAB2cTqg,2074
+sarvamai/core/client_wrapper.py,sha256=j1HRTQZXiKW0Td4aplwANc5Z357tVVTJbmmZfR0OMvs,2101
 sarvamai/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
+sarvamai/core/events.py,sha256=j7VWXgMpOsjCXdzY22wIhI7Q-v5InZ4WchRzA88x_Sk,856
 sarvamai/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 sarvamai/core/http_client.py,sha256=cKs2w0ybDBk1wHQf-fTALm_MmvaMe3cZKcYJxqmCxkE,19539
 sarvamai/core/http_response.py,sha256=Argerx6pFZdCpe637AT3EejZw5zlMGgsshZq6rb7krQ,1005
@@ -16,7 +17,7 @@ sarvamai/core/query_encoder.py,sha256=ekulqNd0j8TgD7ox-Qbz7liqX8-KP9blvT9DsRCenY
 sarvamai/core/remove_none_from_dict.py,sha256=EU9SGgYidWq7SexuJbNs4-PZ-5Bl3Vppd864mS6vQZw,342
 sarvamai/core/request_options.py,sha256=h0QUNCFVdCW_7GclVySCAY2w4NhtXVBUCmHgmzaxpcg,1681
 sarvamai/core/serialization.py,sha256=ECL3bvv_0i7U4uvPidZCNel--MUbA0iq0aGcNKi3kws,9818
-sarvamai/environment.py,sha256=86s94KkJUHmMqmGiehN2SiNYxHoBFgIxCHH41T_BZNs,158
+sarvamai/environment.py,sha256=hdwTU767BqRgSMLiAOocY_Vpw8V2N_hAy3yhoK8VFS0,402
 sarvamai/errors/__init__.py,sha256=Rjcid1bNnxJ95F5ACvzU2L5aYJNuJh4OpRG1JSyj8G0,575
 sarvamai/errors/bad_request_error.py,sha256=_EbO8mWqN9kFZPvIap8qa1lL_EWkRcsZe1HKV9GDWJY,264
 sarvamai/errors/forbidden_error.py,sha256=QO1kKlhClAPES6zsEK7g9pglWnxn3KWaOCAawWOg6Aw,263
@@ -49,13 +50,13 @@ sarvamai/requests/translation_response.py,sha256=zEL7LhnLIxwAz89NOkjkSDdnR1LaUa8
 sarvamai/requests/transliteration_response.py,sha256=9ZYsJ-9MSjyRWatEbpsI69Hx86JVACCZCU3BEvN77M4,451
 sarvamai/speech_to_text/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 sarvamai/speech_to_text/client.py,sha256=FnZnTi0XDOnVcD4pxcXmb5brVpU6kOmIM9-CwYyuaWM,11482
-sarvamai/speech_to_text/raw_client.py,sha256=k9J82zyF1oCHqkE_sGZUuxABPs1q6rVdLX3J-yByixE,23674
+sarvamai/speech_to_text/raw_client.py,sha256=Y047f5kF0yGfHOpoCtrN3O7nNhh7QQWzZn-_c322gXU,23938
 sarvamai/text/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 sarvamai/text/client.py,sha256=0D15RtAD5EvSFSLdzCZEBPQ5K4nGJv2jf0BI1Mlr2mg,28668
-sarvamai/text/raw_client.py,sha256=LQT2yNlNxP7NVw7bKZ6vkUy1tkDs1-IXaf7sjFY5UkI,44227
+sarvamai/text/raw_client.py,sha256=pj00d9dcZLc6w_W_FaFCLDizBitbPamAY1Sma0ySFgc,44623
 sarvamai/text_to_speech/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 sarvamai/text_to_speech/client.py,sha256=1T_P7dpVmZBzUNPbKFGBDPXRwpDV-5zY6KcEOQoLrs8,9215
-sarvamai/text_to_speech/raw_client.py,sha256=VIgGQ8WKQOY9RTAEfboWeCe1zZM1l8wZ3jOUAnYrxxg,14312
+sarvamai/text_to_speech/raw_client.py,sha256=8vpIMUBJ1GxIf1SKZdLWFKQ305XXpTFqX3ohCzOclA0,14444
 sarvamai/types/__init__.py,sha256=K3_vaMlZnhW4W0tO5jUHZegxHpg_7wOyB1iDcnjJ2lo,4022
 sarvamai/types/chat_completion_request_assistant_message.py,sha256=EUkQKNCaBvrL2PIOIN9Q4GXIV7uJ6C86k-BreA-83uE,660
 sarvamai/types/chat_completion_request_message.py,sha256=4M8Rkc1kNHfORTNDFrc8wbq8eR37kq5nrsd-wNvL4kw,1673
@@ -101,6 +102,6 @@ sarvamai/types/translatiterate_target_language.py,sha256=S8yfYS-BtEvSdQw3sBbQckb
 sarvamai/types/transliterate_source_language.py,sha256=6BVDfik4WSY40DiQPFH32SYjXJBMzrJCouxOJNd4f9A,283
 sarvamai/types/transliteration_response.py,sha256=RZcsRP73Si24t4LF8MZCw_qZyVbX3gvMz88O41oMk8s,851
 sarvamai/version.py,sha256=Qkp3Ee9YH-O9RTix90e0i7iNrFAGN-QDt2AFwGA4n8k,75
-sarvamai-0.1.5a0.dist-info/METADATA,sha256=j2wI-jSnp0cuLckrfhhnSkdQbM939oEuCIeWqbUsZrw,4792
-sarvamai-0.1.5a0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-sarvamai-0.1.5a0.dist-info/RECORD,,
+sarvamai-0.1.5a1.dist-info/METADATA,sha256=pLMlW-BprjGituUPdhqATkpPg94jeUXS7BKgH3VqgOI,4792
+sarvamai-0.1.5a1.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+sarvamai-0.1.5a1.dist-info/RECORD,,