sarvamai 0.1.5a6__tar.gz → 0.1.5a8__tar.gz

sarvamai-0.1.5a8/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.1
+Name: sarvamai
+Version: 0.1.5a8
+Summary: 
+Requires-Python: >=3.8,<4.0
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: MacOS
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: OS Independent
+Classifier: Operating System :: POSIX
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Requires-Dist: httpx (>=0.21.2)
+Requires-Dist: pydantic (>=1.9.2)
+Requires-Dist: pydantic-core (>=2.18.2,<3.0.0)
+Requires-Dist: typing_extensions (>=4.0.0)
+Requires-Dist: websockets (==12.0)
+Description-Content-Type: text/markdown
+
+

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/pyproject.toml

@@ -3,7 +3,7 @@ name = "sarvamai"
 
 [tool.poetry]
 name = "sarvamai"
-version = "0.1.5a6"
+version = "0.1.5a8"
 description = ""
 readme = "README.md"
 authors = []
@@ -39,14 +39,15 @@ httpx = ">=0.21.2"
 pydantic = ">= 1.9.2"
 pydantic-core = "^2.18.2"
 typing_extensions = ">= 4.0.0"
+websockets = "12.0"
 
-[tool.poetry.dev-dependencies]
-mypy = "1.0.1"
+[tool.poetry.group.dev.dependencies]
+mypy = "==1.13.0"
 pytest = "^7.4.0"
 pytest-asyncio = "^0.23.5"
 python-dateutil = "^2.9.0"
 types-python-dateutil = "^2.9.0.20240316"
-ruff = "^0.5.6"
+ruff = "==0.11.5"
 
 [tool.pytest.ini_options]
 testpaths = [ "tests" ]
@@ -58,6 +59,26 @@ plugins = ["pydantic.mypy"]
 [tool.ruff]
 line-length = 120
 
+[tool.ruff.lint]
+select = [
+    "E",  # pycodestyle errors
+    "F",  # pyflakes
+    "I",  # isort
+]
+ignore = [
+    "E402",  # Module level import not at top of file
+    "E501",  # Line too long
+    "E711",  # Comparison to `None` should be `cond is not None`
+    "E712",  # Avoid equality comparisons to `True`; use `if ...:` checks
+    "E721",  # Use `is` and `is not` for type comparisons, or `isinstance()` for insinstance checks
+    "E722",  # Do not use bare `except`
+    "E731",  # Do not assign a `lambda` expression, use a `def`
+    "F821",  # Undefined name
+    "F841"   # Local variable ... is assigned to but never used
+]
+
+[tool.ruff.lint.isort]
+section-order = ["future", "standard-library", "third-party", "first-party"]
 
 [build-system]
 requires = ["poetry-core"]

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/src/sarvamai/__init__.py

@@ -1,6 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 from .types import (
+    AudioData,
+    AudioDataEncoding,
+    AudioMessage,
     ChatCompletionRequestAssistantMessage,
     ChatCompletionRequestMessage,
     ChatCompletionRequestMessage_Assistant,
@@ -15,9 +20,12 @@ from .types import (
     DiarizedEntry,
     DiarizedTranscript,
     ErrorCode,
+    ErrorData,
     ErrorDetails,
     ErrorMessage,
+    EventsData,
     FinishReason,
+    Format,
     LanguageIdentificationResponse,
     NumeralsFormat,
     ReasoningEffort,
@@ -27,6 +35,9 @@ from .types import (
     SpeechToTextLanguage,
     SpeechToTextModel,
     SpeechToTextResponse,
+    SpeechToTextStreamingResponse,
+    SpeechToTextStreamingResponseData,
+    SpeechToTextStreamingResponseType,
     SpeechToTextTranslateLanguage,
     SpeechToTextTranslateModel,
     SpeechToTextTranslateResponse,
@@ -37,6 +48,8 @@ from .types import (
     TextToSpeechResponse,
     TextToSpeechSpeaker,
     TimestampsModel,
+    TranscriptionData,
+    TranscriptionMetrics,
     TranslateMode,
     TranslateModel,
     TranslateSourceLanguage,
@@ -56,10 +69,12 @@ from .errors import (
     TooManyRequestsError,
     UnprocessableEntityError,
 )
-from . import chat, speech_to_text, text, text_to_speech
+from . import chat, speech_to_text, speech_to_text_streaming, text, text_to_speech
 from .client import AsyncSarvamAI, SarvamAI
 from .environment import SarvamAIEnvironment
 from .requests import (
+    AudioDataParams,
+    AudioMessageParams,
     ChatCompletionRequestAssistantMessageParams,
     ChatCompletionRequestMessageParams,
     ChatCompletionRequestMessage_AssistantParams,
@@ -73,21 +88,33 @@ from .requests import (
     CreateChatCompletionResponseParams,
     DiarizedEntryParams,
     DiarizedTranscriptParams,
+    ErrorDataParams,
     ErrorDetailsParams,
     ErrorMessageParams,
+    EventsDataParams,
     LanguageIdentificationResponseParams,
     SpeechToTextResponseParams,
+    SpeechToTextStreamingResponseDataParams,
+    SpeechToTextStreamingResponseParams,
     SpeechToTextTranslateResponseParams,
     StopConfigurationParams,
     TextToSpeechResponseParams,
     TimestampsModelParams,
+    TranscriptionDataParams,
+    TranscriptionMetricsParams,
     TranslationResponseParams,
     TransliterationResponseParams,
 )
+from .speech_to_text_streaming import SpeechToTextStreamingLanguageCode, SpeechToTextStreamingModel
 from .version import __version__
 
 __all__ = [
     "AsyncSarvamAI",
+    "AudioData",
+    "AudioDataEncoding",
+    "AudioDataParams",
+    "AudioMessage",
+    "AudioMessageParams",
     "BadRequestError",
     "ChatCompletionRequestAssistantMessage",
     "ChatCompletionRequestAssistantMessageParams",
@@ -116,12 +143,17 @@ __all__ = [
     "DiarizedTranscript",
     "DiarizedTranscriptParams",
     "ErrorCode",
+    "ErrorData",
+    "ErrorDataParams",
     "ErrorDetails",
     "ErrorDetailsParams",
     "ErrorMessage",
     "ErrorMessageParams",
+    "EventsData",
+    "EventsDataParams",
     "FinishReason",
     "ForbiddenError",
+    "Format",
     "InternalServerError",
     "LanguageIdentificationResponse",
     "LanguageIdentificationResponseParams",
@@ -137,6 +169,13 @@ __all__ = [
     "SpeechToTextModel",
     "SpeechToTextResponse",
     "SpeechToTextResponseParams",
+    "SpeechToTextStreamingLanguageCode",
+    "SpeechToTextStreamingModel",
+    "SpeechToTextStreamingResponse",
+    "SpeechToTextStreamingResponseData",
+    "SpeechToTextStreamingResponseDataParams",
+    "SpeechToTextStreamingResponseParams",
+    "SpeechToTextStreamingResponseType",
     "SpeechToTextTranslateLanguage",
     "SpeechToTextTranslateModel",
     "SpeechToTextTranslateResponse",
@@ -152,6 +191,10 @@ __all__ = [
     "TimestampsModel",
     "TimestampsModelParams",
     "TooManyRequestsError",
+    "TranscriptionData",
+    "TranscriptionDataParams",
+    "TranscriptionMetrics",
+    "TranscriptionMetricsParams",
     "TranslateMode",
     "TranslateModel",
     "TranslateSourceLanguage",
@@ -168,6 +211,7 @@ __all__ = [
     "__version__",
     "chat",
     "speech_to_text",
+    "speech_to_text_streaming",
     "text",
     "text_to_speech",
 ]

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/src/sarvamai/chat/__init__.py

@@ -1,2 +1,4 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/src/sarvamai/chat/client.py

@@ -1,15 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
-from .raw_client import RawChatClient
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
 from ..requests.chat_completion_request_message import ChatCompletionRequestMessageParams
-from ..types.reasoning_effort import ReasoningEffort
 from ..requests.stop_configuration import StopConfigurationParams
-from ..core.request_options import RequestOptions
 from ..types.create_chat_completion_response import CreateChatCompletionResponse
-from ..core.client_wrapper import AsyncClientWrapper
-from .raw_client import AsyncRawChatClient
+from ..types.reasoning_effort import ReasoningEffort
+from .raw_client import AsyncRawChatClient, RawChatClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -117,7 +116,7 @@ class ChatClient:
             messages=[{"content": "content", "role": "assistant"}],
         )
         """
-        response = self._raw_client.completions(
+        _response = self._raw_client.completions(
             messages=messages,
             temperature=temperature,
             top_p=top_p,
@@ -132,7 +131,7 @@ class ChatClient:
             wiki_grounding=wiki_grounding,
             request_options=request_options,
         )
-        return response.data
+        return _response.data
 
 
 class AsyncChatClient:
@@ -245,7 +244,7 @@ class AsyncChatClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.completions(
+        _response = await self._raw_client.completions(
             messages=messages,
             temperature=temperature,
             top_p=top_p,
@@ -260,4 +259,4 @@ class AsyncChatClient:
             wiki_grounding=wiki_grounding,
             request_options=request_options,
         )
-        return response.data
+        return _response.data

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/src/sarvamai/chat/raw_client.py

@@ -1,24 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
-from ..requests.chat_completion_request_message import ChatCompletionRequestMessageParams
-from ..types.reasoning_effort import ReasoningEffort
-from ..requests.stop_configuration import StopConfigurationParams
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.pydantic_utilities import parse_obj_as
 from ..core.request_options import RequestOptions
-from ..core.http_response import HttpResponse
-from ..types.create_chat_completion_response import CreateChatCompletionResponse
 from ..core.serialization import convert_and_respect_annotation_metadata
-from ..core.pydantic_utilities import parse_obj_as
 from ..errors.bad_request_error import BadRequestError
 from ..errors.forbidden_error import ForbiddenError
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..errors.too_many_requests_error import TooManyRequestsError
 from ..errors.internal_server_error import InternalServerError
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper
-from ..core.http_response import AsyncHttpResponse
+from ..errors.too_many_requests_error import TooManyRequestsError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..requests.chat_completion_request_message import ChatCompletionRequestMessageParams
+from ..requests.stop_configuration import StopConfigurationParams
+from ..types.create_chat_completion_response import CreateChatCompletionResponse
+from ..types.reasoning_effort import ReasoningEffort
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -106,6 +105,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(
@@ -144,58 +144,63 @@ class RawChatClient:
                 return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 422:
                 raise UnprocessableEntityError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 429:
                 raise TooManyRequestsError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
 class AsyncRawChatClient:
@@ -280,6 +285,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(
@@ -318,55 +324,60 @@ class AsyncRawChatClient:
                 return AsyncHttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 403:
                 raise ForbiddenError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 422:
                 raise UnprocessableEntityError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 429:
                 raise TooManyRequestsError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             if _response.status_code == 500:
                 raise InternalServerError(
-                    typing.cast(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
                         typing.Optional[typing.Any],
                         parse_obj_as(
                             type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
-                    )
+                    ),
                 )
             _response_json = _response.json()
         except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/src/sarvamai/client.py

@@ -1,20 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing
-from .environment import SarvamAIEnvironment
 import os
+import typing
+
 import httpx
+from .chat.client import AsyncChatClient, ChatClient
 from .core.api_error import ApiError
-from .core.client_wrapper import SyncClientWrapper
-from .text.client import TextClient
-from .speech_to_text.client import SpeechToTextClient
-from .text_to_speech.client import TextToSpeechClient
-from .chat.client import ChatClient
-from .core.client_wrapper import AsyncClientWrapper
-from .text.client import AsyncTextClient
-from .speech_to_text.client import AsyncSpeechToTextClient
-from .text_to_speech.client import AsyncTextToSpeechClient
-from .chat.client import AsyncChatClient
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .environment import SarvamAIEnvironment
+from .speech_to_text.client import AsyncSpeechToTextClient, SpeechToTextClient
+from .speech_to_text_streaming.client import AsyncSpeechToTextStreamingClient, SpeechToTextStreamingClient
+from .text.client import AsyncTextClient, TextClient
+from .text_to_speech.client import AsyncTextToSpeechClient, TextToSpeechClient
 
 
 class SarvamAI:
@@ -23,9 +20,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 +51,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 +65,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
@@ -85,6 +78,7 @@ class SarvamAI:
         self.speech_to_text = SpeechToTextClient(client_wrapper=self._client_wrapper)
         self.text_to_speech = TextToSpeechClient(client_wrapper=self._client_wrapper)
         self.chat = ChatClient(client_wrapper=self._client_wrapper)
+        self.speech_to_text_streaming = SpeechToTextStreamingClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncSarvamAI:
@@ -93,9 +87,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 +118,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 +132,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 +145,4 @@ 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")
+        self.speech_to_text_streaming = AsyncSpeechToTextStreamingClient(client_wrapper=self._client_wrapper)

{sarvamai-0.1.5a6 → sarvamai-0.1.5a8}/src/sarvamai/core/__init__.py

@@ -1,8 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
 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 +30,8 @@ __all__ = [
     "AsyncHttpClient",
     "AsyncHttpResponse",
     "BaseClientWrapper",
+    "EventEmitterMixin",
+    "EventType",
     "FieldMetadata",
     "File",
     "HttpClient",

sarvamai-0.1.5a8/src/sarvamai/core/api_error.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from typing import Any, Dict, Optional
+
+
+class ApiError(Exception):
+    headers: Optional[Dict[str, str]]
+    status_code: Optional[int]
+    body: Any
+
+    def __init__(
+        self,
+        *,
+        headers: Optional[Dict[str, str]] = None,
+        status_code: Optional[int] = None,
+        body: Any = None,
+    ) -> None:
+        self.headers = headers
+        self.status_code = status_code
+        self.body = body
+
+    def __str__(self) -> str:
+        return f"headers: {self.headers}, status_code: {self.status_code}, body: {self.body}"