mistralai 1.9.7__py3-none-any.whl → 1.9.10__py3-none-any.whl

mistralai_azure/models/usageinfo.py

@@ -1,19 +1,82 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai_azure.types import BaseModel
-from typing_extensions import TypedDict
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+import pydantic
+from pydantic import ConfigDict, model_serializer
+from typing import Any, Dict, Optional
+from typing_extensions import NotRequired, TypedDict
 
 
 class UsageInfoTypedDict(TypedDict):
-    prompt_tokens: int
-    completion_tokens: int
-    total_tokens: int
+    prompt_tokens: NotRequired[int]
+    completion_tokens: NotRequired[int]
+    total_tokens: NotRequired[int]
+    prompt_audio_seconds: NotRequired[Nullable[int]]
 
 
 class UsageInfo(BaseModel):
-    prompt_tokens: int
+    model_config = ConfigDict(
+        populate_by_name=True, arbitrary_types_allowed=True, extra="allow"
+    )
+    __pydantic_extra__: Dict[str, Any] = pydantic.Field(init=False)
 
-    completion_tokens: int
+    prompt_tokens: Optional[int] = 0
 
-    total_tokens: int
+    completion_tokens: Optional[int] = 0
+
+    total_tokens: Optional[int] = 0
+
+    prompt_audio_seconds: OptionalNullable[int] = UNSET
+
+    @property
+    def additional_properties(self):
+        return self.__pydantic_extra__
+
+    @additional_properties.setter
+    def additional_properties(self, value):
+        self.__pydantic_extra__ = value  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "prompt_tokens",
+            "completion_tokens",
+            "total_tokens",
+            "prompt_audio_seconds",
+        ]
+        nullable_fields = ["prompt_audio_seconds"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        for k, v in serialized.items():
+            m[k] = v
+
+        return m

mistralai_azure/models/usermessage.py

@@ -39,7 +39,7 @@ class UserMessage(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_azure/ocr.py

@@ -0,0 +1,271 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from .basesdk import BaseSDK
+from mistralai_azure import models, utils
+from mistralai_azure._hooks import HookContext
+from mistralai_azure.types import Nullable, OptionalNullable, UNSET
+from typing import Any, List, Mapping, Optional, Union
+
+
+class Ocr(BaseSDK):
+    def process(
+        self,
+        *,
+        model: Nullable[str],
+        document: Union[models.Document, models.DocumentTypedDict],
+        id: Optional[str] = None,
+        pages: OptionalNullable[List[int]] = UNSET,
+        include_image_base64: OptionalNullable[bool] = UNSET,
+        image_limit: OptionalNullable[int] = UNSET,
+        image_min_size: OptionalNullable[int] = UNSET,
+        bbox_annotation_format: OptionalNullable[
+            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+        ] = UNSET,
+        document_annotation_format: OptionalNullable[
+            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+        ] = UNSET,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> Optional[models.OCRResponse]:
+        r"""OCR
+
+        :param model:
+        :param document: Document to run OCR on
+        :param id:
+        :param pages: Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0
+        :param include_image_base64: Include image URLs in response
+        :param image_limit: Max images to extract
+        :param image_min_size: Minimum height and width of image to extract
+        :param bbox_annotation_format: Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field
+        :param document_annotation_format: Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.OCRRequest(
+            model=model,
+            id=id,
+            document=utils.get_pydantic_model(document, models.Document),
+            pages=pages,
+            include_image_base64=include_image_base64,
+            image_limit=image_limit,
+            image_min_size=image_min_size,
+            bbox_annotation_format=utils.get_pydantic_model(
+                bbox_annotation_format, OptionalNullable[models.ResponseFormat]
+            ),
+            document_annotation_format=utils.get_pydantic_model(
+                document_annotation_format, OptionalNullable[models.ResponseFormat]
+            ),
+        )
+
+        req = self._build_request(
+            method="POST",
+            path="/ocr",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.OCRRequest
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="ocr_v1_ocr_post",
+                oauth2_scopes=[],
+                security_source=self.sdk_configuration.security,
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, Optional[models.OCRResponse])
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = utils.unmarshal_json(
+                http_res.text, models.HTTPValidationErrorData
+            )
+            raise models.HTTPValidationError(data=response_data)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = utils.stream_to_text(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    async def process_async(
+        self,
+        *,
+        model: Nullable[str],
+        document: Union[models.Document, models.DocumentTypedDict],
+        id: Optional[str] = None,
+        pages: OptionalNullable[List[int]] = UNSET,
+        include_image_base64: OptionalNullable[bool] = UNSET,
+        image_limit: OptionalNullable[int] = UNSET,
+        image_min_size: OptionalNullable[int] = UNSET,
+        bbox_annotation_format: OptionalNullable[
+            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+        ] = UNSET,
+        document_annotation_format: OptionalNullable[
+            Union[models.ResponseFormat, models.ResponseFormatTypedDict]
+        ] = UNSET,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ) -> Optional[models.OCRResponse]:
+        r"""OCR
+
+        :param model:
+        :param document: Document to run OCR on
+        :param id:
+        :param pages: Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0
+        :param include_image_base64: Include image URLs in response
+        :param image_limit: Max images to extract
+        :param image_min_size: Minimum height and width of image to extract
+        :param bbox_annotation_format: Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field
+        :param document_annotation_format: Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field
+        :param retries: Override the default retry configuration for this method
+        :param server_url: Override the default server URL for this method
+        :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.OCRRequest(
+            model=model,
+            id=id,
+            document=utils.get_pydantic_model(document, models.Document),
+            pages=pages,
+            include_image_base64=include_image_base64,
+            image_limit=image_limit,
+            image_min_size=image_min_size,
+            bbox_annotation_format=utils.get_pydantic_model(
+                bbox_annotation_format, OptionalNullable[models.ResponseFormat]
+            ),
+            document_annotation_format=utils.get_pydantic_model(
+                document_annotation_format, OptionalNullable[models.ResponseFormat]
+            ),
+        )
+
+        req = self._build_request_async(
+            method="POST",
+            path="/ocr",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=True,
+            request_has_path_params=False,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            get_serialized_body=lambda: utils.serialize_request_body(
+                request, False, False, "json", models.OCRRequest
+            ),
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="ocr_v1_ocr_post",
+                oauth2_scopes=[],
+                security_source=self.sdk_configuration.security,
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, Optional[models.OCRResponse])
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = utils.unmarshal_json(
+                http_res.text, models.HTTPValidationErrorData
+            )
+            raise models.HTTPValidationError(data=response_data)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = await utils.stream_to_text_async(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )

mistralai_azure/sdk.py

@@ -1,27 +1,33 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-import weakref
-from typing import Any, Callable, Dict, Optional, Union, cast
-
-import httpx
-
-from mistralai_azure import models, utils
-from mistralai_azure._hooks import SDKHooks
-from mistralai_azure.chat import Chat
-from mistralai_azure.types import UNSET, OptionalNullable
-
 from .basesdk import BaseSDK
 from .httpclient import AsyncHttpClient, ClientOwner, HttpClient, close_clients
 from .sdkconfiguration import SDKConfiguration
 from .utils.logger import Logger, get_default_logger
 from .utils.retries import RetryConfig
+import httpx
+import importlib
+from mistralai_azure import models, utils
+from mistralai_azure._hooks import SDKHooks
+from mistralai_azure.types import OptionalNullable, UNSET
+from typing import Any, Callable, Dict, Optional, TYPE_CHECKING, Union, cast
+import weakref
+
+if TYPE_CHECKING:
+    from mistralai_azure.chat import Chat
+    from mistralai_azure.ocr import Ocr
 
 
 class MistralAzure(BaseSDK):
     r"""Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it."""
 
-    chat: Chat
+    chat: "Chat"
     r"""Chat Completion API."""
+    ocr: "Ocr"
+    _sub_sdk_map = {
+        "chat": ("mistralai_azure.chat", "Chat"),
+        "ocr": ("mistralai_azure.ocr", "Ocr"),
+    }
 
     def __init__(
         self,
@@ -101,6 +107,9 @@ class MistralAzure(BaseSDK):
 
         hooks = SDKHooks()
 
+        # pylint: disable=protected-access
+        self.sdk_configuration.__dict__["_hooks"] = hooks
+
         current_server_url, *_ = self.sdk_configuration.get_server_details()
         server_url, self.sdk_configuration.client = hooks.sdk_init(
             current_server_url, client
@@ -108,9 +117,6 @@ class MistralAzure(BaseSDK):
         if current_server_url != server_url:
             self.sdk_configuration.server_url = server_url
 
-        # pylint: disable=protected-access
-        self.sdk_configuration.__dict__["_hooks"] = hooks
-
         weakref.finalize(
             self,
             close_clients,
@@ -121,10 +127,32 @@ class MistralAzure(BaseSDK):
             self.sdk_configuration.async_client_supplied,
         )
 
-        self._init_sdks()
+    def __getattr__(self, name: str):
+        if name in self._sub_sdk_map:
+            module_path, class_name = self._sub_sdk_map[name]
+            try:
+                module = importlib.import_module(module_path)
+                klass = getattr(module, class_name)
+                instance = klass(self.sdk_configuration)
+                setattr(self, name, instance)
+                return instance
+            except ImportError as e:
+                raise AttributeError(
+                    f"Failed to import module {module_path} for attribute {name}: {e}"
+                ) from e
+            except AttributeError as e:
+                raise AttributeError(
+                    f"Failed to find class {class_name} in module {module_path} for attribute {name}: {e}"
+                ) from e
+
+        raise AttributeError(
+            f"'{type(self).__name__}' object has no attribute '{name}'"
+        )
 
-    def _init_sdks(self):
-        self.chat = Chat(self.sdk_configuration)
+    def __dir__(self):
+        default_attrs = list(super().__dir__())
+        lazy_attrs = list(self._sub_sdk_map.keys())
+        return sorted(list(set(default_attrs + lazy_attrs)))
 
     def __enter__(self):
         return self

mistralai_azure/sdkconfiguration.py

@@ -1,6 +1,5 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from ._hooks import SDKHooks
 from ._version import (
     __gen_version__,
     __openapi_doc_version__,
@@ -42,9 +41,6 @@ class SDKConfiguration:
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 
-    def __post_init__(self):
-        self._hooks = SDKHooks()
-
     def get_server_details(self) -> Tuple[str, Dict[str, str]]:
         if self.server_url is not None and self.server_url:
             return remove_suffix(self.server_url, "/"), {}
@@ -55,6 +51,3 @@ class SDKConfiguration:
             raise ValueError(f'Invalid server "{self.server}"')
 
         return SERVERS[self.server], {}
-
-    def get_hooks(self) -> SDKHooks:
-        return self._hooks

mistralai_azure/types/basemodel.py

@@ -2,7 +2,7 @@
 
 from pydantic import ConfigDict, model_serializer
 from pydantic import BaseModel as PydanticBaseModel
-from typing import TYPE_CHECKING, Literal, Optional, TypeVar, Union, NewType
+from typing import TYPE_CHECKING, Literal, Optional, TypeVar, Union
 from typing_extensions import TypeAliasType, TypeAlias
 
 
@@ -35,5 +35,5 @@ else:
         "OptionalNullable", Union[Optional[Nullable[T]], Unset], type_params=(T,)
     )
 
-UnrecognizedInt = NewType("UnrecognizedInt", int)
-UnrecognizedStr = NewType("UnrecognizedStr", str)
+UnrecognizedInt: TypeAlias = int
+UnrecognizedStr: TypeAlias = str