mistralai 1.5.0__tar.gz → 1.5.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.5.0
+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)
@@ -467,6 +467,10 @@ The documentation for the GCP SDK is available [here](https://github.com/mistral
 * [archive](https://github.com/mistralai/client-python/blob/master/docs/sdks/models/README.md#archive) - Archive Fine Tuned Model
 * [unarchive](https://github.com/mistralai/client-python/blob/master/docs/sdks/models/README.md#unarchive) - Unarchive Fine Tuned Model
 
+### [ocr](https://github.com/mistralai/client-python/blob/master/docs/sdks/ocr/README.md)
+
+* [process](https://github.com/mistralai/client-python/blob/master/docs/sdks/ocr/README.md#process) - OCR
+
 </details>
 <!-- End Available Resources and Operations [operations] -->
 

{mistralai-1.5.0 → 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)
@@ -442,6 +442,10 @@ The documentation for the GCP SDK is available [here](https://github.com/mistral
 * [archive](https://github.com/mistralai/client-python/blob/master/docs/sdks/models/README.md#archive) - Archive Fine Tuned Model
 * [unarchive](https://github.com/mistralai/client-python/blob/master/docs/sdks/models/README.md#unarchive) - Unarchive Fine Tuned Model
 
+### [ocr](https://github.com/mistralai/client-python/blob/master/docs/sdks/ocr/README.md)
+
+* [process](https://github.com/mistralai/client-python/blob/master/docs/sdks/ocr/README.md#process) - OCR
+
 </details>
 <!-- End Available Resources and Operations [operations] -->
 

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

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

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

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.5.0"
+__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.0 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.0 → mistralai-1.5.2}/src/mistralai/chat.py

@@ -3,7 +3,7 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
-from mistralai.types import Nullable, OptionalNullable, UNSET
+from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import eventstreaming, get_security_from_env
 from typing import Any, List, Mapping, Optional, Union
 
@@ -96,7 +96,7 @@ class Chat(BaseSDK):
     def complete(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         messages: Union[List[models.Messages], List[models.MessagesTypedDict]],
         temperature: OptionalNullable[float] = UNSET,
         top_p: Optional[float] = None,
@@ -253,7 +253,7 @@ class Chat(BaseSDK):
     async def complete_async(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         messages: Union[List[models.Messages], List[models.MessagesTypedDict]],
         temperature: OptionalNullable[float] = UNSET,
         top_p: Optional[float] = None,
@@ -410,7 +410,7 @@ class Chat(BaseSDK):
     def stream(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         messages: Union[
             List[models.ChatCompletionStreamRequestMessages],
             List[models.ChatCompletionStreamRequestMessagesTypedDict],
@@ -585,7 +585,7 @@ class Chat(BaseSDK):
     async def stream_async(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         messages: Union[
             List[models.ChatCompletionStreamRequestMessages],
             List[models.ChatCompletionStreamRequestMessagesTypedDict],

{mistralai-1.5.0 → mistralai-1.5.2}/src/mistralai/classifiers.py

@@ -3,7 +3,7 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
-from mistralai.types import Nullable, OptionalNullable, UNSET
+from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from typing import Any, Mapping, Optional, Union
 
@@ -14,11 +14,11 @@ class Classifiers(BaseSDK):
     def moderate(
         self,
         *,
+        model: str,
         inputs: Union[
             models.ClassificationRequestInputs,
             models.ClassificationRequestInputsTypedDict,
         ],
-        model: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -26,8 +26,8 @@ class Classifiers(BaseSDK):
     ) -> models.ClassificationResponse:
         r"""Moderations
 
+        :param model: ID of the model to use.
         :param inputs: Text to classify.
-        :param model:
         :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
@@ -42,8 +42,8 @@ class Classifiers(BaseSDK):
             base_url = server_url
 
         request = models.ClassificationRequest(
-            inputs=inputs,
             model=model,
+            inputs=inputs,
         )
 
         req = self._build_request(
@@ -115,11 +115,11 @@ class Classifiers(BaseSDK):
     async def moderate_async(
         self,
         *,
+        model: str,
         inputs: Union[
             models.ClassificationRequestInputs,
             models.ClassificationRequestInputsTypedDict,
         ],
-        model: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -127,8 +127,8 @@ class Classifiers(BaseSDK):
     ) -> models.ClassificationResponse:
         r"""Moderations
 
+        :param model: ID of the model to use.
         :param inputs: Text to classify.
-        :param model:
         :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
@@ -143,8 +143,8 @@ class Classifiers(BaseSDK):
             base_url = server_url
 
         request = models.ClassificationRequest(
-            inputs=inputs,
             model=model,
+            inputs=inputs,
         )
 
         req = self._build_request_async(
@@ -216,11 +216,12 @@ class Classifiers(BaseSDK):
     def moderate_chat(
         self,
         *,
+        model: str,
         inputs: Union[
-            models.ChatClassificationRequestInputs,
-            models.ChatClassificationRequestInputsTypedDict,
+            models.ChatModerationRequestInputs,
+            models.ChatModerationRequestInputsTypedDict,
         ],
-        model: Nullable[str],
+        truncate_for_context_length: Optional[bool] = False,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -228,8 +229,9 @@ class Classifiers(BaseSDK):
     ) -> models.ClassificationResponse:
         r"""Moderations Chat
 
-        :param inputs: Chat to classify
         :param model:
+        :param inputs: Chat to classify
+        :param truncate_for_context_length:
         :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
@@ -243,11 +245,10 @@ class Classifiers(BaseSDK):
         if server_url is not None:
             base_url = server_url
 
-        request = models.ChatClassificationRequest(
-            inputs=utils.get_pydantic_model(
-                inputs, models.ChatClassificationRequestInputs
-            ),
+        request = models.ChatModerationRequest(
             model=model,
+            inputs=utils.get_pydantic_model(inputs, models.ChatModerationRequestInputs),
+            truncate_for_context_length=truncate_for_context_length,
         )
 
         req = self._build_request(
@@ -264,7 +265,7 @@ class Classifiers(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request, False, False, "json", models.ChatClassificationRequest
+                request, False, False, "json", models.ChatModerationRequest
             ),
             timeout_ms=timeout_ms,
         )
@@ -319,11 +320,12 @@ class Classifiers(BaseSDK):
     async def moderate_chat_async(
         self,
         *,
+        model: str,
         inputs: Union[
-            models.ChatClassificationRequestInputs,
-            models.ChatClassificationRequestInputsTypedDict,
+            models.ChatModerationRequestInputs,
+            models.ChatModerationRequestInputsTypedDict,
         ],
-        model: Nullable[str],
+        truncate_for_context_length: Optional[bool] = False,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -331,8 +333,9 @@ class Classifiers(BaseSDK):
     ) -> models.ClassificationResponse:
         r"""Moderations Chat
 
-        :param inputs: Chat to classify
         :param model:
+        :param inputs: Chat to classify
+        :param truncate_for_context_length:
         :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
@@ -346,11 +349,10 @@ class Classifiers(BaseSDK):
         if server_url is not None:
             base_url = server_url
 
-        request = models.ChatClassificationRequest(
-            inputs=utils.get_pydantic_model(
-                inputs, models.ChatClassificationRequestInputs
-            ),
+        request = models.ChatModerationRequest(
             model=model,
+            inputs=utils.get_pydantic_model(inputs, models.ChatModerationRequestInputs),
+            truncate_for_context_length=truncate_for_context_length,
         )
 
         req = self._build_request_async(
@@ -367,7 +369,7 @@ class Classifiers(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request, False, False, "json", models.ChatClassificationRequest
+                request, False, False, "json", models.ChatModerationRequest
             ),
             timeout_ms=timeout_ms,
         )

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

@@ -14,9 +14,8 @@ class Embeddings(BaseSDK):
     def create(
         self,
         *,
+        model: str,
         inputs: Union[models.Inputs, models.InputsTypedDict],
-        model: Optional[str] = "mistral-embed",
-        encoding_format: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -26,9 +25,8 @@ class Embeddings(BaseSDK):
 
         Embeddings
 
-        :param inputs: Text to embed.
         :param model: ID of the model to use.
-        :param encoding_format: The format to return the embeddings in.
+        :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
@@ -43,9 +41,8 @@ class Embeddings(BaseSDK):
             base_url = server_url
 
         request = models.EmbeddingRequest(
-            inputs=inputs,
             model=model,
-            encoding_format=encoding_format,
+            inputs=inputs,
         )
 
         req = self._build_request(
@@ -117,9 +114,8 @@ class Embeddings(BaseSDK):
     async def create_async(
         self,
         *,
+        model: str,
         inputs: Union[models.Inputs, models.InputsTypedDict],
-        model: Optional[str] = "mistral-embed",
-        encoding_format: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -129,9 +125,8 @@ class Embeddings(BaseSDK):
 
         Embeddings
 
-        :param inputs: Text to embed.
         :param model: ID of the model to use.
-        :param encoding_format: The format to return the embeddings in.
+        :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
@@ -146,9 +141,8 @@ class Embeddings(BaseSDK):
             base_url = server_url
 
         request = models.EmbeddingRequest(
-            inputs=inputs,
             model=model,
-            encoding_format=encoding_format,
+            inputs=inputs,
         )
 
         req = self._build_request_async(

{mistralai-1.5.0 → 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.0 → mistralai-1.5.2}/src/mistralai/fim.py

@@ -3,7 +3,7 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
-from mistralai.types import Nullable, OptionalNullable, UNSET
+from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import eventstreaming, get_security_from_env
 from typing import Any, Mapping, Optional, Union
 
@@ -14,7 +14,7 @@ class Fim(BaseSDK):
     def complete(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         prompt: str,
         temperature: OptionalNullable[float] = UNSET,
         top_p: Optional[float] = 1,
@@ -143,7 +143,7 @@ class Fim(BaseSDK):
     async def complete_async(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         prompt: str,
         temperature: OptionalNullable[float] = UNSET,
         top_p: Optional[float] = 1,
@@ -272,7 +272,7 @@ class Fim(BaseSDK):
     def stream(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         prompt: str,
         temperature: OptionalNullable[float] = UNSET,
         top_p: Optional[float] = 1,
@@ -407,7 +407,7 @@ class Fim(BaseSDK):
     async def stream_async(
         self,
         *,
-        model: Nullable[str],
+        model: str,
         prompt: str,
         temperature: OptionalNullable[float] = UNSET,
         top_p: Optional[float] = 1,

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

@@ -39,16 +39,6 @@ from .batchjobin import BatchJobIn, BatchJobInTypedDict
 from .batchjobout import BatchJobOut, BatchJobOutObject, BatchJobOutTypedDict
 from .batchjobsout import BatchJobsOut, BatchJobsOutObject, BatchJobsOutTypedDict
 from .batchjobstatus import BatchJobStatus
-from .chatclassificationrequest import (
-    ChatClassificationRequest,
-    ChatClassificationRequestInputs,
-    ChatClassificationRequestInputsTypedDict,
-    ChatClassificationRequestTypedDict,
-    One,
-    OneTypedDict,
-    Two,
-    TwoTypedDict,
-)
 from .chatcompletionchoice import (
     ChatCompletionChoice,
     ChatCompletionChoiceTypedDict,
@@ -78,6 +68,16 @@ from .chatcompletionstreamrequest import (
     ChatCompletionStreamRequestToolChoiceTypedDict,
     ChatCompletionStreamRequestTypedDict,
 )
+from .chatmoderationrequest import (
+    ChatModerationRequest,
+    ChatModerationRequestInputs,
+    ChatModerationRequestInputsTypedDict,
+    ChatModerationRequestTypedDict,
+    One,
+    OneTypedDict,
+    Two,
+    TwoTypedDict,
+)
 from .checkpointout import CheckpointOut, CheckpointOutTypedDict
 from .classificationobject import ClassificationObject, ClassificationObjectTypedDict
 from .classificationrequest import (
@@ -115,6 +115,11 @@ from .detailedjobout import (
     DetailedJobOutStatus,
     DetailedJobOutTypedDict,
 )
+from .documenturlchunk import (
+    DocumentURLChunk,
+    DocumentURLChunkType,
+    DocumentURLChunkTypedDict,
+)
 from .embeddingrequest import (
     EmbeddingRequest,
     EmbeddingRequestTypedDict,
@@ -274,7 +279,13 @@ from .listfilesout import ListFilesOut, ListFilesOutTypedDict
 from .metricout import MetricOut, MetricOutTypedDict
 from .modelcapabilities import ModelCapabilities, ModelCapabilitiesTypedDict
 from .modellist import Data, DataTypedDict, ModelList, ModelListTypedDict
-from .prediction import Prediction, PredictionType, PredictionTypedDict
+from .ocrimageobject import OCRImageObject, OCRImageObjectTypedDict
+from .ocrpagedimensions import OCRPageDimensions, OCRPageDimensionsTypedDict
+from .ocrpageobject import OCRPageObject, OCRPageObjectTypedDict
+from .ocrrequest import Document, DocumentTypedDict, OCRRequest, OCRRequestTypedDict
+from .ocrresponse import OCRResponse, OCRResponseTypedDict
+from .ocrusageinfo import OCRUsageInfo, OCRUsageInfoTypedDict
+from .prediction import Prediction, PredictionTypedDict
 from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .responseformats import ResponseFormats
@@ -386,10 +397,6 @@ __all__ = [
     "BatchJobsOut",
     "BatchJobsOutObject",
     "BatchJobsOutTypedDict",
-    "ChatClassificationRequest",
-    "ChatClassificationRequestInputs",
-    "ChatClassificationRequestInputsTypedDict",
-    "ChatClassificationRequestTypedDict",
     "ChatCompletionChoice",
     "ChatCompletionChoiceTypedDict",
     "ChatCompletionRequest",
@@ -406,6 +413,10 @@ __all__ = [
     "ChatCompletionStreamRequestToolChoice",
     "ChatCompletionStreamRequestToolChoiceTypedDict",
     "ChatCompletionStreamRequestTypedDict",
+    "ChatModerationRequest",
+    "ChatModerationRequestInputs",
+    "ChatModerationRequestInputsTypedDict",
+    "ChatModerationRequestTypedDict",
     "CheckpointOut",
     "CheckpointOutTypedDict",
     "ClassificationObject",
@@ -445,6 +456,11 @@ __all__ = [
     "DetailedJobOutRepositoriesTypedDict",
     "DetailedJobOutStatus",
     "DetailedJobOutTypedDict",
+    "Document",
+    "DocumentTypedDict",
+    "DocumentURLChunk",
+    "DocumentURLChunkType",
+    "DocumentURLChunkTypedDict",
     "EmbeddingRequest",
     "EmbeddingRequestTypedDict",
     "EmbeddingResponse",
@@ -568,11 +584,22 @@ __all__ = [
     "ModelCapabilitiesTypedDict",
     "ModelList",
     "ModelListTypedDict",
+    "OCRImageObject",
+    "OCRImageObjectTypedDict",
+    "OCRPageDimensions",
+    "OCRPageDimensionsTypedDict",
+    "OCRPageObject",
+    "OCRPageObjectTypedDict",
+    "OCRRequest",
+    "OCRRequestTypedDict",
+    "OCRResponse",
+    "OCRResponseTypedDict",
+    "OCRUsageInfo",
+    "OCRUsageInfoTypedDict",
     "Object",
     "One",
     "OneTypedDict",
     "Prediction",
-    "PredictionType",
     "PredictionTypedDict",
     "QueryParamStatus",
     "ReferenceChunk",

{mistralai-1.5.0 → mistralai-1.5.2}/src/mistralai/models/assistantmessage.py

@@ -26,6 +26,7 @@ class AssistantMessageTypedDict(TypedDict):
     content: NotRequired[Nullable[AssistantMessageContentTypedDict]]
     tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
     prefix: NotRequired[bool]
+    r"""Set this to `true` when adding an assistant message as prefix to condition the model response. The role of the prefix message is to force the model to start its answer by the content of the message."""
     role: NotRequired[AssistantMessageRole]
 
 
@@ -35,6 +36,7 @@ class AssistantMessage(BaseModel):
     tool_calls: OptionalNullable[List[ToolCall]] = UNSET
 
     prefix: Optional[bool] = False
+    r"""Set this to `true` when adding an assistant message as prefix to condition the model response. The role of the prefix message is to force the model to start its answer by the content of the message."""
 
     role: Optional[AssistantMessageRole] = "assistant"
 

{mistralai-1.5.0 → mistralai-1.5.2}/src/mistralai/models/chatcompletionrequest.py

@@ -59,7 +59,7 @@ ChatCompletionRequestToolChoice = TypeAliasType(
 
 
 class ChatCompletionRequestTypedDict(TypedDict):
-    model: Nullable[str]
+    model: str
     r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
     messages: List[MessagesTypedDict]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
@@ -90,7 +90,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
 
 
 class ChatCompletionRequest(BaseModel):
-    model: Nullable[str]
+    model: str
     r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
 
     messages: List[Messages]
@@ -152,14 +152,7 @@ class ChatCompletionRequest(BaseModel):
             "prediction",
             "safe_prompt",
         ]
-        nullable_fields = [
-            "model",
-            "temperature",
-            "max_tokens",
-            "random_seed",
-            "tools",
-            "n",
-        ]
+        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
         null_default_fields = []
 
         serialized = handler(self)

{mistralai-1.5.0 → mistralai-1.5.2}/src/mistralai/models/chatcompletionstreamrequest.py

@@ -63,7 +63,7 @@ ChatCompletionStreamRequestToolChoice = TypeAliasType(
 
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):
-    model: Nullable[str]
+    model: str
     r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
     messages: List[ChatCompletionStreamRequestMessagesTypedDict]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
@@ -93,7 +93,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
 
 
 class ChatCompletionStreamRequest(BaseModel):
-    model: Nullable[str]
+    model: str
     r"""ID of the model to use. You can use the [List Available Models](/api/#tag/models/operation/list_models_v1_models_get) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
 
     messages: List[ChatCompletionStreamRequestMessages]
@@ -154,14 +154,7 @@ class ChatCompletionStreamRequest(BaseModel):
             "prediction",
             "safe_prompt",
         ]
-        nullable_fields = [
-            "model",
-            "temperature",
-            "max_tokens",
-            "random_seed",
-            "tools",
-            "n",
-        ]
+        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
         null_default_fields = []
 
         serialized = handler(self)