mistralai 1.0.0rc2__py3-none-any.whl → 1.0.2__py3-none-any.whl

mistralai_azure/sdkconfiguration.py

@@ -29,9 +29,9 @@ class SDKConfiguration:
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.0.0-alpha.13"
-    gen_version: str = "2.386.0"
-    user_agent: str = "speakeasy-sdk/python 1.0.0-alpha.13 2.386.0 0.0.2 mistralai_azure"
+    sdk_version: str = "1.0.1"
+    gen_version: str = "2.399.0"
+    user_agent: str = "speakeasy-sdk/python 1.0.1 2.399.0 0.0.2 mistralai_azure"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 

mistralai_gcp/chat.py

@@ -33,7 +33,7 @@ class Chat(BaseSDK):
 
         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.
 
-        :param model: ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions.
+        :param model: 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.
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param temperature: What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both.
         :param top_p: Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.
@@ -146,7 +146,7 @@ class Chat(BaseSDK):
 
         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.
 
-        :param model: ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions.
+        :param model: 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.
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param temperature: What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both.
         :param top_p: Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.
@@ -257,7 +257,7 @@ class Chat(BaseSDK):
     ) -> Optional[models.ChatCompletionResponse]:
         r"""Chat Completion
 
-        :param model: ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions.
+        :param model: 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.
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param temperature: What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both.
         :param top_p: Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.
@@ -367,7 +367,7 @@ class Chat(BaseSDK):
     ) -> Optional[models.ChatCompletionResponse]:
         r"""Chat Completion
 
-        :param model: ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions.
+        :param model: 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.
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
         :param temperature: What sampling temperature to use, between 0.0 and 1.0. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both.
         :param top_p: Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.

mistralai_gcp/models/__init__.py

@@ -14,18 +14,18 @@ from .fimcompletionrequest import FIMCompletionRequest, FIMCompletionRequestStop
 from .fimcompletionresponse import FIMCompletionResponse, FIMCompletionResponseTypedDict
 from .fimcompletionstreamrequest import FIMCompletionStreamRequest, FIMCompletionStreamRequestStop, FIMCompletionStreamRequestStopTypedDict, FIMCompletionStreamRequestTypedDict
 from .function import Function, FunctionTypedDict
-from .functioncall import FunctionCall, FunctionCallTypedDict
+from .functioncall import Arguments, ArgumentsTypedDict, FunctionCall, FunctionCallTypedDict
 from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
 from .responseformat import ResponseFormat, ResponseFormatTypedDict, ResponseFormats
 from .sdkerror import SDKError
 from .security import Security, SecurityTypedDict
 from .systemmessage import Content, ContentTypedDict, Role, SystemMessage, SystemMessageTypedDict
 from .textchunk import TextChunk, TextChunkTypedDict
-from .tool import Tool, ToolTypedDict
-from .toolcall import ToolCall, ToolCallTypedDict
+from .tool import Tool, ToolToolTypes, ToolTypedDict
+from .toolcall import ToolCall, ToolCallTypedDict, ToolTypes
 from .toolmessage import ToolMessage, ToolMessageRole, ToolMessageTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from .usermessage import UserMessage, UserMessageContent, UserMessageContentTypedDict, UserMessageRole, UserMessageTypedDict
 from .validationerror import Loc, LocTypedDict, ValidationError, ValidationErrorTypedDict
 
-__all__ = ["AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceFinishReason", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestMessages", "ChatCompletionRequestMessagesTypedDict", "ChatCompletionRequestStop", "ChatCompletionRequestStopTypedDict", "ChatCompletionRequestToolChoice", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "FIMCompletionRequest", "FIMCompletionRequestStop", "FIMCompletionRequestStopTypedDict", "FIMCompletionRequestTypedDict", "FIMCompletionResponse", "FIMCompletionResponseTypedDict", "FIMCompletionStreamRequest", "FIMCompletionStreamRequestStop", "FIMCompletionStreamRequestStopTypedDict", "FIMCompletionStreamRequestTypedDict", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "Role", "SDKError", "Security", "SecurityTypedDict", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict"]
+__all__ = ["Arguments", "ArgumentsTypedDict", "AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceFinishReason", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestMessages", "ChatCompletionRequestMessagesTypedDict", "ChatCompletionRequestStop", "ChatCompletionRequestStopTypedDict", "ChatCompletionRequestToolChoice", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "FIMCompletionRequest", "FIMCompletionRequestStop", "FIMCompletionRequestStopTypedDict", "FIMCompletionRequestTypedDict", "FIMCompletionResponse", "FIMCompletionResponseTypedDict", "FIMCompletionStreamRequest", "FIMCompletionStreamRequestStop", "FIMCompletionStreamRequestStopTypedDict", "FIMCompletionStreamRequestTypedDict", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "Role", "SDKError", "Security", "SecurityTypedDict", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolToolTypes", "ToolTypedDict", "ToolTypes", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict"]

mistralai_gcp/models/chatcompletionchoice.py

@@ -3,20 +3,19 @@
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
 from mistralai_gcp.types import BaseModel
-from typing import Literal, Optional, TypedDict
-from typing_extensions import NotRequired
+from typing import Literal, TypedDict
 
 
 ChatCompletionChoiceFinishReason = Literal["stop", "length", "model_length", "error", "tool_calls"]
 
 class ChatCompletionChoiceTypedDict(TypedDict):
     index: int
+    message: AssistantMessageTypedDict
     finish_reason: ChatCompletionChoiceFinishReason
-    message: NotRequired[AssistantMessageTypedDict]
     
 
 class ChatCompletionChoice(BaseModel):
     index: int
+    message: AssistantMessage
     finish_reason: ChatCompletionChoiceFinishReason
-    message: Optional[AssistantMessage] = None
     

mistralai_gcp/models/chatcompletionrequest.py

@@ -14,11 +14,25 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+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]]
+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]
+
+
+ChatCompletionRequestMessages = 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"))]
+
+
 ChatCompletionRequestToolChoice = Literal["auto", "none", "any"]
 
 class ChatCompletionRequestTypedDict(TypedDict):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    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[ChatCompletionRequestMessagesTypedDict]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: NotRequired[float]
@@ -42,7 +56,7 @@ class ChatCompletionRequestTypedDict(TypedDict):
 
 class ChatCompletionRequest(BaseModel):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    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[ChatCompletionRequestMessages]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: Optional[float] = 0.7
@@ -89,17 +103,3 @@ class ChatCompletionRequest(BaseModel):
 
         return m
         
-
-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]]
-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]
-
-
-ChatCompletionRequestMessages = 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"))]
-

mistralai_gcp/models/chatcompletionstreamrequest.py

@@ -14,11 +14,25 @@ from typing import List, Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+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]]
+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]
+
+
+Messages = 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"))]
+
+
 ToolChoice = Literal["auto", "none", "any"]
 
 class ChatCompletionStreamRequestTypedDict(TypedDict):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    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."""
     temperature: NotRequired[float]
@@ -41,7 +55,7 @@ class ChatCompletionStreamRequestTypedDict(TypedDict):
 
 class ChatCompletionStreamRequest(BaseModel):
     model: Nullable[str]
-    r"""ID of the model to use. You can use the [List Available Models](/api#operation/listModels) API to see all of your available models, or see our [Model overview](/models) for model descriptions."""
+    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]
     r"""The prompt(s) to generate completions for, encoded as a list of dict with role and content."""
     temperature: Optional[float] = 0.7
@@ -87,17 +101,3 @@ class ChatCompletionStreamRequest(BaseModel):
 
         return m
         
-
-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]]
-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]
-
-
-Messages = 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"))]
-

mistralai_gcp/models/deltamessage.py

@@ -4,25 +4,25 @@ from __future__ import annotations
 from .toolcall import ToolCall, ToolCallTypedDict
 from mistralai_gcp.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from pydantic import model_serializer
-from typing import Optional, TypedDict
+from typing import List, Optional, TypedDict
 from typing_extensions import NotRequired
 
 
 class DeltaMessageTypedDict(TypedDict):
     role: NotRequired[str]
-    content: NotRequired[str]
-    tool_calls: NotRequired[Nullable[ToolCallTypedDict]]
+    content: NotRequired[Nullable[str]]
+    tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
     
 
 class DeltaMessage(BaseModel):
     role: Optional[str] = None
-    content: Optional[str] = None
-    tool_calls: OptionalNullable[ToolCall] = UNSET
+    content: OptionalNullable[str] = UNSET
+    tool_calls: OptionalNullable[List[ToolCall]] = UNSET
     
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["role", "content", "tool_calls"]
-        nullable_fields = ["tool_calls"]
+        nullable_fields = ["content", "tool_calls"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai_gcp/models/fimcompletionrequest.py

@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+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]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
 class FIMCompletionRequestTypedDict(TypedDict):
     model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
@@ -84,11 +92,3 @@ class FIMCompletionRequest(BaseModel):
 
         return m
         
-
-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]]
-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

@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
 from typing_extensions import NotRequired
 
 
+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]]
+r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
+
+
 class FIMCompletionStreamRequestTypedDict(TypedDict):
     model: Nullable[str]
     r"""ID of the model to use. Only compatible for now with:
@@ -82,11 +90,3 @@ class FIMCompletionStreamRequest(BaseModel):
 
         return m
         
-
-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]]
-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

@@ -2,15 +2,21 @@
 
 from __future__ import annotations
 from mistralai_gcp.types import BaseModel
-from typing import TypedDict
+from typing import Any, Dict, TypedDict, Union
+
+
+ArgumentsTypedDict = Union[Dict[str, Any], str]
+
+
+Arguments = Union[Dict[str, Any], str]
 
 
 class FunctionCallTypedDict(TypedDict):
     name: str
-    arguments: str
+    arguments: ArgumentsTypedDict
     
 
 class FunctionCall(BaseModel):
     name: str
-    arguments: str
+    arguments: Arguments
     

mistralai_gcp/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_gcp/models/tool.py

@@ -2,17 +2,21 @@
 
 from __future__ import annotations
 from .function import Function, FunctionTypedDict
-from mistralai_gcp.types import BaseModel
-import pydantic
-from typing import Final, Optional, TypedDict
-from typing_extensions import Annotated
+from mistralai_gcp.types import BaseModel, UnrecognizedStr
+from mistralai_gcp.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing import Literal, Optional, TypedDict, Union
+from typing_extensions import Annotated, NotRequired
 
 
+ToolToolTypes = Union[Literal["function"], UnrecognizedStr]
+
 class ToolTypedDict(TypedDict):
     function: FunctionTypedDict
+    type: NotRequired[ToolToolTypes]
     
 
 class Tool(BaseModel):
     function: Function
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "function" # type: ignore
+    type: Annotated[Optional[ToolToolTypes], PlainValidator(validate_open_enum(False))] = "function"
     

mistralai_gcp/models/toolcall.py

@@ -2,19 +2,23 @@
 
 from __future__ import annotations
 from .functioncall import FunctionCall, FunctionCallTypedDict
-from mistralai_gcp.types import BaseModel
-import pydantic
-from typing import Final, Optional, TypedDict
+from mistralai_gcp.types import BaseModel, UnrecognizedStr
+from mistralai_gcp.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing import Literal, Optional, TypedDict, Union
 from typing_extensions import Annotated, NotRequired
 
 
+ToolTypes = Union[Literal["function"], UnrecognizedStr]
+
 class ToolCallTypedDict(TypedDict):
     function: FunctionCallTypedDict
     id: NotRequired[str]
+    type: NotRequired[ToolTypes]
     
 
 class ToolCall(BaseModel):
     function: FunctionCall
     id: Optional[str] = "null"
-    TYPE: Annotated[Final[Optional[str]], pydantic.Field(alias="type")] = "function" # type: ignore
+    type: Annotated[Optional[ToolTypes], PlainValidator(validate_open_enum(False))] = "function"
     

mistralai_gcp/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_gcp/models/validationerror.py

@@ -5,6 +5,12 @@ from mistralai_gcp.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_gcp/sdkconfiguration.py

@@ -29,9 +29,9 @@ class SDKConfiguration:
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.0.0-alpha.12"
-    gen_version: str = "2.386.0"
-    user_agent: str = "speakeasy-sdk/python 1.0.0-alpha.12 2.386.0 0.0.2 mistralai-gcp"
+    sdk_version: str = "1.0.1"
+    gen_version: str = "2.399.0"
+    user_agent: str = "speakeasy-sdk/python 1.0.1 2.399.0 0.0.2 mistralai-gcp"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None