sarvamai 0.1.4__tar.gz → 0.1.5a0__tar.gz

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

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

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

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

{sarvamai-0.1.4 → sarvamai-0.1.5a0}/src/sarvamai/chat/client.py

@@ -33,6 +33,7 @@ class ChatClient:
     def completions(
         self,
         *,
+        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -52,6 +53,9 @@ 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.
 
@@ -115,10 +119,12 @@ 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,
@@ -154,6 +160,7 @@ class AsyncChatClient:
     async def completions(
         self,
         *,
+        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -173,6 +180,9 @@ 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.
 
@@ -241,6 +251,7 @@ class AsyncChatClient:
 
         async def main() -> None:
             await client.chat.completions(
+                authorization="Authorization",
                 messages=[{"content": "content", "role": "assistant"}],
             )
 
@@ -248,6 +259,7 @@ class AsyncChatClient:
         asyncio.run(main())
         """
         response = await self._raw_client.completions(
+            authorization=authorization,
             messages=messages,
             temperature=temperature,
             top_p=top_p,

{sarvamai-0.1.4 → sarvamai-0.1.5a0}/src/sarvamai/chat/raw_client.py

@@ -31,6 +31,7 @@ class RawChatClient:
     def completions(
         self,
         *,
+        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -50,6 +51,9 @@ 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.
 
@@ -129,6 +133,7 @@ class RawChatClient:
             },
             headers={
                 "content-type": "application/json",
+                "Authorization": str(authorization) if authorization is not None else None,
             },
             request_options=request_options,
             omit=OMIT,
@@ -206,6 +211,7 @@ class AsyncRawChatClient:
     async def completions(
         self,
         *,
+        authorization: str,
         messages: typing.Sequence[ChatCompletionRequestMessageParams],
         temperature: typing.Optional[float] = OMIT,
         top_p: typing.Optional[float] = OMIT,
@@ -225,6 +231,9 @@ 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.
 
@@ -304,6 +313,7 @@ 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.4 → sarvamai-0.1.5a0}/src/sarvamai/core/client_wrapper.py

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