sarvamai 0.1.5a1__tar.gz → 0.1.5a4__tar.gz

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sarvamai
-Version: 0.1.5a1
+Version: 0.1.5a4
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers
@@ -58,7 +58,7 @@ client = SarvamAI(
 client.text.translate(
     input="input",
     source_language_code="auto",
-    target_language_code="en-IN",
+    target_language_code="bn-IN",
 )
 ```
 
@@ -80,7 +80,7 @@ async def main() -> None:
     await client.text.translate(
         input="input",
         source_language_code="auto",
-        target_language_code="en-IN",
+        target_language_code="bn-IN",
     )
 
 

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/README.md

@@ -32,7 +32,7 @@ client = SarvamAI(
 client.text.translate(
     input="input",
     source_language_code="auto",
-    target_language_code="en-IN",
+    target_language_code="bn-IN",
 )
 ```
 
@@ -54,7 +54,7 @@ async def main() -> None:
     await client.text.translate(
         input="input",
         source_language_code="auto",
-        target_language_code="en-IN",
+        target_language_code="bn-IN",
     )
 
 

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/pyproject.toml

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

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/src/sarvamai/__init__.py

@@ -39,12 +39,12 @@ from .types import (
     TimestampsModel,
     TranslateMode,
     TranslateModel,
-    TranslatePostprocessing,
     TranslateSourceLanguage,
     TranslateSpeakerGender,
     TranslateTargetLanguage,
     TranslationResponse,
     TranslatiterateTargetLanguage,
+    TransliterateMode,
     TransliterateSourceLanguage,
     TransliterationResponse,
 )
@@ -154,13 +154,13 @@ __all__ = [
     "TooManyRequestsError",
     "TranslateMode",
     "TranslateModel",
-    "TranslatePostprocessing",
     "TranslateSourceLanguage",
     "TranslateSpeakerGender",
     "TranslateTargetLanguage",
     "TranslationResponse",
     "TranslationResponseParams",
     "TranslatiterateTargetLanguage",
+    "TransliterateMode",
     "TransliterateSourceLanguage",
     "TransliterationResponse",
     "TransliterationResponseParams",

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/src/sarvamai/chat/client.py

@@ -33,7 +33,6 @@ class ChatClient:
     def completions(
         self,
         *,
-        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -53,9 +52,6 @@ class ChatClient:
 
         Parameters
         ----------
-        authorization : str
-            Bearer token for authentication. Format: "Bearer <token>"
-
         messages : typing.Sequence[ChatCompletionRequestMessageParams]
             A list of messages comprising the conversation so far.
 
@@ -119,12 +115,10 @@ class ChatClient:
             api_subscription_key="YOUR_API_SUBSCRIPTION_KEY",
         )
         client.chat.completions(
-            authorization="Authorization",
             messages=[{"content": "content", "role": "assistant"}],
         )
         """
         response = self._raw_client.completions(
-            authorization=authorization,
             messages=messages,
             temperature=temperature,
             top_p=top_p,
@@ -160,7 +154,6 @@ class AsyncChatClient:
     async def completions(
         self,
         *,
-        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -180,9 +173,6 @@ class AsyncChatClient:
 
         Parameters
         ----------
-        authorization : str
-            Bearer token for authentication. Format: "Bearer <token>"
-
         messages : typing.Sequence[ChatCompletionRequestMessageParams]
             A list of messages comprising the conversation so far.
 
@@ -251,7 +241,6 @@ class AsyncChatClient:
 
         async def main() -> None:
             await client.chat.completions(
-                authorization="Authorization",
                 messages=[{"content": "content", "role": "assistant"}],
             )
 
@@ -259,7 +248,6 @@ class AsyncChatClient:
         asyncio.run(main())
         """
         response = await self._raw_client.completions(
-            authorization=authorization,
             messages=messages,
             temperature=temperature,
             top_p=top_p,

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/src/sarvamai/chat/raw_client.py

@@ -31,7 +31,6 @@ class RawChatClient:
     def completions(
         self,
         *,
-        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -51,9 +50,6 @@ class RawChatClient:
 
         Parameters
         ----------
-        authorization : str
-            Bearer token for authentication. Format: "Bearer <token>"
-
         messages : typing.Sequence[ChatCompletionRequestMessageParams]
             A list of messages comprising the conversation so far.
 
@@ -111,7 +107,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(
@@ -134,7 +129,6 @@ class RawChatClient:
             },
             headers={
                 "content-type": "application/json",
-                "Authorization": str(authorization) if authorization is not None else None,
             },
             request_options=request_options,
             omit=OMIT,
@@ -212,7 +206,6 @@ class AsyncRawChatClient:
     async def completions(
         self,
         *,
-        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -232,9 +225,6 @@ class AsyncRawChatClient:
 
         Parameters
         ----------
-        authorization : str
-            Bearer token for authentication. Format: "Bearer <token>"
-
         messages : typing.Sequence[ChatCompletionRequestMessageParams]
             A list of messages comprising the conversation so far.
 
@@ -292,7 +282,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(
@@ -315,7 +304,6 @@ class AsyncRawChatClient:
             },
             headers={
                 "content-type": "application/json",
-                "Authorization": str(authorization) if authorization is not None else None,
             },
             request_options=request_options,
             omit=OMIT,

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/src/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-0.1.5a1 → sarvamai-0.1.5a4}/src/sarvamai/core/__init__.py

@@ -3,7 +3,6 @@
 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
@@ -28,8 +27,6 @@ __all__ = [
     "AsyncHttpClient",
     "AsyncHttpResponse",
     "BaseClientWrapper",
-    "EventEmitterMixin",
-    "EventType",
     "FieldMetadata",
     "File",
     "HttpClient",

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/src/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.5a1",
+            "User-Agent": "sarvamai/0.1.5a4",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "sarvamai",
-            "X-Fern-SDK-Version": "0.1.5a1",
+            "X-Fern-SDK-Version": "0.1.5a4",
         }
         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-0.1.5a4/src/sarvamai/environment.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+
+
+class SarvamAIEnvironment(enum.Enum):
+    PRODUCTION = "https://api.sarvam.ai"

{sarvamai-0.1.5a1 → sarvamai-0.1.5a4}/src/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,