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

mistralai_azure/models/chatcompletionstreamrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -17,8 +18,9 @@ from mistralai_azure.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from mistralai_azure.utils import get_discriminator
+from mistralai_azure.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -91,6 +93,8 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
     parallel_tool_calls: NotRequired[bool]
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -138,6 +142,11 @@ class ChatCompletionStreamRequest(BaseModel):
 
     parallel_tool_calls: Optional[bool] = None
 
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
+
     safe_prompt: Optional[bool] = None
     r"""Whether to inject a safety prompt before all conversations."""
 
@@ -159,16 +168,24 @@ class ChatCompletionStreamRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
             "safe_prompt",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
 
         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/completionresponsestreamchoice.py

@@ -38,7 +38,7 @@ class CompletionResponseStreamChoice(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/deltamessage.py

@@ -46,7 +46,7 @@ class DeltaMessage(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/documenturlchunk.py

@@ -0,0 +1,62 @@
+"""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 import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+DocumentURLChunkType = Literal["document_url"]
+
+
+class DocumentURLChunkTypedDict(TypedDict):
+    document_url: str
+    document_name: NotRequired[Nullable[str]]
+    r"""The filename of the document"""
+    type: NotRequired[DocumentURLChunkType]
+
+
+class DocumentURLChunk(BaseModel):
+    document_url: str
+
+    document_name: OptionalNullable[str] = UNSET
+    r"""The filename of the document"""
+
+    type: Optional[DocumentURLChunkType] = "document_url"
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["document_name", "type"]
+        nullable_fields = ["document_name"]
+        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/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)