sarvamai 0.1.6__py3-none-any.whl → 0.1.7a0__py3-none-any.whl

sarvamai/types/__init__.py

@@ -4,6 +4,9 @@
 
 from .audio_data import AudioData
 from .audio_message import AudioMessage
+from .base_job_parameters import BaseJobParameters
+from .bulk_job_callback import BulkJobCallback
+from .bulk_job_init_response_v_1 import BulkJobInitResponseV1
 from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessage
 from .chat_completion_request_message import (
     ChatCompletionRequestMessage,
@@ -25,8 +28,14 @@ from .error_data import ErrorData
 from .error_details import ErrorDetails
 from .error_message import ErrorMessage
 from .events_data import EventsData
+from .file_signed_url_details import FileSignedUrlDetails
+from .files_download_response import FilesDownloadResponse
+from .files_request import FilesRequest
+from .files_upload_response import FilesUploadResponse
 from .finish_reason import FinishReason
 from .format import Format
+from .job_state import JobState
+from .job_status_v_1 import JobStatusV1
 from .language_identification_response import LanguageIdentificationResponse
 from .numerals_format import NumeralsFormat
 from .reasoning_effort import ReasoningEffort
@@ -34,6 +43,7 @@ from .response_type import ResponseType
 from .role import Role
 from .sarvam_model_ids import SarvamModelIds
 from .speech_sample_rate import SpeechSampleRate
+from .speech_to_text_job_parameters import SpeechToTextJobParameters
 from .speech_to_text_language import SpeechToTextLanguage
 from .speech_to_text_model import SpeechToTextModel
 from .speech_to_text_response import SpeechToTextResponse
@@ -48,6 +58,10 @@ from .speech_to_text_translate_streaming_response import SpeechToTextTranslateSt
 from .speech_to_text_translate_transcription_data import SpeechToTextTranslateTranscriptionData
 from .spoken_form_numerals_format import SpokenFormNumeralsFormat
 from .stop_configuration import StopConfiguration
+from .storage_container_type import StorageContainerType
+from .task_detail_v_1 import TaskDetailV1
+from .task_file_details import TaskFileDetails
+from .task_state import TaskState
 from .text_to_speech_language import TextToSpeechLanguage
 from .text_to_speech_model import TextToSpeechModel
 from .text_to_speech_response import TextToSpeechResponse
@@ -68,6 +82,9 @@ from .transliteration_response import TransliterationResponse
 __all__ = [
     "AudioData",
     "AudioMessage",
+    "BaseJobParameters",
+    "BulkJobCallback",
+    "BulkJobInitResponseV1",
     "ChatCompletionRequestAssistantMessage",
     "ChatCompletionRequestMessage",
     "ChatCompletionRequestMessage_Assistant",
@@ -87,8 +104,14 @@ __all__ = [
     "ErrorDetails",
     "ErrorMessage",
     "EventsData",
+    "FileSignedUrlDetails",
+    "FilesDownloadResponse",
+    "FilesRequest",
+    "FilesUploadResponse",
     "FinishReason",
     "Format",
+    "JobState",
+    "JobStatusV1",
     "LanguageIdentificationResponse",
     "NumeralsFormat",
     "ReasoningEffort",
@@ -96,6 +119,7 @@ __all__ = [
     "Role",
     "SarvamModelIds",
     "SpeechSampleRate",
+    "SpeechToTextJobParameters",
     "SpeechToTextLanguage",
     "SpeechToTextModel",
     "SpeechToTextResponse",
@@ -110,6 +134,10 @@ __all__ = [
     "SpeechToTextTranslateTranscriptionData",
     "SpokenFormNumeralsFormat",
     "StopConfiguration",
+    "StorageContainerType",
+    "TaskDetailV1",
+    "TaskFileDetails",
+    "TaskState",
     "TextToSpeechLanguage",
     "TextToSpeechModel",
     "TextToSpeechResponse",

sarvamai/types/base_job_parameters.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class BaseJobParameters(UniversalBaseModel):
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/bulk_job_callback.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class BulkJobCallback(UniversalBaseModel):
+    url: str = pydantic.Field()
+    """
+    Webhook url to call upon job completion
+    """
+
+    auth_token: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Authorization token required for the callback Url
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/bulk_job_init_response_v_1.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .base_job_parameters import BaseJobParameters
+from .bulk_job_callback import BulkJobCallback
+from .job_state import JobState
+from .storage_container_type import StorageContainerType
+
+
+class BulkJobInitResponseV1(UniversalBaseModel):
+    job_id: str = pydantic.Field()
+    """
+    Job UUID.
+    """
+
+    storage_container_type: StorageContainerType = pydantic.Field()
+    """
+    Storage Container Type
+    """
+
+    job_parameters: BaseJobParameters
+    job_state: JobState
+    owner_id: str
+    callback: typing.Optional[BulkJobCallback] = pydantic.Field(default=None)
+    """
+    Parameters for callback URL
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/file_signed_url_details.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class FileSignedUrlDetails(UniversalBaseModel):
+    file_url: str
+    file_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/files_download_response.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .file_signed_url_details import FileSignedUrlDetails
+from .job_state import JobState
+from .storage_container_type import StorageContainerType
+
+
+class FilesDownloadResponse(UniversalBaseModel):
+    job_id: str
+    job_state: JobState
+    download_urls: typing.Dict[str, FileSignedUrlDetails]
+    storage_container_type: StorageContainerType
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/files_request.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class FilesRequest(UniversalBaseModel):
+    job_id: str
+    files: typing.List[str]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/files_upload_response.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .file_signed_url_details import FileSignedUrlDetails
+from .job_state import JobState
+from .storage_container_type import StorageContainerType
+
+
+class FilesUploadResponse(UniversalBaseModel):
+    job_id: str
+    job_state: JobState
+    upload_urls: typing.Dict[str, FileSignedUrlDetails]
+    storage_container_type: StorageContainerType
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/job_state.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+JobState = typing.Union[typing.Literal["Accepted", "Pending", "Running", "Completed", "Failed"], typing.Any]

sarvamai/types/job_status_v_1.py

@@ -0,0 +1,80 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .job_state import JobState
+from .storage_container_type import StorageContainerType
+from .task_detail_v_1 import TaskDetailV1
+
+
+class JobStatusV1(UniversalBaseModel):
+    job_state: JobState = pydantic.Field()
+    """
+    Job State
+    """
+
+    created_at: str = pydantic.Field()
+    """
+    Created At
+    """
+
+    updated_at: str = pydantic.Field()
+    """
+    Updated At
+    """
+
+    job_id: str = pydantic.Field()
+    """
+    Job Id
+    """
+
+    total_files: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Total Files
+    """
+
+    successful_files_count: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Success Count
+    """
+
+    failed_files_count: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Failed Count
+    """
+
+    owner_id: str = pydantic.Field()
+    """
+    Owner of the job
+    """
+
+    storage_container_type: StorageContainerType = pydantic.Field()
+    """
+    Storage Container Type
+    """
+
+    error_message: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Error Message
+    """
+
+    job_details: typing.Optional[typing.List[TaskDetailV1]] = pydantic.Field(default=None)
+    """
+    Job details at file level.
+    """
+
+    num_internal_retries: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of internal retries
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/speech_to_text_job_parameters.py

@@ -0,0 +1,44 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .speech_to_text_model import SpeechToTextModel
+from .speech_to_text_translate_language import SpeechToTextTranslateLanguage
+
+
+class SpeechToTextJobParameters(UniversalBaseModel):
+    language_code: typing.Optional[SpeechToTextTranslateLanguage] = pydantic.Field(default=None)
+    """
+    Language code
+    """
+
+    model: typing.Optional[SpeechToTextModel] = pydantic.Field(default=None)
+    """
+    Model to be used for speech to text
+    """
+
+    with_timestamps: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether to include timestamps in the response
+    """
+
+    with_diarization: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Enables speaker diarization, which identifies and separates different speakers in the audio. In beta mode
+    """
+
+    num_speakers: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of speakers to be detected in the audio. This is used when with_diarization is true.
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/storage_container_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+StorageContainerType = typing.Union[typing.Literal["Azure", "Local", "Google", "Azure_V1"], typing.Any]

sarvamai/types/task_detail_v_1.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .task_file_details import TaskFileDetails
+from .task_state import TaskState
+
+
+class TaskDetailV1(UniversalBaseModel):
+    inputs: typing.Optional[typing.List[TaskFileDetails]] = None
+    outputs: typing.Optional[typing.List[TaskFileDetails]] = None
+    state: typing.Optional[TaskState] = None
+    error_message: typing.Optional[str] = None
+    exception_name: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/task_file_details.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class TaskFileDetails(UniversalBaseModel):
+    file_name: str
+    file_id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

sarvamai/types/task_state.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TaskState = typing.Union[typing.Literal["Success", "API Error", "Internal Server Error"], typing.Any]