mistralai 1.5.0__tar.gz → 1.5.1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.5.0
+Version: 1.5.1
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.8
@@ -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.1}/README-PYPI.md

@@ -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.1}/pyproject.toml

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

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

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.5.0"
+__version__: str = "1.5.1"
 __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.1 2.497.0 0.0.2 mistralai"
 
 try:
     if __package__ is not None:

{mistralai-1.5.0 → mistralai-1.5.1}/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.1}/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.1}/src/mistralai/embeddings.py

@@ -16,7 +16,6 @@ class Embeddings(BaseSDK):
         *,
         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,
@@ -28,7 +27,6 @@ class Embeddings(BaseSDK):
 
         :param inputs: Text to embed.
         :param model: ID of the model to use.
-        :param encoding_format: The format to return the embeddings in.
         :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(
@@ -119,7 +116,6 @@ class Embeddings(BaseSDK):
         *,
         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,
@@ -131,7 +127,6 @@ class Embeddings(BaseSDK):
 
         :param inputs: Text to embed.
         :param model: ID of the model to use.
-        :param encoding_format: The format to return the embeddings in.
         :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.1}/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.1}/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,7 @@ from .detailedjobout import (
     DetailedJobOutStatus,
     DetailedJobOutTypedDict,
 )
+from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
 from .embeddingrequest import (
     EmbeddingRequest,
     EmbeddingRequestTypedDict,
@@ -274,7 +275,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 +393,6 @@ __all__ = [
     "BatchJobsOut",
     "BatchJobsOutObject",
     "BatchJobsOutTypedDict",
-    "ChatClassificationRequest",
-    "ChatClassificationRequestInputs",
-    "ChatClassificationRequestInputsTypedDict",
-    "ChatClassificationRequestTypedDict",
     "ChatCompletionChoice",
     "ChatCompletionChoiceTypedDict",
     "ChatCompletionRequest",
@@ -406,6 +409,10 @@ __all__ = [
     "ChatCompletionStreamRequestToolChoice",
     "ChatCompletionStreamRequestToolChoiceTypedDict",
     "ChatCompletionStreamRequestTypedDict",
+    "ChatModerationRequest",
+    "ChatModerationRequestInputs",
+    "ChatModerationRequestInputsTypedDict",
+    "ChatModerationRequestTypedDict",
     "CheckpointOut",
     "CheckpointOutTypedDict",
     "ClassificationObject",
@@ -445,6 +452,10 @@ __all__ = [
     "DetailedJobOutRepositoriesTypedDict",
     "DetailedJobOutStatus",
     "DetailedJobOutTypedDict",
+    "Document",
+    "DocumentTypedDict",
+    "DocumentURLChunk",
+    "DocumentURLChunkTypedDict",
     "EmbeddingRequest",
     "EmbeddingRequestTypedDict",
     "EmbeddingResponse",
@@ -568,11 +579,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.1}/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.1}/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.1}/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)

mistralai-1.5.1/src/mistralai/models/chatmoderationrequest.py

@@ -0,0 +1,86 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .systemmessage import SystemMessage, SystemMessageTypedDict
+from .toolmessage import ToolMessage, ToolMessageTypedDict
+from .usermessage import UserMessage, UserMessageTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import get_discriminator
+import pydantic
+from pydantic import Discriminator, Tag
+from typing import List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+TwoTypedDict = TypeAliasType(
+    "TwoTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
+
+
+Two = 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")),
+]
+
+
+OneTypedDict = TypeAliasType(
+    "OneTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
+
+
+One = 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")),
+]
+
+
+ChatModerationRequestInputsTypedDict = TypeAliasType(
+    "ChatModerationRequestInputsTypedDict",
+    Union[List[OneTypedDict], List[List[TwoTypedDict]]],
+)
+r"""Chat to classify"""
+
+
+ChatModerationRequestInputs = TypeAliasType(
+    "ChatModerationRequestInputs", Union[List[One], List[List[Two]]]
+)
+r"""Chat to classify"""
+
+
+class ChatModerationRequestTypedDict(TypedDict):
+    model: str
+    inputs: ChatModerationRequestInputsTypedDict
+    r"""Chat to classify"""
+    truncate_for_context_length: NotRequired[bool]
+
+
+class ChatModerationRequest(BaseModel):
+    model: str
+
+    inputs: Annotated[ChatModerationRequestInputs, pydantic.Field(alias="input")]
+    r"""Chat to classify"""
+
+    truncate_for_context_length: Optional[bool] = False