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

letta/agents/messages/types/messages_list_response_item.py

@@ -1,13 +1,120 @@
 # 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
+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 MessagesListResponseItem_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 MessagesListResponseItem_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 MessagesListResponseItem_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 MessagesListResponseItem_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 MessagesListResponseItem_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 MessagesListResponseItem_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
+
 
 MessagesListResponseItem = typing.Union[
-    SystemMessageOutput, UserMessageOutput, ReasoningMessage, ToolCallMessage, ToolReturnMessage, AssistantMessageOutput
+    MessagesListResponseItem_SystemMessage,
+    MessagesListResponseItem_UserMessage,
+    MessagesListResponseItem_ReasoningMessage,
+    MessagesListResponseItem_ToolCallMessage,
+    MessagesListResponseItem_ToolReturnMessage,
+    MessagesListResponseItem_AssistantMessage,
 ]

letta/agents/types/__init__.py

@@ -3,23 +3,28 @@
 from .agents_get_agent_variables_response import AgentsGetAgentVariablesResponse
 from .agents_migrate_response import AgentsMigrateResponse
 from .agents_search_deployed_agents_request_combinator import AgentsSearchDeployedAgentsRequestCombinator
-from .agents_search_deployed_agents_request_search_item import AgentsSearchDeployedAgentsRequestSearchItem
-from .agents_search_deployed_agents_request_search_item_direction import (
-    AgentsSearchDeployedAgentsRequestSearchItemDirection,
+from .agents_search_deployed_agents_request_search_item import (
+    AgentsSearchDeployedAgentsRequestSearchItem,
+    AgentsSearchDeployedAgentsRequestSearchItem_Name,
+    AgentsSearchDeployedAgentsRequestSearchItem_OrderBy,
+    AgentsSearchDeployedAgentsRequestSearchItem_Version,
 )
-from .agents_search_deployed_agents_request_search_item_direction_direction import (
-    AgentsSearchDeployedAgentsRequestSearchItemDirectionDirection,
+from .agents_search_deployed_agents_request_search_item_name import AgentsSearchDeployedAgentsRequestSearchItemName
+from .agents_search_deployed_agents_request_search_item_name_operator import (
+    AgentsSearchDeployedAgentsRequestSearchItemNameOperator,
 )
-from .agents_search_deployed_agents_request_search_item_direction_value import (
-    AgentsSearchDeployedAgentsRequestSearchItemDirectionValue,
+from .agents_search_deployed_agents_request_search_item_order_by import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderBy,
 )
-from .agents_search_deployed_agents_request_search_item_operator import (
-    AgentsSearchDeployedAgentsRequestSearchItemOperator,
+from .agents_search_deployed_agents_request_search_item_order_by_direction import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection,
 )
-from .agents_search_deployed_agents_request_search_item_operator_operator import (
-    AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator,
+from .agents_search_deployed_agents_request_search_item_order_by_value import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByValue,
+)
+from .agents_search_deployed_agents_request_search_item_version import (
+    AgentsSearchDeployedAgentsRequestSearchItemVersion,
 )
-from .agents_search_deployed_agents_request_search_item_zero import AgentsSearchDeployedAgentsRequestSearchItemZero
 from .create_agent_request_tool_rules_item import CreateAgentRequestToolRulesItem
 from .update_agent_tool_rules_item import UpdateAgentToolRulesItem
 
@@ -28,12 +33,15 @@ __all__ = [
     "AgentsMigrateResponse",
     "AgentsSearchDeployedAgentsRequestCombinator",
     "AgentsSearchDeployedAgentsRequestSearchItem",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirection",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirectionDirection",
-    "AgentsSearchDeployedAgentsRequestSearchItemDirectionValue",
-    "AgentsSearchDeployedAgentsRequestSearchItemOperator",
-    "AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator",
-    "AgentsSearchDeployedAgentsRequestSearchItemZero",
+    "AgentsSearchDeployedAgentsRequestSearchItemName",
+    "AgentsSearchDeployedAgentsRequestSearchItemNameOperator",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderBy",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection",
+    "AgentsSearchDeployedAgentsRequestSearchItemOrderByValue",
+    "AgentsSearchDeployedAgentsRequestSearchItemVersion",
+    "AgentsSearchDeployedAgentsRequestSearchItem_Name",
+    "AgentsSearchDeployedAgentsRequestSearchItem_OrderBy",
+    "AgentsSearchDeployedAgentsRequestSearchItem_Version",
     "CreateAgentRequestToolRulesItem",
     "UpdateAgentToolRulesItem",
 ]

letta/agents/types/agents_search_deployed_agents_request_search_item.py

@@ -1,16 +1,67 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+from ...core.pydantic_utilities import UniversalBaseModel
 import typing
-from .agents_search_deployed_agents_request_search_item_zero import AgentsSearchDeployedAgentsRequestSearchItemZero
-from .agents_search_deployed_agents_request_search_item_operator import (
-    AgentsSearchDeployedAgentsRequestSearchItemOperator,
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from .agents_search_deployed_agents_request_search_item_name_operator import (
+    AgentsSearchDeployedAgentsRequestSearchItemNameOperator,
 )
-from .agents_search_deployed_agents_request_search_item_direction import (
-    AgentsSearchDeployedAgentsRequestSearchItemDirection,
+from .agents_search_deployed_agents_request_search_item_order_by_value import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByValue,
 )
+from .agents_search_deployed_agents_request_search_item_order_by_direction import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection,
+)
+
+
+class AgentsSearchDeployedAgentsRequestSearchItem_Version(UniversalBaseModel):
+    field: typing.Literal["version"] = "version"
+    value: 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 AgentsSearchDeployedAgentsRequestSearchItem_Name(UniversalBaseModel):
+    field: typing.Literal["name"] = "name"
+    operator: AgentsSearchDeployedAgentsRequestSearchItemNameOperator
+    value: 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 AgentsSearchDeployedAgentsRequestSearchItem_OrderBy(UniversalBaseModel):
+    field: typing.Literal["order_by"] = "order_by"
+    value: AgentsSearchDeployedAgentsRequestSearchItemOrderByValue
+    direction: AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection
+
+    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
+
 
 AgentsSearchDeployedAgentsRequestSearchItem = typing.Union[
-    AgentsSearchDeployedAgentsRequestSearchItemZero,
-    AgentsSearchDeployedAgentsRequestSearchItemOperator,
-    AgentsSearchDeployedAgentsRequestSearchItemDirection,
+    AgentsSearchDeployedAgentsRequestSearchItem_Version,
+    AgentsSearchDeployedAgentsRequestSearchItem_Name,
+    AgentsSearchDeployedAgentsRequestSearchItem_OrderBy,
 ]

letta/agents/types/{agents_search_deployed_agents_request_search_item_operator.py → agents_search_deployed_agents_request_search_item_name.py}

@@ -1,17 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ...core.pydantic_utilities import UniversalBaseModel
-import typing
-from .agents_search_deployed_agents_request_search_item_operator_operator import (
-    AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator,
+from .agents_search_deployed_agents_request_search_item_name_operator import (
+    AgentsSearchDeployedAgentsRequestSearchItemNameOperator,
 )
 from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 import pydantic
 
 
-class AgentsSearchDeployedAgentsRequestSearchItemOperator(UniversalBaseModel):
-    field: typing.Literal["name"] = "name"
-    operator: AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator
+class AgentsSearchDeployedAgentsRequestSearchItemName(UniversalBaseModel):
+    operator: AgentsSearchDeployedAgentsRequestSearchItemNameOperator
     value: str
 
     if IS_PYDANTIC_V2:

letta/agents/types/{agents_search_deployed_agents_request_search_item_operator_operator.py → agents_search_deployed_agents_request_search_item_name_operator.py}

@@ -2,6 +2,6 @@
 
 import typing
 
-AgentsSearchDeployedAgentsRequestSearchItemOperatorOperator = typing.Union[
+AgentsSearchDeployedAgentsRequestSearchItemNameOperator = typing.Union[
     typing.Literal["eq", "neq", "contains"], typing.Any
 ]

letta/agents/types/agents_search_deployed_agents_request_search_item_order_by.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ...core.pydantic_utilities import UniversalBaseModel
+from .agents_search_deployed_agents_request_search_item_order_by_value import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByValue,
+)
+from .agents_search_deployed_agents_request_search_item_order_by_direction import (
+    AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection,
+)
+from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class AgentsSearchDeployedAgentsRequestSearchItemOrderBy(UniversalBaseModel):
+    value: AgentsSearchDeployedAgentsRequestSearchItemOrderByValue
+    direction: AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection
+
+    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

letta/agents/types/agents_search_deployed_agents_request_search_item_order_by_direction.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+AgentsSearchDeployedAgentsRequestSearchItemOrderByDirection = typing.Union[typing.Literal["asc", "desc"], typing.Any]

letta/agents/types/{agents_search_deployed_agents_request_search_item_direction_value.py → agents_search_deployed_agents_request_search_item_order_by_value.py}

@@ -2,6 +2,6 @@
 
 import typing
 
-AgentsSearchDeployedAgentsRequestSearchItemDirectionValue = typing.Union[
+AgentsSearchDeployedAgentsRequestSearchItemOrderByValue = typing.Union[
     typing.Literal["created_at", "updated_at"], typing.Any
 ]

letta/agents/types/{agents_search_deployed_agents_request_search_item_zero.py → agents_search_deployed_agents_request_search_item_version.py}

@@ -1,13 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ...core.pydantic_utilities import UniversalBaseModel
-import typing
 from ...core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 import pydantic
 
 
-class AgentsSearchDeployedAgentsRequestSearchItemZero(UniversalBaseModel):
-    field: typing.Literal["version"] = "version"
+class AgentsSearchDeployedAgentsRequestSearchItemVersion(UniversalBaseModel):
     value: str
 
     if IS_PYDANTIC_V2:

letta/client.py

@@ -39,7 +39,7 @@ class Letta:
 
 
 
-    token : typing.Union[str, typing.Callable[[], str]]
+    token : typing.Optional[str]
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -63,7 +63,7 @@ class Letta:
         *,
         base_url: typing.Optional[str] = None,
         environment: LettaEnvironment = LettaEnvironment.LETTA_CLOUD,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[str] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.Client] = None,
@@ -106,7 +106,7 @@ class AsyncLetta:
 
 
 
-    token : typing.Union[str, typing.Callable[[], str]]
+    token : typing.Optional[str]
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -130,7 +130,7 @@ class AsyncLetta:
         *,
         base_url: typing.Optional[str] = None,
         environment: LettaEnvironment = LettaEnvironment.LETTA_CLOUD,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[str] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.AsyncClient] = None,

letta/core/client_wrapper.py

@@ -7,14 +7,8 @@ from .http_client import AsyncHttpClient
 
 
 class BaseClientWrapper:
-    def __init__(
-        self,
-        *,
-        token: typing.Union[str, typing.Callable[[], str]],
-        base_url: str,
-        timeout: typing.Optional[float] = None,
-    ):
-        self._token = token
+    def __init__(self, *, token: typing.Optional[str] = None, base_url: str, timeout: typing.Optional[float] = None):
+        self.token = token
         self._base_url = base_url
         self._timeout = timeout
 
@@ -22,17 +16,12 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.4",
+            "X-Fern-SDK-Version": "0.1.6",
         }
-        headers["Authorization"] = f"Bearer {self._get_token()}"
+        if self.token is not None:
+            headers["Authorization"] = f"Bearer {self.token}"
         return headers
 
-    def _get_token(self) -> str:
-        if isinstance(self._token, str):
-            return self._token
-        else:
-            return self._token()
-
     def get_base_url(self) -> str:
         return self._base_url
 
@@ -44,7 +33,7 @@ class SyncClientWrapper(BaseClientWrapper):
     def __init__(
         self,
         *,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[str] = None,
         base_url: str,
         timeout: typing.Optional[float] = None,
         httpx_client: httpx.Client,
@@ -62,7 +51,7 @@ class AsyncClientWrapper(BaseClientWrapper):
     def __init__(
         self,
         *,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[str] = None,
         base_url: str,
         timeout: typing.Optional[float] = None,
         httpx_client: httpx.AsyncClient,

letta/types/__init__.py

@@ -3,6 +3,7 @@
 from .action_model import ActionModel
 from .action_parameters_model import ActionParametersModel
 from .action_response_model import ActionResponseModel
+from .agent_environment_variable import AgentEnvironmentVariable
 from .agent_state import AgentState
 from .agent_state_tool_rules_item import AgentStateToolRulesItem
 from .agent_type import AgentType
@@ -49,6 +50,15 @@ from .job import Job
 from .job_status import JobStatus
 from .letta_request import LettaRequest
 from .letta_response import LettaResponse
+from .letta_response_messages_item import (
+    LettaResponseMessagesItem,
+    LettaResponseMessagesItem_AssistantMessage,
+    LettaResponseMessagesItem_ReasoningMessage,
+    LettaResponseMessagesItem_SystemMessage,
+    LettaResponseMessagesItem_ToolCallMessage,
+    LettaResponseMessagesItem_ToolReturnMessage,
+    LettaResponseMessagesItem_UserMessage,
+)
 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
@@ -119,6 +129,7 @@ __all__ = [
     "ActionModel",
     "ActionParametersModel",
     "ActionResponseModel",
+    "AgentEnvironmentVariable",
     "AgentState",
     "AgentStateToolRulesItem",
     "AgentType",
@@ -165,6 +176,13 @@ __all__ = [
     "JobStatus",
     "LettaRequest",
     "LettaResponse",
+    "LettaResponseMessagesItem",
+    "LettaResponseMessagesItem_AssistantMessage",
+    "LettaResponseMessagesItem_ReasoningMessage",
+    "LettaResponseMessagesItem_SystemMessage",
+    "LettaResponseMessagesItem_ToolCallMessage",
+    "LettaResponseMessagesItem_ToolReturnMessage",
+    "LettaResponseMessagesItem_UserMessage",
     "LettaSchemasLettaMessageToolCall",
     "LettaSchemasMessageMessage",
     "LettaSchemasOpenaiChatCompletionRequestTool",

letta/types/agent_environment_variable.py

@@ -0,0 +1,68 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+import pydantic
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class AgentEnvironmentVariable(UniversalBaseModel):
+    created_by_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the user that made this object.
+    """
+
+    last_updated_by_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the user that made this object.
+    """
+
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    """
+    The timestamp when the object was created.
+    """
+
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    """
+    The timestamp when the object was last updated.
+    """
+
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The human-friendly ID of the Agent-env
+    """
+
+    key: str = pydantic.Field()
+    """
+    The name of the environment variable.
+    """
+
+    value: str = pydantic.Field()
+    """
+    The value of the environment variable.
+    """
+
+    description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    An optional description of the environment variable.
+    """
+
+    organization_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The ID of the organization this environment variable belongs to.
+    """
+
+    agent_id: str = pydantic.Field()
+    """
+    The ID of the agent this environment variable belongs to.
+    """
+
+    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

letta/types/agent_state.py

@@ -13,6 +13,7 @@ from ..core.serialization import FieldMetadata
 from .memory import Memory
 from .letta_schemas_tool_tool import LettaSchemasToolTool
 from .source import Source
+from .agent_environment_variable import AgentEnvironmentVariable
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -21,15 +22,15 @@ class AgentState(UniversalBaseModel):
     Representation of an agent's state. This is the state of the agent at a given time, and is persisted in the DB backend. The state has all the information needed to recreate a persisted agent.
 
     Parameters:
-    id (str): The unique identifier of the agent.
-    name (str): The name of the agent (must be unique to the user).
-    created_at (datetime): The datetime the agent was created.
-    message_ids (List[str]): The ids of the messages in the agent's in-context memory.
-    memory (Memory): The in-context memory of the agent.
-    tools (List[str]): The tools used by the agent. This includes any memory editing functions specified in `memory`.
-    system (str): The system prompt used by the agent.
-    llm_config (LLMConfig): The LLM configuration used by the agent.
-    embedding_config (EmbeddingConfig): The embedding configuration used by the agent.
+        id (str): The unique identifier of the agent.
+        name (str): The name of the agent (must be unique to the user).
+        created_at (datetime): The datetime the agent was created.
+        message_ids (List[str]): The ids of the messages in the agent's in-context memory.
+        memory (Memory): The in-context memory of the agent.
+        tools (List[str]): The tools used by the agent. This includes any memory editing functions specified in `memory`.
+        system (str): The system prompt used by the agent.
+        llm_config (LLMConfig): The LLM configuration used by the agent.
+        embedding_config (EmbeddingConfig): The embedding configuration used by the agent.
     """
 
     created_by_id: typing.Optional[str] = pydantic.Field(default=None)
@@ -129,6 +130,11 @@ class AgentState(UniversalBaseModel):
     The tags associated with the agent.
     """
 
+    tool_exec_environment_variables: typing.List[AgentEnvironmentVariable] = pydantic.Field()
+    """
+    The environment variables for tool execution specific to this agent.
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

letta/types/assistant_message_output.py

@@ -2,15 +2,14 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
-import typing
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 import pydantic
 
 
 class AssistantMessageOutput(UniversalBaseModel):
     id: str
     date: dt.datetime
-    message_type: typing.Literal["assistant_message"] = "assistant_message"
     assistant_message: str
 
     if IS_PYDANTIC_V2:

letta/types/block.py

@@ -13,15 +13,15 @@ class Block(UniversalBaseModel):
     A Block represents a reserved section of the LLM's context window which is editable. `Block` objects contained in the `Memory` object, which is able to edit the Block values.
 
     Parameters:
-    label (str): The label of the block (e.g. 'human', 'persona'). This defines a category for the block.
-    value (str): The value of the block. This is the string that is represented in the context window.
-    limit (int): The character limit of the block.
-    is*template (bool): Whether the block is a template (e.g. saved human/persona options). Non-template blocks are not stored in the database and are ephemeral, while templated blocks are stored in the database.
-    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.
-    user_id (str): The unique identifier of the user associated with the block.
+        label (str): The label of the block (e.g. 'human', 'persona'). This defines a category for the block.
+        value (str): The value of the block. This is the string that is represented in the context window.
+        limit (int): The character limit of the block.
+        is_template (bool): Whether the block is a template (e.g. saved human/persona options). Non-template blocks are not stored in the database and are ephemeral, while templated blocks are stored in the database.
+        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.
+        user_id (str): The unique identifier of the user associated with the block.
     """
 
     value: str = pydantic.Field()