mistralai 1.10.0__py3-none-any.whl → 1.11.1__py3-none-any.whl

mistralai/models/audiochunk.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-AudioChunkType = Literal["input_audio"]
+AudioChunkType = Literal["input_audio",]
 
 
 class AudioChunkTypedDict(TypedDict):

mistralai/models/audioencoding.py

@@ -0,0 +1,18 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+AudioEncoding = Union[
+    Literal[
+        "pcm_s16le",
+        "pcm_s32le",
+        "pcm_f16le",
+        "pcm_f32le",
+        "pcm_mulaw",
+        "pcm_alaw",
+    ],
+    UnrecognizedStr,
+]

mistralai/models/audioformat.py

@@ -0,0 +1,17 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .audioencoding import AudioEncoding
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class AudioFormatTypedDict(TypedDict):
+    encoding: AudioEncoding
+    sample_rate: int
+
+
+class AudioFormat(BaseModel):
+    encoding: AudioEncoding
+
+    sample_rate: int

mistralai/models/basemodelcard.py

@@ -12,7 +12,7 @@ from typing import List, Literal, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-BaseModelCardType = Literal["base"]
+BaseModelCardType = Literal["base",]
 
 
 class BaseModelCardTypedDict(TypedDict):

mistralai/models/batchjobin.py

@@ -2,18 +2,18 @@
 
 from __future__ import annotations
 from .apiendpoint import APIEndpoint
+from .batchrequest import BatchRequest, BatchRequestTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
-from mistralai.utils import validate_open_enum
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Dict, List, Optional
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class BatchJobInTypedDict(TypedDict):
-    input_files: List[str]
-    r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""
     endpoint: APIEndpoint
+    input_files: NotRequired[Nullable[List[str]]]
+    r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""
+    requests: NotRequired[Nullable[List[BatchRequestTypedDict]]]
     model: NotRequired[Nullable[str]]
     r"""The model to be used for batch inference."""
     agent_id: NotRequired[Nullable[str]]
@@ -25,10 +25,12 @@ class BatchJobInTypedDict(TypedDict):
 
 
 class BatchJobIn(BaseModel):
-    input_files: List[str]
+    endpoint: APIEndpoint
+
+    input_files: OptionalNullable[List[str]] = UNSET
     r"""The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```"""
 
-    endpoint: Annotated[APIEndpoint, PlainValidator(validate_open_enum(False))]
+    requests: OptionalNullable[List[BatchRequest]] = UNSET
 
     model: OptionalNullable[str] = UNSET
     r"""The model to be used for batch inference."""
@@ -44,8 +46,15 @@ class BatchJobIn(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["model", "agent_id", "metadata", "timeout_hours"]
-        nullable_fields = ["model", "agent_id", "metadata"]
+        optional_fields = [
+            "input_files",
+            "requests",
+            "model",
+            "agent_id",
+            "metadata",
+            "timeout_hours",
+        ]
+        nullable_fields = ["input_files", "requests", "model", "agent_id", "metadata"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/batchjobout.py

@@ -9,7 +9,7 @@ from typing import Any, Dict, List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-BatchJobOutObject = Literal["batch"]
+BatchJobOutObject = Literal["batch",]
 
 
 class BatchJobOutTypedDict(TypedDict):
@@ -29,6 +29,7 @@ class BatchJobOutTypedDict(TypedDict):
     agent_id: NotRequired[Nullable[str]]
     output_file: NotRequired[Nullable[str]]
     error_file: NotRequired[Nullable[str]]
+    outputs: NotRequired[Nullable[List[Dict[str, Any]]]]
     started_at: NotRequired[Nullable[int]]
     completed_at: NotRequired[Nullable[int]]
 
@@ -66,6 +67,8 @@ class BatchJobOut(BaseModel):
 
     error_file: OptionalNullable[str] = UNSET
 
+    outputs: OptionalNullable[List[Dict[str, Any]]] = UNSET
+
     started_at: OptionalNullable[int] = UNSET
 
     completed_at: OptionalNullable[int] = UNSET
@@ -79,6 +82,7 @@ class BatchJobOut(BaseModel):
             "agent_id",
             "output_file",
             "error_file",
+            "outputs",
             "started_at",
             "completed_at",
         ]
@@ -88,6 +92,7 @@ class BatchJobOut(BaseModel):
             "agent_id",
             "output_file",
             "error_file",
+            "outputs",
             "started_at",
             "completed_at",
         ]

mistralai/models/batchjobsout.py

@@ -7,7 +7,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-BatchJobsOutObject = Literal["list"]
+BatchJobsOutObject = Literal["list",]
 
 
 class BatchJobsOutTypedDict(TypedDict):

mistralai/models/batchrequest.py

@@ -0,0 +1,48 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import Any, Dict
+from typing_extensions import NotRequired, TypedDict
+
+
+class BatchRequestTypedDict(TypedDict):
+    body: Dict[str, Any]
+    custom_id: NotRequired[Nullable[str]]
+
+
+class BatchRequest(BaseModel):
+    body: Dict[str, Any]
+
+    custom_id: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["custom_id"]
+        nullable_fields = ["custom_id"]
+        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/models/chatcompletionchoice.py

@@ -3,14 +3,19 @@
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
 from mistralai.types import BaseModel, UnrecognizedStr
-from mistralai.utils import validate_open_enum
-from pydantic.functional_validators import PlainValidator
 from typing import Literal, Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import TypedDict
 
 
 FinishReason = Union[
-    Literal["stop", "length", "model_length", "error", "tool_calls"], UnrecognizedStr
+    Literal[
+        "stop",
+        "length",
+        "model_length",
+        "error",
+        "tool_calls",
+    ],
+    UnrecognizedStr,
 ]
 
 
@@ -25,4 +30,4 @@ class ChatCompletionChoice(BaseModel):
 
     message: AssistantMessage
 
-    finish_reason: Annotated[FinishReason, PlainValidator(validate_open_enum(False))]
+    finish_reason: FinishReason

mistralai/models/chatcompletionrequest.py

@@ -12,9 +12,8 @@ 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, validate_open_enum
+from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -153,9 +152,7 @@ class ChatCompletionRequest(BaseModel):
     parallel_tool_calls: Optional[bool] = None
     r"""Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel."""
 
-    prompt_mode: Annotated[
-        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    prompt_mode: OptionalNullable[MistralPromptMode] = 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

mistralai/models/chatcompletionstreamrequest.py

@@ -12,9 +12,8 @@ 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, validate_open_enum
+from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Any, Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -155,9 +154,7 @@ class ChatCompletionStreamRequest(BaseModel):
     parallel_tool_calls: Optional[bool] = None
     r"""Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel."""
 
-    prompt_mode: Annotated[
-        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
-    ] = UNSET
+    prompt_mode: OptionalNullable[MistralPromptMode] = 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

mistralai/models/classificationrequest.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
-from typing import List, Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
+from pydantic import model_serializer
+from typing import Any, Dict, List, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 ClassificationRequestInputsTypedDict = TypeAliasType(
@@ -24,6 +25,7 @@ class ClassificationRequestTypedDict(TypedDict):
     r"""ID of the model to use."""
     inputs: ClassificationRequestInputsTypedDict
     r"""Text to classify."""
+    metadata: NotRequired[Nullable[Dict[str, Any]]]
 
 
 class ClassificationRequest(BaseModel):
@@ -32,3 +34,35 @@ class ClassificationRequest(BaseModel):
 
     inputs: Annotated[ClassificationRequestInputs, pydantic.Field(alias="input")]
     r"""Text to classify."""
+
+    metadata: OptionalNullable[Dict[str, Any]] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["metadata"]
+        nullable_fields = ["metadata"]
+        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/models/classifierdetailedjobout.py

@@ -29,7 +29,9 @@ ClassifierDetailedJobOutStatus = Literal[
     "CANCELLATION_REQUESTED",
 ]
 
-ClassifierDetailedJobOutObject = Literal["job"]
+
+ClassifierDetailedJobOutObject = Literal["job",]
+
 
 ClassifierDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 
@@ -37,7 +39,7 @@ ClassifierDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 ClassifierDetailedJobOutIntegrations = WandbIntegrationOut
 
 
-ClassifierDetailedJobOutJobType = Literal["classifier"]
+ClassifierDetailedJobOutJobType = Literal["classifier",]
 
 
 class ClassifierDetailedJobOutTypedDict(TypedDict):

mistralai/models/classifierftmodelout.py

@@ -12,9 +12,10 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ClassifierFTModelOutObject = Literal["model"]
+ClassifierFTModelOutObject = Literal["model",]
 
-ClassifierFTModelOutModelType = Literal["classifier"]
+
+ClassifierFTModelOutModelType = Literal["classifier",]
 
 
 class ClassifierFTModelOutTypedDict(TypedDict):

mistralai/models/classifierjobout.py

@@ -27,16 +27,18 @@ ClassifierJobOutStatus = Literal[
 ]
 r"""The current status of the fine-tuning job."""
 
-ClassifierJobOutObject = Literal["job"]
+
+ClassifierJobOutObject = Literal["job",]
 r"""The object type of the fine-tuning job."""
 
+
 ClassifierJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 
 
 ClassifierJobOutIntegrations = WandbIntegrationOut
 
 
-ClassifierJobOutJobType = Literal["classifier"]
+ClassifierJobOutJobType = Literal["classifier",]
 r"""The type of job (`FT` for fine-tuning)."""
 
 

mistralai/models/codeinterpretertool.py

@@ -6,7 +6,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-CodeInterpreterToolType = Literal["code_interpreter"]
+CodeInterpreterToolType = Literal["code_interpreter",]
 
 
 class CodeInterpreterToolTypedDict(TypedDict):

mistralai/models/completiondetailedjobout.py

@@ -29,7 +29,9 @@ CompletionDetailedJobOutStatus = Literal[
     "CANCELLATION_REQUESTED",
 ]
 
-CompletionDetailedJobOutObject = Literal["job"]
+
+CompletionDetailedJobOutObject = Literal["job",]
+
 
 CompletionDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 
@@ -37,7 +39,8 @@ CompletionDetailedJobOutIntegrationsTypedDict = WandbIntegrationOutTypedDict
 CompletionDetailedJobOutIntegrations = WandbIntegrationOut
 
 
-CompletionDetailedJobOutJobType = Literal["completion"]
+CompletionDetailedJobOutJobType = Literal["completion",]
+
 
 CompletionDetailedJobOutRepositoriesTypedDict = GithubRepositoryOutTypedDict
 

mistralai/models/completionftmodelout.py

@@ -11,9 +11,10 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-CompletionFTModelOutObject = Literal["model"]
+CompletionFTModelOutObject = Literal["model",]
 
-ModelType = Literal["completion"]
+
+ModelType = Literal["completion",]
 
 
 class CompletionFTModelOutTypedDict(TypedDict):

mistralai/models/completionjobout.py

@@ -28,18 +28,21 @@ Status = Literal[
 ]
 r"""The current status of the fine-tuning job."""
 
-CompletionJobOutObject = Literal["job"]
+
+CompletionJobOutObject = Literal["job",]
 r"""The object type of the fine-tuning job."""
 
+
 IntegrationsTypedDict = WandbIntegrationOutTypedDict
 
 
 Integrations = WandbIntegrationOut
 
 
-JobType = Literal["completion"]
+JobType = Literal["completion",]
 r"""The type of job (`FT` for fine-tuning)."""
 
+
 RepositoriesTypedDict = GithubRepositoryOutTypedDict
 
 

mistralai/models/completionresponsestreamchoice.py

@@ -3,15 +3,19 @@
 from __future__ import annotations
 from .deltamessage import DeltaMessage, DeltaMessageTypedDict
 from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL, UnrecognizedStr
-from mistralai.utils import validate_open_enum
 from pydantic import model_serializer
-from pydantic.functional_validators import PlainValidator
 from typing import Literal, Union
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import TypedDict
 
 
 CompletionResponseStreamChoiceFinishReason = Union[
-    Literal["stop", "length", "error", "tool_calls"], UnrecognizedStr
+    Literal[
+        "stop",
+        "length",
+        "error",
+        "tool_calls",
+    ],
+    UnrecognizedStr,
 ]
 
 
@@ -26,10 +30,7 @@ class CompletionResponseStreamChoice(BaseModel):
 
     delta: DeltaMessage
 
-    finish_reason: Annotated[
-        Nullable[CompletionResponseStreamChoiceFinishReason],
-        PlainValidator(validate_open_enum(False)),
-    ]
+    finish_reason: Nullable[CompletionResponseStreamChoiceFinishReason]
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):

mistralai/models/conversationappendrequest.py

@@ -8,7 +8,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationAppendRequestHandoffExecution = Literal["client", "server"]
+ConversationAppendRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationAppendRequestTypedDict(TypedDict):

mistralai/models/conversationappendstreamrequest.py

@@ -8,7 +8,10 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationAppendStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationAppendStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationAppendStreamRequestTypedDict(TypedDict):

mistralai/models/conversationhistory.py

@@ -12,7 +12,8 @@ from typing import List, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ConversationHistoryObject = Literal["conversation.history"]
+ConversationHistoryObject = Literal["conversation.history",]
+
 
 EntriesTypedDict = TypeAliasType(
     "EntriesTypedDict",

mistralai/models/conversationmessages.py

@@ -7,7 +7,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationMessagesObject = Literal["conversation.messages"]
+ConversationMessagesObject = Literal["conversation.messages",]
 
 
 class ConversationMessagesTypedDict(TypedDict):

mistralai/models/conversationrequest.py

@@ -16,7 +16,11 @@ from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-HandoffExecution = Literal["client", "server"]
+HandoffExecution = Literal[
+    "client",
+    "server",
+]
+
 
 ToolsTypedDict = TypeAliasType(
     "ToolsTypedDict",

mistralai/models/conversationresponse.py

@@ -11,7 +11,8 @@ from typing import List, Literal, Optional, Union
 from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ConversationResponseObject = Literal["conversation.response"]
+ConversationResponseObject = Literal["conversation.response",]
+
 
 OutputsTypedDict = TypeAliasType(
     "OutputsTypedDict",

mistralai/models/conversationrestartrequest.py

@@ -9,7 +9,10 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationRestartRequestHandoffExecution = Literal["client", "server"]
+ConversationRestartRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationRestartRequestTypedDict(TypedDict):

mistralai/models/conversationrestartstreamrequest.py

@@ -9,7 +9,10 @@ from typing import Any, Dict, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-ConversationRestartStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationRestartStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
 
 
 class ConversationRestartStreamRequestTypedDict(TypedDict):

mistralai/models/conversationstreamrequest.py

@@ -16,7 +16,11 @@ from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-ConversationStreamRequestHandoffExecution = Literal["client", "server"]
+ConversationStreamRequestHandoffExecution = Literal[
+    "client",
+    "server",
+]
+
 
 ConversationStreamRequestToolsTypedDict = TypeAliasType(
     "ConversationStreamRequestToolsTypedDict",

mistralai/models/documentlibrarytool.py

@@ -6,7 +6,7 @@ from typing import List, Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-DocumentLibraryToolType = Literal["document_library"]
+DocumentLibraryToolType = Literal["document_library",]
 
 
 class DocumentLibraryToolTypedDict(TypedDict):

mistralai/models/documenturlchunk.py

@@ -7,7 +7,7 @@ from typing import Literal, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-DocumentURLChunkType = Literal["document_url"]
+DocumentURLChunkType = Literal["document_url",]
 
 
 class DocumentURLChunkTypedDict(TypedDict):

mistralai/models/embeddingdtype.py

@@ -4,4 +4,10 @@ from __future__ import annotations
 from typing import Literal
 
 
-EmbeddingDtype = Literal["float", "int8", "uint8", "binary", "ubinary"]
+EmbeddingDtype = Literal[
+    "float",
+    "int8",
+    "uint8",
+    "binary",
+    "ubinary",
+]