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

sarvamai/chat/raw_client.py

@@ -111,7 +111,6 @@ 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(
@@ -292,7 +291,6 @@ 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.
 
-from .environment import SarvamAIEnvironment
 import typing
+from .environment import SarvamAIEnvironment
 import os
 import httpx
 from .core.api_error import ApiError
@@ -23,6 +23,9 @@ 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
 
@@ -54,6 +57,7 @@ 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,
@@ -68,7 +72,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(
-            environment=environment,
+            base_url=_get_base_url(base_url=base_url, environment=environment),
             api_subscription_key=api_subscription_key,
             httpx_client=httpx_client
             if httpx_client is not None
@@ -89,6 +93,9 @@ 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
 
@@ -120,6 +127,7 @@ 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,
@@ -134,7 +142,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(
-            environment=environment,
+            base_url=_get_base_url(base_url=base_url, environment=environment),
             api_subscription_key=api_subscription_key,
             httpx_client=httpx_client
             if httpx_client is not None
@@ -147,3 +155,12 @@ 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/client_wrapper.py

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

sarvamai/environment.py

@@ -1,14 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
+import enum
 
 
-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")
+class SarvamAIEnvironment(enum.Enum):
+    PRODUCTION = "https://api.sarvam.ai"

sarvamai/speech_to_text/raw_client.py

@@ -80,7 +80,6 @@ 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,
@@ -211,7 +210,6 @@ 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,
@@ -352,7 +350,6 @@ 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,
@@ -483,7 +480,6 @@ 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,7 +145,6 @@ class RawTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "translate",
-            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -250,7 +249,6 @@ class RawTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "text-lid",
-            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -424,7 +422,6 @@ class RawTextClient:
         """
         _response = self._client_wrapper.httpx_client.request(
             "transliterate",
-            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -619,7 +616,6 @@ class AsyncRawTextClient:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "translate",
-            base_url=self._client_wrapper.get_environment().base,
             method="POST",
             json={
                 "input": input,
@@ -724,7 +720,6 @@ 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,
@@ -898,7 +893,6 @@ 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,7 +97,6 @@ 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,
@@ -255,7 +254,6 @@ 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.4a0.dist-info → sarvamai-0.1.5a0.dist-info}/METADATA

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

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

@@ -1,11 +1,11 @@
 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=GMvPA408o9eNJa-34apLRUpa53j8wkqR1dTj2vaIxFg,17957
-sarvamai/client.py,sha256=g9hNkKQeENNu7LoXXYZIjNJI_klN0ajb9kyRNHzeg4U,6235
+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/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-sarvamai/core/client_wrapper.py,sha256=kv2Zk2Pz-OoaD2yvpqYtv1lRZgKjPjojLIKmd5phP4o,2101
+sarvamai/core/client_wrapper.py,sha256=EbW-_PB6M6SvRkHTxRiBdKcfjiVWejX_afBkAB2cTqg,2074
 sarvamai/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 sarvamai/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 sarvamai/core/http_client.py,sha256=cKs2w0ybDBk1wHQf-fTALm_MmvaMe3cZKcYJxqmCxkE,19539
@@ -16,7 +16,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=hdwTU767BqRgSMLiAOocY_Vpw8V2N_hAy3yhoK8VFS0,402
+sarvamai/environment.py,sha256=86s94KkJUHmMqmGiehN2SiNYxHoBFgIxCHH41T_BZNs,158
 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 +49,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=Y047f5kF0yGfHOpoCtrN3O7nNhh7QQWzZn-_c322gXU,23938
+sarvamai/speech_to_text/raw_client.py,sha256=k9J82zyF1oCHqkE_sGZUuxABPs1q6rVdLX3J-yByixE,23674
 sarvamai/text/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 sarvamai/text/client.py,sha256=0D15RtAD5EvSFSLdzCZEBPQ5K4nGJv2jf0BI1Mlr2mg,28668
-sarvamai/text/raw_client.py,sha256=pj00d9dcZLc6w_W_FaFCLDizBitbPamAY1Sma0ySFgc,44623
+sarvamai/text/raw_client.py,sha256=LQT2yNlNxP7NVw7bKZ6vkUy1tkDs1-IXaf7sjFY5UkI,44227
 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=8vpIMUBJ1GxIf1SKZdLWFKQ305XXpTFqX3ohCzOclA0,14444
+sarvamai/text_to_speech/raw_client.py,sha256=VIgGQ8WKQOY9RTAEfboWeCe1zZM1l8wZ3jOUAnYrxxg,14312
 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 +101,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.4a0.dist-info/METADATA,sha256=SHJFwnXz3GzexaRGLTfT4f4YQi0-CPLPxhRS76aiom0,4792
-sarvamai-0.1.4a0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-sarvamai-0.1.4a0.dist-info/RECORD,,
+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,,