mistralai 1.9.11__py3-none-any.whl → 1.10.1__py3-none-any.whl

mistralai_gcp/models/__init__.py

@@ -1,122 +1,154 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from .assistantmessage import (
-    AssistantMessage,
-    AssistantMessageContent,
-    AssistantMessageContentTypedDict,
-    AssistantMessageRole,
-    AssistantMessageTypedDict,
-)
-from .chatcompletionchoice import (
-    ChatCompletionChoice,
-    ChatCompletionChoiceFinishReason,
-    ChatCompletionChoiceTypedDict,
-)
-from .chatcompletionrequest import (
-    ChatCompletionRequest,
-    ChatCompletionRequestMessages,
-    ChatCompletionRequestMessagesTypedDict,
-    ChatCompletionRequestStop,
-    ChatCompletionRequestStopTypedDict,
-    ChatCompletionRequestToolChoice,
-    ChatCompletionRequestToolChoiceTypedDict,
-    ChatCompletionRequestTypedDict,
-)
-from .chatcompletionresponse import (
-    ChatCompletionResponse,
-    ChatCompletionResponseTypedDict,
-)
-from .chatcompletionstreamrequest import (
-    ChatCompletionStreamRequest,
-    ChatCompletionStreamRequestToolChoice,
-    ChatCompletionStreamRequestToolChoiceTypedDict,
-    ChatCompletionStreamRequestTypedDict,
-    Messages,
-    MessagesTypedDict,
-    Stop,
-    StopTypedDict,
-)
-from .completionchunk import CompletionChunk, CompletionChunkTypedDict
-from .completionevent import CompletionEvent, CompletionEventTypedDict
-from .completionresponsestreamchoice import (
-    CompletionResponseStreamChoice,
-    CompletionResponseStreamChoiceTypedDict,
-    FinishReason,
-)
-from .contentchunk import ContentChunk, ContentChunkTypedDict
-from .deltamessage import Content, ContentTypedDict, DeltaMessage, DeltaMessageTypedDict
-from .fimcompletionrequest import (
-    FIMCompletionRequest,
-    FIMCompletionRequestStop,
-    FIMCompletionRequestStopTypedDict,
-    FIMCompletionRequestTypedDict,
-)
-from .fimcompletionresponse import FIMCompletionResponse, FIMCompletionResponseTypedDict
-from .fimcompletionstreamrequest import (
-    FIMCompletionStreamRequest,
-    FIMCompletionStreamRequestStop,
-    FIMCompletionStreamRequestStopTypedDict,
-    FIMCompletionStreamRequestTypedDict,
-)
-from .function import Function, FunctionTypedDict
-from .functioncall import (
-    Arguments,
-    ArgumentsTypedDict,
-    FunctionCall,
-    FunctionCallTypedDict,
-)
-from .functionname import FunctionName, FunctionNameTypedDict
-from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
-from .imageurl import ImageURL, ImageURLTypedDict
-from .imageurlchunk import (
-    ImageURLChunk,
-    ImageURLChunkImageURL,
-    ImageURLChunkImageURLTypedDict,
-    ImageURLChunkType,
-    ImageURLChunkTypedDict,
-)
-from .jsonschema import JSONSchema, JSONSchemaTypedDict
-from .prediction import Prediction, PredictionTypedDict
-from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
-from .responseformat import ResponseFormat, ResponseFormatTypedDict
-from .responseformats import ResponseFormats
-from .sdkerror import SDKError
-from .security import Security, SecurityTypedDict
-from .systemmessage import (
-    Role,
-    SystemMessage,
-    SystemMessageContent,
-    SystemMessageContentTypedDict,
-    SystemMessageTypedDict,
-)
-from .textchunk import TextChunk, TextChunkTypedDict, Type
-from .tool import Tool, ToolTypedDict
-from .toolcall import ToolCall, ToolCallTypedDict
-from .toolchoice import ToolChoice, ToolChoiceTypedDict
-from .toolchoiceenum import ToolChoiceEnum
-from .toolmessage import (
-    ToolMessage,
-    ToolMessageContent,
-    ToolMessageContentTypedDict,
-    ToolMessageRole,
-    ToolMessageTypedDict,
-)
-from .tooltypes import ToolTypes
-from .usageinfo import UsageInfo, UsageInfoTypedDict
-from .usermessage import (
-    UserMessage,
-    UserMessageContent,
-    UserMessageContentTypedDict,
-    UserMessageRole,
-    UserMessageTypedDict,
-)
-from .validationerror import (
-    Loc,
-    LocTypedDict,
-    ValidationError,
-    ValidationErrorTypedDict,
-)
+from .mistralgcperror import MistralGcpError
+from typing import TYPE_CHECKING
+from importlib import import_module
+import builtins
+import sys
 
+if TYPE_CHECKING:
+    from .assistantmessage import (
+        AssistantMessage,
+        AssistantMessageContent,
+        AssistantMessageContentTypedDict,
+        AssistantMessageRole,
+        AssistantMessageTypedDict,
+    )
+    from .chatcompletionchoice import (
+        ChatCompletionChoice,
+        ChatCompletionChoiceFinishReason,
+        ChatCompletionChoiceTypedDict,
+    )
+    from .chatcompletionrequest import (
+        ChatCompletionRequest,
+        ChatCompletionRequestMessages,
+        ChatCompletionRequestMessagesTypedDict,
+        ChatCompletionRequestStop,
+        ChatCompletionRequestStopTypedDict,
+        ChatCompletionRequestToolChoice,
+        ChatCompletionRequestToolChoiceTypedDict,
+        ChatCompletionRequestTypedDict,
+    )
+    from .chatcompletionresponse import (
+        ChatCompletionResponse,
+        ChatCompletionResponseTypedDict,
+    )
+    from .chatcompletionstreamrequest import (
+        ChatCompletionStreamRequest,
+        ChatCompletionStreamRequestToolChoice,
+        ChatCompletionStreamRequestToolChoiceTypedDict,
+        ChatCompletionStreamRequestTypedDict,
+        Messages,
+        MessagesTypedDict,
+        Stop,
+        StopTypedDict,
+    )
+    from .completionchunk import CompletionChunk, CompletionChunkTypedDict
+    from .completionevent import CompletionEvent, CompletionEventTypedDict
+    from .completionresponsestreamchoice import (
+        CompletionResponseStreamChoice,
+        CompletionResponseStreamChoiceTypedDict,
+        FinishReason,
+    )
+    from .contentchunk import ContentChunk, ContentChunkTypedDict
+    from .deltamessage import (
+        Content,
+        ContentTypedDict,
+        DeltaMessage,
+        DeltaMessageTypedDict,
+    )
+    from .fimcompletionrequest import (
+        FIMCompletionRequest,
+        FIMCompletionRequestStop,
+        FIMCompletionRequestStopTypedDict,
+        FIMCompletionRequestTypedDict,
+    )
+    from .fimcompletionresponse import (
+        FIMCompletionResponse,
+        FIMCompletionResponseTypedDict,
+    )
+    from .fimcompletionstreamrequest import (
+        FIMCompletionStreamRequest,
+        FIMCompletionStreamRequestStop,
+        FIMCompletionStreamRequestStopTypedDict,
+        FIMCompletionStreamRequestTypedDict,
+    )
+    from .function import Function, FunctionTypedDict
+    from .functioncall import (
+        Arguments,
+        ArgumentsTypedDict,
+        FunctionCall,
+        FunctionCallTypedDict,
+    )
+    from .functionname import FunctionName, FunctionNameTypedDict
+    from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
+    from .imageurl import ImageURL, ImageURLTypedDict
+    from .imageurlchunk import (
+        ImageURLChunk,
+        ImageURLChunkImageURL,
+        ImageURLChunkImageURLTypedDict,
+        ImageURLChunkType,
+        ImageURLChunkTypedDict,
+    )
+    from .jsonschema import JSONSchema, JSONSchemaTypedDict
+    from .mistralpromptmode import MistralPromptMode
+    from .no_response_error import NoResponseError
+    from .prediction import Prediction, PredictionTypedDict
+    from .referencechunk import (
+        ReferenceChunk,
+        ReferenceChunkType,
+        ReferenceChunkTypedDict,
+    )
+    from .responseformat import ResponseFormat, ResponseFormatTypedDict
+    from .responseformats import ResponseFormats
+    from .responsevalidationerror import ResponseValidationError
+    from .sdkerror import SDKError
+    from .security import Security, SecurityTypedDict
+    from .systemmessage import (
+        Role,
+        SystemMessage,
+        SystemMessageContent,
+        SystemMessageContentTypedDict,
+        SystemMessageTypedDict,
+    )
+    from .systemmessagecontentchunks import (
+        SystemMessageContentChunks,
+        SystemMessageContentChunksTypedDict,
+    )
+    from .textchunk import TextChunk, TextChunkTypedDict, Type
+    from .thinkchunk import (
+        ThinkChunk,
+        ThinkChunkType,
+        ThinkChunkTypedDict,
+        Thinking,
+        ThinkingTypedDict,
+    )
+    from .tool import Tool, ToolTypedDict
+    from .toolcall import ToolCall, ToolCallTypedDict
+    from .toolchoice import ToolChoice, ToolChoiceTypedDict
+    from .toolchoiceenum import ToolChoiceEnum
+    from .toolmessage import (
+        ToolMessage,
+        ToolMessageContent,
+        ToolMessageContentTypedDict,
+        ToolMessageRole,
+        ToolMessageTypedDict,
+    )
+    from .tooltypes import ToolTypes
+    from .usageinfo import UsageInfo, UsageInfoTypedDict
+    from .usermessage import (
+        UserMessage,
+        UserMessageContent,
+        UserMessageContentTypedDict,
+        UserMessageRole,
+        UserMessageTypedDict,
+    )
+    from .validationerror import (
+        Loc,
+        LocTypedDict,
+        ValidationError,
+        ValidationErrorTypedDict,
+    )
 
 __all__ = [
     "Arguments",
@@ -187,6 +219,9 @@ __all__ = [
     "LocTypedDict",
     "Messages",
     "MessagesTypedDict",
+    "MistralGcpError",
+    "MistralPromptMode",
+    "NoResponseError",
     "Prediction",
     "PredictionTypedDict",
     "ReferenceChunk",
@@ -195,6 +230,7 @@ __all__ = [
     "ResponseFormat",
     "ResponseFormatTypedDict",
     "ResponseFormats",
+    "ResponseValidationError",
     "Role",
     "SDKError",
     "Security",
@@ -203,10 +239,17 @@ __all__ = [
     "StopTypedDict",
     "SystemMessage",
     "SystemMessageContent",
+    "SystemMessageContentChunks",
+    "SystemMessageContentChunksTypedDict",
     "SystemMessageContentTypedDict",
     "SystemMessageTypedDict",
     "TextChunk",
     "TextChunkTypedDict",
+    "ThinkChunk",
+    "ThinkChunkType",
+    "ThinkChunkTypedDict",
+    "Thinking",
+    "ThinkingTypedDict",
     "Tool",
     "ToolCall",
     "ToolCallTypedDict",
@@ -231,3 +274,165 @@ __all__ = [
     "ValidationError",
     "ValidationErrorTypedDict",
 ]
+
+_dynamic_imports: dict[str, str] = {
+    "AssistantMessage": ".assistantmessage",
+    "AssistantMessageContent": ".assistantmessage",
+    "AssistantMessageContentTypedDict": ".assistantmessage",
+    "AssistantMessageRole": ".assistantmessage",
+    "AssistantMessageTypedDict": ".assistantmessage",
+    "ChatCompletionChoice": ".chatcompletionchoice",
+    "ChatCompletionChoiceFinishReason": ".chatcompletionchoice",
+    "ChatCompletionChoiceTypedDict": ".chatcompletionchoice",
+    "ChatCompletionRequest": ".chatcompletionrequest",
+    "ChatCompletionRequestMessages": ".chatcompletionrequest",
+    "ChatCompletionRequestMessagesTypedDict": ".chatcompletionrequest",
+    "ChatCompletionRequestStop": ".chatcompletionrequest",
+    "ChatCompletionRequestStopTypedDict": ".chatcompletionrequest",
+    "ChatCompletionRequestToolChoice": ".chatcompletionrequest",
+    "ChatCompletionRequestToolChoiceTypedDict": ".chatcompletionrequest",
+    "ChatCompletionRequestTypedDict": ".chatcompletionrequest",
+    "ChatCompletionResponse": ".chatcompletionresponse",
+    "ChatCompletionResponseTypedDict": ".chatcompletionresponse",
+    "ChatCompletionStreamRequest": ".chatcompletionstreamrequest",
+    "ChatCompletionStreamRequestToolChoice": ".chatcompletionstreamrequest",
+    "ChatCompletionStreamRequestToolChoiceTypedDict": ".chatcompletionstreamrequest",
+    "ChatCompletionStreamRequestTypedDict": ".chatcompletionstreamrequest",
+    "Messages": ".chatcompletionstreamrequest",
+    "MessagesTypedDict": ".chatcompletionstreamrequest",
+    "Stop": ".chatcompletionstreamrequest",
+    "StopTypedDict": ".chatcompletionstreamrequest",
+    "CompletionChunk": ".completionchunk",
+    "CompletionChunkTypedDict": ".completionchunk",
+    "CompletionEvent": ".completionevent",
+    "CompletionEventTypedDict": ".completionevent",
+    "CompletionResponseStreamChoice": ".completionresponsestreamchoice",
+    "CompletionResponseStreamChoiceTypedDict": ".completionresponsestreamchoice",
+    "FinishReason": ".completionresponsestreamchoice",
+    "ContentChunk": ".contentchunk",
+    "ContentChunkTypedDict": ".contentchunk",
+    "Content": ".deltamessage",
+    "ContentTypedDict": ".deltamessage",
+    "DeltaMessage": ".deltamessage",
+    "DeltaMessageTypedDict": ".deltamessage",
+    "FIMCompletionRequest": ".fimcompletionrequest",
+    "FIMCompletionRequestStop": ".fimcompletionrequest",
+    "FIMCompletionRequestStopTypedDict": ".fimcompletionrequest",
+    "FIMCompletionRequestTypedDict": ".fimcompletionrequest",
+    "FIMCompletionResponse": ".fimcompletionresponse",
+    "FIMCompletionResponseTypedDict": ".fimcompletionresponse",
+    "FIMCompletionStreamRequest": ".fimcompletionstreamrequest",
+    "FIMCompletionStreamRequestStop": ".fimcompletionstreamrequest",
+    "FIMCompletionStreamRequestStopTypedDict": ".fimcompletionstreamrequest",
+    "FIMCompletionStreamRequestTypedDict": ".fimcompletionstreamrequest",
+    "Function": ".function",
+    "FunctionTypedDict": ".function",
+    "Arguments": ".functioncall",
+    "ArgumentsTypedDict": ".functioncall",
+    "FunctionCall": ".functioncall",
+    "FunctionCallTypedDict": ".functioncall",
+    "FunctionName": ".functionname",
+    "FunctionNameTypedDict": ".functionname",
+    "HTTPValidationError": ".httpvalidationerror",
+    "HTTPValidationErrorData": ".httpvalidationerror",
+    "ImageURL": ".imageurl",
+    "ImageURLTypedDict": ".imageurl",
+    "ImageURLChunk": ".imageurlchunk",
+    "ImageURLChunkImageURL": ".imageurlchunk",
+    "ImageURLChunkImageURLTypedDict": ".imageurlchunk",
+    "ImageURLChunkType": ".imageurlchunk",
+    "ImageURLChunkTypedDict": ".imageurlchunk",
+    "JSONSchema": ".jsonschema",
+    "JSONSchemaTypedDict": ".jsonschema",
+    "MistralPromptMode": ".mistralpromptmode",
+    "NoResponseError": ".no_response_error",
+    "Prediction": ".prediction",
+    "PredictionTypedDict": ".prediction",
+    "ReferenceChunk": ".referencechunk",
+    "ReferenceChunkType": ".referencechunk",
+    "ReferenceChunkTypedDict": ".referencechunk",
+    "ResponseFormat": ".responseformat",
+    "ResponseFormatTypedDict": ".responseformat",
+    "ResponseFormats": ".responseformats",
+    "ResponseValidationError": ".responsevalidationerror",
+    "SDKError": ".sdkerror",
+    "Security": ".security",
+    "SecurityTypedDict": ".security",
+    "Role": ".systemmessage",
+    "SystemMessage": ".systemmessage",
+    "SystemMessageContent": ".systemmessage",
+    "SystemMessageContentTypedDict": ".systemmessage",
+    "SystemMessageTypedDict": ".systemmessage",
+    "SystemMessageContentChunks": ".systemmessagecontentchunks",
+    "SystemMessageContentChunksTypedDict": ".systemmessagecontentchunks",
+    "TextChunk": ".textchunk",
+    "TextChunkTypedDict": ".textchunk",
+    "Type": ".textchunk",
+    "ThinkChunk": ".thinkchunk",
+    "ThinkChunkType": ".thinkchunk",
+    "ThinkChunkTypedDict": ".thinkchunk",
+    "Thinking": ".thinkchunk",
+    "ThinkingTypedDict": ".thinkchunk",
+    "Tool": ".tool",
+    "ToolTypedDict": ".tool",
+    "ToolCall": ".toolcall",
+    "ToolCallTypedDict": ".toolcall",
+    "ToolChoice": ".toolchoice",
+    "ToolChoiceTypedDict": ".toolchoice",
+    "ToolChoiceEnum": ".toolchoiceenum",
+    "ToolMessage": ".toolmessage",
+    "ToolMessageContent": ".toolmessage",
+    "ToolMessageContentTypedDict": ".toolmessage",
+    "ToolMessageRole": ".toolmessage",
+    "ToolMessageTypedDict": ".toolmessage",
+    "ToolTypes": ".tooltypes",
+    "UsageInfo": ".usageinfo",
+    "UsageInfoTypedDict": ".usageinfo",
+    "UserMessage": ".usermessage",
+    "UserMessageContent": ".usermessage",
+    "UserMessageContentTypedDict": ".usermessage",
+    "UserMessageRole": ".usermessage",
+    "UserMessageTypedDict": ".usermessage",
+    "Loc": ".validationerror",
+    "LocTypedDict": ".validationerror",
+    "ValidationError": ".validationerror",
+    "ValidationErrorTypedDict": ".validationerror",
+}
+
+
+def dynamic_import(modname, retries=3):
+    for attempt in range(retries):
+        try:
+            return import_module(modname, __package__)
+        except KeyError:
+            # Clear any half-initialized module and retry
+            sys.modules.pop(modname, None)
+            if attempt == retries - 1:
+                break
+    raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
+
+def __getattr__(attr_name: str) -> object:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__} "
+        )
+
+    try:
+        module = dynamic_import(module_name)
+        result = getattr(module, attr_name)
+        return result
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = builtins.list(_dynamic_imports.keys())
+    return builtins.sorted(lazy_attrs)

mistralai_gcp/models/assistantmessage.py

@@ -56,7 +56,7 @@ class AssistantMessage(BaseModel):
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_gcp/models/chatcompletionrequest.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 from .assistantmessage import AssistantMessage, AssistantMessageTypedDict
+from .mistralpromptmode import MistralPromptMode
 from .prediction import Prediction, PredictionTypedDict
 from .responseformat import ResponseFormat, ResponseFormatTypedDict
 from .systemmessage import SystemMessage, SystemMessageTypedDict
@@ -17,8 +18,9 @@ from mistralai_gcp.types import (
     UNSET,
     UNSET_SENTINEL,
 )
-from mistralai_gcp.utils import get_discriminator
+from mistralai_gcp.utils import get_discriminator, validate_open_enum
 from pydantic import Discriminator, Tag, model_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -61,11 +63,13 @@ ChatCompletionRequestToolChoiceTypedDict = TypeAliasType(
     "ChatCompletionRequestToolChoiceTypedDict",
     Union[ToolChoiceTypedDict, ToolChoiceEnum],
 )
+r"""Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `any` or `required` means the model must call one or more tools. Specifying a particular tool via `{\"type\": \"function\", \"function\": {\"name\": \"my_function\"}}` forces the model to call that tool."""
 
 
 ChatCompletionRequestToolChoice = TypeAliasType(
     "ChatCompletionRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
 )
+r"""Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `any` or `required` means the model must call one or more tools. Specifying a particular tool via `{\"type\": \"function\", \"function\": {\"name\": \"my_function\"}}` forces the model to call that tool."""
 
 
 class ChatCompletionRequestTypedDict(TypedDict):
@@ -86,16 +90,23 @@ class ChatCompletionRequestTypedDict(TypedDict):
     random_seed: NotRequired[Nullable[int]]
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
     response_format: NotRequired[ResponseFormatTypedDict]
+    r"""Specify the format that the model must output. By default it will use `{ \"type\": \"text\" }`. Setting to `{ \"type\": \"json_object\" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message. Setting to `{ \"type\": \"json_schema\" }` enables JSON schema mode, which guarantees the message the model generates is in JSON and follows the schema you provide."""
     tools: NotRequired[Nullable[List[ToolTypedDict]]]
+    r"""A list of tools the model may call. Use this to provide a list of functions the model may generate JSON inputs for."""
     tool_choice: NotRequired[ChatCompletionRequestToolChoiceTypedDict]
+    r"""Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `any` or `required` means the model must call one or more tools. Specifying a particular tool via `{\"type\": \"function\", \"function\": {\"name\": \"my_function\"}}` forces the model to call that tool."""
     presence_penalty: NotRequired[float]
-    r"""presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative."""
+    r"""The `presence_penalty` determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative."""
     frequency_penalty: NotRequired[float]
-    r"""frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition."""
+    r"""The `frequency_penalty` penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition."""
     n: NotRequired[Nullable[int]]
     r"""Number of completions to return for each request, input tokens are only billed once."""
     prediction: NotRequired[PredictionTypedDict]
+    r"""Enable users to specify an expected completion, optimizing response times by leveraging known or predictable content."""
     parallel_tool_calls: NotRequired[bool]
+    r"""Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel."""
+    prompt_mode: NotRequired[Nullable[MistralPromptMode]]
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
 
 class ChatCompletionRequest(BaseModel):
@@ -124,23 +135,33 @@ class ChatCompletionRequest(BaseModel):
     r"""The seed to use for random sampling. If set, different calls will generate deterministic results."""
 
     response_format: Optional[ResponseFormat] = None
+    r"""Specify the format that the model must output. By default it will use `{ \"type\": \"text\" }`. Setting to `{ \"type\": \"json_object\" }` enables JSON mode, which guarantees the message the model generates is in JSON. When using JSON mode you MUST also instruct the model to produce JSON yourself with a system or a user message. Setting to `{ \"type\": \"json_schema\" }` enables JSON schema mode, which guarantees the message the model generates is in JSON and follows the schema you provide."""
 
     tools: OptionalNullable[List[Tool]] = UNSET
+    r"""A list of tools the model may call. Use this to provide a list of functions the model may generate JSON inputs for."""
 
     tool_choice: Optional[ChatCompletionRequestToolChoice] = None
+    r"""Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `any` or `required` means the model must call one or more tools. Specifying a particular tool via `{\"type\": \"function\", \"function\": {\"name\": \"my_function\"}}` forces the model to call that tool."""
 
     presence_penalty: Optional[float] = None
-    r"""presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative."""
+    r"""The `presence_penalty` determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative."""
 
     frequency_penalty: Optional[float] = None
-    r"""frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition."""
+    r"""The `frequency_penalty` penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition."""
 
     n: OptionalNullable[int] = UNSET
     r"""Number of completions to return for each request, input tokens are only billed once."""
 
     prediction: Optional[Prediction] = None
+    r"""Enable users to specify an expected completion, optimizing response times by leveraging known or predictable content."""
 
     parallel_tool_calls: Optional[bool] = None
+    r"""Whether to enable parallel function calling during tool use, when enabled the model can call multiple tools in parallel."""
+
+    prompt_mode: Annotated[
+        OptionalNullable[MistralPromptMode], PlainValidator(validate_open_enum(False))
+    ] = UNSET
+    r"""Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used."""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
@@ -159,15 +180,23 @@ class ChatCompletionRequest(BaseModel):
             "n",
             "prediction",
             "parallel_tool_calls",
+            "prompt_mode",
+        ]
+        nullable_fields = [
+            "temperature",
+            "max_tokens",
+            "random_seed",
+            "tools",
+            "n",
+            "prompt_mode",
         ]
-        nullable_fields = ["temperature", "max_tokens", "random_seed", "tools", "n"]
         null_default_fields = []
 
         serialized = handler(self)
 
         m = {}
 
-        for n, f in self.model_fields.items():
+        for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
             serialized.pop(k, None)

mistralai_gcp/models/chatcompletionresponse.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai_gcp.types import BaseModel
-from typing import List, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class ChatCompletionResponseTypedDict(TypedDict):
@@ -13,8 +13,8 @@ class ChatCompletionResponseTypedDict(TypedDict):
     object: str
     model: str
     usage: UsageInfoTypedDict
-    created: NotRequired[int]
-    choices: NotRequired[List[ChatCompletionChoiceTypedDict]]
+    created: int
+    choices: List[ChatCompletionChoiceTypedDict]
 
 
 class ChatCompletionResponse(BaseModel):
@@ -26,6 +26,6 @@ class ChatCompletionResponse(BaseModel):
 
     usage: UsageInfo
 
-    created: Optional[int] = None
+    created: int
 
-    choices: Optional[List[ChatCompletionChoice]] = None
+    choices: List[ChatCompletionChoice]