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

mistralai_azure/models/systemmessage.py

@@ -4,13 +4,17 @@ from __future__ import annotations
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai_azure.types import BaseModel
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-SystemMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+SystemMessageContentTypedDict = TypeAliasType(
+    "SystemMessageContentTypedDict", Union[str, List[TextChunkTypedDict]]
+)
 
 
-SystemMessageContent = Union[str, List[TextChunk]]
+SystemMessageContent = TypeAliasType(
+    "SystemMessageContent", Union[str, List[TextChunk]]
+)
 
 
 Role = Literal["system"]

mistralai_azure/models/textchunk.py

@@ -2,11 +2,8 @@
 
 from __future__ import annotations
 from mistralai_azure.types import BaseModel
-from mistralai_azure.utils import validate_const
-import pydantic
-from pydantic.functional_validators import AfterValidator
 from typing import Literal, Optional
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 Type = Literal["text"]
@@ -14,13 +11,10 @@ Type = Literal["text"]
 
 class TextChunkTypedDict(TypedDict):
     text: str
-    type: Type
+    type: NotRequired[Type]
 
 
 class TextChunk(BaseModel):
     text: str
 
-    TYPE: Annotated[
-        Annotated[Optional[Type], AfterValidator(validate_const("text"))],
-        pydantic.Field(alias="type"),
-    ] = "text"
+    type: Optional[Type] = "text"

mistralai_azure/models/toolmessage.py

@@ -11,13 +11,15 @@ from mistralai_azure.types import (
 )
 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
 
 
-ToolMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+ToolMessageContentTypedDict = TypeAliasType(
+    "ToolMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-ToolMessageContent = Union[str, List[ContentChunk]]
+ToolMessageContent = TypeAliasType("ToolMessageContent", Union[str, List[ContentChunk]])
 
 
 ToolMessageRole = Literal["tool"]

mistralai_azure/models/usermessage.py

@@ -5,13 +5,15 @@ from .contentchunk import ContentChunk, ContentChunkTypedDict
 from mistralai_azure.types import BaseModel, Nullable, 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
 
 
-UserMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+UserMessageContentTypedDict = TypeAliasType(
+    "UserMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-UserMessageContent = Union[str, List[ContentChunk]]
+UserMessageContent = TypeAliasType("UserMessageContent", Union[str, List[ContentChunk]])
 
 
 UserMessageRole = Literal["user"]

mistralai_azure/models/validationerror.py

@@ -3,13 +3,13 @@
 from __future__ import annotations
 from mistralai_azure.types import BaseModel
 from typing import List, Union
-from typing_extensions import TypedDict
+from typing_extensions import TypeAliasType, TypedDict
 
 
-LocTypedDict = Union[str, int]
+LocTypedDict = TypeAliasType("LocTypedDict", Union[str, int])
 
 
-Loc = Union[str, int]
+Loc = TypeAliasType("Loc", Union[str, int])
 
 
 class ValidationErrorTypedDict(TypedDict):

mistralai_azure/sdkconfiguration.py

@@ -29,8 +29,8 @@ class SDKConfiguration:
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
     sdk_version: str = "1.2.3"
-    gen_version: str = "2.460.1"
-    user_agent: str = "speakeasy-sdk/python 1.2.3 2.460.1 0.0.2 mistralai_azure"
+    gen_version: str = "2.470.1"
+    user_agent: str = "speakeasy-sdk/python 1.2.3 2.470.1 0.0.2 mistralai_azure"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 

mistralai_azure/utils/annotations.py

@@ -1,30 +1,55 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from enum import Enum
-from typing import Any
+from typing import Any, Optional
 
 def get_discriminator(model: Any, fieldname: str, key: str) -> str:
-    if isinstance(model, dict):
-        try:
-            return f'{model.get(key)}'
-        except AttributeError as e:
-            raise ValueError(f'Could not find discriminator key {key} in {model}') from e
+    """
+    Recursively search for the discriminator attribute in a model.
 
-    if hasattr(model, fieldname):
-        attr = getattr(model, fieldname)
+    Args:
+        model (Any): The model to search within.
+        fieldname (str): The name of the field to search for.
+        key (str): The key to search for in dictionaries.
 
-        if isinstance(attr, Enum):
-            return f'{attr.value}'
+    Returns:
+        str: The name of the discriminator attribute.
 
-        return f'{attr}'
+    Raises:
+        ValueError: If the discriminator attribute is not found.
+    """
+    upper_fieldname = fieldname.upper()
 
-    fieldname = fieldname.upper()
-    if hasattr(model, fieldname):
-        attr = getattr(model, fieldname)
+    def get_field_discriminator(field: Any) -> Optional[str]:
+        """Search for the discriminator attribute in a given field."""
 
-        if isinstance(attr, Enum):
-            return f'{attr.value}'
+        if isinstance(field, dict):
+            if key in field:
+                return f'{field[key]}'
 
-        return f'{attr}'
+        if hasattr(field, fieldname):
+            attr = getattr(field, fieldname)
+            if isinstance(attr, Enum):
+                return f'{attr.value}'
+            return f'{attr}'
+
+        if hasattr(field, upper_fieldname):
+            attr = getattr(field, upper_fieldname)
+            if isinstance(attr, Enum):
+                return f'{attr.value}'
+            return f'{attr}'
+
+        return None
+
+
+    if isinstance(model, list):
+        for field in model:
+            discriminator = get_field_discriminator(field)
+            if discriminator is not None:
+                return discriminator
+
+    discriminator = get_field_discriminator(model)
+    if discriminator is not None:
+        return discriminator
 
     raise ValueError(f'Could not find discriminator field {fieldname} in {model}')

mistralai_azure/utils/eventstreaming.py

@@ -2,12 +2,72 @@
 
 import re
 import json
-from typing import Callable, TypeVar, Optional, Generator, AsyncGenerator, Tuple
+from typing import (
+    Callable,
+    Generic,
+    TypeVar,
+    Optional,
+    Generator,
+    AsyncGenerator,
+    Tuple,
+)
 import httpx
 
 T = TypeVar("T")
 
 
+class EventStream(Generic[T]):
+    response: httpx.Response
+    generator: Generator[T, None, None]
+
+    def __init__(
+        self,
+        response: httpx.Response,
+        decoder: Callable[[str], T],
+        sentinel: Optional[str] = None,
+    ):
+        self.response = response
+        self.generator = stream_events(response, decoder, sentinel)
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        return next(self.generator)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.response.close()
+
+
+class EventStreamAsync(Generic[T]):
+    response: httpx.Response
+    generator: AsyncGenerator[T, None]
+
+    def __init__(
+        self,
+        response: httpx.Response,
+        decoder: Callable[[str], T],
+        sentinel: Optional[str] = None,
+    ):
+        self.response = response
+        self.generator = stream_events_async(response, decoder, sentinel)
+
+    def __aiter__(self):
+        return self
+
+    async def __anext__(self):
+        return await self.generator.__anext__()
+
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.response.aclose()
+
+
 class ServerEvent:
     id: Optional[str] = None
     event: Optional[str] = None

mistralai_gcp/chat.py

@@ -5,7 +5,7 @@ from mistralai_gcp import models, utils
 from mistralai_gcp._hooks import HookContext
 from mistralai_gcp.types import Nullable, OptionalNullable, UNSET
 from mistralai_gcp.utils import eventstreaming
-from typing import Any, AsyncGenerator, Generator, List, Optional, Union
+from typing import Any, List, Optional, Union
 
 
 class Chat(BaseSDK):
@@ -40,7 +40,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.
@@ -132,7 +132,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]",
@@ -185,7 +185,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.
@@ -277,7 +277,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_gcp/fim.py

@@ -5,7 +5,7 @@ from mistralai_gcp import models, utils
 from mistralai_gcp._hooks import HookContext
 from mistralai_gcp.types import Nullable, OptionalNullable, UNSET
 from mistralai_gcp.utils import eventstreaming
-from typing import Any, AsyncGenerator, Generator, Optional, Union
+from typing import Any, Optional, Union
 
 
 class Fim(BaseSDK):
@@ -32,7 +32,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.
@@ -112,7 +112,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]",
@@ -157,7 +157,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.
@@ -237,7 +237,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_gcp/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_gcp/models/assistantmessage.py

@@ -12,13 +12,17 @@ from mistralai_gcp.types import (
 )
 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_gcp/models/chatcompletionrequest.py

@@ -19,23 +19,30 @@ from mistralai_gcp.types import (
 from mistralai_gcp.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
 
 
-ChatCompletionRequestStopTypedDict = Union[str, List[str]]
+ChatCompletionRequestStopTypedDict = TypeAliasType(
+    "ChatCompletionRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionRequestStop = Union[str, List[str]]
+ChatCompletionRequestStop = TypeAliasType(
+    "ChatCompletionRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-ChatCompletionRequestMessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+ChatCompletionRequestMessagesTypedDict = TypeAliasType(
+    "ChatCompletionRequestMessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 ChatCompletionRequestMessages = Annotated[
@@ -49,10 +56,15 @@ ChatCompletionRequestMessages = Annotated[
 ]
 
 
-ChatCompletionRequestToolChoiceTypedDict = Union[ToolChoiceTypedDict, ToolChoiceEnum]
+ChatCompletionRequestToolChoiceTypedDict = TypeAliasType(
+    "ChatCompletionRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-ChatCompletionRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+ChatCompletionRequestToolChoice = TypeAliasType(
+    "ChatCompletionRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class ChatCompletionRequestTypedDict(TypedDict):

mistralai_gcp/models/chatcompletionstreamrequest.py

@@ -19,23 +19,26 @@ from mistralai_gcp.types import (
 from mistralai_gcp.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
 
 
-StopTypedDict = Union[str, List[str]]
+StopTypedDict = TypeAliasType("StopTypedDict", Union[str, List[str]])
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-Stop = Union[str, List[str]]
+Stop = TypeAliasType("Stop", Union[str, List[str]])
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-MessagesTypedDict = Union[
-    SystemMessageTypedDict,
-    UserMessageTypedDict,
-    AssistantMessageTypedDict,
-    ToolMessageTypedDict,
-]
+MessagesTypedDict = TypeAliasType(
+    "MessagesTypedDict",
+    Union[
+        SystemMessageTypedDict,
+        UserMessageTypedDict,
+        AssistantMessageTypedDict,
+        ToolMessageTypedDict,
+    ],
+)
 
 
 Messages = Annotated[
@@ -49,12 +52,15 @@ Messages = Annotated[
 ]
 
 
-ChatCompletionStreamRequestToolChoiceTypedDict = Union[
-    ToolChoiceTypedDict, ToolChoiceEnum
-]
+ChatCompletionStreamRequestToolChoiceTypedDict = TypeAliasType(
+    "ChatCompletionStreamRequestToolChoiceTypedDict",
+    Union[ToolChoiceTypedDict, ToolChoiceEnum],
+)
 
 
-ChatCompletionStreamRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
+ChatCompletionStreamRequestToolChoice = TypeAliasType(
+    "ChatCompletionStreamRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
+)
 
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):

mistralai_gcp/models/contentchunk.py

@@ -6,10 +6,12 @@ from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai_gcp.utils import get_discriminator
 from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import Annotated
+from typing_extensions import Annotated, TypeAliasType
 
 
-ContentChunkTypedDict = Union[TextChunkTypedDict, ReferenceChunkTypedDict]
+ContentChunkTypedDict = TypeAliasType(
+    "ContentChunkTypedDict", Union[TextChunkTypedDict, ReferenceChunkTypedDict]
+)
 
 
 ContentChunk = Annotated[

mistralai_gcp/models/deltamessage.py

@@ -12,13 +12,15 @@ from mistralai_gcp.types import (
 )
 from pydantic import model_serializer
 from typing import List, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
+ContentTypedDict = TypeAliasType(
+    "ContentTypedDict", Union[str, List[ContentChunkTypedDict]]
+)
 
 
-Content = Union[str, List[ContentChunk]]
+Content = TypeAliasType("Content", Union[str, List[ContentChunk]])
 
 
 class DeltaMessageTypedDict(TypedDict):

mistralai_gcp/models/fimcompletionrequest.py

@@ -10,14 +10,18 @@ from mistralai_gcp.types import (
 )
 from pydantic import model_serializer
 from typing import List, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-FIMCompletionRequestStopTypedDict = Union[str, List[str]]
+FIMCompletionRequestStopTypedDict = TypeAliasType(
+    "FIMCompletionRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-FIMCompletionRequestStop = Union[str, List[str]]
+FIMCompletionRequestStop = TypeAliasType(
+    "FIMCompletionRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 

mistralai_gcp/models/fimcompletionstreamrequest.py

@@ -10,14 +10,18 @@ from mistralai_gcp.types import (
 )
 from pydantic import model_serializer
 from typing import List, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-FIMCompletionStreamRequestStopTypedDict = Union[str, List[str]]
+FIMCompletionStreamRequestStopTypedDict = TypeAliasType(
+    "FIMCompletionStreamRequestStopTypedDict", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 
-FIMCompletionStreamRequestStop = Union[str, List[str]]
+FIMCompletionStreamRequestStop = TypeAliasType(
+    "FIMCompletionStreamRequestStop", Union[str, List[str]]
+)
 r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
 
 

mistralai_gcp/models/functioncall.py

@@ -3,13 +3,13 @@
 from __future__ import annotations
 from mistralai_gcp.types import BaseModel
 from typing import Any, Dict, Union
-from typing_extensions import TypedDict
+from typing_extensions import TypeAliasType, TypedDict
 
 
-ArgumentsTypedDict = Union[Dict[str, Any], str]
+ArgumentsTypedDict = TypeAliasType("ArgumentsTypedDict", Union[Dict[str, Any], str])
 
 
-Arguments = Union[Dict[str, Any], str]
+Arguments = TypeAliasType("Arguments", Union[Dict[str, Any], str])
 
 
 class FunctionCallTypedDict(TypedDict):

mistralai_gcp/models/referencechunk.py

@@ -2,11 +2,8 @@
 
 from __future__ import annotations
 from mistralai_gcp.types import BaseModel
-from mistralai_gcp.utils import validate_const
-import pydantic
-from pydantic.functional_validators import AfterValidator
 from typing import List, Literal, Optional
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 ReferenceChunkType = Literal["reference"]
@@ -14,15 +11,10 @@ ReferenceChunkType = Literal["reference"]
 
 class ReferenceChunkTypedDict(TypedDict):
     reference_ids: List[int]
-    type: ReferenceChunkType
+    type: NotRequired[ReferenceChunkType]
 
 
 class ReferenceChunk(BaseModel):
     reference_ids: List[int]
 
-    TYPE: Annotated[
-        Annotated[
-            Optional[ReferenceChunkType], AfterValidator(validate_const("reference"))
-        ],
-        pydantic.Field(alias="type"),
-    ] = "reference"
+    type: Optional[ReferenceChunkType] = "reference"

mistralai_gcp/models/systemmessage.py

@@ -4,13 +4,17 @@ from __future__ import annotations
 from .textchunk import TextChunk, TextChunkTypedDict
 from mistralai_gcp.types import BaseModel
 from typing import List, Literal, Optional, Union
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 
-SystemMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+SystemMessageContentTypedDict = TypeAliasType(
+    "SystemMessageContentTypedDict", Union[str, List[TextChunkTypedDict]]
+)
 
 
-SystemMessageContent = Union[str, List[TextChunk]]
+SystemMessageContent = TypeAliasType(
+    "SystemMessageContent", Union[str, List[TextChunk]]
+)
 
 
 Role = Literal["system"]