letta-client 0.1.17__py3-none-any.whl → 0.1.21__py3-none-any.whl

letta_client/types/__init__.py

@@ -10,37 +10,26 @@ from .agent_type import AgentType
 from .app_auth_scheme import AppAuthScheme
 from .app_auth_scheme_auth_mode import AppAuthSchemeAuthMode
 from .app_model import AppModel
-from .assistant_file import AssistantFile
-from .assistant_message_input import AssistantMessageInput
-from .assistant_message_output import AssistantMessageOutput
+from .assistant_message import AssistantMessage
+from .assistant_message_content import AssistantMessageContent
 from .auth_request import AuthRequest
 from .auth_response import AuthResponse
 from .auth_scheme_field import AuthSchemeField
 from .block import Block
 from .block_update import BlockUpdate
-from .chat_completion_request import ChatCompletionRequest
-from .chat_completion_request_function_call import ChatCompletionRequestFunctionCall
-from .chat_completion_request_messages_item import ChatCompletionRequestMessagesItem
-from .chat_completion_request_stop import ChatCompletionRequestStop
-from .chat_completion_request_tool_choice import ChatCompletionRequestToolChoice
-from .chat_completion_response import ChatCompletionResponse
+from .chat_completion_message_tool_call import ChatCompletionMessageToolCall
 from .child_tool_rule import ChildToolRule
-from .choice import Choice
 from .conditional_tool_rule import ConditionalToolRule
 from .conflict_error_body import ConflictErrorBody
 from .context_window_overview import ContextWindowOverview
-from .create_assistant_file_request import CreateAssistantFileRequest
-from .create_assistant_request import CreateAssistantRequest
 from .create_block import CreateBlock
-from .delete_assistant_file_response import DeleteAssistantFileResponse
-from .delete_assistant_response import DeleteAssistantResponse
 from .e_2_b_sandbox_config import E2BSandboxConfig
 from .embedding_config import EmbeddingConfig
 from .embedding_config_embedding_endpoint_type import EmbeddingConfigEmbeddingEndpointType
 from .file_metadata import FileMetadata
-from .function_call_input import FunctionCallInput
-from .function_call_output import FunctionCallOutput
-from .function_schema import FunctionSchema
+from .function import Function
+from .function_definition import FunctionDefinition
+from .function_tool import FunctionTool
 from .health import Health
 from .http_validation_error import HttpValidationError
 from .init_tool_rule import InitToolRule
@@ -48,50 +37,27 @@ from .internal_server_error_body import InternalServerErrorBody
 from .job import Job
 from .job_status import JobStatus
 from .job_type import JobType
-from .letta_message_union import (
-    LettaMessageUnion,
-    LettaMessageUnion_AssistantMessage,
-    LettaMessageUnion_ReasoningMessage,
-    LettaMessageUnion_SystemMessage,
-    LettaMessageUnion_ToolCallMessage,
-    LettaMessageUnion_ToolReturnMessage,
-    LettaMessageUnion_UserMessage,
-)
+from .letta_message_union import LettaMessageUnion
 from .letta_request import LettaRequest
 from .letta_request_config import LettaRequestConfig
 from .letta_response import LettaResponse
-from .letta_schemas_letta_message_tool_call import LettaSchemasLettaMessageToolCall
-from .letta_schemas_message_message import LettaSchemasMessageMessage
-from .letta_schemas_openai_chat_completion_request_tool import LettaSchemasOpenaiChatCompletionRequestTool
-from .letta_schemas_openai_chat_completion_request_tool_call import LettaSchemasOpenaiChatCompletionRequestToolCall
-from .letta_schemas_openai_chat_completion_request_tool_call_function import (
-    LettaSchemasOpenaiChatCompletionRequestToolCallFunction,
-)
-from .letta_schemas_openai_chat_completion_response_message import LettaSchemasOpenaiChatCompletionResponseMessage
-from .letta_schemas_openai_chat_completion_response_tool_call import LettaSchemasOpenaiChatCompletionResponseToolCall
-from .letta_schemas_openai_chat_completions_tool_call_function import LettaSchemasOpenaiChatCompletionsToolCallFunction
-from .letta_schemas_openai_chat_completions_tool_call_input import LettaSchemasOpenaiChatCompletionsToolCallInput
-from .letta_schemas_openai_chat_completions_tool_call_output import LettaSchemasOpenaiChatCompletionsToolCallOutput
-from .letta_schemas_tool_tool import LettaSchemasToolTool
 from .letta_usage_statistics import LettaUsageStatistics
 from .llm_config import LlmConfig
 from .llm_config_model_endpoint_type import LlmConfigModelEndpointType
 from .local_sandbox_config import LocalSandboxConfig
-from .log_prob_token import LogProbToken
 from .memory import Memory
-from .message_content_log_prob import MessageContentLogProb
+from .message import Message
 from .message_create import MessageCreate
+from .message_create_content import MessageCreateContent
 from .message_create_role import MessageCreateRole
 from .message_role import MessageRole
 from .not_found_error_body import NotFoundErrorBody
 from .not_found_error_body_message import NotFoundErrorBodyMessage
-from .open_ai_assistant import OpenAiAssistant
 from .organization import Organization
 from .organization_create import OrganizationCreate
 from .passage import Passage
 from .provider import Provider
 from .reasoning_message import ReasoningMessage
-from .response_format import ResponseFormat
 from .run import Run
 from .sandbox_config import SandboxConfig
 from .sandbox_config_create import SandboxConfigCreate
@@ -103,17 +69,16 @@ from .sandbox_environment_variable_create import SandboxEnvironmentVariableCreat
 from .sandbox_environment_variable_update import SandboxEnvironmentVariableUpdate
 from .sandbox_type import SandboxType
 from .source import Source
-from .system_message_input import SystemMessageInput
-from .system_message_output import SystemMessageOutput
+from .system_message import SystemMessage
+from .system_message_content import SystemMessageContent
 from .terminal_tool_rule import TerminalToolRule
+from .text_content import TextContent
+from .tool import Tool
+from .tool_call import ToolCall
 from .tool_call_delta import ToolCallDelta
-from .tool_call_function_output import ToolCallFunctionOutput
 from .tool_call_message import ToolCallMessage
 from .tool_call_message_tool_call import ToolCallMessageToolCall
 from .tool_create import ToolCreate
-from .tool_function_choice import ToolFunctionChoice
-from .tool_input import ToolInput
-from .tool_message import ToolMessage
 from .tool_return_message import ToolReturnMessage
 from .tool_return_message_status import ToolReturnMessageStatus
 from .tool_rule_type import ToolRuleType
@@ -121,9 +86,8 @@ from .tool_type import ToolType
 from .usage_statistics import UsageStatistics
 from .user import User
 from .user_create import UserCreate
-from .user_message_input import UserMessageInput
-from .user_message_input_content import UserMessageInputContent
-from .user_message_output import UserMessageOutput
+from .user_message import UserMessage
+from .user_message_content import UserMessageContent
 from .user_update import UserUpdate
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
@@ -139,37 +103,26 @@ __all__ = [
     "AppAuthScheme",
     "AppAuthSchemeAuthMode",
     "AppModel",
-    "AssistantFile",
-    "AssistantMessageInput",
-    "AssistantMessageOutput",
+    "AssistantMessage",
+    "AssistantMessageContent",
     "AuthRequest",
     "AuthResponse",
     "AuthSchemeField",
     "Block",
     "BlockUpdate",
-    "ChatCompletionRequest",
-    "ChatCompletionRequestFunctionCall",
-    "ChatCompletionRequestMessagesItem",
-    "ChatCompletionRequestStop",
-    "ChatCompletionRequestToolChoice",
-    "ChatCompletionResponse",
+    "ChatCompletionMessageToolCall",
     "ChildToolRule",
-    "Choice",
     "ConditionalToolRule",
     "ConflictErrorBody",
     "ContextWindowOverview",
-    "CreateAssistantFileRequest",
-    "CreateAssistantRequest",
     "CreateBlock",
-    "DeleteAssistantFileResponse",
-    "DeleteAssistantResponse",
     "E2BSandboxConfig",
     "EmbeddingConfig",
     "EmbeddingConfigEmbeddingEndpointType",
     "FileMetadata",
-    "FunctionCallInput",
-    "FunctionCallOutput",
-    "FunctionSchema",
+    "Function",
+    "FunctionDefinition",
+    "FunctionTool",
     "Health",
     "HttpValidationError",
     "InitToolRule",
@@ -178,45 +131,26 @@ __all__ = [
     "JobStatus",
     "JobType",
     "LettaMessageUnion",
-    "LettaMessageUnion_AssistantMessage",
-    "LettaMessageUnion_ReasoningMessage",
-    "LettaMessageUnion_SystemMessage",
-    "LettaMessageUnion_ToolCallMessage",
-    "LettaMessageUnion_ToolReturnMessage",
-    "LettaMessageUnion_UserMessage",
     "LettaRequest",
     "LettaRequestConfig",
     "LettaResponse",
-    "LettaSchemasLettaMessageToolCall",
-    "LettaSchemasMessageMessage",
-    "LettaSchemasOpenaiChatCompletionRequestTool",
-    "LettaSchemasOpenaiChatCompletionRequestToolCall",
-    "LettaSchemasOpenaiChatCompletionRequestToolCallFunction",
-    "LettaSchemasOpenaiChatCompletionResponseMessage",
-    "LettaSchemasOpenaiChatCompletionResponseToolCall",
-    "LettaSchemasOpenaiChatCompletionsToolCallFunction",
-    "LettaSchemasOpenaiChatCompletionsToolCallInput",
-    "LettaSchemasOpenaiChatCompletionsToolCallOutput",
-    "LettaSchemasToolTool",
     "LettaUsageStatistics",
     "LlmConfig",
     "LlmConfigModelEndpointType",
     "LocalSandboxConfig",
-    "LogProbToken",
     "Memory",
-    "MessageContentLogProb",
+    "Message",
     "MessageCreate",
+    "MessageCreateContent",
     "MessageCreateRole",
     "MessageRole",
     "NotFoundErrorBody",
     "NotFoundErrorBodyMessage",
-    "OpenAiAssistant",
     "Organization",
     "OrganizationCreate",
     "Passage",
     "Provider",
     "ReasoningMessage",
-    "ResponseFormat",
     "Run",
     "SandboxConfig",
     "SandboxConfigCreate",
@@ -228,17 +162,16 @@ __all__ = [
     "SandboxEnvironmentVariableUpdate",
     "SandboxType",
     "Source",
-    "SystemMessageInput",
-    "SystemMessageOutput",
+    "SystemMessage",
+    "SystemMessageContent",
     "TerminalToolRule",
+    "TextContent",
+    "Tool",
+    "ToolCall",
     "ToolCallDelta",
-    "ToolCallFunctionOutput",
     "ToolCallMessage",
     "ToolCallMessageToolCall",
     "ToolCreate",
-    "ToolFunctionChoice",
-    "ToolInput",
-    "ToolMessage",
     "ToolReturnMessage",
     "ToolReturnMessageStatus",
     "ToolRuleType",
@@ -246,9 +179,8 @@ __all__ = [
     "UsageStatistics",
     "User",
     "UserCreate",
-    "UserMessageInput",
-    "UserMessageInputContent",
-    "UserMessageOutput",
+    "UserMessage",
+    "UserMessageContent",
     "UserUpdate",
     "ValidationError",
     "ValidationErrorLocItem",

letta_client/types/agent_state.py

@@ -8,10 +8,8 @@ from .agent_state_tool_rules_item import AgentStateToolRulesItem
 from .agent_type import AgentType
 from .llm_config import LlmConfig
 from .embedding_config import EmbeddingConfig
-import typing_extensions
-from ..core.serialization import FieldMetadata
 from .memory import Memory
-from .letta_schemas_tool_tool import LettaSchemasToolTool
+from .tool import Tool
 from .source import Source
 from .agent_environment_variable import AgentEnvironmentVariable
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
@@ -98,9 +96,7 @@ class AgentState(UncheckedBaseModel):
     The description of the agent.
     """
 
-    metadata: typing_extensions.Annotated[
-        typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="metadata_")
-    ] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     The metadata of the agent.
     """
@@ -110,7 +106,7 @@ class AgentState(UncheckedBaseModel):
     The in-context memory of the agent.
     """
 
-    tools: typing.List[LettaSchemasToolTool] = pydantic.Field()
+    tools: typing.List[Tool] = pydantic.Field()
     """
     The tools used by the agent.
     """

letta_client/types/{letta_schemas_openai_chat_completion_response_tool_call.py → assistant_message.py}

@@ -1,16 +1,18 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.unchecked_base_model import UncheckedBaseModel
+import datetime as dt
 import typing
-from .function_call_output import FunctionCallOutput
+from .assistant_message_content import AssistantMessageContent
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class LettaSchemasOpenaiChatCompletionResponseToolCall(UncheckedBaseModel):
+class AssistantMessage(UncheckedBaseModel):
     id: str
-    type: typing.Optional[typing.Literal["function"]] = None
-    function: FunctionCallOutput
+    date: dt.datetime
+    message_type: typing.Literal["assistant_message"] = "assistant_message"
+    content: AssistantMessageContent
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

letta_client/types/assistant_message_content.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .text_content import TextContent
+
+AssistantMessageContent = typing.Union[str, typing.List[TextContent]]

letta_client/types/block.py

@@ -3,8 +3,6 @@
 from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 import typing
-import typing_extensions
-from ..core.serialization import FieldMetadata
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -20,7 +18,7 @@ class Block(UncheckedBaseModel):
         label (str): The label of the block (e.g. 'human', 'persona'). This defines a category for the block.
         template_name (str): The name of the block template (if it is a template).
         description (str): Description of the block.
-        metadata_ (Dict): Metadata of the block.
+        metadata (Dict): Metadata of the block.
         user_id (str): The unique identifier of the user associated with the block.
     """
 
@@ -54,9 +52,7 @@ class Block(UncheckedBaseModel):
     Description of the block.
     """
 
-    metadata: typing_extensions.Annotated[
-        typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="metadata_")
-    ] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Metadata of the block.
     """

letta_client/types/block_update.py

@@ -3,8 +3,6 @@
 from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 import pydantic
-import typing_extensions
-from ..core.serialization import FieldMetadata
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -43,9 +41,7 @@ class BlockUpdate(UncheckedBaseModel):
     Description of the block.
     """
 
-    metadata: typing_extensions.Annotated[
-        typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="metadata_")
-    ] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Metadata of the block.
     """

letta_client/types/{assistant_message_output.py → chat_completion_message_tool_call.py}

@@ -1,16 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.unchecked_base_model import UncheckedBaseModel
-import datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .function import Function
 import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class AssistantMessageOutput(UncheckedBaseModel):
+class ChatCompletionMessageToolCall(UncheckedBaseModel):
     id: str
-    date: dt.datetime
-    assistant_message: str
+    function: Function
+    type: typing.Literal["function"] = "function"
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

letta_client/types/context_window_overview.py

@@ -3,8 +3,8 @@
 from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 import typing
-from .letta_schemas_openai_chat_completion_request_tool import LettaSchemasOpenaiChatCompletionRequestTool
-from .letta_schemas_message_message import LettaSchemasMessageMessage
+from .function_tool import FunctionTool
+from .message import Message
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -83,9 +83,7 @@ class ContextWindowOverview(UncheckedBaseModel):
     The number of tokens in the functions definitions.
     """
 
-    functions_definitions: typing.Optional[typing.List[LettaSchemasOpenaiChatCompletionRequestTool]] = pydantic.Field(
-        default=None
-    )
+    functions_definitions: typing.Optional[typing.List[FunctionTool]] = pydantic.Field(default=None)
     """
     The content of the functions definitions.
     """
@@ -95,7 +93,7 @@ class ContextWindowOverview(UncheckedBaseModel):
     The number of tokens in the messages list.
     """
 
-    messages: typing.List[LettaSchemasMessageMessage] = pydantic.Field()
+    messages: typing.List[Message] = pydantic.Field()
     """
     The messages in the context window.
     """

letta_client/types/create_block.py

@@ -3,8 +3,6 @@
 from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 import typing
-import typing_extensions
-from ..core.serialization import FieldMetadata
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -39,9 +37,7 @@ class CreateBlock(UncheckedBaseModel):
     Description of the block.
     """
 
-    metadata: typing_extensions.Annotated[
-        typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="metadata_")
-    ] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     Metadata of the block.
     """

letta_client/types/{function_call_output.py → function.py}

@@ -6,7 +6,7 @@ import typing
 import pydantic
 
 
-class FunctionCallOutput(UncheckedBaseModel):
+class Function(UncheckedBaseModel):
     arguments: str
     name: str
 

letta_client/types/{function_schema.py → function_definition.py}

@@ -6,10 +6,11 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class FunctionSchema(UncheckedBaseModel):
+class FunctionDefinition(UncheckedBaseModel):
     name: str
     description: typing.Optional[str] = None
     parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    strict: typing.Optional[bool] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

letta_client/types/{function_call_input.py → function_tool.py}

@@ -1,13 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.unchecked_base_model import UncheckedBaseModel
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .function_definition import FunctionDefinition
 import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class FunctionCallInput(UncheckedBaseModel):
-    name: str
+class FunctionTool(UncheckedBaseModel):
+    function: FunctionDefinition
+    type: typing.Literal["function"] = "function"
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

letta_client/types/job.py

@@ -5,8 +5,6 @@ import typing
 import pydantic
 import datetime as dt
 from .job_status import JobStatus
-import typing_extensions
-from ..core.serialization import FieldMetadata
 from .job_type import JobType
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
@@ -53,9 +51,7 @@ class Job(UncheckedBaseModel):
     The unix timestamp of when the job was completed.
     """
 
-    metadata: typing_extensions.Annotated[
-        typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="metadata_")
-    ] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     The metadata of the job.
     """

letta_client/types/letta_message_union.py

@@ -1,125 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
-import datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from .tool_call_message_tool_call import ToolCallMessageToolCall
-from .tool_return_message_status import ToolReturnMessageStatus
-import typing_extensions
-from ..core.unchecked_base_model import UnionMetadata
-
-
-class LettaMessageUnion_SystemMessage(UncheckedBaseModel):
-    message_type: typing.Literal["system_message"] = "system_message"
-    id: str
-    date: dt.datetime
-    message: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaMessageUnion_UserMessage(UncheckedBaseModel):
-    message_type: typing.Literal["user_message"] = "user_message"
-    id: str
-    date: dt.datetime
-    message: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaMessageUnion_ReasoningMessage(UncheckedBaseModel):
-    message_type: typing.Literal["reasoning_message"] = "reasoning_message"
-    id: str
-    date: dt.datetime
-    reasoning: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaMessageUnion_ToolCallMessage(UncheckedBaseModel):
-    message_type: typing.Literal["tool_call_message"] = "tool_call_message"
-    id: str
-    date: dt.datetime
-    tool_call: ToolCallMessageToolCall
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaMessageUnion_ToolReturnMessage(UncheckedBaseModel):
-    message_type: typing.Literal["tool_return_message"] = "tool_return_message"
-    id: str
-    date: dt.datetime
-    tool_return: str
-    status: ToolReturnMessageStatus
-    tool_call_id: str
-    stdout: typing.Optional[typing.List[str]] = None
-    stderr: typing.Optional[typing.List[str]] = None
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaMessageUnion_AssistantMessage(UncheckedBaseModel):
-    message_type: typing.Literal["assistant_message"] = "assistant_message"
-    id: str
-    date: dt.datetime
-    assistant_message: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-LettaMessageUnion = typing_extensions.Annotated[
-    typing.Union[
-        LettaMessageUnion_SystemMessage,
-        LettaMessageUnion_UserMessage,
-        LettaMessageUnion_ReasoningMessage,
-        LettaMessageUnion_ToolCallMessage,
-        LettaMessageUnion_ToolReturnMessage,
-        LettaMessageUnion_AssistantMessage,
-    ],
-    UnionMetadata(discriminant="message_type"),
+from .system_message import SystemMessage
+from .user_message import UserMessage
+from .reasoning_message import ReasoningMessage
+from .tool_call_message import ToolCallMessage
+from .tool_return_message import ToolReturnMessage
+from .assistant_message import AssistantMessage
+
+LettaMessageUnion = typing.Union[
+    SystemMessage, UserMessage, ReasoningMessage, ToolCallMessage, ToolReturnMessage, AssistantMessage
 ]

letta_client/types/letta_usage_statistics.py

@@ -17,6 +17,7 @@ class LettaUsageStatistics(UncheckedBaseModel):
         step_count (int): The number of steps taken by the agent.
     """
 
+    message_type: typing.Optional[typing.Literal["usage_statistics"]] = None
     completion_tokens: typing.Optional[int] = pydantic.Field(default=None)
     """
     The number of tokens generated by the agent.