mistralai 1.0.0rc1__py3-none-any.whl → 1.0.1__py3-none-any.whl

mistralai/models/retrievefileout.py

@@ -57,18 +57,13 @@ class RetrieveFileOut(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/security.py

@@ -3,14 +3,14 @@
 from __future__ import annotations
 from mistralai.types import BaseModel
 from mistralai.utils import FieldMetadata, SecurityMetadata
-from typing import TypedDict
-from typing_extensions import Annotated
+from typing import Optional, TypedDict
+from typing_extensions import Annotated, NotRequired
 
 
 class SecurityTypedDict(TypedDict):
-    api_key: str
+    api_key: NotRequired[str]
     
 
 class Security(BaseModel):
-    api_key: Annotated[str, FieldMetadata(security=SecurityMetadata(scheme=True, scheme_type="http", sub_type="bearer", field_name="Authorization"))]
+    api_key: Annotated[Optional[str], FieldMetadata(security=SecurityMetadata(scheme=True, scheme_type="http", sub_type="bearer", field_name="Authorization"))] = None
     

mistralai/models/systemmessage.py

@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+
+
+Content = Union[str, List[ContentChunk]]
+
+
 Role = Literal["system"]
 
 class SystemMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class SystemMessage(BaseModel):
     content: Content
     role: Optional[Role] = "system"
     
-
-ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
-
-
-Content = Union[str, List[ContentChunk]]
-

mistralai/models/toolmessage.py

@@ -36,18 +36,13 @@ class ToolMessage(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/trainingparameters.py

@@ -34,18 +34,13 @@ class TrainingParameters(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/trainingparametersin.py

@@ -42,18 +42,13 @@ class TrainingParametersIn(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/updateftmodelin.py

@@ -30,18 +30,13 @@ class UpdateFTModelIn(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/uploadfileout.py

@@ -57,18 +57,13 @@ class UploadFileOut(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/usermessage.py

@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+
+
+UserMessageContent = Union[str, List[TextChunk]]
+
+
 UserMessageRole = Literal["user"]
 
 class UserMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class UserMessage(BaseModel):
     content: UserMessageContent
     role: Optional[UserMessageRole] = "user"
     
-
-UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
-
-
-UserMessageContent = Union[str, List[TextChunk]]
-

mistralai/models/validationerror.py

@@ -5,6 +5,12 @@ from mistralai.types import BaseModel
 from typing import List, TypedDict, Union
 
 
+LocTypedDict = Union[str, int]
+
+
+Loc = Union[str, int]
+
+
 class ValidationErrorTypedDict(TypedDict):
     loc: List[LocTypedDict]
     msg: str
@@ -16,9 +22,3 @@ class ValidationError(BaseModel):
     msg: str
     type: str
     
-
-LocTypedDict = Union[str, int]
-
-
-Loc = Union[str, int]
-

mistralai/models/wandbintegration.py

@@ -42,18 +42,13 @@ class WandbIntegration(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models/wandbintegrationout.py

@@ -38,18 +38,13 @@ class WandbIntegrationOut(BaseModel):
             k = f.alias or n
             val = serialized.get(k)
 
+            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 (
-                    k in optional_fields
-                    and k in nullable_fields
-                    and (
-                        self.__pydantic_fields_set__.intersection({n})
-                        or k in null_default_fields
-                    )  # pylint: disable=no-member
-                )
+                not k in optional_fields or (optional_nullable and is_set)
             ):
                 m[k] = val
 

mistralai/models_.py

@@ -1,10 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from .basesdk import BaseSDK
-from mistralai import models
+from mistralai import models, utils
 from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
-import mistralai.utils as utils
+from mistralai.utils import get_security_from_env
 from typing import Any, Optional
 
 class Models(BaseSDK):
@@ -62,7 +62,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = self.do_request(
-            hook_ctx=HookContext(operation_id="list_models_v1_models_get", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="list_models_v1_models_get", oauth2_scopes=[], security_source=get_security_from_env(self.sdk_configuration.security, models.Security)),
             request=req,
             error_status_codes=["422","4XX","5XX"],
             retry_config=retry_config
@@ -133,7 +133,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = await self.do_request_async(
-            hook_ctx=HookContext(operation_id="list_models_v1_models_get", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="list_models_v1_models_get", oauth2_scopes=[], security_source=get_security_from_env(self.sdk_configuration.security, models.Security)),
             request=req,
             error_status_codes=["422","4XX","5XX"],
             retry_config=retry_config
@@ -164,7 +164,7 @@ class Models(BaseSDK):
 
         Retrieve a model information.
 
-        :param model_id: 
+        :param model_id: The ID of the model to retrieve.
         :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
@@ -211,7 +211,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = self.do_request(
-            hook_ctx=HookContext(operation_id="retrieve_model_v1_models__model_id__get", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="retrieve_model_v1_models__model_id__get", oauth2_scopes=[], security_source=get_security_from_env(self.sdk_configuration.security, models.Security)),
             request=req,
             error_status_codes=["422","4XX","5XX"],
             retry_config=retry_config
@@ -242,7 +242,7 @@ class Models(BaseSDK):
 
         Retrieve a model information.
 
-        :param model_id: 
+        :param model_id: The ID of the model to retrieve.
         :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
@@ -289,7 +289,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = await self.do_request_async(
-            hook_ctx=HookContext(operation_id="retrieve_model_v1_models__model_id__get", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="retrieve_model_v1_models__model_id__get", oauth2_scopes=[], security_source=get_security_from_env(self.sdk_configuration.security, models.Security)),
             request=req,
             error_status_codes=["422","4XX","5XX"],
             retry_config=retry_config
@@ -320,7 +320,7 @@ class Models(BaseSDK):
 
         Delete a fine-tuned model.
 
-        :param model_id: 
+        :param model_id: The ID of the model to delete.
         :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
@@ -367,7 +367,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = self.do_request(
-            hook_ctx=HookContext(operation_id="delete_model_v1_models__model_id__delete", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="delete_model_v1_models__model_id__delete", oauth2_scopes=[], security_source=get_security_from_env(self.sdk_configuration.security, models.Security)),
             request=req,
             error_status_codes=["422","4XX","5XX"],
             retry_config=retry_config
@@ -398,7 +398,7 @@ class Models(BaseSDK):
 
         Delete a fine-tuned model.
 
-        :param model_id: 
+        :param model_id: The ID of the model to delete.
         :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
@@ -445,7 +445,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = await self.do_request_async(
-            hook_ctx=HookContext(operation_id="delete_model_v1_models__model_id__delete", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="delete_model_v1_models__model_id__delete", oauth2_scopes=[], security_source=get_security_from_env(self.sdk_configuration.security, models.Security)),
             request=req,
             error_status_codes=["422","4XX","5XX"],
             retry_config=retry_config
@@ -478,7 +478,7 @@ class Models(BaseSDK):
 
         Update a model name or description.
 
-        :param model_id: 
+        :param model_id: The ID of the model to update.
         :param name: 
         :param description: 
         :param retries: Override the default retry configuration for this method
@@ -532,7 +532,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = self.do_request(
-            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model", 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
@@ -561,7 +561,7 @@ class Models(BaseSDK):
 
         Update a model name or description.
 
-        :param model_id: 
+        :param model_id: The ID of the model to update.
         :param name: 
         :param description: 
         :param retries: Override the default retry configuration for this method
@@ -615,7 +615,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = await self.do_request_async(
-            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_update_fine_tuned_model", 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
@@ -642,7 +642,7 @@ class Models(BaseSDK):
 
         Archive a fine-tuned model.
 
-        :param model_id: 
+        :param model_id: The ID of the model to archive.
         :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
@@ -689,7 +689,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = self.do_request(
-            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model", 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
@@ -716,7 +716,7 @@ class Models(BaseSDK):
 
         Archive a fine-tuned model.
 
-        :param model_id: 
+        :param model_id: The ID of the model to archive.
         :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
@@ -763,7 +763,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = await self.do_request_async(
-            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_archive_fine_tuned_model", 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
@@ -790,7 +790,7 @@ class Models(BaseSDK):
 
         Un-archive a fine-tuned model.
 
-        :param model_id: 
+        :param model_id: The ID of the model to unarchive.
         :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
@@ -837,7 +837,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = self.do_request(
-            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model", 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
@@ -864,7 +864,7 @@ class Models(BaseSDK):
 
         Un-archive a fine-tuned model.
 
-        :param model_id: 
+        :param model_id: The ID of the model to unarchive.
         :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
@@ -911,7 +911,7 @@ class Models(BaseSDK):
             ])                
         
         http_res = await self.do_request_async(
-            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model", oauth2_scopes=[], security_source=self.sdk_configuration.security),
+            hook_ctx=HookContext(operation_id="jobs_api_routes_fine_tuning_unarchive_fine_tuned_model", 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

mistralai/sdk.py

@@ -3,10 +3,12 @@
 from .basesdk import BaseSDK
 from .httpclient import AsyncHttpClient, HttpClient
 from .sdkconfiguration import SDKConfiguration
+from .utils.logger import Logger, NoOpLogger
 from .utils.retries import RetryConfig
 import httpx
-from mistralai import models
+from mistralai import models, utils
 from mistralai._hooks import SDKHooks
+from mistralai.agents import Agents
 from mistralai.chat import Chat
 from mistralai.embeddings import Embeddings
 from mistralai.files import Files
@@ -14,7 +16,6 @@ from mistralai.fim import Fim
 from mistralai.fine_tuning import FineTuning
 from mistralai.models_ import Models
 from mistralai.types import OptionalNullable, UNSET
-import mistralai.utils as utils
 from typing import Any, Callable, Dict, Optional, Union
 
 class Mistral(BaseSDK):
@@ -24,23 +25,25 @@ class Mistral(BaseSDK):
     files: Files
     r"""Files API"""
     fine_tuning: FineTuning
-    r"""Fine-tuning API"""
     chat: Chat
     r"""Chat Completion API."""
     fim: Fim
     r"""Fill-in-the-middle API."""
+    agents: Agents
+    r"""Agents API."""
     embeddings: Embeddings
     r"""Embeddings API."""
     def __init__(
         self,
-        api_key: Union[str, Callable[[], str]],
+        api_key: Optional[Union[Optional[str], Callable[[], Optional[str]]]] = None,
         server: Optional[str] = None,
         server_url: Optional[str] = None,
         url_params: Optional[Dict[str, str]] = None,
         client: Optional[HttpClient] = None,
         async_client: Optional[AsyncHttpClient] = None,
         retry_config: OptionalNullable[RetryConfig] = UNSET,
-        timeout_ms: Optional[int] = None
+        timeout_ms: Optional[int] = None,
+        debug_logger: Optional[Logger] = None
     ) -> None:
         r"""Instantiates the SDK configuring it with the provided parameters.
 
@@ -63,6 +66,9 @@ class Mistral(BaseSDK):
         if async_client is None:
             async_client = httpx.AsyncClient()
 
+        if debug_logger is None:
+            debug_logger = NoOpLogger()
+
         assert issubclass(
             type(async_client), AsyncHttpClient
         ), "The provided async_client must implement the AsyncHttpClient protocol."
@@ -85,7 +91,8 @@ class Mistral(BaseSDK):
             server_url=server_url,
             server=server,
             retry_config=retry_config,
-            timeout_ms=timeout_ms
+            timeout_ms=timeout_ms,
+            debug_logger=debug_logger
         ))
 
         hooks = SDKHooks()
@@ -107,5 +114,6 @@ class Mistral(BaseSDK):
         self.fine_tuning = FineTuning(self.sdk_configuration)
         self.chat = Chat(self.sdk_configuration)
         self.fim = Fim(self.sdk_configuration)
+        self.agents = Agents(self.sdk_configuration)
         self.embeddings = Embeddings(self.sdk_configuration)
     

mistralai/sdkconfiguration.py

@@ -3,7 +3,7 @@
 
 from ._hooks import SDKHooks
 from .httpclient import AsyncHttpClient, HttpClient
-from .utils import RetryConfig, remove_suffix
+from .utils import Logger, RetryConfig, remove_suffix
 from dataclasses import dataclass
 from mistralai import models
 from mistralai.types import OptionalNullable, UNSET
@@ -23,14 +23,15 @@ SERVERS = {
 class SDKConfiguration:
     client: HttpClient
     async_client: AsyncHttpClient
+    debug_logger: Logger
     security: Optional[Union[models.Security,Callable[[], models.Security]]] = None
     server_url: Optional[str] = ""
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.0.0rc1"
-    gen_version: str = "2.382.2"
-    user_agent: str = "speakeasy-sdk/python 1.0.0rc1 2.382.2 0.0.2 mistralai"
+    sdk_version: str = "1.0.1"
+    gen_version: str = "2.390.6"
+    user_agent: str = "speakeasy-sdk/python 1.0.1 2.390.6 0.0.2 mistralai"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 

mistralai/types/basemodel.py

@@ -2,8 +2,8 @@
 
 from pydantic import ConfigDict, model_serializer
 from pydantic import BaseModel as PydanticBaseModel
-from typing import Literal, Optional, TypeVar, Union, NewType
-from typing_extensions import TypeAliasType
+from typing import TYPE_CHECKING, Literal, Optional, TypeVar, Union, NewType
+from typing_extensions import TypeAliasType, TypeAlias
 
 
 class BaseModel(PydanticBaseModel):
@@ -26,10 +26,14 @@ UNSET_SENTINEL = "~?~unset~?~sentinel~?~"
 
 
 T = TypeVar("T")
-Nullable = TypeAliasType("Nullable", Union[T, None], type_params=(T,))
-OptionalNullable = TypeAliasType(
-    "OptionalNullable", Union[Optional[Nullable[T]], Unset], type_params=(T,)
-)
+if TYPE_CHECKING:
+    Nullable: TypeAlias = Union[T, None]
+    OptionalNullable: TypeAlias = Union[Optional[Nullable[T]], Unset]
+else:
+    Nullable = TypeAliasType("Nullable", Union[T, None], type_params=(T,))
+    OptionalNullable = TypeAliasType(
+        "OptionalNullable", Union[Optional[Nullable[T]], Unset], type_params=(T,)
+    )
 
 UnrecognizedInt = NewType("UnrecognizedInt", int)
 UnrecognizedStr = NewType("UnrecognizedStr", str)

mistralai/utils/__init__.py

@@ -35,6 +35,7 @@ from .serializers import (
 )
 from .url import generate_url, template_url, remove_suffix
 from .values import get_global_from_env, match_content_type, match_status_codes, match_response
+from .logger import Logger, get_body_content, NoOpLogger
 
 __all__ = [
     "BackoffStrategy",
@@ -42,6 +43,7 @@ __all__ = [
     "find_metadata",
     "FormMetadata",
     "generate_url",
+    "get_body_content",
     "get_discriminator",
     "get_global_from_env",
     "get_headers",
@@ -51,11 +53,13 @@ __all__ = [
     "get_security",
     "get_security_from_env",
     "HeaderMetadata",
+    "Logger",
     "marshal_json",
     "match_content_type",
     "match_status_codes",
     "match_response",
     "MultipartFormMetadata",
+    "NoOpLogger",
     "OpenEnumMeta",
     "PathParamMetadata",
     "QueryParamMetadata",