mistralai 1.0.3__py3-none-any.whl → 1.1.0__py3-none-any.whl

mistralai/models/chatcompletionstreamrequest.py

@@ -5,12 +5,14 @@ from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
 from .tool import Tool, ToolTypedDict
+from .toolchoice import ToolChoice, ToolChoiceTypedDict
+from .toolchoiceenum import ToolChoiceEnum
 from .toolmessage import ToolMessage, ToolMessageTypedDict
 from .usermessage import UserMessage, UserMessageTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from typing import List, Literal, Optional, TypedDict, Union
+from typing import List, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
@@ -22,13 +24,32 @@ ChatCompletionStreamRequestStop = Union[str, List[str]]
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionStreamRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
+ChatCompletionStreamRequestMessagesTypedDict = Union[
+    SystemMessageTypedDict,
+    UserMessageTypedDict,
+    AssistantMessageTypedDict,
+    ToolMessageTypedDict,
+]
 
 
-ChatCompletionStreamRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
+ChatCompletionStreamRequestMessages = Annotated[
+    Union[
+        Annotated[AssistantMessage, Tag("assistant")],
+        Annotated[SystemMessage, Tag("system")],
+        Annotated[ToolMessage, Tag("tool")],
+        Annotated[UserMessage, Tag("user")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "role", "role")),
+]
 
 
-ChatCompletionStreamRequestToolChoice = Literal["auto", "none", "any"]
+ChatCompletionStreamRequestToolChoiceTypedDict = Union[
+    ToolChoiceTypedDict, ToolChoiceEnum
+]
+
+
+ChatCompletionStreamRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):
     model: Nullable[str]
@@ -50,38 +71,62 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
     response_format: NotRequired[ResponseFormatTypedDict]
     tools: NotRequired[Nullable[List[ToolTypedDict]]]
-    tool_choice: NotRequired[ChatCompletionStreamRequestToolChoice]
+    tool_choice: NotRequired[ChatCompletionStreamRequestToolChoiceTypedDict]
     safe_prompt: NotRequired[bool]
     r"""Whether to inject a safety prompt before all conversations."""
-    
+
 
 class ChatCompletionStreamRequest(BaseModel):
     model: Nullable[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]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
+
     temperature: Optional[float] = 0.7
     r"""What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both."""
+
     top_p: Optional[float] = 1
     r"""Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both."""
+
     max_tokens: OptionalNullable[int] = UNSET
     r"""The maximum number of tokens to generate in the completion. The token count of your prompt plus `max_tokens` cannot exceed the model's context length."""
+
     min_tokens: OptionalNullable[int] = UNSET
     r"""The minimum number of tokens to generate in the completion."""
+
     stream: Optional[bool] = True
+
     stop: Optional[ChatCompletionStreamRequestStop] = None
     r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
     random_seed: OptionalNullable[int] = UNSET
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
+
     response_format: Optional[ResponseFormat] = None
+
     tools: OptionalNullable[List[Tool]] = UNSET
-    tool_choice: Optional[ChatCompletionStreamRequestToolChoice] = "auto"
+
+    tool_choice: Optional[ChatCompletionStreamRequestToolChoice] = None
+
     safe_prompt: Optional[bool] = False
     r"""Whether to inject a safety prompt before all conversations."""
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["temperature", "top_p", "max_tokens", "min_tokens", "stream", "stop", "random_seed", "response_format", "tools", "tool_choice", "safe_prompt"]
+        optional_fields = [
+            "temperature",
+            "top_p",
+            "max_tokens",
+            "min_tokens",
+            "stream",
+            "stop",
+            "random_seed",
+            "response_format",
+            "tools",
+            "tool_choice",
+            "safe_prompt",
+        ]
         nullable_fields = ["model", "max_tokens", "min_tokens", "random_seed", "tools"]
         null_default_fields = []
 
@@ -92,9 +137,13 @@ class ChatCompletionStreamRequest(BaseModel):
         for n, f in 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
+            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
@@ -104,4 +153,3 @@ class ChatCompletionStreamRequest(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/checkpointout.py

@@ -13,13 +13,14 @@ class CheckpointOutTypedDict(TypedDict):
     r"""The step number that the checkpoint was created at."""
     created_at: int
     r"""The UNIX timestamp (in seconds) for when the checkpoint was created."""
-    
+
 
 class CheckpointOut(BaseModel):
     metrics: MetricOut
     r"""Metrics at the step number during the fine-tuning job. Use these metrics to assess if the training is going smoothly (loss should decrease, token accuracy should increase)."""
+
     step_number: int
     r"""The step number that the checkpoint was created at."""
+
     created_at: int
     r"""The UNIX timestamp (in seconds) for when the checkpoint was created."""
-    

mistralai/models/completionchunk.py

@@ -1,7 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .completionresponsestreamchoice import CompletionResponseStreamChoice, CompletionResponseStreamChoiceTypedDict
+from .completionresponsestreamchoice import (
+    CompletionResponseStreamChoice,
+    CompletionResponseStreamChoiceTypedDict,
+)
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai.types import BaseModel
 from typing import List, Optional, TypedDict
@@ -15,13 +18,17 @@ class CompletionChunkTypedDict(TypedDict):
     object: NotRequired[str]
     created: NotRequired[int]
     usage: NotRequired[UsageInfoTypedDict]
-    
+
 
 class CompletionChunk(BaseModel):
     id: str
+
     model: str
+
     choices: List[CompletionResponseStreamChoice]
+
     object: Optional[str] = None
+
     created: Optional[int] = None
+
     usage: Optional[UsageInfo] = None
-    

mistralai/models/completionevent.py

@@ -8,8 +8,7 @@ from typing import TypedDict
 
 class CompletionEventTypedDict(TypedDict):
     data: CompletionChunkTypedDict
-    
+
 
 class CompletionEvent(BaseModel):
     data: CompletionChunk
-    

mistralai/models/completionresponsestreamchoice.py

@@ -7,19 +7,24 @@ from pydantic import model_serializer
 from typing import Literal, TypedDict
 
 
-CompletionResponseStreamChoiceFinishReason = Literal["stop", "length", "error", "tool_calls"]
+CompletionResponseStreamChoiceFinishReason = Literal[
+    "stop", "length", "error", "tool_calls"
+]
+
 
 class CompletionResponseStreamChoiceTypedDict(TypedDict):
     index: int
     delta: DeltaMessageTypedDict
     finish_reason: Nullable[CompletionResponseStreamChoiceFinishReason]
-    
+
 
 class CompletionResponseStreamChoice(BaseModel):
     index: int
+
     delta: DeltaMessage
+
     finish_reason: Nullable[CompletionResponseStreamChoiceFinishReason]
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = []
@@ -33,9 +38,13 @@ class CompletionResponseStreamChoice(BaseModel):
         for n, f in 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
+            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
@@ -45,4 +54,3 @@ class CompletionResponseStreamChoice(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/contentchunk.py

@@ -1,17 +1,20 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
-import pydantic
-from typing import Final, Optional, TypedDict
+from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
+from .textchunk import TextChunk, TextChunkTypedDict
+from mistralai.utils import get_discriminator
+from pydantic import Discriminator, Tag
+from typing import Union
 from typing_extensions import Annotated
 
 
-class ContentChunkTypedDict(TypedDict):
-    text: str
-    
+ContentChunkTypedDict = Union[TextChunkTypedDict, ImageURLChunkTypedDict]
 
-class ContentChunk(BaseModel):
-    text: str
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "text" # type: ignore
-    
+
+ContentChunk = Annotated[
+    Union[
+        Annotated[ImageURLChunk, Tag("image_url")], Annotated[TextChunk, Tag("text")]
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]

mistralai/models/delete_model_v1_models_model_id_deleteop.py

@@ -10,9 +10,10 @@ from typing_extensions import Annotated
 class DeleteModelV1ModelsModelIDDeleteRequestTypedDict(TypedDict):
     model_id: str
     r"""The ID of the model to delete."""
-    
+
 
 class DeleteModelV1ModelsModelIDDeleteRequest(BaseModel):
-    model_id: Annotated[str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))]
+    model_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
     r"""The ID of the model to delete."""
-    

mistralai/models/deletefileout.py

@@ -12,13 +12,14 @@ class DeleteFileOutTypedDict(TypedDict):
     r"""The object type that was deleted"""
     deleted: bool
     r"""The deletion status."""
-    
+
 
 class DeleteFileOut(BaseModel):
     id: str
     r"""The ID of the deleted file."""
+
     object: str
     r"""The object type that was deleted"""
+
     deleted: bool
     r"""The deletion status."""
-    

mistralai/models/deletemodelout.py

@@ -13,13 +13,14 @@ class DeleteModelOutTypedDict(TypedDict):
     r"""The object type that was deleted"""
     deleted: NotRequired[bool]
     r"""The deletion status"""
-    
+
 
 class DeleteModelOut(BaseModel):
     id: str
     r"""The ID of the deleted model."""
+
     object: Optional[str] = "model"
     r"""The object type that was deleted"""
+
     deleted: Optional[bool] = True
     r"""The deletion status"""
-    

mistralai/models/deltamessage.py

@@ -12,13 +12,15 @@ class DeltaMessageTypedDict(TypedDict):
     role: NotRequired[str]
     content: NotRequired[Nullable[str]]
     tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
-    
+
 
 class DeltaMessage(BaseModel):
     role: Optional[str] = None
+
     content: OptionalNullable[str] = UNSET
+
     tool_calls: OptionalNullable[List[ToolCall]] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["role", "content", "tool_calls"]
@@ -32,9 +34,13 @@ class DeltaMessage(BaseModel):
         for n, f in 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
+            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
@@ -44,4 +50,3 @@ class DeltaMessage(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/detailedjobout.py

@@ -15,7 +15,18 @@ from typing import Final, List, Literal, Optional, TypedDict
 from typing_extensions import Annotated, NotRequired
 
 
-DetailedJobOutStatus = Literal["QUEUED", "STARTED", "VALIDATING", "VALIDATED", "RUNNING", "FAILED_VALIDATION", "FAILED", "SUCCESS", "CANCELLED", "CANCELLATION_REQUESTED"]
+DetailedJobOutStatus = Literal[
+    "QUEUED",
+    "STARTED",
+    "VALIDATING",
+    "VALIDATED",
+    "RUNNING",
+    "FAILED_VALIDATION",
+    "FAILED",
+    "SUCCESS",
+    "CANCELLED",
+    "CANCELLATION_REQUESTED",
+]
 
 DetailedJobOutObject = Literal["job"]
 
@@ -52,35 +63,73 @@ class DetailedJobOutTypedDict(TypedDict):
     events: NotRequired[List[EventOutTypedDict]]
     r"""Event items are created every time the status of a fine-tuning job changes. The timestamped list of all events is accessible here."""
     checkpoints: NotRequired[List[CheckpointOutTypedDict]]
-    
+
 
 class DetailedJobOut(BaseModel):
     id: str
+
     auto_start: bool
+
     hyperparameters: TrainingParameters
+
     model: FineTuneableModel
     r"""The name of the model to fine-tune."""
+
     status: DetailedJobOutStatus
+
     job_type: str
+
     created_at: int
+
     modified_at: int
+
     training_files: List[str]
+
     validation_files: OptionalNullable[List[str]] = UNSET
+
+    # fmt: off
     OBJECT: Annotated[Final[Optional[DetailedJobOutObject]], pydantic.Field(alias="object")] = "job" # type: ignore
+    # fmt: on
+
     fine_tuned_model: OptionalNullable[str] = UNSET
+
     suffix: OptionalNullable[str] = UNSET
+
     integrations: OptionalNullable[List[DetailedJobOutIntegrations]] = UNSET
+
     trained_tokens: OptionalNullable[int] = UNSET
+
     repositories: Optional[List[DetailedJobOutRepositories]] = None
+
     metadata: OptionalNullable[JobMetadataOut] = UNSET
+
     events: Optional[List[EventOut]] = None
     r"""Event items are created every time the status of a fine-tuning job changes. The timestamped list of all events is accessible here."""
+
     checkpoints: Optional[List[CheckpointOut]] = None
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["validation_files", "object", "fine_tuned_model", "suffix", "integrations", "trained_tokens", "repositories", "metadata", "events", "checkpoints"]
-        nullable_fields = ["validation_files", "fine_tuned_model", "suffix", "integrations", "trained_tokens", "metadata"]
+        optional_fields = [
+            "validation_files",
+            "object",
+            "fine_tuned_model",
+            "suffix",
+            "integrations",
+            "trained_tokens",
+            "repositories",
+            "metadata",
+            "events",
+            "checkpoints",
+        ]
+        nullable_fields = [
+            "validation_files",
+            "fine_tuned_model",
+            "suffix",
+            "integrations",
+            "trained_tokens",
+            "metadata",
+        ]
         null_default_fields = []
 
         serialized = handler(self)
@@ -90,9 +139,13 @@ class DetailedJobOut(BaseModel):
         for n, f in 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
+            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
@@ -102,4 +155,3 @@ class DetailedJobOut(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/embeddingrequest.py

@@ -23,16 +23,18 @@ class EmbeddingRequestTypedDict(TypedDict):
     r"""ID of the model to use."""
     encoding_format: NotRequired[Nullable[str]]
     r"""The format to return the embeddings in."""
-    
+
 
 class EmbeddingRequest(BaseModel):
     inputs: Annotated[Inputs, pydantic.Field(alias="input")]
     r"""Text to embed."""
+
     model: str
     r"""ID of the model to use."""
+
     encoding_format: OptionalNullable[str] = UNSET
     r"""The format to return the embeddings in."""
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["encoding_format"]
@@ -46,9 +48,13 @@ class EmbeddingRequest(BaseModel):
         for n, f in 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
+            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
@@ -58,4 +64,3 @@ class EmbeddingRequest(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/embeddingresponse.py

@@ -13,12 +13,15 @@ class EmbeddingResponseTypedDict(TypedDict):
     model: str
     usage: UsageInfoTypedDict
     data: List[EmbeddingResponseDataTypedDict]
-    
+
 
 class EmbeddingResponse(BaseModel):
     id: str
+
     object: str
+
     model: str
+
     usage: UsageInfo
+
     data: List[EmbeddingResponseData]
-    

mistralai/models/embeddingresponsedata.py

@@ -10,10 +10,11 @@ class EmbeddingResponseDataTypedDict(TypedDict):
     object: NotRequired[str]
     embedding: NotRequired[List[float]]
     index: NotRequired[int]
-    
+
 
 class EmbeddingResponseData(BaseModel):
     object: Optional[str] = None
+
     embedding: Optional[List[float]] = None
+
     index: Optional[int] = None
-    

mistralai/models/eventout.py

@@ -13,15 +13,17 @@ class EventOutTypedDict(TypedDict):
     created_at: int
     r"""The UNIX timestamp (in seconds) of the event."""
     data: NotRequired[Nullable[Dict[str, Any]]]
-    
+
 
 class EventOut(BaseModel):
     name: str
     r"""The name of the event."""
+
     created_at: int
     r"""The UNIX timestamp (in seconds) of the event."""
+
     data: OptionalNullable[Dict[str, Any]] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["data"]
@@ -35,9 +37,13 @@ class EventOut(BaseModel):
         for n, f in 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
+            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
@@ -47,4 +53,3 @@ class EventOut(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/files_api_routes_delete_fileop.py

@@ -9,8 +9,9 @@ from typing_extensions import Annotated
 
 class FilesAPIRoutesDeleteFileRequestTypedDict(TypedDict):
     file_id: str
-    
+
 
 class FilesAPIRoutesDeleteFileRequest(BaseModel):
-    file_id: Annotated[str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))]
-    
+    file_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/files_api_routes_retrieve_fileop.py

@@ -9,8 +9,9 @@ from typing_extensions import Annotated
 
 class FilesAPIRoutesRetrieveFileRequestTypedDict(TypedDict):
     file_id: str
-    
+
 
 class FilesAPIRoutesRetrieveFileRequest(BaseModel):
-    file_id: Annotated[str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))]
-    
+    file_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]