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

letta/__init__.py

@@ -4,6 +4,7 @@ from .types import (
     ActionModel,
     ActionParametersModel,
     ActionResponseModel,
+    AgentEnvironmentVariable,
     AgentState,
     AgentStateToolRulesItem,
     AgentType,
@@ -50,6 +51,13 @@ from .types import (
     JobStatus,
     LettaRequest,
     LettaResponse,
+    LettaResponseMessagesItem,
+    LettaResponseMessagesItem_AssistantMessage,
+    LettaResponseMessagesItem_ReasoningMessage,
+    LettaResponseMessagesItem_SystemMessage,
+    LettaResponseMessagesItem_ToolCallMessage,
+    LettaResponseMessagesItem_ToolReturnMessage,
+    LettaResponseMessagesItem_UserMessage,
     LettaSchemasLettaMessageToolCall,
     LettaSchemasMessageMessage,
     LettaSchemasOpenaiChatCompletionRequestTool,
@@ -121,12 +129,15 @@ from .agents import (
     AgentsMigrateResponse,
     AgentsSearchDeployedAgentsRequestCombinator,
     AgentsSearchDeployedAgentsRequestSearchItem,
-    AgentsSearchDeployedAgentsRequestSearchItemDirection,
-    AgentsSearchDeployedAgentsRequestSearchItemDirectionDirection,
-    AgentsSearchDeployedAgentsRequestSearchItemDirectionValue,
-    AgentsSearchDeployedAgentsRequestSearchItemOperator,
-    AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator,
-    AgentsSearchDeployedAgentsRequestSearchItemZero,
+    AgentsSearchDeployedAgentsRequestSearchItemName,
+    AgentsSearchDeployedAgentsRequestSearchItemNameOperator,
+    AgentsSearchDeployedAgentsRequestSearchItemOrderBy,
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection,
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByValue,
+    AgentsSearchDeployedAgentsRequestSearchItemVersion,
+    AgentsSearchDeployedAgentsRequestSearchItem_Name,
+    AgentsSearchDeployedAgentsRequestSearchItem_OrderBy,
+    AgentsSearchDeployedAgentsRequestSearchItem_Version,
     CreateAgentRequestToolRulesItem,
     UpdateAgentToolRulesItem,
 )
@@ -138,6 +149,7 @@ __all__ = [
     "ActionModel",
     "ActionParametersModel",
     "ActionResponseModel",
+    "AgentEnvironmentVariable",
     "AgentState",
     "AgentStateToolRulesItem",
     "AgentType",
@@ -145,12 +157,15 @@ __all__ = [
     "AgentsMigrateResponse",
     "AgentsSearchDeployedAgentsRequestCombinator",
     "AgentsSearchDeployedAgentsRequestSearchItem",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirection",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirectionDirection",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirectionValue",
-    "AgentsSearchDeployedAgentsRequestSearchItemOperator",
-    "AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator",
-    "AgentsSearchDeployedAgentsRequestSearchItemZero",
+    "AgentsSearchDeployedAgentsRequestSearchItemName",
+    "AgentsSearchDeployedAgentsRequestSearchItemNameOperator",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderBy",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderByValue",
+    "AgentsSearchDeployedAgentsRequestSearchItemVersion",
+    "AgentsSearchDeployedAgentsRequestSearchItem_Name",
+    "AgentsSearchDeployedAgentsRequestSearchItem_OrderBy",
+    "AgentsSearchDeployedAgentsRequestSearchItem_Version",
     "AppAuthScheme",
     "AppAuthSchemeAuthMode",
     "AppModel",
@@ -200,6 +215,13 @@ __all__ = [
     "LettaEnvironment",
     "LettaRequest",
     "LettaResponse",
+    "LettaResponseMessagesItem",
+    "LettaResponseMessagesItem_AssistantMessage",
+    "LettaResponseMessagesItem_ReasoningMessage",
+    "LettaResponseMessagesItem_SystemMessage",
+    "LettaResponseMessagesItem_ToolCallMessage",
+    "LettaResponseMessagesItem_ToolReturnMessage",
+    "LettaResponseMessagesItem_UserMessage",
     "LettaSchemasLettaMessageToolCall",
     "LettaSchemasMessageMessage",
     "LettaSchemasOpenaiChatCompletionRequestTool",

letta/agents/__init__.py

@@ -5,33 +5,69 @@ from .types import (
     AgentsMigrateResponse,
     AgentsSearchDeployedAgentsRequestCombinator,
     AgentsSearchDeployedAgentsRequestSearchItem,
-    AgentsSearchDeployedAgentsRequestSearchItemDirection,
-    AgentsSearchDeployedAgentsRequestSearchItemDirectionDirection,
-    AgentsSearchDeployedAgentsRequestSearchItemDirectionValue,
-    AgentsSearchDeployedAgentsRequestSearchItemOperator,
-    AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator,
-    AgentsSearchDeployedAgentsRequestSearchItemZero,
+    AgentsSearchDeployedAgentsRequestSearchItemName,
+    AgentsSearchDeployedAgentsRequestSearchItemNameOperator,
+    AgentsSearchDeployedAgentsRequestSearchItemOrderBy,
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection,
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByValue,
+    AgentsSearchDeployedAgentsRequestSearchItemVersion,
+    AgentsSearchDeployedAgentsRequestSearchItem_Name,
+    AgentsSearchDeployedAgentsRequestSearchItem_OrderBy,
+    AgentsSearchDeployedAgentsRequestSearchItem_Version,
     CreateAgentRequestToolRulesItem,
     UpdateAgentToolRulesItem,
 )
 from . import archival_memory, context, memory, memory_blocks, messages, recall_memory, sources, tools
-from .messages import LettaStreamingResponse, MessagesListResponse, MessagesListResponseItem
+from .messages import (
+    LettaStreamingResponse,
+    LettaStreamingResponse_AssistantMessage,
+    LettaStreamingResponse_ReasoningMessage,
+    LettaStreamingResponse_SystemMessage,
+    LettaStreamingResponse_ToolCallMessage,
+    LettaStreamingResponse_ToolReturnMessage,
+    LettaStreamingResponse_UsageStatistics,
+    LettaStreamingResponse_UserMessage,
+    MessagesListResponse,
+    MessagesListResponseItem,
+    MessagesListResponseItem_AssistantMessage,
+    MessagesListResponseItem_ReasoningMessage,
+    MessagesListResponseItem_SystemMessage,
+    MessagesListResponseItem_ToolCallMessage,
+    MessagesListResponseItem_ToolReturnMessage,
+    MessagesListResponseItem_UserMessage,
+)
 
 __all__ = [
     "AgentsGetAgentVariablesResponse",
     "AgentsMigrateResponse",
     "AgentsSearchDeployedAgentsRequestCombinator",
     "AgentsSearchDeployedAgentsRequestSearchItem",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirection",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirectionDirection",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirectionValue",
-    "AgentsSearchDeployedAgentsRequestSearchItemOperator",
-    "AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator",
-    "AgentsSearchDeployedAgentsRequestSearchItemZero",
+    "AgentsSearchDeployedAgentsRequestSearchItemName",
+    "AgentsSearchDeployedAgentsRequestSearchItemNameOperator",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderBy",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderByValue",
+    "AgentsSearchDeployedAgentsRequestSearchItemVersion",
+    "AgentsSearchDeployedAgentsRequestSearchItem_Name",
+    "AgentsSearchDeployedAgentsRequestSearchItem_OrderBy",
+    "AgentsSearchDeployedAgentsRequestSearchItem_Version",
     "CreateAgentRequestToolRulesItem",
     "LettaStreamingResponse",
+    "LettaStreamingResponse_AssistantMessage",
+    "LettaStreamingResponse_ReasoningMessage",
+    "LettaStreamingResponse_SystemMessage",
+    "LettaStreamingResponse_ToolCallMessage",
+    "LettaStreamingResponse_ToolReturnMessage",
+    "LettaStreamingResponse_UsageStatistics",
+    "LettaStreamingResponse_UserMessage",
     "MessagesListResponse",
     "MessagesListResponseItem",
+    "MessagesListResponseItem_AssistantMessage",
+    "MessagesListResponseItem_ReasoningMessage",
+    "MessagesListResponseItem_SystemMessage",
+    "MessagesListResponseItem_ToolCallMessage",
+    "MessagesListResponseItem_ToolReturnMessage",
+    "MessagesListResponseItem_UserMessage",
     "UpdateAgentToolRulesItem",
     "archival_memory",
     "context",

letta/agents/client.py

@@ -68,6 +68,8 @@ class AgentsClient:
         name: typing.Optional[str] = None,
         tags: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         match_all_tags: typing.Optional[bool] = None,
+        cursor: typing.Optional[int] = None,
+        limit: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[AgentState]:
         """
@@ -85,6 +87,12 @@ class AgentsClient:
         match_all_tags : typing.Optional[bool]
             If True, only returns agents that match ALL given tags. Otherwise, return agents that have ANY of the passed in tags.
 
+        cursor : typing.Optional[int]
+            Cursor for pagination
+
+        limit : typing.Optional[int]
+            Limit for pagination
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -109,6 +117,8 @@ class AgentsClient:
                 "name": name,
                 "tags": tags,
                 "match_all_tags": match_all_tags,
+                "cursor": cursor,
+                "limit": limit,
             },
             request_options=request_options,
         )
@@ -161,6 +171,7 @@ class AgentsClient:
         embedding_chunk_size: typing.Optional[int] = OMIT,
         from_template: typing.Optional[str] = OMIT,
         project_id: typing.Optional[str] = OMIT,
+        tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         user_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
@@ -235,6 +246,9 @@ class AgentsClient:
         project_id : typing.Optional[str]
             The project id that the agent will be associated with.
 
+        tool_exec_environment_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
+            The environment variables for tool execution specific to this agent.
+
         user_id : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
@@ -297,6 +311,7 @@ class AgentsClient:
                 "embedding_chunk_size": embedding_chunk_size,
                 "from_template": from_template,
                 "project_id": project_id,
+                "tool_exec_environment_variables": tool_exec_environment_variables,
                 "user_id": user_id,
             },
             headers={
@@ -459,6 +474,7 @@ class AgentsClient:
         message_ids: typing.Optional[typing.Sequence[str]] = OMIT,
         description: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
         """
@@ -504,6 +520,9 @@ class AgentsClient:
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             The metadata of the agent.
 
+        tool_exec_environment_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
+            The environment variables for tool execution specific to this agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -545,6 +564,7 @@ class AgentsClient:
                 "message_ids": message_ids,
                 "description": description,
                 "metadata_": metadata,
+                "tool_exec_environment_variables": tool_exec_environment_variables,
             },
             headers={
                 "content-type": "application/json",
@@ -1289,6 +1309,8 @@ class AsyncAgentsClient:
         name: typing.Optional[str] = None,
         tags: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         match_all_tags: typing.Optional[bool] = None,
+        cursor: typing.Optional[int] = None,
+        limit: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[AgentState]:
         """
@@ -1306,6 +1328,12 @@ class AsyncAgentsClient:
         match_all_tags : typing.Optional[bool]
             If True, only returns agents that match ALL given tags. Otherwise, return agents that have ANY of the passed in tags.
 
+        cursor : typing.Optional[int]
+            Cursor for pagination
+
+        limit : typing.Optional[int]
+            Limit for pagination
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1338,6 +1366,8 @@ class AsyncAgentsClient:
                 "name": name,
                 "tags": tags,
                 "match_all_tags": match_all_tags,
+                "cursor": cursor,
+                "limit": limit,
             },
             request_options=request_options,
         )
@@ -1390,6 +1420,7 @@ class AsyncAgentsClient:
         embedding_chunk_size: typing.Optional[int] = OMIT,
         from_template: typing.Optional[str] = OMIT,
         project_id: typing.Optional[str] = OMIT,
+        tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         user_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
@@ -1464,6 +1495,9 @@ class AsyncAgentsClient:
         project_id : typing.Optional[str]
             The project id that the agent will be associated with.
 
+        tool_exec_environment_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
+            The environment variables for tool execution specific to this agent.
+
         user_id : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
@@ -1534,6 +1568,7 @@ class AsyncAgentsClient:
                 "embedding_chunk_size": embedding_chunk_size,
                 "from_template": from_template,
                 "project_id": project_id,
+                "tool_exec_environment_variables": tool_exec_environment_variables,
                 "user_id": user_id,
             },
             headers={
@@ -1712,6 +1747,7 @@ class AsyncAgentsClient:
         message_ids: typing.Optional[typing.Sequence[str]] = OMIT,
         description: typing.Optional[str] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        tool_exec_environment_variables: typing.Optional[typing.Dict[str, typing.Optional[str]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AgentState:
         """
@@ -1757,6 +1793,9 @@ class AsyncAgentsClient:
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             The metadata of the agent.
 
+        tool_exec_environment_variables : typing.Optional[typing.Dict[str, typing.Optional[str]]]
+            The environment variables for tool execution specific to this agent.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1806,6 +1845,7 @@ class AsyncAgentsClient:
                 "message_ids": message_ids,
                 "description": description,
                 "metadata_": metadata,
+                "tool_exec_environment_variables": tool_exec_environment_variables,
             },
             headers={
                 "content-type": "application/json",

letta/agents/messages/__init__.py

@@ -1,5 +1,39 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import LettaStreamingResponse, MessagesListResponse, MessagesListResponseItem
+from .types import (
+    LettaStreamingResponse,
+    LettaStreamingResponse_AssistantMessage,
+    LettaStreamingResponse_ReasoningMessage,
+    LettaStreamingResponse_SystemMessage,
+    LettaStreamingResponse_ToolCallMessage,
+    LettaStreamingResponse_ToolReturnMessage,
+    LettaStreamingResponse_UsageStatistics,
+    LettaStreamingResponse_UserMessage,
+    MessagesListResponse,
+    MessagesListResponseItem,
+    MessagesListResponseItem_AssistantMessage,
+    MessagesListResponseItem_ReasoningMessage,
+    MessagesListResponseItem_SystemMessage,
+    MessagesListResponseItem_ToolCallMessage,
+    MessagesListResponseItem_ToolReturnMessage,
+    MessagesListResponseItem_UserMessage,
+)
 
-__all__ = ["LettaStreamingResponse", "MessagesListResponse", "MessagesListResponseItem"]
+__all__ = [
+    "LettaStreamingResponse",
+    "LettaStreamingResponse_AssistantMessage",
+    "LettaStreamingResponse_ReasoningMessage",
+    "LettaStreamingResponse_SystemMessage",
+    "LettaStreamingResponse_ToolCallMessage",
+    "LettaStreamingResponse_ToolReturnMessage",
+    "LettaStreamingResponse_UsageStatistics",
+    "LettaStreamingResponse_UserMessage",
+    "MessagesListResponse",
+    "MessagesListResponseItem",
+    "MessagesListResponseItem_AssistantMessage",
+    "MessagesListResponseItem_ReasoningMessage",
+    "MessagesListResponseItem_SystemMessage",
+    "MessagesListResponseItem_ToolCallMessage",
+    "MessagesListResponseItem_ToolReturnMessage",
+    "MessagesListResponseItem_UserMessage",
+]

letta/agents/messages/client.py

@@ -345,6 +345,25 @@ class MessagesClient:
         ------
         typing.Iterator[LettaStreamingResponse]
             Successful response
+
+        Examples
+        --------
+        from letta import Letta, MessageCreate
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        response = client.agents.messages.stream(
+            agent_id="agent_id",
+            messages=[
+                MessageCreate(
+                    role="user",
+                    text="text",
+                )
+            ],
+        )
+        for chunk in response:
+            yield chunk
         """
         with self._client_wrapper.httpx_client.stream(
             f"v1/agents/{jsonable_encoder(agent_id)}/messages/stream",
@@ -737,6 +756,33 @@ class AsyncMessagesClient:
         ------
         typing.AsyncIterator[LettaStreamingResponse]
             Successful response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta import AsyncLetta, MessageCreate
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            response = await client.agents.messages.stream(
+                agent_id="agent_id",
+                messages=[
+                    MessageCreate(
+                        role="user",
+                        text="text",
+                    )
+                ],
+            )
+            async for chunk in response:
+                yield chunk
+
+
+        asyncio.run(main())
         """
         async with self._client_wrapper.httpx_client.stream(
             f"v1/agents/{jsonable_encoder(agent_id)}/messages/stream",

letta/agents/messages/types/__init__.py

@@ -1,7 +1,41 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .letta_streaming_response import LettaStreamingResponse
+from .letta_streaming_response import (
+    LettaStreamingResponse,
+    LettaStreamingResponse_AssistantMessage,
+    LettaStreamingResponse_ReasoningMessage,
+    LettaStreamingResponse_SystemMessage,
+    LettaStreamingResponse_ToolCallMessage,
+    LettaStreamingResponse_ToolReturnMessage,
+    LettaStreamingResponse_UsageStatistics,
+    LettaStreamingResponse_UserMessage,
+)
 from .messages_list_response import MessagesListResponse
-from .messages_list_response_item import MessagesListResponseItem
+from .messages_list_response_item import (
+    MessagesListResponseItem,
+    MessagesListResponseItem_AssistantMessage,
+    MessagesListResponseItem_ReasoningMessage,
+    MessagesListResponseItem_SystemMessage,
+    MessagesListResponseItem_ToolCallMessage,
+    MessagesListResponseItem_ToolReturnMessage,
+    MessagesListResponseItem_UserMessage,
+)
 
-__all__ = ["LettaStreamingResponse", "MessagesListResponse", "MessagesListResponseItem"]
+__all__ = [
+    "LettaStreamingResponse",
+    "LettaStreamingResponse_AssistantMessage",
+    "LettaStreamingResponse_ReasoningMessage",
+    "LettaStreamingResponse_SystemMessage",
+    "LettaStreamingResponse_ToolCallMessage",
+    "LettaStreamingResponse_ToolReturnMessage",
+    "LettaStreamingResponse_UsageStatistics",
+    "LettaStreamingResponse_UserMessage",
+    "MessagesListResponse",
+    "MessagesListResponseItem",
+    "MessagesListResponseItem_AssistantMessage",
+    "MessagesListResponseItem_ReasoningMessage",
+    "MessagesListResponseItem_SystemMessage",
+    "MessagesListResponseItem_ToolCallMessage",
+    "MessagesListResponseItem_ToolReturnMessage",
+    "MessagesListResponseItem_UserMessage",
+]

letta/agents/messages/types/letta_streaming_response.py

@@ -1,20 +1,138 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+from ....core.pydantic_utilities import UniversalBaseModel
 import typing
-from ....types.system_message_output import SystemMessageOutput
-from ....types.user_message_output import UserMessageOutput
-from ....types.reasoning_message import ReasoningMessage
-from ....types.tool_call_message import ToolCallMessage
-from ....types.tool_return_message import ToolReturnMessage
-from ....types.assistant_message_output import AssistantMessageOutput
-from ....types.letta_usage_statistics import LettaUsageStatistics
+import datetime as dt
+from ....core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ....types.tool_call_message_tool_call import ToolCallMessageToolCall
+from ....types.tool_return_message_status import ToolReturnMessageStatus
+
+
+class LettaStreamingResponse_SystemMessage(UniversalBaseModel):
+    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 LettaStreamingResponse_UserMessage(UniversalBaseModel):
+    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 LettaStreamingResponse_ReasoningMessage(UniversalBaseModel):
+    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 LettaStreamingResponse_ToolCallMessage(UniversalBaseModel):
+    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 LettaStreamingResponse_ToolReturnMessage(UniversalBaseModel):
+    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 LettaStreamingResponse_AssistantMessage(UniversalBaseModel):
+    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
+
+
+class LettaStreamingResponse_UsageStatistics(UniversalBaseModel):
+    message_type: typing.Literal["usage_statistics"] = "usage_statistics"
+    completion_tokens: typing.Optional[int] = None
+    prompt_tokens: typing.Optional[int] = None
+    total_tokens: typing.Optional[int] = None
+    step_count: typing.Optional[int] = 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
+
 
 LettaStreamingResponse = typing.Union[
-    SystemMessageOutput,
-    UserMessageOutput,
-    ReasoningMessage,
-    ToolCallMessage,
-    ToolReturnMessage,
-    AssistantMessageOutput,
-    LettaUsageStatistics,
+    LettaStreamingResponse_SystemMessage,
+    LettaStreamingResponse_UserMessage,
+    LettaStreamingResponse_ReasoningMessage,
+    LettaStreamingResponse_ToolCallMessage,
+    LettaStreamingResponse_ToolReturnMessage,
+    LettaStreamingResponse_AssistantMessage,
+    LettaStreamingResponse_UsageStatistics,
 ]