sarvamai 0.1.12a0__tar.gz → 0.1.13__tar.gz

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/PKG-INFO

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

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/pyproject.toml

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

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/__init__.py

@@ -44,7 +44,7 @@ from .types import (
     FinishReason,
     FlushSignal,
     JobState,
-    JobStatusV1,
+    JobStatusV1Response,
     LanguageIdentificationResponse,
     NumeralsFormat,
     PingSignal,
@@ -149,7 +149,7 @@ from .requests import (
     FilesRequestParams,
     FilesUploadResponseParams,
     FlushSignalParams,
-    JobStatusV1Params,
+    JobStatusV1ResponseParams,
     LanguageIdentificationResponseParams,
     PingSignalParams,
     SendTextDataParams,
@@ -266,8 +266,8 @@ __all__ = [
     "ForbiddenError",
     "InternalServerError",
     "JobState",
-    "JobStatusV1",
-    "JobStatusV1Params",
+    "JobStatusV1Response",
+    "JobStatusV1ResponseParams",
     "LanguageIdentificationResponse",
     "LanguageIdentificationResponseParams",
     "NumeralsFormat",

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/core/client_wrapper.py

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

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/requests/__init__.py

@@ -38,7 +38,7 @@ from .files_download_response import FilesDownloadResponseParams
 from .files_request import FilesRequestParams
 from .files_upload_response import FilesUploadResponseParams
 from .flush_signal import FlushSignalParams
-from .job_status_v_1 import JobStatusV1Params
+from .job_status_v_1_response import JobStatusV1ResponseParams
 from .language_identification_response import LanguageIdentificationResponseParams
 from .ping_signal import PingSignalParams
 from .send_text import SendTextParams
@@ -97,7 +97,7 @@ __all__ = [
     "FilesRequestParams",
     "FilesUploadResponseParams",
     "FlushSignalParams",
-    "JobStatusV1Params",
+    "JobStatusV1ResponseParams",
     "LanguageIdentificationResponseParams",
     "PingSignalParams",
     "SendTextDataParams",

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/requests/bulk_job_init_response_v_1.py

@@ -4,7 +4,6 @@ import typing_extensions
 from ..types.job_state import JobState
 from ..types.storage_container_type import StorageContainerType
 from .base_job_parameters import BaseJobParametersParams
-from .bulk_job_callback import BulkJobCallbackParams
 
 
 class BulkJobInitResponseV1Params(typing_extensions.TypedDict):
@@ -20,8 +19,3 @@ class BulkJobInitResponseV1Params(typing_extensions.TypedDict):
 
     job_parameters: BaseJobParametersParams
     job_state: JobState
-    owner_id: str
-    callback: typing_extensions.NotRequired[BulkJobCallbackParams]
-    """
-    Parameters for callback URL
-    """

sarvamai-0.1.12a0/src/sarvamai/requests/job_status_v_1.py → sarvamai-0.1.13/src/sarvamai/requests/job_status_v_1_response.py

@@ -8,7 +8,7 @@ from ..types.storage_container_type import StorageContainerType
 from .task_detail_v_1 import TaskDetailV1Params
 
 
-class JobStatusV1Params(typing_extensions.TypedDict):
+class JobStatusV1ResponseParams(typing_extensions.TypedDict):
     job_state: JobState
     """
     Job State
@@ -44,11 +44,6 @@ class JobStatusV1Params(typing_extensions.TypedDict):
     Failed Count
     """
 
-    owner_id: str
-    """
-    Owner of the job
-    """
-
     storage_container_type: StorageContainerType
     """
     Storage Container Type
@@ -63,8 +58,3 @@ class JobStatusV1Params(typing_extensions.TypedDict):
     """
     Job details at file level.
     """
-
-    num_internal_retries: typing_extensions.NotRequired[int]
-    """
-    Number of internal retries
-    """

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/speech_to_text_job/client.py

@@ -9,7 +9,7 @@ from ..requests.speech_to_text_job_parameters import SpeechToTextJobParametersPa
 from ..types.bulk_job_init_response_v_1 import BulkJobInitResponseV1
 from ..types.files_download_response import FilesDownloadResponse
 from ..types.files_upload_response import FilesUploadResponse
-from ..types.job_status_v_1 import JobStatusV1
+from ..types.job_status_v_1_response import JobStatusV1Response
 from ..types.speech_to_text_model import SpeechToTextModel
 from ..types.speech_to_text_language import SpeechToTextLanguage
 from .raw_client import AsyncRawSpeechToTextJobClient, RawSpeechToTextJobClient
@@ -72,15 +72,13 @@ class SpeechToTextJobClient:
         )
         """
         _response = self._raw_client.initialise(
-            job_parameters=job_parameters,
-            callback=callback,
-            request_options=request_options,
+            job_parameters=job_parameters, callback=callback, request_options=request_options
         )
         return _response.data
 
     def get_status(
         self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> JobStatusV1:
+    ) -> JobStatusV1Response:
         """
         Get the status of a speech to text bulk job V1
 
@@ -94,7 +92,7 @@ class SpeechToTextJobClient:
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
             Successful Response
 
         Examples
@@ -117,7 +115,7 @@ class SpeechToTextJobClient:
         *,
         ptu_id: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> JobStatusV1:
+    ) -> JobStatusV1Response:
         """
         Start a speech to text bulk job V1
 
@@ -133,7 +131,7 @@ class SpeechToTextJobClient:
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
             Successful Response
 
         Examples
@@ -147,17 +145,11 @@ class SpeechToTextJobClient:
             job_id="job_id",
         )
         """
-        _response = self._raw_client.start(
-            job_id, ptu_id=ptu_id, request_options=request_options
-        )
+        _response = self._raw_client.start(job_id, ptu_id=ptu_id, request_options=request_options)
         return _response.data
 
     def get_upload_links(
-        self,
-        *,
-        job_id: str,
-        files: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, job_id: str, files: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> FilesUploadResponse:
         """
         Start a speech to text bulk job V1
@@ -188,17 +180,11 @@ class SpeechToTextJobClient:
             files=["files"],
         )
         """
-        _response = self._raw_client.get_upload_links(
-            job_id=job_id, files=files, request_options=request_options
-        )
+        _response = self._raw_client.get_upload_links(job_id=job_id, files=files, request_options=request_options)
         return _response.data
 
     def get_download_links(
-        self,
-        *,
-        job_id: str,
-        files: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, job_id: str, files: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> FilesDownloadResponse:
         """
         Start a speech to text bulk job V1
@@ -229,9 +215,7 @@ class SpeechToTextJobClient:
             files=["files"],
         )
         """
-        _response = self._raw_client.get_download_links(
-            job_id=job_id, files=files, request_options=request_options
-        )
+        _response = self._raw_client.get_download_links(job_id=job_id, files=files, request_options=request_options)
         return _response.data
 
     def create_job(
@@ -366,15 +350,13 @@ class AsyncSpeechToTextJobClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.initialise(
-            job_parameters=job_parameters,
-            callback=callback,
-            request_options=request_options,
+            job_parameters=job_parameters, callback=callback, request_options=request_options
         )
         return _response.data
 
     async def get_status(
         self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> JobStatusV1:
+    ) -> JobStatusV1Response:
         """
         Get the status of a speech to text bulk job V1
 
@@ -388,7 +370,7 @@ class AsyncSpeechToTextJobClient:
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
             Successful Response
 
         Examples
@@ -410,9 +392,7 @@ class AsyncSpeechToTextJobClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.get_status(
-            job_id, request_options=request_options
-        )
+        _response = await self._raw_client.get_status(job_id, request_options=request_options)
         return _response.data
 
     async def start(
@@ -421,7 +401,7 @@ class AsyncSpeechToTextJobClient:
         *,
         ptu_id: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> JobStatusV1:
+    ) -> JobStatusV1Response:
         """
         Start a speech to text bulk job V1
 
@@ -437,7 +417,7 @@ class AsyncSpeechToTextJobClient:
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
             Successful Response
 
         Examples
@@ -459,17 +439,11 @@ class AsyncSpeechToTextJobClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.start(
-            job_id, ptu_id=ptu_id, request_options=request_options
-        )
+        _response = await self._raw_client.start(job_id, ptu_id=ptu_id, request_options=request_options)
         return _response.data
 
     async def get_upload_links(
-        self,
-        *,
-        job_id: str,
-        files: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, job_id: str, files: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> FilesUploadResponse:
         """
         Start a speech to text bulk job V1
@@ -508,17 +482,11 @@ class AsyncSpeechToTextJobClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.get_upload_links(
-            job_id=job_id, files=files, request_options=request_options
-        )
+        _response = await self._raw_client.get_upload_links(job_id=job_id, files=files, request_options=request_options)
         return _response.data
 
     async def get_download_links(
-        self,
-        *,
-        job_id: str,
-        files: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, job_id: str, files: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> FilesDownloadResponse:
         """
         Start a speech to text bulk job V1

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/speech_to_text_job/job.py

@@ -5,7 +5,7 @@ import time
 import typing
 import httpx
 
-from ..types import JobStatusV1
+from ..types import JobStatusV1Response
 
 if typing.TYPE_CHECKING:
     from .client import AsyncSpeechToTextJobClient, SpeechToTextJobClient
@@ -44,7 +44,9 @@ class AsyncSpeechToTextJob:
         """
         return self._job_id
 
-    async def upload_files(self, file_paths: typing.Sequence[str]) -> bool:
+    async def upload_files(
+        self, file_paths: typing.Sequence[str], timeout: float = 60.0
+    ) -> bool:
         """
         Upload input audio files for the speech-to-text job.
 
@@ -53,6 +55,9 @@ class AsyncSpeechToTextJob:
         file_paths : Sequence[str]
             List of full paths to local audio files.
 
+        timeout : float, optional
+            The maximum time to wait for the upload to complete (in seconds),
+            by default 60.0
         Returns
         -------
         bool
@@ -62,7 +67,8 @@ class AsyncSpeechToTextJob:
             job_id=self._job_id,
             files=[os.path.basename(p) for p in file_paths],
         )
-        async with httpx.AsyncClient() as session:
+        client_timeout = httpx.Timeout(timeout=timeout)
+        async with httpx.AsyncClient(timeout=client_timeout) as session:
             for path in file_paths:
                 file_name = os.path.basename(path)
                 url = upload_links.upload_urls[file_name].file_url
@@ -86,7 +92,7 @@ class AsyncSpeechToTextJob:
 
     async def wait_until_complete(
         self, poll_interval: int = 5, timeout: int = 600
-    ) -> JobStatusV1:
+    ) -> JobStatusV1Response:
         """
         Polls job status until it completes or fails.
 
@@ -100,7 +106,7 @@ class AsyncSpeechToTextJob:
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
             Final job status.
 
         Raises
@@ -178,23 +184,23 @@ class AsyncSpeechToTextJob:
                     f.write(response.content)
         return True
 
-    async def get_status(self) -> JobStatusV1:
+    async def get_status(self) -> JobStatusV1Response:
         """
         Retrieve the current status of the job.
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
         """
         return await self._client.get_status(self._job_id)
 
-    async def start(self) -> JobStatusV1:
+    async def start(self) -> JobStatusV1Response:
         """
         Start the speech-to-text job processing.
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
         """
         return await self._client.start(job_id=self._job_id)
 
@@ -277,7 +283,9 @@ class SpeechToTextJob:
         """
         return self._job_id
 
-    def upload_files(self, file_paths: typing.Sequence[str]) -> bool:
+    async def upload_files(
+        self, file_paths: typing.Sequence[str], timeout: float = 60.0
+    ) -> bool:
         """
         Upload input audio files for the speech-to-text job.
 
@@ -286,6 +294,9 @@ class SpeechToTextJob:
         file_paths : Sequence[str]
             List of full paths to local audio files.
 
+        timeout : float, optional
+            The maximum time to wait for the upload to complete (in seconds),
+            by default 60.0
         Returns
         -------
         bool
@@ -294,7 +305,8 @@ class SpeechToTextJob:
         upload_links = self._client.get_upload_links(
             job_id=self._job_id, files=[os.path.basename(p) for p in file_paths]
         )
-        with httpx.Client() as client:
+        client_timeout = httpx.Timeout(timeout=timeout)
+        with httpx.Client(timeout=client_timeout) as client:
             for path in file_paths:
                 file_name = os.path.basename(path)
                 url = upload_links.upload_urls[file_name].file_url
@@ -315,7 +327,7 @@ class SpeechToTextJob:
 
     def wait_until_complete(
         self, poll_interval: int = 5, timeout: int = 600
-    ) -> JobStatusV1:
+    ) -> JobStatusV1Response:
         """
         Polls job status until it completes or fails.
 
@@ -329,7 +341,7 @@ class SpeechToTextJob:
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
             Final job status.
 
         Raises
@@ -407,23 +419,23 @@ class SpeechToTextJob:
                     f.write(response.content)
         return True
 
-    def get_status(self) -> JobStatusV1:
+    def get_status(self) -> JobStatusV1Response:
         """
         Retrieve the current status of the job.
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
         """
         return self._client.get_status(self._job_id)
 
-    def start(self) -> JobStatusV1:
+    def start(self) -> JobStatusV1Response:
         """
         Start the speech-to-text job processing.
 
         Returns
         -------
-        JobStatusV1
+        JobStatusV1Response
         """
         return self._client.start(job_id=self._job_id)
 

{sarvamai-0.1.12a0 → sarvamai-0.1.13}/src/sarvamai/speech_to_text_job/raw_client.py

@@ -21,7 +21,7 @@ from ..requests.speech_to_text_job_parameters import SpeechToTextJobParametersPa
 from ..types.bulk_job_init_response_v_1 import BulkJobInitResponseV1
 from ..types.files_download_response import FilesDownloadResponse
 from ..types.files_upload_response import FilesUploadResponse
-from ..types.job_status_v_1 import JobStatusV1
+from ..types.job_status_v_1_response import JobStatusV1Response
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -158,7 +158,7 @@ class RawSpeechToTextJobClient:
 
     def get_status(
         self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> HttpResponse[JobStatusV1]:
+    ) -> HttpResponse[JobStatusV1Response]:
         """
         Get the status of a speech to text bulk job V1
 
@@ -172,7 +172,7 @@ class RawSpeechToTextJobClient:
 
         Returns
         -------
-        HttpResponse[JobStatusV1]
+        HttpResponse[JobStatusV1Response]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -184,9 +184,9 @@ class RawSpeechToTextJobClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    JobStatusV1,
+                    JobStatusV1Response,
                     parse_obj_as(
-                        type_=JobStatusV1,  # type: ignore
+                        type_=JobStatusV1Response,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -268,7 +268,7 @@ class RawSpeechToTextJobClient:
         *,
         ptu_id: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[JobStatusV1]:
+    ) -> HttpResponse[JobStatusV1Response]:
         """
         Start a speech to text bulk job V1
 
@@ -284,7 +284,7 @@ class RawSpeechToTextJobClient:
 
         Returns
         -------
-        HttpResponse[JobStatusV1]
+        HttpResponse[JobStatusV1Response]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -299,9 +299,9 @@ class RawSpeechToTextJobClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    JobStatusV1,
+                    JobStatusV1Response,
                     parse_obj_as(
-                        type_=JobStatusV1,  # type: ignore
+                        type_=JobStatusV1Response,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -739,7 +739,7 @@ class AsyncRawSpeechToTextJobClient:
 
     async def get_status(
         self, job_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[JobStatusV1]:
+    ) -> AsyncHttpResponse[JobStatusV1Response]:
         """
         Get the status of a speech to text bulk job V1
 
@@ -753,7 +753,7 @@ class AsyncRawSpeechToTextJobClient:
 
         Returns
         -------
-        AsyncHttpResponse[JobStatusV1]
+        AsyncHttpResponse[JobStatusV1Response]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -765,9 +765,9 @@ class AsyncRawSpeechToTextJobClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    JobStatusV1,
+                    JobStatusV1Response,
                     parse_obj_as(
-                        type_=JobStatusV1,  # type: ignore
+                        type_=JobStatusV1Response,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -849,7 +849,7 @@ class AsyncRawSpeechToTextJobClient:
         *,
         ptu_id: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[JobStatusV1]:
+    ) -> AsyncHttpResponse[JobStatusV1Response]:
         """
         Start a speech to text bulk job V1
 
@@ -865,7 +865,7 @@ class AsyncRawSpeechToTextJobClient:
 
         Returns
         -------
-        AsyncHttpResponse[JobStatusV1]
+        AsyncHttpResponse[JobStatusV1Response]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -880,9 +880,9 @@ class AsyncRawSpeechToTextJobClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    JobStatusV1,
+                    JobStatusV1Response,
                     parse_obj_as(
-                        type_=JobStatusV1,  # type: ignore
+                        type_=JobStatusV1Response,  # type: ignore
                         object_=_response.json(),
                     ),
                 )