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

mistralai/documents.py

@@ -3,6 +3,10 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
+from mistralai.models import (
+    documentupdatein as models_documentupdatein,
+    file as models_file,
+)
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -76,6 +80,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -92,7 +97,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_list_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -183,6 +188,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -199,7 +205,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_list_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -230,7 +236,7 @@ class Documents(BaseSDK):
         self,
         *,
         library_id: str,
-        file: Union[models.File, models.FileTypedDict],
+        file: Union[models_file.File, models_file.FileTypedDict],
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -241,7 +247,15 @@ class Documents(BaseSDK):
         Given a library, upload a new document to that library. It is queued for processing, it status will change it has been processed. The processing has to be completed in order be discoverable for the library search
 
         :param library_id:
-        :param file: The File object (not file name) to be uploaded.  To upload a file and specify a custom file name you should format your request as such:  ```bash  file=@path/to/your/file.jsonl;filename=custom_name.jsonl  ```  Otherwise, you can just keep the original file name:  ```bash  file=@path/to/your/file.jsonl  ```
+        :param file: The File object (not file name) to be uploaded.
+            To upload a file and specify a custom file name you should format your request as such:
+            ```bash
+            file=@path/to/your/file.jsonl;filename=custom_name.jsonl
+            ```
+            Otherwise, you can just keep the original file name:
+            ```bash
+            file=@path/to/your/file.jsonl
+            ```
         :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
@@ -284,6 +298,7 @@ class Documents(BaseSDK):
                 "multipart",
                 models.LibrariesDocumentsUploadV1DocumentUpload,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -300,7 +315,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_upload_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -331,7 +346,7 @@ class Documents(BaseSDK):
         self,
         *,
         library_id: str,
-        file: Union[models.File, models.FileTypedDict],
+        file: Union[models_file.File, models_file.FileTypedDict],
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -342,7 +357,15 @@ class Documents(BaseSDK):
         Given a library, upload a new document to that library. It is queued for processing, it status will change it has been processed. The processing has to be completed in order be discoverable for the library search
 
         :param library_id:
-        :param file: The File object (not file name) to be uploaded.  To upload a file and specify a custom file name you should format your request as such:  ```bash  file=@path/to/your/file.jsonl;filename=custom_name.jsonl  ```  Otherwise, you can just keep the original file name:  ```bash  file=@path/to/your/file.jsonl  ```
+        :param file: The File object (not file name) to be uploaded.
+            To upload a file and specify a custom file name you should format your request as such:
+            ```bash
+            file=@path/to/your/file.jsonl;filename=custom_name.jsonl
+            ```
+            Otherwise, you can just keep the original file name:
+            ```bash
+            file=@path/to/your/file.jsonl
+            ```
         :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
@@ -385,6 +408,7 @@ class Documents(BaseSDK):
                 "multipart",
                 models.LibrariesDocumentsUploadV1DocumentUpload,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -401,7 +425,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_upload_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -477,6 +501,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -493,7 +518,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -569,6 +594,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -585,7 +611,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -619,7 +645,10 @@ class Documents(BaseSDK):
         document_id: str,
         name: OptionalNullable[str] = UNSET,
         attributes: OptionalNullable[
-            Union[Dict[str, models.Attributes], Dict[str, models.AttributesTypedDict]]
+            Union[
+                Dict[str, models_documentupdatein.Attributes],
+                Dict[str, models_documentupdatein.AttributesTypedDict],
+            ]
         ] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -678,6 +707,7 @@ class Documents(BaseSDK):
                 "json",
                 models.DocumentUpdateIn,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -694,7 +724,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_update_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -728,7 +758,10 @@ class Documents(BaseSDK):
         document_id: str,
         name: OptionalNullable[str] = UNSET,
         attributes: OptionalNullable[
-            Union[Dict[str, models.Attributes], Dict[str, models.AttributesTypedDict]]
+            Union[
+                Dict[str, models_documentupdatein.Attributes],
+                Dict[str, models_documentupdatein.AttributesTypedDict],
+            ]
         ] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -787,6 +820,7 @@ class Documents(BaseSDK):
                 "json",
                 models.DocumentUpdateIn,
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -803,7 +837,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_update_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -879,6 +913,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -895,7 +930,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_delete_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -971,6 +1006,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -987,7 +1023,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_delete_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1063,6 +1099,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1079,7 +1116,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_text_content_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1155,6 +1192,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1171,7 +1209,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_text_content_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1247,6 +1285,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1263,7 +1302,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_status_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1339,6 +1378,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1355,7 +1395,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_status_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1431,6 +1471,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1447,7 +1488,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_signed_url_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1523,6 +1564,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1539,7 +1581,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_signed_url_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1615,6 +1657,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1631,7 +1674,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_extracted_text_signed_url_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1707,6 +1750,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1723,7 +1767,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_get_extracted_text_signed_url_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1799,6 +1843,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1815,7 +1860,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_reprocess_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -1891,6 +1936,7 @@ class Documents(BaseSDK):
             accept_header_value="application/json",
             http_headers=http_headers,
             security=self.sdk_configuration.security,
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -1907,7 +1953,7 @@ class Documents(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="libraries_documents_reprocess_v1",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/embeddings.py

@@ -3,10 +3,15 @@
 from .basesdk import BaseSDK
 from mistralai import models, utils
 from mistralai._hooks import HookContext
+from mistralai.models import (
+    embeddingdtype as models_embeddingdtype,
+    embeddingrequest as models_embeddingrequest,
+    encodingformat as models_encodingformat,
+)
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
-from typing import Any, Mapping, Optional, Union
+from typing import Any, Dict, Mapping, Optional, Union
 
 
 class Embeddings(BaseSDK):
@@ -17,11 +22,13 @@ class Embeddings(BaseSDK):
         *,
         model: str,
         inputs: Union[
-            models.EmbeddingRequestInputs, models.EmbeddingRequestInputsTypedDict
+            models_embeddingrequest.EmbeddingRequestInputs,
+            models_embeddingrequest.EmbeddingRequestInputsTypedDict,
         ],
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         output_dimension: OptionalNullable[int] = UNSET,
-        output_dtype: Optional[models.EmbeddingDtype] = None,
-        encoding_format: Optional[models.EncodingFormat] = None,
+        output_dtype: Optional[models_embeddingdtype.EmbeddingDtype] = None,
+        encoding_format: Optional[models_encodingformat.EncodingFormat] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -33,6 +40,7 @@ class Embeddings(BaseSDK):
 
         :param model: The ID of the model to be used for embedding.
         :param inputs: The text content to be embedded, can be a string or an array of strings for fast processing in bulk.
+        :param metadata:
         :param output_dimension: The dimension of the output embeddings when feature available. If not provided, a default output dimension will be used.
         :param output_dtype:
         :param encoding_format:
@@ -53,6 +61,7 @@ class Embeddings(BaseSDK):
 
         request = models.EmbeddingRequest(
             model=model,
+            metadata=metadata,
             inputs=inputs,
             output_dimension=output_dimension,
             output_dtype=output_dtype,
@@ -75,6 +84,7 @@ class Embeddings(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.EmbeddingRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -91,7 +101,7 @@ class Embeddings(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="embeddings_v1_embeddings_post",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),
@@ -123,11 +133,13 @@ class Embeddings(BaseSDK):
         *,
         model: str,
         inputs: Union[
-            models.EmbeddingRequestInputs, models.EmbeddingRequestInputsTypedDict
+            models_embeddingrequest.EmbeddingRequestInputs,
+            models_embeddingrequest.EmbeddingRequestInputsTypedDict,
         ],
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         output_dimension: OptionalNullable[int] = UNSET,
-        output_dtype: Optional[models.EmbeddingDtype] = None,
-        encoding_format: Optional[models.EncodingFormat] = None,
+        output_dtype: Optional[models_embeddingdtype.EmbeddingDtype] = None,
+        encoding_format: Optional[models_encodingformat.EncodingFormat] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -139,6 +151,7 @@ class Embeddings(BaseSDK):
 
         :param model: The ID of the model to be used for embedding.
         :param inputs: The text content to be embedded, can be a string or an array of strings for fast processing in bulk.
+        :param metadata:
         :param output_dimension: The dimension of the output embeddings when feature available. If not provided, a default output dimension will be used.
         :param output_dtype:
         :param encoding_format:
@@ -159,6 +172,7 @@ class Embeddings(BaseSDK):
 
         request = models.EmbeddingRequest(
             model=model,
+            metadata=metadata,
             inputs=inputs,
             output_dimension=output_dimension,
             output_dtype=output_dtype,
@@ -181,6 +195,7 @@ class Embeddings(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request, False, False, "json", models.EmbeddingRequest
             ),
+            allow_empty_value=None,
             timeout_ms=timeout_ms,
         )
 
@@ -197,7 +212,7 @@ class Embeddings(BaseSDK):
                 config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="embeddings_v1_embeddings_post",
-                oauth2_scopes=[],
+                oauth2_scopes=None,
                 security_source=get_security_from_env(
                     self.sdk_configuration.security, models.Security
                 ),

mistralai/extra/README.md

@@ -34,7 +34,7 @@ class Chat(BaseSDK):
 
 3. Now build the SDK with the custom code:
 ```bash
-rm -rf dist; poetry build; python3 -m pip install ~/client-python/dist/mistralai-1.4.1-py3-none-any.whl --force-reinstall
+rm -rf dist; uv build; uv pip install --reinstall ~/client-python/dist/mistralai-1.4.1-py3-none-any.whl
 ```
 
 4. And now you should be able to call the custom method:

mistralai/extra/mcp/auth.py

@@ -1,9 +1,8 @@
-from typing import Optional
+import logging
 
-from authlib.oauth2.rfc8414 import AuthorizationServerMetadata
-from authlib.integrations.httpx_client import AsyncOAuth2Client as AsyncOAuth2ClientBase
 import httpx
-import logging
+from authlib.integrations.httpx_client import AsyncOAuth2Client as AsyncOAuth2ClientBase
+from authlib.oauth2.rfc8414 import AuthorizationServerMetadata
 
 from mistralai.types import BaseModel
 
@@ -16,8 +15,8 @@ class Oauth2AuthorizationScheme(BaseModel):
     authorization_url: str
     token_url: str
     scope: list[str]
-    description: Optional[str] = None
-    refresh_url: Optional[str] = None
+    description: str | None = None
+    refresh_url: str | None = None
 
 
 class OAuthParams(BaseModel):
@@ -42,7 +41,7 @@ class AsyncOAuth2Client(AsyncOAuth2ClientBase):
 
 async def get_well_known_authorization_server_metadata(
     server_url: str,
-) -> Optional[AuthorizationServerMetadata]:
+) -> AuthorizationServerMetadata | None:
     """Fetch the metadata from the well-known location.
 
     This should be available on MCP servers as described by the specification:
@@ -123,10 +122,10 @@ async def dynamic_client_registration(
 async def build_oauth_params(
     server_url: str,
     redirect_url: str,
-    client_id: Optional[str] = None,
-    client_secret: Optional[str] = None,
-    scope: Optional[list[str]] = None,
-    async_client: Optional[httpx.AsyncClient] = None,
+    client_id: str | None = None,
+    client_secret: str | None = None,
+    scope: list[str] | None = None,
+    async_client: httpx.AsyncClient | None = None,
 ) -> OAuthParams:
     """Get issuer metadata and build the oauth required params."""
     metadata = await get_oauth_server_metadata(server_url=server_url)

mistralai/extra/mcp/base.py

@@ -1,11 +1,14 @@
-from typing import Optional, Union
 import logging
 import typing
+from collections.abc import Sequence
 from contextlib import AsyncExitStack
-from typing import Protocol, Any
+from typing import Any, Protocol
 
-from mcp import ClientSession
-from mcp.types import ListPromptsResult, EmbeddedResource, ImageContent, TextContent
+from mcp import ClientSession  # pyright: ignore[reportMissingImports]
+from mcp.types import (  # pyright: ignore[reportMissingImports]
+    ContentBlock,
+    ListPromptsResult,
+)
 
 from mistralai.extra.exceptions import MCPException
 from mistralai.models import (
@@ -20,8 +23,8 @@ logger = logging.getLogger(__name__)
 
 
 class MCPSystemPrompt(typing.TypedDict):
-    description: Optional[str]
-    messages: list[Union[SystemMessageTypedDict, AssistantMessageTypedDict]]
+    description: str | None
+    messages: list[SystemMessageTypedDict | AssistantMessageTypedDict]
 
 
 class MCPClientProtocol(Protocol):
@@ -29,7 +32,7 @@ class MCPClientProtocol(Protocol):
 
     _name: str
 
-    async def initialize(self, exit_stack: Optional[AsyncExitStack]) -> None:
+    async def initialize(self, exit_stack: AsyncExitStack | None) -> None:
         ...
 
     async def aclose(self) -> None:
@@ -39,7 +42,7 @@ class MCPClientProtocol(Protocol):
         ...
 
     async def execute_tool(
-        self, name: str, arguments: dict
+        self, name: str, arguments: dict[str, Any]
     ) -> list[TextChunkTypedDict]:
         ...
 
@@ -57,20 +60,18 @@ class MCPClientBase(MCPClientProtocol):
 
     _session: ClientSession
 
-    def __init__(self, name: Optional[str] = None):
+    def __init__(self, name: str | None = None):
         self._name = name or self.__class__.__name__
-        self._exit_stack: Optional[AsyncExitStack] = None
+        self._exit_stack: AsyncExitStack | None = None
         self._is_initialized = False
 
-    def _convert_content(
-        self, mcp_content: Union[TextContent, ImageContent, EmbeddedResource]
-    ) -> TextChunkTypedDict:
+    def _convert_content(self, mcp_content: ContentBlock) -> TextChunkTypedDict:
         if not mcp_content.type == "text":
             raise MCPException("Only supporting text tool responses for now.")
         return {"type": "text", "text": mcp_content.text}
 
     def _convert_content_list(
-        self, mcp_contents: list[Union[TextContent, ImageContent, EmbeddedResource]]
+        self, mcp_contents: Sequence[ContentBlock]
     ) -> list[TextChunkTypedDict]:
         content_chunks = []
         for mcp_content in mcp_contents:
@@ -108,7 +109,7 @@ class MCPClientBase(MCPClientProtocol):
             "description": prompt_result.description,
             "messages": [
                 typing.cast(
-                    Union[SystemMessageTypedDict, AssistantMessageTypedDict],
+                    SystemMessageTypedDict | AssistantMessageTypedDict,
                     {
                         "role": message.role,
                         "content": self._convert_content(mcp_content=message.content),
@@ -121,7 +122,7 @@ class MCPClientBase(MCPClientProtocol):
     async def list_system_prompts(self) -> ListPromptsResult:
         return await self._session.list_prompts()
 
-    async def initialize(self, exit_stack: Optional[AsyncExitStack] = None) -> None:
+    async def initialize(self, exit_stack: AsyncExitStack | None = None) -> None:
         """Initialize the MCP session."""
         # client is already initialized so return
         if self._is_initialized: