sarvamai 0.1.22a4__py3-none-any.whl → 0.1.23a1__py3-none-any.whl

sarvamai/core/http_sse/_decoders.py

@@ -0,0 +1,61 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from typing import List, Optional
+
+from ._models import ServerSentEvent
+
+
+class SSEDecoder:
+    def __init__(self) -> None:
+        self._event = ""
+        self._data: List[str] = []
+        self._last_event_id = ""
+        self._retry: Optional[int] = None
+
+    def decode(self, line: str) -> Optional[ServerSentEvent]:
+        # See: https://html.spec.whatwg.org/multipage/server-sent-events.html#event-stream-interpretation  # noqa: E501
+
+        if not line:
+            if not self._event and not self._data and not self._last_event_id and self._retry is None:
+                return None
+
+            sse = ServerSentEvent(
+                event=self._event,
+                data="\n".join(self._data),
+                id=self._last_event_id,
+                retry=self._retry,
+            )
+
+            # NOTE: as per the SSE spec, do not reset last_event_id.
+            self._event = ""
+            self._data = []
+            self._retry = None
+
+            return sse
+
+        if line.startswith(":"):
+            return None
+
+        fieldname, _, value = line.partition(":")
+
+        if value.startswith(" "):
+            value = value[1:]
+
+        if fieldname == "event":
+            self._event = value
+        elif fieldname == "data":
+            self._data.append(value)
+        elif fieldname == "id":
+            if "\0" in value:
+                pass
+            else:
+                self._last_event_id = value
+        elif fieldname == "retry":
+            try:
+                self._retry = int(value)
+            except (TypeError, ValueError):
+                pass
+        else:
+            pass  # Field is ignored.
+
+        return None

sarvamai/core/http_sse/_exceptions.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import httpx
+
+
+class SSEError(httpx.TransportError):
+    pass

sarvamai/core/http_sse/_models.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import json
+from dataclasses import dataclass
+from typing import Any, Optional
+
+
+@dataclass(frozen=True)
+class ServerSentEvent:
+    event: str = "message"
+    data: str = ""
+    id: str = ""
+    retry: Optional[int] = None
+
+    def json(self) -> Any:
+        """Parse the data field as JSON."""
+        return json.loads(self.data)

sarvamai/core/jsonable_encoder.py

@@ -30,6 +30,10 @@ DictIntStrAny = Dict[Union[int, str], Any]
 
 def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any], Any]]] = None) -> Any:
     custom_encoder = custom_encoder or {}
+    # Generated SDKs use Ellipsis (`...`) as the sentinel value for "OMIT".
+    # OMIT values should be excluded from serialized payloads.
+    if obj is Ellipsis:
+        return None
     if custom_encoder:
         if type(obj) in custom_encoder:
             return custom_encoder[type(obj)](obj)
@@ -70,6 +74,8 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
         allowed_keys = set(obj.keys())
         for key, value in obj.items():
             if key in allowed_keys:
+                if value is Ellipsis:
+                    continue
                 encoded_key = jsonable_encoder(key, custom_encoder=custom_encoder)
                 encoded_value = jsonable_encoder(value, custom_encoder=custom_encoder)
                 encoded_dict[encoded_key] = encoded_value
@@ -77,6 +83,8 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
     if isinstance(obj, (list, set, frozenset, GeneratorType, tuple)):
         encoded_list = []
         for item in obj:
+            if item is Ellipsis:
+                continue
             encoded_list.append(jsonable_encoder(item, custom_encoder=custom_encoder))
         return encoded_list
 

sarvamai/core/pydantic_utilities.py

@@ -2,6 +2,7 @@
 
 # nopycln: file
 import datetime as dt
+import inspect
 from collections import defaultdict
 from typing import Any, Callable, ClassVar, Dict, List, Mapping, Optional, Set, Tuple, Type, TypeVar, Union, cast
 
@@ -37,7 +38,36 @@ Model = TypeVar("Model", bound=pydantic.BaseModel)
 
 
 def parse_obj_as(type_: Type[T], object_: Any) -> T:
-    dealiased_object = convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
+    # convert_and_respect_annotation_metadata is required for TypedDict aliasing.
+    #
+    # For Pydantic models, whether we should pre-dealias depends on how the model encodes aliasing:
+    # - If the model uses real Pydantic aliases (pydantic.Field(alias=...)), then we must pass wire keys through
+    #   unchanged so Pydantic can validate them.
+    # - If the model encodes aliasing only via FieldMetadata annotations, then we MUST pre-dealias because Pydantic
+    #   will not recognize those aliases during validation.
+    if inspect.isclass(type_) and issubclass(type_, pydantic.BaseModel):
+        has_pydantic_aliases = False
+        if IS_PYDANTIC_V2:
+            for field_name, field_info in getattr(type_, "model_fields", {}).items():  # type: ignore[attr-defined]
+                alias = getattr(field_info, "alias", None)
+                if alias is not None and alias != field_name:
+                    has_pydantic_aliases = True
+                    break
+        else:
+            for field in getattr(type_, "__fields__", {}).values():
+                alias = getattr(field, "alias", None)
+                name = getattr(field, "name", None)
+                if alias is not None and name is not None and alias != name:
+                    has_pydantic_aliases = True
+                    break
+
+        dealiased_object = (
+            object_
+            if has_pydantic_aliases
+            else convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
+        )
+    else:
+        dealiased_object = convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
     if IS_PYDANTIC_V2:
         adapter = pydantic.TypeAdapter(type_)  # type: ignore[attr-defined]
         return adapter.validate_python(dealiased_object)
@@ -59,9 +89,46 @@ class UniversalBaseModel(pydantic.BaseModel):
             protected_namespaces=(),
         )
 
+        @pydantic.model_validator(mode="before")  # type: ignore[attr-defined]
+        @classmethod
+        def _coerce_field_names_to_aliases(cls, data: Any) -> Any:
+            """
+            Accept Python field names in input by rewriting them to their Pydantic aliases,
+            while avoiding silent collisions when a key could refer to multiple fields.
+            """
+            if not isinstance(data, Mapping):
+                return data
+
+            fields = getattr(cls, "model_fields", {})  # type: ignore[attr-defined]
+            name_to_alias: Dict[str, str] = {}
+            alias_to_name: Dict[str, str] = {}
+
+            for name, field_info in fields.items():
+                alias = getattr(field_info, "alias", None) or name
+                name_to_alias[name] = alias
+                if alias != name:
+                    alias_to_name[alias] = name
+
+            # Detect ambiguous keys: a key that is an alias for one field and a name for another.
+            ambiguous_keys = set(alias_to_name.keys()).intersection(set(name_to_alias.keys()))
+            for key in ambiguous_keys:
+                if key in data and name_to_alias[key] not in data:
+                    raise ValueError(
+                        f"Ambiguous input key '{key}': it is both a field name and an alias. "
+                        "Provide the explicit alias key to disambiguate."
+                    )
+
+            original_keys = set(data.keys())
+            rewritten: Dict[str, Any] = dict(data)
+            for name, alias in name_to_alias.items():
+                if alias != name and name in original_keys and alias not in rewritten:
+                    rewritten[alias] = rewritten.pop(name)
+
+            return rewritten
+
         @pydantic.model_serializer(mode="plain", when_used="json")  # type: ignore[attr-defined]
         def serialize_model(self) -> Any:  # type: ignore[name-defined]
-            serialized = self.model_dump()
+            serialized = self.dict()  # type: ignore[attr-defined]
             data = {k: serialize_datetime(v) if isinstance(v, dt.datetime) else v for k, v in serialized.items()}
             return data
 
@@ -71,6 +138,40 @@ class UniversalBaseModel(pydantic.BaseModel):
             smart_union = True
             json_encoders = {dt.datetime: serialize_datetime}
 
+        @pydantic.root_validator(pre=True)
+        def _coerce_field_names_to_aliases(cls, values: Any) -> Any:
+            """
+            Pydantic v1 equivalent of _coerce_field_names_to_aliases.
+            """
+            if not isinstance(values, Mapping):
+                return values
+
+            fields = getattr(cls, "__fields__", {})
+            name_to_alias: Dict[str, str] = {}
+            alias_to_name: Dict[str, str] = {}
+
+            for name, field in fields.items():
+                alias = getattr(field, "alias", None) or name
+                name_to_alias[name] = alias
+                if alias != name:
+                    alias_to_name[alias] = name
+
+            ambiguous_keys = set(alias_to_name.keys()).intersection(set(name_to_alias.keys()))
+            for key in ambiguous_keys:
+                if key in values and name_to_alias[key] not in values:
+                    raise ValueError(
+                        f"Ambiguous input key '{key}': it is both a field name and an alias. "
+                        "Provide the explicit alias key to disambiguate."
+                    )
+
+            original_keys = set(values.keys())
+            rewritten: Dict[str, Any] = dict(values)
+            for name, alias in name_to_alias.items():
+                if alias != name and name in original_keys and alias not in rewritten:
+                    rewritten[alias] = rewritten.pop(name)
+
+            return rewritten
+
     @classmethod
     def model_construct(cls: Type["Model"], _fields_set: Optional[Set[str]] = None, **values: Any) -> "Model":
         dealiased_object = convert_and_respect_annotation_metadata(object_=values, annotation=cls, direction="read")
@@ -147,7 +248,10 @@ class UniversalBaseModel(pydantic.BaseModel):
 
             dict_dump = super().dict(**kwargs_with_defaults_exclude_unset_include_fields)
 
-        return convert_and_respect_annotation_metadata(object_=dict_dump, annotation=self.__class__, direction="write")
+        return cast(
+            Dict[str, Any],
+            convert_and_respect_annotation_metadata(object_=dict_dump, annotation=self.__class__, direction="write"),
+        )
 
 
 def _union_list_of_pydantic_dicts(source: List[Any], destination: List[Any]) -> List[Any]:
@@ -217,7 +321,9 @@ def universal_root_validator(
 ) -> Callable[[AnyCallable], AnyCallable]:
     def decorator(func: AnyCallable) -> AnyCallable:
         if IS_PYDANTIC_V2:
-            return cast(AnyCallable, pydantic.model_validator(mode="before" if pre else "after")(func))  # type: ignore[attr-defined]
+            # In Pydantic v2, for RootModel we always use "before" mode
+            # The custom validators transform the input value before the model is created
+            return cast(AnyCallable, pydantic.model_validator(mode="before")(func))  # type: ignore[attr-defined]
         return cast(AnyCallable, pydantic.root_validator(pre=pre)(func))  # type: ignore[call-overload]
 
     return decorator

sarvamai/errors/__init__.py

@@ -2,12 +2,46 @@
 
 # isort: skip_file
 
-from .bad_request_error import BadRequestError
-from .forbidden_error import ForbiddenError
-from .internal_server_error import InternalServerError
-from .service_unavailable_error import ServiceUnavailableError
-from .too_many_requests_error import TooManyRequestsError
-from .unprocessable_entity_error import UnprocessableEntityError
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .bad_request_error import BadRequestError
+    from .forbidden_error import ForbiddenError
+    from .internal_server_error import InternalServerError
+    from .service_unavailable_error import ServiceUnavailableError
+    from .too_many_requests_error import TooManyRequestsError
+    from .unprocessable_entity_error import UnprocessableEntityError
+_dynamic_imports: typing.Dict[str, str] = {
+    "BadRequestError": ".bad_request_error",
+    "ForbiddenError": ".forbidden_error",
+    "InternalServerError": ".internal_server_error",
+    "ServiceUnavailableError": ".service_unavailable_error",
+    "TooManyRequestsError": ".too_many_requests_error",
+    "UnprocessableEntityError": ".unprocessable_entity_error",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = [
     "BadRequestError",

sarvamai/errors/bad_request_error.py

@@ -6,5 +6,5 @@ from ..core.api_error import ApiError
 
 
 class BadRequestError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Any, headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=400, headers=headers, body=body)

sarvamai/errors/forbidden_error.py

@@ -6,5 +6,5 @@ from ..core.api_error import ApiError
 
 
 class ForbiddenError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Any, headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=403, headers=headers, body=body)

sarvamai/errors/internal_server_error.py

@@ -6,5 +6,5 @@ from ..core.api_error import ApiError
 
 
 class InternalServerError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Any, headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=500, headers=headers, body=body)

sarvamai/errors/service_unavailable_error.py

@@ -6,5 +6,5 @@ from ..core.api_error import ApiError
 
 
 class ServiceUnavailableError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Any, headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=503, headers=headers, body=body)

sarvamai/errors/too_many_requests_error.py

@@ -6,5 +6,5 @@ from ..core.api_error import ApiError
 
 
 class TooManyRequestsError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Any, headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=429, headers=headers, body=body)

sarvamai/errors/unprocessable_entity_error.py

@@ -6,5 +6,5 @@ from ..core.api_error import ApiError
 
 
 class UnprocessableEntityError(ApiError):
-    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Any, headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=422, headers=headers, body=body)

sarvamai/requests/__init__.py

@@ -2,68 +2,156 @@
 
 # isort: skip_file
 
-from .audio_data import AudioDataParams
-from .audio_message import AudioMessageParams
-from .audio_output import AudioOutputParams
-from .audio_output_data import AudioOutputDataParams
-from .base_job_parameters import BaseJobParametersParams
-from .bulk_job_callback import BulkJobCallbackParams
-from .bulk_job_init_response_v_1 import BulkJobInitResponseV1Params
-from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessageParams
-from .chat_completion_request_message import (
-    ChatCompletionRequestMessageParams,
-    ChatCompletionRequestMessage_AssistantParams,
-    ChatCompletionRequestMessage_SystemParams,
-    ChatCompletionRequestMessage_UserParams,
-)
-from .chat_completion_request_system_message import ChatCompletionRequestSystemMessageParams
-from .chat_completion_request_user_message import ChatCompletionRequestUserMessageParams
-from .chat_completion_response_message import ChatCompletionResponseMessageParams
-from .choice import ChoiceParams
-from .completion_usage import CompletionUsageParams
-from .config_message import ConfigMessageParams
-from .configure_connection import ConfigureConnectionParams
-from .configure_connection_data import ConfigureConnectionDataParams
-from .create_chat_completion_response import CreateChatCompletionResponseParams
-from .diarized_entry import DiarizedEntryParams
-from .diarized_transcript import DiarizedTranscriptParams
-from .error_data import ErrorDataParams
-from .error_details import ErrorDetailsParams
-from .error_message import ErrorMessageParams
-from .error_response import ErrorResponseParams
-from .error_response_data import ErrorResponseDataParams
-from .event_response import EventResponseParams
-from .event_response_data import EventResponseDataParams
-from .events_data import EventsDataParams
-from .file_signed_url_details import FileSignedUrlDetailsParams
-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_response import JobStatusV1ResponseParams
-from .language_identification_response import LanguageIdentificationResponseParams
-from .ping_signal import PingSignalParams
-from .send_text import SendTextParams
-from .send_text_data import SendTextDataParams
-from .speech_to_text_job_parameters import SpeechToTextJobParametersParams
-from .speech_to_text_response import SpeechToTextResponseParams
-from .speech_to_text_response_data import SpeechToTextResponseDataParams
-from .speech_to_text_streaming_response import SpeechToTextStreamingResponseParams
-from .speech_to_text_transcription_data import SpeechToTextTranscriptionDataParams
-from .speech_to_text_translate_job_parameters import SpeechToTextTranslateJobParametersParams
-from .speech_to_text_translate_response import SpeechToTextTranslateResponseParams
-from .speech_to_text_translate_response_data import SpeechToTextTranslateResponseDataParams
-from .speech_to_text_translate_streaming_response import SpeechToTextTranslateStreamingResponseParams
-from .speech_to_text_translate_transcription_data import SpeechToTextTranslateTranscriptionDataParams
-from .stop_configuration import StopConfigurationParams
-from .stt_flush_signal import SttFlushSignalParams
-from .task_detail_v_1 import TaskDetailV1Params
-from .task_file_details import TaskFileDetailsParams
-from .text_to_speech_response import TextToSpeechResponseParams
-from .timestamps_model import TimestampsModelParams
-from .transcription_metrics import TranscriptionMetricsParams
-from .translation_response import TranslationResponseParams
-from .transliteration_response import TransliterationResponseParams
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .audio_data import AudioDataParams
+    from .audio_message import AudioMessageParams
+    from .audio_output import AudioOutputParams
+    from .audio_output_data import AudioOutputDataParams
+    from .base_job_parameters import BaseJobParametersParams
+    from .bulk_job_callback import BulkJobCallbackParams
+    from .bulk_job_init_response_v_1 import BulkJobInitResponseV1Params
+    from .chat_completion_request_assistant_message import ChatCompletionRequestAssistantMessageParams
+    from .chat_completion_request_message import (
+        ChatCompletionRequestMessageParams,
+        ChatCompletionRequestMessage_AssistantParams,
+        ChatCompletionRequestMessage_SystemParams,
+        ChatCompletionRequestMessage_UserParams,
+    )
+    from .chat_completion_request_system_message import ChatCompletionRequestSystemMessageParams
+    from .chat_completion_request_user_message import ChatCompletionRequestUserMessageParams
+    from .chat_completion_response_message import ChatCompletionResponseMessageParams
+    from .choice import ChoiceParams
+    from .completion_usage import CompletionUsageParams
+    from .config_message import ConfigMessageParams
+    from .configure_connection import ConfigureConnectionParams
+    from .configure_connection_data import ConfigureConnectionDataParams
+    from .create_chat_completion_response import CreateChatCompletionResponseParams
+    from .diarized_entry import DiarizedEntryParams
+    from .diarized_transcript import DiarizedTranscriptParams
+    from .error_data import ErrorDataParams
+    from .error_details import ErrorDetailsParams
+    from .error_message import ErrorMessageParams
+    from .error_response import ErrorResponseParams
+    from .error_response_data import ErrorResponseDataParams
+    from .event_response import EventResponseParams
+    from .event_response_data import EventResponseDataParams
+    from .events_data import EventsDataParams
+    from .file_signed_url_details import FileSignedUrlDetailsParams
+    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_response import JobStatusV1ResponseParams
+    from .language_identification_response import LanguageIdentificationResponseParams
+    from .ping_signal import PingSignalParams
+    from .send_text import SendTextParams
+    from .send_text_data import SendTextDataParams
+    from .speech_to_text_job_parameters import SpeechToTextJobParametersParams
+    from .speech_to_text_response import SpeechToTextResponseParams
+    from .speech_to_text_response_data import SpeechToTextResponseDataParams
+    from .speech_to_text_streaming_response import SpeechToTextStreamingResponseParams
+    from .speech_to_text_transcription_data import SpeechToTextTranscriptionDataParams
+    from .speech_to_text_translate_job_parameters import SpeechToTextTranslateJobParametersParams
+    from .speech_to_text_translate_response import SpeechToTextTranslateResponseParams
+    from .speech_to_text_translate_response_data import SpeechToTextTranslateResponseDataParams
+    from .speech_to_text_translate_streaming_response import SpeechToTextTranslateStreamingResponseParams
+    from .speech_to_text_translate_transcription_data import SpeechToTextTranslateTranscriptionDataParams
+    from .stop_configuration import StopConfigurationParams
+    from .stt_flush_signal import SttFlushSignalParams
+    from .task_detail_v_1 import TaskDetailV1Params
+    from .task_file_details import TaskFileDetailsParams
+    from .text_to_speech_response import TextToSpeechResponseParams
+    from .timestamps_model import TimestampsModelParams
+    from .transcription_metrics import TranscriptionMetricsParams
+    from .translation_response import TranslationResponseParams
+    from .transliteration_response import TransliterationResponseParams
+_dynamic_imports: typing.Dict[str, str] = {
+    "AudioDataParams": ".audio_data",
+    "AudioMessageParams": ".audio_message",
+    "AudioOutputDataParams": ".audio_output_data",
+    "AudioOutputParams": ".audio_output",
+    "BaseJobParametersParams": ".base_job_parameters",
+    "BulkJobCallbackParams": ".bulk_job_callback",
+    "BulkJobInitResponseV1Params": ".bulk_job_init_response_v_1",
+    "ChatCompletionRequestAssistantMessageParams": ".chat_completion_request_assistant_message",
+    "ChatCompletionRequestMessageParams": ".chat_completion_request_message",
+    "ChatCompletionRequestMessage_AssistantParams": ".chat_completion_request_message",
+    "ChatCompletionRequestMessage_SystemParams": ".chat_completion_request_message",
+    "ChatCompletionRequestMessage_UserParams": ".chat_completion_request_message",
+    "ChatCompletionRequestSystemMessageParams": ".chat_completion_request_system_message",
+    "ChatCompletionRequestUserMessageParams": ".chat_completion_request_user_message",
+    "ChatCompletionResponseMessageParams": ".chat_completion_response_message",
+    "ChoiceParams": ".choice",
+    "CompletionUsageParams": ".completion_usage",
+    "ConfigMessageParams": ".config_message",
+    "ConfigureConnectionDataParams": ".configure_connection_data",
+    "ConfigureConnectionParams": ".configure_connection",
+    "CreateChatCompletionResponseParams": ".create_chat_completion_response",
+    "DiarizedEntryParams": ".diarized_entry",
+    "DiarizedTranscriptParams": ".diarized_transcript",
+    "ErrorDataParams": ".error_data",
+    "ErrorDetailsParams": ".error_details",
+    "ErrorMessageParams": ".error_message",
+    "ErrorResponseDataParams": ".error_response_data",
+    "ErrorResponseParams": ".error_response",
+    "EventResponseDataParams": ".event_response_data",
+    "EventResponseParams": ".event_response",
+    "EventsDataParams": ".events_data",
+    "FileSignedUrlDetailsParams": ".file_signed_url_details",
+    "FilesDownloadResponseParams": ".files_download_response",
+    "FilesRequestParams": ".files_request",
+    "FilesUploadResponseParams": ".files_upload_response",
+    "FlushSignalParams": ".flush_signal",
+    "JobStatusV1ResponseParams": ".job_status_v_1_response",
+    "LanguageIdentificationResponseParams": ".language_identification_response",
+    "PingSignalParams": ".ping_signal",
+    "SendTextDataParams": ".send_text_data",
+    "SendTextParams": ".send_text",
+    "SpeechToTextJobParametersParams": ".speech_to_text_job_parameters",
+    "SpeechToTextResponseDataParams": ".speech_to_text_response_data",
+    "SpeechToTextResponseParams": ".speech_to_text_response",
+    "SpeechToTextStreamingResponseParams": ".speech_to_text_streaming_response",
+    "SpeechToTextTranscriptionDataParams": ".speech_to_text_transcription_data",
+    "SpeechToTextTranslateJobParametersParams": ".speech_to_text_translate_job_parameters",
+    "SpeechToTextTranslateResponseDataParams": ".speech_to_text_translate_response_data",
+    "SpeechToTextTranslateResponseParams": ".speech_to_text_translate_response",
+    "SpeechToTextTranslateStreamingResponseParams": ".speech_to_text_translate_streaming_response",
+    "SpeechToTextTranslateTranscriptionDataParams": ".speech_to_text_translate_transcription_data",
+    "StopConfigurationParams": ".stop_configuration",
+    "SttFlushSignalParams": ".stt_flush_signal",
+    "TaskDetailV1Params": ".task_detail_v_1",
+    "TaskFileDetailsParams": ".task_file_details",
+    "TextToSpeechResponseParams": ".text_to_speech_response",
+    "TimestampsModelParams": ".timestamps_model",
+    "TranscriptionMetricsParams": ".transcription_metrics",
+    "TranslationResponseParams": ".translation_response",
+    "TransliterationResponseParams": ".transliteration_response",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
 
 __all__ = [
     "AudioDataParams",

sarvamai/requests/audio_data.py

@@ -3,7 +3,6 @@
 import typing
 
 import typing_extensions
-from ..types.audio_data_input_audio_codec import AudioDataInputAudioCodec
 
 
 class AudioDataParams(typing_extensions.TypedDict):
@@ -27,8 +26,3 @@ class AudioDataParams(typing_extensions.TypedDict):
     """
     Audio encoding format
     """
-
-    input_audio_codec: typing_extensions.NotRequired[AudioDataInputAudioCodec]
-    """
-    Audio codec/format of the input file. Our API automatically detects all codec formats, but for PCM files specifically (pcm_s16le, pcm_l16, pcm_raw), you must pass this parameter. PCM files supports sample rate 16000 and 8000.
-    """

sarvamai/requests/error_response_data.py

@@ -12,7 +12,7 @@ class ErrorResponseDataParams(typing_extensions.TypedDict):
     Optional error code for programmatic error handling
     """
 
-    details: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    details: typing_extensions.NotRequired[typing.Dict[str, typing.Any]]
     """
     Additional error details and context information
     """

sarvamai/requests/file_signed_url_details.py

@@ -7,4 +7,4 @@ import typing_extensions
 
 class FileSignedUrlDetailsParams(typing_extensions.TypedDict):
     file_url: str
-    file_metadata: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    file_metadata: typing_extensions.NotRequired[typing.Dict[str, typing.Any]]

sarvamai/requests/speech_to_text_transcription_data.py

@@ -17,12 +17,12 @@ class SpeechToTextTranscriptionDataParams(typing_extensions.TypedDict):
     Transcript of the provided speech in original language
     """
 
-    timestamps: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    timestamps: typing_extensions.NotRequired[typing.Dict[str, typing.Any]]
     """
     Timestamp information (if available)
     """
 
-    diarized_transcript: typing_extensions.NotRequired[typing.Dict[str, typing.Optional[typing.Any]]]
+    diarized_transcript: typing_extensions.NotRequired[typing.Dict[str, typing.Any]]
     """
     Diarized transcript of the provided speech
     """