mistralai 1.5.1__tar.gz → 1.5.2__tar.gz

{mistralai-1.5.1 → mistralai-1.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.5.1
+Version: 1.5.2
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.8
@@ -267,10 +267,10 @@ with Mistral(
     api_key=os.getenv("MISTRAL_API_KEY", ""),
 ) as mistral:
 
-    res = mistral.embeddings.create(inputs=[
+    res = mistral.embeddings.create(model="mistral-embed", inputs=[
         "Embed this sentence.",
         "As well as this one.",
-    ], model="mistral-embed")
+    ])
 
     # Handle response
     print(res)
@@ -290,10 +290,10 @@ async def main():
         api_key=os.getenv("MISTRAL_API_KEY", ""),
     ) as mistral:
 
-        res = await mistral.embeddings.create_async(inputs=[
+        res = await mistral.embeddings.create_async(model="mistral-embed", inputs=[
             "Embed this sentence.",
             "As well as this one.",
-        ], model="mistral-embed")
+        ])
 
         # Handle response
         print(res)

{mistralai-1.5.1 → mistralai-1.5.2}/README-PYPI.md

@@ -242,10 +242,10 @@ with Mistral(
     api_key=os.getenv("MISTRAL_API_KEY", ""),
 ) as mistral:
 
-    res = mistral.embeddings.create(inputs=[
+    res = mistral.embeddings.create(model="mistral-embed", inputs=[
         "Embed this sentence.",
         "As well as this one.",
-    ], model="mistral-embed")
+    ])
 
     # Handle response
     print(res)
@@ -265,10 +265,10 @@ async def main():
         api_key=os.getenv("MISTRAL_API_KEY", ""),
     ) as mistral:
 
-        res = await mistral.embeddings.create_async(inputs=[
+        res = await mistral.embeddings.create_async(model="mistral-embed", inputs=[
             "Embed this sentence.",
             "As well as this one.",
-        ], model="mistral-embed")
+        ])
 
         # Handle response
         print(res)

{mistralai-1.5.1 → mistralai-1.5.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mistralai"
-version = "1.5.1"
+version = "1.5.2"
 description = "Python Client SDK for the Mistral AI API."
 authors = ["Mistral"]
 readme = "README-PYPI.md"

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.5.1"
+__version__: str = "1.5.2"
 __openapi_doc_version__: str = "0.0.2"
 __gen_version__: str = "2.497.0"
-__user_agent__: str = "speakeasy-sdk/python 1.5.1 2.497.0 0.0.2 mistralai"
+__user_agent__: str = "speakeasy-sdk/python 1.5.2 2.497.0 0.0.2 mistralai"
 
 try:
     if __package__ is not None:

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/embeddings.py

@@ -14,8 +14,8 @@ class Embeddings(BaseSDK):
     def create(
         self,
         *,
+        model: str,
         inputs: Union[models.Inputs, models.InputsTypedDict],
-        model: Optional[str] = "mistral-embed",
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -25,8 +25,8 @@ class Embeddings(BaseSDK):
 
         Embeddings
 
-        :param inputs: Text to embed.
         :param model: ID of the model to use.
+        :param inputs: Text to embed.
         :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
@@ -114,8 +114,8 @@ class Embeddings(BaseSDK):
     async def create_async(
         self,
         *,
+        model: str,
         inputs: Union[models.Inputs, models.InputsTypedDict],
-        model: Optional[str] = "mistral-embed",
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -125,8 +125,8 @@ class Embeddings(BaseSDK):
 
         Embeddings
 
-        :param inputs: Text to embed.
         :param model: ID of the model to use.
+        :param inputs: Text to embed.
         :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

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/extra/utils/response_format.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel
-from typing import TypeVar, Any, Type
+from typing import TypeVar, Any, Type, Dict
 from ...models import JSONSchema, ResponseFormat
 from ._pydantic_helper import rec_strict_json_schema
 
@@ -7,7 +7,7 @@ CustomPydanticModel = TypeVar("CustomPydanticModel", bound=BaseModel)
 
 
 def response_format_from_pydantic_model(
-    model: type[CustomPydanticModel],
+    model: Type[CustomPydanticModel],
 ) -> ResponseFormat:
     """Generate a strict JSON schema from a pydantic model."""
     model_schema = rec_strict_json_schema(model.model_json_schema())
@@ -18,7 +18,7 @@ def response_format_from_pydantic_model(
 
 
 def pydantic_model_from_json(
-    json_data: dict[str, Any], pydantic_model: Type[CustomPydanticModel]
+    json_data: Dict[str, Any], pydantic_model: Type[CustomPydanticModel]
 ) -> CustomPydanticModel:
     """Parse a JSON schema into a pydantic model."""
     return pydantic_model.model_validate(json_data)

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/models/__init__.py

@@ -115,7 +115,11 @@ from .detailedjobout import (
     DetailedJobOutStatus,
     DetailedJobOutTypedDict,
 )
-from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
+from .documenturlchunk import (
+    DocumentURLChunk,
+    DocumentURLChunkType,
+    DocumentURLChunkTypedDict,
+)
 from .embeddingrequest import (
     EmbeddingRequest,
     EmbeddingRequestTypedDict,
@@ -455,6 +459,7 @@ __all__ = [
     "Document",
     "DocumentTypedDict",
     "DocumentURLChunk",
+    "DocumentURLChunkType",
     "DocumentURLChunkTypedDict",
     "EmbeddingRequest",
     "EmbeddingRequestTypedDict",

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/models/documenturlchunk.py

@@ -2,38 +2,32 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import validate_const
-import pydantic
 from pydantic import model_serializer
-from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
+
+
+DocumentURLChunkType = Literal["document_url"]
 
 
 class DocumentURLChunkTypedDict(TypedDict):
     document_url: str
-    type: Literal["document_url"]
     document_name: NotRequired[Nullable[str]]
     r"""The filename of the document"""
+    type: NotRequired[DocumentURLChunkType]
 
 
 class DocumentURLChunk(BaseModel):
     document_url: str
 
-    TYPE: Annotated[
-        Annotated[
-            Optional[Literal["document_url"]],
-            AfterValidator(validate_const("document_url")),
-        ],
-        pydantic.Field(alias="type"),
-    ] = "document_url"
-
     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 = ["type", "document_name"]
+        optional_fields = ["document_name", "type"]
         nullable_fields = ["document_name"]
         null_default_fields = []
 

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/models/embeddingrequest.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 import pydantic
-from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+from typing import List, Union
+from typing_extensions import Annotated, TypeAliasType, TypedDict
 
 
 InputsTypedDict = TypeAliasType("InputsTypedDict", Union[str, List[str]])
@@ -16,15 +16,15 @@ r"""Text to embed."""
 
 
 class EmbeddingRequestTypedDict(TypedDict):
+    model: str
+    r"""ID of the model to use."""
     inputs: InputsTypedDict
     r"""Text to embed."""
-    model: NotRequired[str]
-    r"""ID of the model to use."""
 
 
 class EmbeddingRequest(BaseModel):
+    model: str
+    r"""ID of the model to use."""
+
     inputs: Annotated[Inputs, pydantic.Field(alias="input")]
     r"""Text to embed."""
-
-    model: Optional[str] = "mistral-embed"
-    r"""ID of the model to use."""

{mistralai-1.5.1 → mistralai-1.5.2}/src/mistralai/models/filepurpose.py

@@ -5,4 +5,4 @@ from mistralai.types import UnrecognizedStr
 from typing import Literal, Union
 
 
-FilePurpose = Union[Literal["fine-tune", "batch"], UnrecognizedStr]
+FilePurpose = Union[Literal["fine-tune", "batch", "ocr"], UnrecognizedStr]