mirror of
https://github.com/Ladebeze66/llm_ticket3.git
synced 2025-12-18 07:27:46 +01:00
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
|
|
|
|
from __future__ import annotations
|
|
from .contentchunk import ContentChunk, ContentChunkTypedDict
|
|
from .toolcall import ToolCall, ToolCallTypedDict
|
|
from mistralai_azure.types import (
|
|
BaseModel,
|
|
Nullable,
|
|
OptionalNullable,
|
|
UNSET,
|
|
UNSET_SENTINEL,
|
|
)
|
|
from pydantic import model_serializer
|
|
from typing import List, Literal, Optional, Union
|
|
from typing_extensions import NotRequired, TypeAliasType, TypedDict
|
|
|
|
|
|
AssistantMessageContentTypedDict = TypeAliasType(
|
|
"AssistantMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
|
|
)
|
|
|
|
|
|
AssistantMessageContent = TypeAliasType(
|
|
"AssistantMessageContent", Union[str, List[ContentChunk]]
|
|
)
|
|
|
|
|
|
AssistantMessageRole = Literal["assistant"]
|
|
|
|
|
|
class AssistantMessageTypedDict(TypedDict):
|
|
content: NotRequired[Nullable[AssistantMessageContentTypedDict]]
|
|
tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
|
|
prefix: NotRequired[bool]
|
|
r"""Set this to `true` when adding an assistant message as prefix to condition the model response. The role of the prefix message is to force the model to start its answer by the content of the message."""
|
|
role: NotRequired[AssistantMessageRole]
|
|
|
|
|
|
class AssistantMessage(BaseModel):
|
|
content: OptionalNullable[AssistantMessageContent] = UNSET
|
|
|
|
tool_calls: OptionalNullable[List[ToolCall]] = UNSET
|
|
|
|
prefix: Optional[bool] = False
|
|
r"""Set this to `true` when adding an assistant message as prefix to condition the model response. The role of the prefix message is to force the model to start its answer by the content of the message."""
|
|
|
|
role: Optional[AssistantMessageRole] = "assistant"
|
|
|
|
@model_serializer(mode="wrap")
|
|
def serialize_model(self, handler):
|
|
optional_fields = ["content", "tool_calls", "prefix", "role"]
|
|
nullable_fields = ["content", "tool_calls"]
|
|
null_default_fields = []
|
|
|
|
serialized = handler(self)
|
|
|
|
m = {}
|
|
|
|
for n, f in self.model_fields.items():
|
|
k = f.alias or n
|
|
val = serialized.get(k)
|
|
serialized.pop(k, None)
|
|
|
|
optional_nullable = k in optional_fields and k in nullable_fields
|
|
is_set = (
|
|
self.__pydantic_fields_set__.intersection({n})
|
|
or k in null_default_fields
|
|
) # pylint: disable=no-member
|
|
|
|
if val is not None and val != UNSET_SENTINEL:
|
|
m[k] = val
|
|
elif val != UNSET_SENTINEL and (
|
|
not k in optional_fields or (optional_nullable and is_set)
|
|
):
|
|
m[k] = val
|
|
|
|
return m
|