letta-client 0.1.4__py3-none-any.whl → 0.1.9__py3-none-any.whl

letta/types/file_metadata.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 import pydantic
 import datetime as dt
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class FileMetadata(UniversalBaseModel):
+class FileMetadata(UncheckedBaseModel):
     """
     Representation of a single FileMetadata
     """
@@ -17,11 +17,6 @@ class FileMetadata(UniversalBaseModel):
     The human-friendly ID of the File
     """
 
-    organization_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The unique identifier of the organization associated with the document.
-    """
-
     source_id: str = pydantic.Field()
     """
     The unique identifier of the source associated with the document.

letta/types/function_call_input.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
 
 
-class FunctionCallInput(UniversalBaseModel):
+class FunctionCallInput(UncheckedBaseModel):
     name: str
 
     if IS_PYDANTIC_V2:

letta/types/function_call_output.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
 
 
-class FunctionCallOutput(UniversalBaseModel):
+class FunctionCallOutput(UncheckedBaseModel):
     arguments: str
     name: str
 

letta/types/function_schema.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class FunctionSchema(UniversalBaseModel):
+class FunctionSchema(UncheckedBaseModel):
     name: str
     description: typing.Optional[str] = None
     parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None

letta/types/health.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
 
 
-class Health(UniversalBaseModel):
+class Health(UncheckedBaseModel):
     """
     Health check response body
     """

letta/types/http_validation_error.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from .validation_error import ValidationError
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class HttpValidationError(UniversalBaseModel):
+class HttpValidationError(UncheckedBaseModel):
     detail: typing.Optional[typing.List[ValidationError]] = None
 
     if IS_PYDANTIC_V2:

letta/types/init_tool_rule.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 import typing
 from .tool_rule_type import ToolRuleType
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class InitToolRule(UniversalBaseModel):
+class InitToolRule(UncheckedBaseModel):
     """
     Represents the initial tool rule configuration.
     """

letta/types/internal_server_error_body.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class InternalServerErrorBody(UniversalBaseModel):
+class InternalServerErrorBody(UncheckedBaseModel):
     message: typing.Literal["Failed to create agent template"] = "Failed to create agent template"
 
     if IS_PYDANTIC_V2:

letta/types/job.py

@@ -1,6 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 import pydantic
 import datetime as dt
@@ -10,16 +10,16 @@ from ..core.serialization import FieldMetadata
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class Job(UniversalBaseModel):
+class Job(UncheckedBaseModel):
     """
     Representation of offline jobs, used for tracking status of data loading tasks (involving parsing and embedding files).
 
     Parameters:
-    id (str): The unique identifier of the job.
-    status (JobStatus): The status of the job.
-    created_at (datetime): The unix timestamp of when the job was created.
-    completed_at (datetime): The unix timestamp of when the job was completed.
-    user_id (str): The unique identifier of the user associated with the.
+        id (str): The unique identifier of the job.
+        status (JobStatus): The status of the job.
+        created_at (datetime): The unix timestamp of when the job was created.
+        completed_at (datetime): The unix timestamp of when the job was completed.
+        user_id (str): The unique identifier of the user associated with the.
     """
 
     created_by_id: typing.Optional[str] = pydantic.Field(default=None)
@@ -64,11 +64,6 @@ class Job(UniversalBaseModel):
     The human-friendly ID of the Job
     """
 
-    user_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The unique identifier of the user associated with the job.
-    """
-
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

letta/types/letta_request.py

@@ -1,18 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from .message_create import MessageCreate
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaRequest(UniversalBaseModel):
+class LettaRequest(UncheckedBaseModel):
     messages: typing.List[MessageCreate] = pydantic.Field()
     """
     The messages to be sent to the agent.
     """
 
+    use_assistant_message: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether the server should parse specific tool call arguments (default `send_message`) as `AssistantMessage` objects.
+    """
+
     assistant_message_tool_name: typing.Optional[str] = pydantic.Field(default=None)
     """
     The name of the designated message tool.

letta/types/letta_response.py

@@ -1,23 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
+from .letta_response_messages_item import LettaResponseMessagesItem
 import pydantic
 from .letta_usage_statistics import LettaUsageStatistics
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaResponse(UniversalBaseModel):
+class LettaResponse(UncheckedBaseModel):
     """
     Response object from an agent interaction, consisting of the new messages generated by the agent and usage statistics.
     The type of the returned messages can be either `Message` or `LettaMessage`, depending on what was specified in the request.
 
     Attributes:
-    messages (List[Union[Message, LettaMessage]]): The messages returned by the agent.
-    usage (LettaUsageStatistics): The usage statistics
+        messages (List[Union[Message, LettaMessage]]): The messages returned by the agent.
+        usage (LettaUsageStatistics): The usage statistics
     """
 
-    messages: typing.List[typing.Optional[typing.Any]] = pydantic.Field()
+    messages: typing.List[LettaResponseMessagesItem] = pydantic.Field()
     """
     The messages returned by the agent.
     """

letta/types/letta_response_messages_item.py

@@ -0,0 +1,125 @@
+# 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 LettaResponseMessagesItem_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 LettaResponseMessagesItem_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 LettaResponseMessagesItem_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 LettaResponseMessagesItem_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 LettaResponseMessagesItem_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 LettaResponseMessagesItem_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
+
+
+LettaResponseMessagesItem = typing_extensions.Annotated[
+    typing.Union[
+        LettaResponseMessagesItem_SystemMessage,
+        LettaResponseMessagesItem_UserMessage,
+        LettaResponseMessagesItem_ReasoningMessage,
+        LettaResponseMessagesItem_ToolCallMessage,
+        LettaResponseMessagesItem_ToolReturnMessage,
+        LettaResponseMessagesItem_AssistantMessage,
+    ],
+    UnionMetadata(discriminant="message_type"),
+]

letta/types/letta_schemas_letta_message_tool_call.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
 
 
-class LettaSchemasLettaMessageToolCall(UniversalBaseModel):
+class LettaSchemasLettaMessageToolCall(UncheckedBaseModel):
     name: str
     arguments: str
     tool_call_id: str

letta/types/letta_schemas_message_message.py

@@ -1,6 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 import pydantic
 import datetime as dt
@@ -9,21 +9,21 @@ from .letta_schemas_openai_chat_completions_tool_call_output import LettaSchemas
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaSchemasMessageMessage(UniversalBaseModel):
+class LettaSchemasMessageMessage(UncheckedBaseModel):
     """
     Letta's internal representation of a message. Includes methods to convert to/from LLM provider formats.
 
     Attributes:
-    id (str): The unique identifier of the message.
-    role (MessageRole): The role of the participant.
-    text (str): The text of the message.
-    user_id (str): The unique identifier of the user.
-    agent_id (str): The unique identifier of the agent.
-    model (str): The model used to make the function call.
-    name (str): The name of the participant.
-    created_at (datetime): The time the message was created.
-    tool_calls (List[ToolCall]): The list of tool calls requested.
-    tool_call_id (str): The id of the tool call.
+        id (str): The unique identifier of the message.
+        role (MessageRole): The role of the participant.
+        text (str): The text of the message.
+        user_id (str): The unique identifier of the user.
+        agent_id (str): The unique identifier of the agent.
+        model (str): The model used to make the function call.
+        name (str): The name of the participant.
+        created_at (datetime): The time the message was created.
+        tool_calls (List[ToolCall]): The list of tool calls requested.
+        tool_call_id (str): The id of the tool call.
     """
 
     created_by_id: typing.Optional[str] = pydantic.Field(default=None)
@@ -61,11 +61,6 @@ class LettaSchemasMessageMessage(UniversalBaseModel):
     The text of the message.
     """
 
-    organization_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The unique identifier of the organization.
-    """
-
     agent_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     The unique identifier of the agent.

letta/types/letta_schemas_openai_chat_completion_request_tool.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from .function_schema import FunctionSchema
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class LettaSchemasOpenaiChatCompletionRequestTool(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionRequestTool(UncheckedBaseModel):
     type: typing.Optional[typing.Literal["function"]] = None
     function: FunctionSchema
 

letta/types/letta_schemas_openai_chat_completion_request_tool_call.py

@@ -1,6 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from .letta_schemas_openai_chat_completion_request_tool_call_function import (
     LettaSchemasOpenaiChatCompletionRequestToolCallFunction,
@@ -9,7 +9,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class LettaSchemasOpenaiChatCompletionRequestToolCall(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionRequestToolCall(UncheckedBaseModel):
     id: str
     type: typing.Optional[typing.Literal["function"]] = None
     function: LettaSchemasOpenaiChatCompletionRequestToolCallFunction

letta/types/letta_schemas_openai_chat_completion_request_tool_call_function.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
 
 
-class LettaSchemasOpenaiChatCompletionRequestToolCallFunction(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionRequestToolCallFunction(UncheckedBaseModel):
     name: str
     arguments: str
 

letta/types/letta_schemas_openai_chat_completion_response_message.py

@@ -1,6 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from .letta_schemas_openai_chat_completion_response_tool_call import LettaSchemasOpenaiChatCompletionResponseToolCall
 from .function_call_output import FunctionCallOutput
@@ -8,7 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class LettaSchemasOpenaiChatCompletionResponseMessage(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionResponseMessage(UncheckedBaseModel):
     content: typing.Optional[str] = None
     tool_calls: typing.Optional[typing.List[LettaSchemasOpenaiChatCompletionResponseToolCall]] = None
     role: str

letta/types/letta_schemas_openai_chat_completion_response_tool_call.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 from .function_call_output import FunctionCallOutput
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class LettaSchemasOpenaiChatCompletionResponseToolCall(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionResponseToolCall(UncheckedBaseModel):
     id: str
     type: typing.Optional[typing.Literal["function"]] = None
     function: FunctionCallOutput

letta/types/letta_schemas_openai_chat_completions_tool_call_function.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
 
-class LettaSchemasOpenaiChatCompletionsToolCallFunction(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionsToolCallFunction(UncheckedBaseModel):
     name: str = pydantic.Field()
     """
     The name of the function to call

letta/types/letta_schemas_openai_chat_completions_tool_call_input.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 import typing
 from .letta_schemas_openai_chat_completions_tool_call_function import LettaSchemasOpenaiChatCompletionsToolCallFunction
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaSchemasOpenaiChatCompletionsToolCallInput(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionsToolCallInput(UncheckedBaseModel):
     id: str = pydantic.Field()
     """
     The ID of the tool call

letta/types/letta_schemas_openai_chat_completions_tool_call_output.py

@@ -1,13 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import pydantic
 import typing
 from .tool_call_function_output import ToolCallFunctionOutput
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaSchemasOpenaiChatCompletionsToolCallOutput(UniversalBaseModel):
+class LettaSchemasOpenaiChatCompletionsToolCallOutput(UncheckedBaseModel):
     id: str = pydantic.Field()
     """
     The ID of the tool call

letta/types/letta_schemas_tool_tool.py

@@ -1,21 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 import pydantic
+from .tool_type import ToolType
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaSchemasToolTool(UniversalBaseModel):
+class LettaSchemasToolTool(UncheckedBaseModel):
     """
     Representation of a tool, which is a function that can be called by the agent.
 
     Parameters:
-    id (str): The unique identifier of the tool.
-    name (str): The name of the function.
-    tags (List[str]): Metadata tags.
-    source_code (str): The source code of the function.
-    json_schema (Dict): The JSON schema of the function.
+        id (str): The unique identifier of the tool.
+        name (str): The name of the function.
+        tags (List[str]): Metadata tags.
+        source_code (str): The source code of the function.
+        json_schema (Dict): The JSON schema of the function.
     """
 
     id: typing.Optional[str] = pydantic.Field(default=None)
@@ -23,6 +24,11 @@ class LettaSchemasToolTool(UniversalBaseModel):
     The human-friendly ID of the Tool
     """
 
+    tool_type: typing.Optional[ToolType] = pydantic.Field(default=None)
+    """
+    The type of the tool.
+    """
+
     description: typing.Optional[str] = pydantic.Field(default=None)
     """
     The description of the tool.
@@ -38,11 +44,6 @@ class LettaSchemasToolTool(UniversalBaseModel):
     The module of the function.
     """
 
-    organization_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The unique identifier of the organization associated with the tool.
-    """
-
     name: typing.Optional[str] = pydantic.Field(default=None)
     """
     The name of the function.
@@ -53,7 +54,7 @@ class LettaSchemasToolTool(UniversalBaseModel):
     Metadata tags.
     """
 
-    source_code: str = pydantic.Field()
+    source_code: typing.Optional[str] = pydantic.Field(default=None)
     """
     The source code of the function.
     """

letta/types/letta_usage_statistics.py

@@ -1,23 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.unchecked_base_model import UncheckedBaseModel
 import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
-class LettaUsageStatistics(UniversalBaseModel):
+class LettaUsageStatistics(UncheckedBaseModel):
     """
     Usage statistics for the agent interaction.
 
     Attributes:
-    completion_tokens (int): The number of tokens generated by the agent.
-    prompt_tokens (int): The number of tokens in the prompt.
-    total_tokens (int): The total number of tokens processed by the agent.
-    step_count (int): The number of steps taken by the agent.
+        completion_tokens (int): The number of tokens generated by the agent.
+        prompt_tokens (int): The number of tokens in the prompt.
+        total_tokens (int): The total number of tokens processed by the agent.
+        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.