mistralai 1.9.6__py3-none-any.whl → 1.9.8__py3-none-any.whl

mistralai_azure/models/filechunk.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import BaseModel
+from mistralai_azure.utils import validate_const
+import pydantic
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, TypedDict
+
+
+class FileChunkTypedDict(TypedDict):
+    file_id: str
+    type: Literal["file"]
+
+
+class FileChunk(BaseModel):
+    file_id: str
+
+    TYPE: Annotated[
+        Annotated[Optional[Literal["file"]], AfterValidator(validate_const("file"))],
+        pydantic.Field(alias="type"),
+    ] = "file"

mistralai_azure/models/imageurl.py

@@ -32,7 +32,7 @@ class ImageURL(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/models/jsonschema.py

@@ -40,7 +40,7 @@ class JSONSchema(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/models/mistralpromptmode.py

@@ -0,0 +1,8 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+MistralPromptMode = Union[Literal["reasoning"], UnrecognizedStr]

mistralai_azure/models/ocrimageobject.py

@@ -0,0 +1,89 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class OCRImageObjectTypedDict(TypedDict):
+    id: str
+    r"""Image ID for extracted image in a page"""
+    top_left_x: Nullable[int]
+    r"""X coordinate of top-left corner of the extracted image"""
+    top_left_y: Nullable[int]
+    r"""Y coordinate of top-left corner of the extracted image"""
+    bottom_right_x: Nullable[int]
+    r"""X coordinate of bottom-right corner of the extracted image"""
+    bottom_right_y: Nullable[int]
+    r"""Y coordinate of bottom-right corner of the extracted image"""
+    image_base64: NotRequired[Nullable[str]]
+    r"""Base64 string of the extracted image"""
+    image_annotation: NotRequired[Nullable[str]]
+    r"""Annotation of the extracted image in json str"""
+
+
+class OCRImageObject(BaseModel):
+    id: str
+    r"""Image ID for extracted image in a page"""
+
+    top_left_x: Nullable[int]
+    r"""X coordinate of top-left corner of the extracted image"""
+
+    top_left_y: Nullable[int]
+    r"""Y coordinate of top-left corner of the extracted image"""
+
+    bottom_right_x: Nullable[int]
+    r"""X coordinate of bottom-right corner of the extracted image"""
+
+    bottom_right_y: Nullable[int]
+    r"""Y coordinate of bottom-right corner of the extracted image"""
+
+    image_base64: OptionalNullable[str] = UNSET
+    r"""Base64 string of the extracted image"""
+
+    image_annotation: OptionalNullable[str] = UNSET
+    r"""Annotation of the extracted image in json str"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["image_base64", "image_annotation"]
+        nullable_fields = [
+            "top_left_x",
+            "top_left_y",
+            "bottom_right_x",
+            "bottom_right_y",
+            "image_base64",
+            "image_annotation",
+        ]
+        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
+
+        return m

mistralai_azure/models/ocrpagedimensions.py

@@ -0,0 +1,25 @@
+"""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
+
+
+class OCRPageDimensionsTypedDict(TypedDict):
+    dpi: int
+    r"""Dots per inch of the page-image"""
+    height: int
+    r"""Height of the image in pixels"""
+    width: int
+    r"""Width of the image in pixels"""
+
+
+class OCRPageDimensions(BaseModel):
+    dpi: int
+    r"""Dots per inch of the page-image"""
+
+    height: int
+    r"""Height of the image in pixels"""
+
+    width: int
+    r"""Width of the image in pixels"""

mistralai_azure/models/ocrpageobject.py

@@ -0,0 +1,64 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .ocrimageobject import OCRImageObject, OCRImageObjectTypedDict
+from .ocrpagedimensions import OCRPageDimensions, OCRPageDimensionsTypedDict
+from mistralai_azure.types import BaseModel, Nullable, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import List
+from typing_extensions import TypedDict
+
+
+class OCRPageObjectTypedDict(TypedDict):
+    index: int
+    r"""The page index in a pdf document starting from 0"""
+    markdown: str
+    r"""The markdown string response of the page"""
+    images: List[OCRImageObjectTypedDict]
+    r"""List of all extracted images in the page"""
+    dimensions: Nullable[OCRPageDimensionsTypedDict]
+    r"""The dimensions of the PDF Page's screenshot image"""
+
+
+class OCRPageObject(BaseModel):
+    index: int
+    r"""The page index in a pdf document starting from 0"""
+
+    markdown: str
+    r"""The markdown string response of the page"""
+
+    images: List[OCRImageObject]
+    r"""List of all extracted images in the page"""
+
+    dimensions: Nullable[OCRPageDimensions]
+    r"""The dimensions of the PDF Page's screenshot image"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = []
+        nullable_fields = ["dimensions"]
+        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
+
+        return m

mistralai_azure/models/ocrrequest.py

@@ -0,0 +1,120 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
+from .filechunk import FileChunk, FileChunkTypedDict
+from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
+from .responseformat import ResponseFormat, ResponseFormatTypedDict
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing import List, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+DocumentTypedDict = TypeAliasType(
+    "DocumentTypedDict",
+    Union[FileChunkTypedDict, ImageURLChunkTypedDict, DocumentURLChunkTypedDict],
+)
+r"""Document to run OCR on"""
+
+
+Document = TypeAliasType("Document", Union[FileChunk, ImageURLChunk, DocumentURLChunk])
+r"""Document to run OCR on"""
+
+
+class OCRRequestTypedDict(TypedDict):
+    model: Nullable[str]
+    document: DocumentTypedDict
+    r"""Document to run OCR on"""
+    id: NotRequired[str]
+    pages: NotRequired[Nullable[List[int]]]
+    r"""Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0"""
+    include_image_base64: NotRequired[Nullable[bool]]
+    r"""Include image URLs in response"""
+    image_limit: NotRequired[Nullable[int]]
+    r"""Max images to extract"""
+    image_min_size: NotRequired[Nullable[int]]
+    r"""Minimum height and width of image to extract"""
+    bbox_annotation_format: NotRequired[Nullable[ResponseFormatTypedDict]]
+    r"""Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field"""
+    document_annotation_format: NotRequired[Nullable[ResponseFormatTypedDict]]
+    r"""Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field"""
+
+
+class OCRRequest(BaseModel):
+    model: Nullable[str]
+
+    document: Document
+    r"""Document to run OCR on"""
+
+    id: Optional[str] = None
+
+    pages: OptionalNullable[List[int]] = UNSET
+    r"""Specific pages user wants to process in various formats: single number, range, or list of both. Starts from 0"""
+
+    include_image_base64: OptionalNullable[bool] = UNSET
+    r"""Include image URLs in response"""
+
+    image_limit: OptionalNullable[int] = UNSET
+    r"""Max images to extract"""
+
+    image_min_size: OptionalNullable[int] = UNSET
+    r"""Minimum height and width of image to extract"""
+
+    bbox_annotation_format: OptionalNullable[ResponseFormat] = UNSET
+    r"""Structured output class for extracting useful information from each extracted bounding box / image from document. Only json_schema is valid for this field"""
+
+    document_annotation_format: OptionalNullable[ResponseFormat] = UNSET
+    r"""Structured output class for extracting useful information from the entire document. Only json_schema is valid for this field"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "id",
+            "pages",
+            "include_image_base64",
+            "image_limit",
+            "image_min_size",
+            "bbox_annotation_format",
+            "document_annotation_format",
+        ]
+        nullable_fields = [
+            "model",
+            "pages",
+            "include_image_base64",
+            "image_limit",
+            "image_min_size",
+            "bbox_annotation_format",
+            "document_annotation_format",
+        ]
+        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
+
+        return m

mistralai_azure/models/ocrresponse.py

@@ -0,0 +1,68 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .ocrpageobject import OCRPageObject, OCRPageObjectTypedDict
+from .ocrusageinfo import OCRUsageInfo, OCRUsageInfoTypedDict
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing import List
+from typing_extensions import NotRequired, TypedDict
+
+
+class OCRResponseTypedDict(TypedDict):
+    pages: List[OCRPageObjectTypedDict]
+    r"""List of OCR info for pages."""
+    model: str
+    r"""The model used to generate the OCR."""
+    usage_info: OCRUsageInfoTypedDict
+    document_annotation: NotRequired[Nullable[str]]
+    r"""Formatted response in the request_format if provided in json str"""
+
+
+class OCRResponse(BaseModel):
+    pages: List[OCRPageObject]
+    r"""List of OCR info for pages."""
+
+    model: str
+    r"""The model used to generate the OCR."""
+
+    usage_info: OCRUsageInfo
+
+    document_annotation: OptionalNullable[str] = UNSET
+    r"""Formatted response in the request_format if provided in json str"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["document_annotation"]
+        nullable_fields = ["document_annotation"]
+        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
+
+        return m

mistralai_azure/models/ocrusageinfo.py

@@ -0,0 +1,57 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai_azure.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+)
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class OCRUsageInfoTypedDict(TypedDict):
+    pages_processed: int
+    r"""Number of pages processed"""
+    doc_size_bytes: NotRequired[Nullable[int]]
+    r"""Document size in bytes"""
+
+
+class OCRUsageInfo(BaseModel):
+    pages_processed: int
+    r"""Number of pages processed"""
+
+    doc_size_bytes: OptionalNullable[int] = UNSET
+    r"""Document size in bytes"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["doc_size_bytes"]
+        nullable_fields = ["doc_size_bytes"]
+        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
+
+        return m

mistralai_azure/models/responseformat.py

@@ -37,7 +37,7 @@ class ResponseFormat(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/models/toolmessage.py

@@ -51,7 +51,7 @@ class ToolMessage(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/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)