mistralai 1.2.2__py3-none-any.whl → 1.2.4__py3-none-any.whl

mistralai/_version.py

@@ -3,7 +3,7 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.2.2"
+__version__: str = "1.2.4"
 
 try:
     if __package__ is not None:

mistralai/agents.py

@@ -5,7 +5,7 @@ from mistralai import models, utils
 from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import eventstreaming, get_security_from_env
-from typing import Any, AsyncGenerator, Generator, List, Optional, Union
+from typing import Any, List, Optional, Union
 
 
 class Agents(BaseSDK):
@@ -336,7 +336,7 @@ class Agents(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[Generator[models.CompletionEvent, None, None]]:
+    ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream Agents completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -428,7 +428,7 @@ class Agents(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events(
+            return eventstreaming.EventStream(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",
@@ -487,7 +487,7 @@ class Agents(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[AsyncGenerator[models.CompletionEvent, None]]:
+    ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream Agents completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -579,7 +579,7 @@ class Agents(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events_async(
+            return eventstreaming.EventStreamAsync(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",

mistralai/chat.py

@@ -5,7 +5,7 @@ from mistralai import models, utils
 from mistralai._hooks import HookContext
 from mistralai.types import Nullable, OptionalNullable, UNSET
 from mistralai.utils import eventstreaming, get_security_from_env
-from typing import Any, AsyncGenerator, Generator, List, Optional, Union
+from typing import Any, List, Optional, Union
 
 
 class Chat(BaseSDK):
@@ -337,7 +337,7 @@ class Chat(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[Generator[models.CompletionEvent, None, None]]:
+    ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream chat completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -435,7 +435,7 @@ class Chat(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events(
+            return eventstreaming.EventStream(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",
@@ -497,7 +497,7 @@ class Chat(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[AsyncGenerator[models.CompletionEvent, None]]:
+    ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream chat completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -595,7 +595,7 @@ class Chat(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events_async(
+            return eventstreaming.EventStreamAsync(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",

mistralai/files.py

@@ -891,3 +891,169 @@ class Files(BaseSDK):
             http_res_text,
             http_res,
         )
+
+    def get_signed_url(
+        self,
+        *,
+        file_id: str,
+        expiry: Optional[int] = 24,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+    ) -> Optional[models.FileSignedURL]:
+        r"""Get Signed Url
+
+        :param file_id:
+        :param expiry: Number of hours before the url becomes invalid. Defaults to 24h
+        :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
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+
+        request = models.FilesAPIRoutesGetSignedURLRequest(
+            file_id=file_id,
+            expiry=expiry,
+        )
+
+        req = self.build_request(
+            method="GET",
+            path="/v1/files/{file_id}/url",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=False,
+            request_has_path_params=True,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            security=self.sdk_configuration.security,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                operation_id="files_api_routes_get_signed_url",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, Optional[models.FileSignedURL])
+        if utils.match_response(http_res, ["4XX", "5XX"], "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = utils.stream_to_text(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )
+
+    async def get_signed_url_async(
+        self,
+        *,
+        file_id: str,
+        expiry: Optional[int] = 24,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+    ) -> Optional[models.FileSignedURL]:
+        r"""Get Signed Url
+
+        :param file_id:
+        :param expiry: Number of hours before the url becomes invalid. Defaults to 24h
+        :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
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+
+        request = models.FilesAPIRoutesGetSignedURLRequest(
+            file_id=file_id,
+            expiry=expiry,
+        )
+
+        req = self.build_request_async(
+            method="GET",
+            path="/v1/files/{file_id}/url",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=False,
+            request_has_path_params=True,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            security=self.sdk_configuration.security,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                operation_id="files_api_routes_get_signed_url",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        if utils.match_response(http_res, "200", "application/json"):
+            return utils.unmarshal_json(http_res.text, Optional[models.FileSignedURL])
+        if utils.match_response(http_res, ["4XX", "5XX"], "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+
+        content_type = http_res.headers.get("Content-Type")
+        http_res_text = await utils.stream_to_text_async(http_res)
+        raise models.SDKError(
+            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
+            http_res.status_code,
+            http_res_text,
+            http_res,
+        )

mistralai/fim.py

@@ -5,7 +5,7 @@ from mistralai import models, utils
 from mistralai._hooks import HookContext
 from mistralai.types import Nullable, OptionalNullable, UNSET
 from mistralai.utils import eventstreaming, get_security_from_env
-from typing import Any, AsyncGenerator, Generator, Optional, Union
+from typing import Any, Optional, Union
 
 
 class Fim(BaseSDK):
@@ -278,7 +278,7 @@ class Fim(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[Generator[models.CompletionEvent, None, None]]:
+    ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
         r"""Stream fim completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -360,7 +360,7 @@ class Fim(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events(
+            return eventstreaming.EventStream(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",
@@ -405,7 +405,7 @@ class Fim(BaseSDK):
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
-    ) -> Optional[AsyncGenerator[models.CompletionEvent, None]]:
+    ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
         r"""Stream fim completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -487,7 +487,7 @@ class Fim(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
-            return eventstreaming.stream_events_async(
+            return eventstreaming.EventStreamAsync(
                 http_res,
                 lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
                 sentinel="[DONE]",

mistralai/httpclient.py

@@ -41,6 +41,9 @@ class HttpClient(Protocol):
     ) -> httpx.Request:
         pass
 
+    def close(self) -> None:
+        pass
+
 
 @runtime_checkable
 class AsyncHttpClient(Protocol):
@@ -76,3 +79,6 @@ class AsyncHttpClient(Protocol):
         extensions: Optional[httpx._types.RequestExtensions] = None,
     ) -> httpx.Request:
         pass
+
+    async def aclose(self) -> None:
+        pass

mistralai/jobs.py

@@ -225,7 +225,7 @@ class Jobs(BaseSDK):
     def create(
         self,
         *,
-        model: models.FineTuneableModel,
+        model: str,
         hyperparameters: Union[
             models.TrainingParametersIn, models.TrainingParametersInTypedDict
         ],
@@ -354,7 +354,7 @@ class Jobs(BaseSDK):
     async def create_async(
         self,
         *,
-        model: models.FineTuneableModel,
+        model: str,
         hyperparameters: Union[
             models.TrainingParametersIn, models.TrainingParametersInTypedDict
         ],

mistralai/models/__init__.py

@@ -133,6 +133,10 @@ from .files_api_routes_download_fileop import (
     FilesAPIRoutesDownloadFileRequest,
     FilesAPIRoutesDownloadFileRequestTypedDict,
 )
+from .files_api_routes_get_signed_urlop import (
+    FilesAPIRoutesGetSignedURLRequest,
+    FilesAPIRoutesGetSignedURLRequestTypedDict,
+)
 from .files_api_routes_list_filesop import (
     FilesAPIRoutesListFilesRequest,
     FilesAPIRoutesListFilesRequestTypedDict,
@@ -148,6 +152,7 @@ from .files_api_routes_upload_fileop import (
     FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict,
 )
 from .fileschema import FileSchema, FileSchemaTypedDict
+from .filesignedurl import FileSignedURL, FileSignedURLTypedDict
 from .fimcompletionrequest import (
     FIMCompletionRequest,
     FIMCompletionRequestStop,
@@ -161,7 +166,6 @@ from .fimcompletionstreamrequest import (
     FIMCompletionStreamRequestStopTypedDict,
     FIMCompletionStreamRequestTypedDict,
 )
-from .finetuneablemodel import FineTuneableModel
 from .ftmodelcapabilitiesout import (
     FTModelCapabilitiesOut,
     FTModelCapabilitiesOutTypedDict,
@@ -269,6 +273,7 @@ from .listfilesout import ListFilesOut, ListFilesOutTypedDict
 from .metricout import MetricOut, MetricOutTypedDict
 from .modelcapabilities import ModelCapabilities, ModelCapabilitiesTypedDict
 from .modellist import Data, DataTypedDict, ModelList, ModelListTypedDict
+from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .responseformats import ResponseFormats
 from .retrieve_model_v1_models_model_id_getop import (
@@ -294,7 +299,13 @@ from .tool import Tool, ToolTypedDict
 from .toolcall import ToolCall, ToolCallTypedDict
 from .toolchoice import ToolChoice, ToolChoiceTypedDict
 from .toolchoiceenum import ToolChoiceEnum
-from .toolmessage import ToolMessage, ToolMessageRole, ToolMessageTypedDict
+from .toolmessage import (
+    ToolMessage,
+    ToolMessageContent,
+    ToolMessageContentTypedDict,
+    ToolMessageRole,
+    ToolMessageTypedDict,
+)
 from .tooltypes import ToolTypes
 from .trainingfile import TrainingFile, TrainingFileTypedDict
 from .trainingparameters import TrainingParameters, TrainingParametersTypedDict
@@ -461,18 +472,21 @@ __all__ = [
     "FilePurpose",
     "FileSchema",
     "FileSchemaTypedDict",
+    "FileSignedURL",
+    "FileSignedURLTypedDict",
     "FileTypedDict",
     "FilesAPIRoutesDeleteFileRequest",
     "FilesAPIRoutesDeleteFileRequestTypedDict",
     "FilesAPIRoutesDownloadFileRequest",
     "FilesAPIRoutesDownloadFileRequestTypedDict",
+    "FilesAPIRoutesGetSignedURLRequest",
+    "FilesAPIRoutesGetSignedURLRequestTypedDict",
     "FilesAPIRoutesListFilesRequest",
     "FilesAPIRoutesListFilesRequestTypedDict",
     "FilesAPIRoutesRetrieveFileRequest",
     "FilesAPIRoutesRetrieveFileRequestTypedDict",
     "FilesAPIRoutesUploadFileMultiPartBodyParams",
     "FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict",
-    "FineTuneableModel",
     "FinishReason",
     "Function",
     "FunctionCall",
@@ -553,6 +567,9 @@ __all__ = [
     "One",
     "OneTypedDict",
     "QueryParamStatus",
+    "ReferenceChunk",
+    "ReferenceChunkType",
+    "ReferenceChunkTypedDict",
     "Repositories",
     "RepositoriesTypedDict",
     "ResponseFormat",
@@ -587,6 +604,8 @@ __all__ = [
     "ToolChoiceEnum",
     "ToolChoiceTypedDict",
     "ToolMessage",
+    "ToolMessageContent",
+    "ToolMessageContentTypedDict",
     "ToolMessageRole",
     "ToolMessageTypedDict",
     "ToolTypedDict",

mistralai/models/agentscompletionrequest.py

@@ -13,23 +13,30 @@ from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-AgentsCompletionRequestStopTypedDict = Union[str, List[str]]
+AgentsCompletionRequestStopTypedDict = TypeAliasType(
+    "AgentsCompletionRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-AgentsCompletionRequestStop = Union[str, List[str]]
+AgentsCompletionRequestStop = TypeAliasType(
+    "AgentsCompletionRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-AgentsCompletionRequestMessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+AgentsCompletionRequestMessagesTypedDict = TypeAliasType(
+    "AgentsCompletionRequestMessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 AgentsCompletionRequestMessages = Annotated[
@@ -43,10 +50,15 @@ AgentsCompletionRequestMessages = Annotated[
 ]
 
 
-AgentsCompletionRequestToolChoiceTypedDict = Union[ToolChoiceTypedDict, ToolChoiceEnum]
+AgentsCompletionRequestToolChoiceTypedDict = TypeAliasType(
+    "AgentsCompletionRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-AgentsCompletionRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+AgentsCompletionRequestToolChoice = TypeAliasType(
+    "AgentsCompletionRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class AgentsCompletionRequestTypedDict(TypedDict):

mistralai/models/agentscompletionstreamrequest.py

@@ -13,23 +13,30 @@ from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_
 from mistralai.utils import get_discriminator
 from pydantic import Discriminator, Tag, model_serializer
 from typing import List, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-AgentsCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+AgentsCompletionStreamRequestStopTypedDict = TypeAliasType(
+    "AgentsCompletionStreamRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-AgentsCompletionStreamRequestStop = Union[str, List[str]]
+AgentsCompletionStreamRequestStop = TypeAliasType(
+    "AgentsCompletionStreamRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-AgentsCompletionStreamRequestMessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+AgentsCompletionStreamRequestMessagesTypedDict = TypeAliasType(
+    "AgentsCompletionStreamRequestMessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 AgentsCompletionStreamRequestMessages = Annotated[
@@ -43,12 +50,15 @@ AgentsCompletionStreamRequestMessages = Annotated[
 ]
 
 
-AgentsCompletionStreamRequestToolChoiceTypedDict = Union[
-    ToolChoiceTypedDict, ToolChoiceEnum
-]
+AgentsCompletionStreamRequestToolChoiceTypedDict = TypeAliasType(
+    "AgentsCompletionStreamRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-AgentsCompletionStreamRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+AgentsCompletionStreamRequestToolChoice = TypeAliasType(
+    "AgentsCompletionStreamRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class AgentsCompletionStreamRequestTypedDict(TypedDict):

mistralai/models/apiendpoint.py

@@ -1,9 +1,17 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from typing import Literal
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
 
 
-APIEndpoint = Literal[
-    "/v1/chat/completions", "/v1/embeddings", "/v1/fim/completions", "/v1/moderations"
+APIEndpoint = Union[
+    Literal[
+        "/v1/chat/completions",
+        "/v1/embeddings",
+        "/v1/fim/completions",
+        "/v1/moderations",
+        "/v1/chat/moderations",
+    ],
+    UnrecognizedStr,
 ]

mistralai/models/assistantmessage.py

@@ -6,13 +6,17 @@ from .toolcall import ToolCall, ToolCallTypedDict
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-AssistantMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+AssistantMessageContentTypedDict = TypeAliasType(
+    "AssistantMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-AssistantMessageContent = Union[str, List[ContentChunk]]
+AssistantMessageContent = TypeAliasType(
+    "AssistantMessageContent", Union[str, List[ContentChunk]]
+)
 
 
 AssistantMessageRole = Literal["assistant"]

mistralai/models/batchjobin.py

@@ -3,9 +3,11 @@
 from __future__ import annotations
 from .apiendpoint import APIEndpoint
 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 NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class BatchJobInTypedDict(TypedDict):
@@ -19,7 +21,7 @@ class BatchJobInTypedDict(TypedDict):
 class BatchJobIn(BaseModel):
     input_files: List[str]
 
-    endpoint: APIEndpoint
+    endpoint: Annotated[APIEndpoint, PlainValidator(validate_open_enum(False))]
 
     model: str