letta-nightly 0.13.0.dev20251031104146__py3-none-any.whl → 0.13.1.dev20251101010313__py3-none-any.whl

letta/schemas/agent.py

@@ -17,6 +17,7 @@ from letta.schemas.letta_base import OrmMetadataBase
 from letta.schemas.llm_config import LLMConfig
 from letta.schemas.memory import Memory
 from letta.schemas.message import Message, MessageCreate
+from letta.schemas.model import EmbeddingModelSettings, ModelSettings
 from letta.schemas.openai.chat_completion_response import UsageStatistics
 from letta.schemas.response_format import ResponseFormatUnion
 from letta.schemas.source import Source
@@ -87,11 +88,19 @@ class AgentState(OrmMetadataBase, validate_assignment=True):
     # agent configuration
     agent_type: AgentType = Field(..., description="The type of agent.")
 
-    # llm information
-    llm_config: LLMConfig = Field(..., description="The LLM configuration used by the agent.")
-    embedding_config: EmbeddingConfig = Field(..., description="The embedding configuration used by the agent.")
+    # model information
+    llm_config: LLMConfig = Field(
+        ..., description="Deprecated: Use `model` field instead. The LLM configuration used by the agent.", deprecated=True
+    )
+    embedding_config: EmbeddingConfig = Field(
+        ..., description="Deprecated: Use `embedding` field instead. The embedding configuration used by the agent.", deprecated=True
+    )
+    model: Optional[ModelSettings] = Field(None, description="The model used by the agent.")
+    embedding: Optional[EmbeddingModelSettings] = Field(None, description="The embedding model used by the agent.")
+
     response_format: Optional[ResponseFormatUnion] = Field(
-        None, description="The response format used by the agent when returning from `send_message`."
+        None,
+        description="The response format used by the agent",
     )
 
     # This is an object representing the in-process state of a running `Agent`
@@ -99,7 +108,7 @@ class AgentState(OrmMetadataBase, validate_assignment=True):
     description: Optional[str] = Field(None, description="The description of the agent.")
     metadata: Optional[Dict] = Field(None, description="The metadata of the agent.")
 
-    memory: Memory = Field(..., description="The in-context memory of the agent.", deprecated=True)
+    memory: Memory = Field(..., description="Deprecated: Use `blocks` field instead. The in-context memory of the agent.", deprecated=True)
     blocks: List[Block] = Field(..., description="The memory blocks used by the agent.")
     tools: List[Tool] = Field(..., description="The tools used by the agent.")
     sources: List[Source] = Field(..., description="The sources used by the agent.")
@@ -117,7 +126,9 @@ class AgentState(OrmMetadataBase, validate_assignment=True):
     base_template_id: Optional[str] = Field(None, description="The base template id of the agent.")
     deployment_id: Optional[str] = Field(None, description="The id of the deployment.")
     entity_id: Optional[str] = Field(None, description="The id of the entity within the template.")
-    identity_ids: List[str] = Field([], description="The ids of the identities associated with this agent.", deprecated=True)
+    identity_ids: List[str] = Field(
+        [], description="Deprecated: Use `identities` field instead. The ids of the identities associated with this agent.", deprecated=True
+    )
     identities: List[Identity] = Field([], description="The identities associated with this agent.")
 
     # An advanced configuration that makes it so this agent does not remember any previous messages
@@ -130,7 +141,9 @@ class AgentState(OrmMetadataBase, validate_assignment=True):
         description="If set to True, memory management will move to a background agent thread.",
     )
 
-    multi_agent_group: Optional[Group] = Field(None, description="The multi-agent group that this agent manages", deprecated=True)
+    multi_agent_group: Optional[Group] = Field(
+        None, description="Deprecated: Use `managed_group` field instead. The multi-agent group that this agent manages.", deprecated=True
+    )
     managed_group: Optional[Group] = Field(None, description="The multi-agent group that this agent manages")
     # Run metrics
     last_run_completion: Optional[datetime] = Field(None, description="The timestamp when the agent last completed a run.")
@@ -202,8 +215,6 @@ class CreateAgent(BaseModel, validate_assignment=True):  #
     tags: Optional[List[str]] = Field(None, description="The tags associated with the agent.")
     system: Optional[str] = Field(None, description="The system prompt used by the agent.")
     agent_type: AgentType = Field(default_factory=lambda: AgentType.memgpt_v2_agent, description="The type of agent.")
-    llm_config: Optional[LLMConfig] = Field(None, description="The LLM configuration used by the agent.")
-    embedding_config: Optional[EmbeddingConfig] = Field(None, description="The embedding configuration used by the agent.")
     # Note: if this is None, then we'll populate with the standard "more human than human" initial message sequence
     # If the client wants to make this empty, then the client can set the arg to an empty list
     initial_message_sequence: Optional[List[MessageCreate]] = Field(
@@ -216,43 +227,78 @@ class CreateAgent(BaseModel, validate_assignment=True):  #
     include_base_tool_rules: Optional[bool] = Field(
         None, description="If true, attaches the Letta base tool rules (e.g. deny all tools not explicitly allowed)."
     )
-    include_default_source: bool = Field(
-        False, description="If true, automatically creates and attaches a default data source for this agent."
+    include_default_source: bool = Field(  # TODO: get rid of this
+        False, description="If true, automatically creates and attaches a default data source for this agent.", deprecated=True
     )
     description: Optional[str] = Field(None, description="The description of the agent.")
     metadata: Optional[Dict] = Field(None, description="The metadata of the agent.")
-    model: Optional[str] = Field(
+
+    # model configuration
+    llm_config: Optional[LLMConfig] = Field(
+        None, description="Deprecated: Use `model` field instead. The LLM configuration used by the agent.", deprecated=True
+    )
+    embedding_config: Optional[EmbeddingConfig] = Field(
+        None, description="Deprecated: Use `embedding` field instead. The embedding configuration used by the agent.", deprecated=True
+    )
+    model: Optional[str | ModelSettings] = Field(  # TODO: make this required  (breaking change)
         None,
-        description="The LLM configuration handle used by the agent, specified in the format "
-        "provider/model-name, as an alternative to specifying llm_config.",
+        description="The model handle or model settings for the agent to use, specified either by a handle or an object. See the model schema for more information.",
     )
-    embedding: Optional[str] = Field(
+    embedding: Optional[str | EmbeddingModelSettings] = Field(
         None, description="The embedding configuration handle used by the agent, specified in the format provider/model-name."
     )
+
     context_window_limit: Optional[int] = Field(None, description="The context window limit used by the agent.")
-    embedding_chunk_size: Optional[int] = Field(DEFAULT_EMBEDDING_CHUNK_SIZE, description="The embedding chunk size used by the agent.")
+    embedding_chunk_size: Optional[int] = Field(
+        DEFAULT_EMBEDDING_CHUNK_SIZE, description="Deprecated: No longer used. The embedding chunk size used by the agent.", deprecated=True
+    )
     max_tokens: Optional[int] = Field(
         None,
-        description="The maximum number of tokens to generate, including reasoning step. If not set, the model will use its default value.",
+        description="Deprecated: Use `model` field to configure max output tokens instead. The maximum number of tokens to generate, including reasoning step.",
+        deprecated=True,
     )
     max_reasoning_tokens: Optional[int] = Field(
-        None, description="The maximum number of tokens to generate for reasoning step. If not set, the model will use its default value."
+        None,
+        description="Deprecated: Use `model` field to configure reasoning tokens instead. The maximum number of tokens to generate for reasoning step.",
+        deprecated=True,
+    )
+    enable_reasoner: Optional[bool] = Field(
+        True,
+        description="Deprecated: Use `model` field to configure reasoning instead. Whether to enable internal extended thinking step for a reasoner model.",
+        deprecated=True,
+    )
+    reasoning: Optional[bool] = Field(
+        None,
+        description="Deprecated: Use `model` field to configure reasoning instead. Whether to enable reasoning for this agent.",
+        deprecated=True,
+    )
+    from_template: Optional[str] = Field(
+        None, description="Deprecated: please use the 'create agents from a template' endpoint instead.", deprecated=True
     )
-    enable_reasoner: Optional[bool] = Field(True, description="Whether to enable internal extended thinking step for a reasoner model.")
-    reasoning: Optional[bool] = Field(None, description="Whether to enable reasoning for this agent.")
-    from_template: Optional[str] = Field(None, description="Deprecated: please use the 'create agents from a template' endpoint instead.")
-    template: bool = Field(False, description="Deprecated: No longer used")
+    template: bool = Field(False, description="Deprecated: No longer used.", deprecated=True)
     project: Optional[str] = Field(
         None,
         deprecated=True,
-        description="Deprecated: Project should now be passed via the X-Project header instead of in the request body. If using the sdk, this can be done via the new x_project field below.",
+        description="Deprecated: Project should now be passed via the X-Project header instead of in the request body. If using the SDK, this can be done via the x_project parameter.",
+    )
+    tool_exec_environment_variables: Optional[Dict[str, str]] = Field(
+        None, description="Deprecated: Use `secrets` field instead. Environment variables for tool execution.", deprecated=True
     )
-    tool_exec_environment_variables: Optional[Dict[str, str]] = Field(None, description="Deprecated: use `secrets` field instead.")
     secrets: Optional[Dict[str, str]] = Field(None, description="The environment variables for tool execution specific to this agent.")
-    memory_variables: Optional[Dict[str, str]] = Field(None, description="The variables that should be set for the agent.")
-    project_id: Optional[str] = Field(None, description="The id of the project the agent belongs to.")
-    template_id: Optional[str] = Field(None, description="The id of the template the agent belongs to.")
-    base_template_id: Optional[str] = Field(None, description="The base template id of the agent.")
+    memory_variables: Optional[Dict[str, str]] = Field(
+        None,
+        description="Deprecated: Only relevant for creating agents from a template. Use the 'create agents from a template' endpoint instead.",
+        deprecated=True,
+    )
+    project_id: Optional[str] = Field(
+        None, description="Deprecated: No longer used. The id of the project the agent belongs to.", deprecated=True
+    )
+    template_id: Optional[str] = Field(
+        None, description="Deprecated: No longer used. The id of the template the agent belongs to.", deprecated=True
+    )
+    base_template_id: Optional[str] = Field(
+        None, description="Deprecated: No longer used. The base template id of the agent.", deprecated=True
+    )
     identity_ids: Optional[List[str]] = Field(None, description="The ids of the identities associated with this agent.")
     message_buffer_autoclear: bool = Field(
         False,
@@ -271,9 +317,14 @@ class CreateAgent(BaseModel, validate_assignment=True):  #
     )
     hidden: Optional[bool] = Field(
         None,
-        description="If set to True, the agent will be hidden.",
+        description="Deprecated: No longer used. If set to True, the agent will be hidden.",
+        deprecated=True,
+    )
+    parallel_tool_calls: Optional[bool] = Field(
+        False,
+        description="Deprecated: Use `model` field to configure parallel tool calls instead. If set to True, enables parallel tool calling.",
+        deprecated=True,
     )
-    parallel_tool_calls: Optional[bool] = Field(False, description="If set to True, enables parallel tool calling. Defaults to False.")
 
     @field_validator("name")
     @classmethod
@@ -355,8 +406,6 @@ class UpdateAgent(BaseModel):
     tags: Optional[List[str]] = Field(None, description="The tags associated with the agent.")
     system: Optional[str] = Field(None, description="The system prompt used by the agent.")
     tool_rules: Optional[List[ToolRule]] = Field(None, description="The tool rules governing the agent.")
-    llm_config: Optional[LLMConfig] = Field(None, description="The LLM configuration used by the agent.")
-    embedding_config: Optional[EmbeddingConfig] = Field(None, description="The embedding configuration used by the agent.")
     message_ids: Optional[List[str]] = Field(None, description="The ids of the messages in the agent's in-context memory.")
     description: Optional[str] = Field(None, description="The description of the agent.")
     metadata: Optional[Dict] = Field(None, description="The metadata of the agent.")
@@ -370,22 +419,42 @@ class UpdateAgent(BaseModel):
         None,
         description="If set to True, the agent will not remember previous messages (though the agent will still retain state via core memory blocks and archival/recall memory). Not recommended unless you have an advanced use case.",
     )
-    model: Optional[str] = Field(
+
+    # model configuration
+    model: Optional[str | ModelSettings] = Field(
         None,
-        description="The LLM configuration handle used by the agent, specified in the format "
-        "provider/model-name, as an alternative to specifying llm_config.",
+        description="The model used by the agent, specified either by a handle or an object. See the model schema for more information.",
     )
-    embedding: Optional[str] = Field(
+    embedding: Optional[str | EmbeddingModelSettings] = Field(
         None, description="The embedding configuration handle used by the agent, specified in the format provider/model-name."
     )
     context_window_limit: Optional[int] = Field(None, description="The context window limit used by the agent.")
+    reasoning: Optional[bool] = Field(
+        None,
+        description="Deprecated: Use `model` field to configure reasoning instead. Whether to enable reasoning for this agent.",
+        deprecated=True,
+    )
+    llm_config: Optional[LLMConfig] = Field(
+        None, description="Deprecated: Use `model` field instead. The LLM configuration used by the agent.", deprecated=True
+    )
+    embedding_config: Optional[EmbeddingConfig] = Field(None, description="The embedding configuration used by the agent.")
+    parallel_tool_calls: Optional[bool] = Field(
+        False,
+        description="Deprecated: Use `model` field to configure parallel tool calls instead. If set to True, enables parallel tool calling.",
+        deprecated=True,
+    )
+    response_format: Optional[ResponseFormatUnion] = Field(
+        None,
+        description="Deprecated: Use `model` field to configure response format instead. The response format for the agent.",
+        deprecated=True,
+    )
     max_tokens: Optional[int] = Field(
         None,
-        description="The maximum number of tokens to generate, including reasoning step. If not set, the model will use its default value.",
+        description="Deprecated: Use `model` field to configure max output tokens instead. The maximum number of tokens to generate, including reasoning step.",
+        deprecated=True,
     )
-    reasoning: Optional[bool] = Field(None, description="Whether to enable reasoning for this agent.")
+
     enable_sleeptime: Optional[bool] = Field(None, description="If set to True, memory management will move to a background agent thread.")
-    response_format: Optional[ResponseFormatUnion] = Field(None, description="The response format for the agent.")
     last_run_completion: Optional[datetime] = Field(None, description="The timestamp when the agent last completed a run.")
     last_run_duration_ms: Optional[int] = Field(None, description="The duration in milliseconds of the agent's last run.")
     timezone: Optional[str] = Field(None, description="The timezone of the agent (IANA format).")
@@ -401,7 +470,6 @@ class UpdateAgent(BaseModel):
         None,
         description="If set to True, the agent will be hidden.",
     )
-    parallel_tool_calls: Optional[bool] = Field(False, description="If set to True, enables parallel tool calling. Defaults to False.")
 
     model_config = ConfigDict(extra="ignore")  # Ignores extra fields
 

letta/schemas/agent_file.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Field
 from letta.helpers.datetime_helpers import get_utc_time
 from letta.schemas.agent import AgentState, CreateAgent
 from letta.schemas.block import Block, CreateBlock
-from letta.schemas.enums import MessageRole
+from letta.schemas.enums import MessageRole, PrimitiveType
 from letta.schemas.file import FileAgent, FileAgentBase, FileMetadata, FileMetadataBase
 from letta.schemas.group import Group, GroupCreate
 from letta.schemas.letta_message import ApprovalReturn
@@ -42,7 +42,7 @@ class ImportResult:
 class MessageSchema(MessageCreate):
     """Message with human-readable ID for agent file"""
 
-    __id_prefix__ = "message"
+    __id_prefix__ = PrimitiveType.MESSAGE.value
     id: str = Field(..., description="Human-readable identifier for this message in the file")
 
     # Override the role field to accept all message roles, not just user/system/assistant
@@ -96,7 +96,7 @@ class MessageSchema(MessageCreate):
 class FileAgentSchema(FileAgentBase):
     """File-Agent relationship with human-readable ID for agent file"""
 
-    __id_prefix__ = "file_agent"
+    __id_prefix__ = PrimitiveType.FILE_AGENT.value
     id: str = Field(..., description="Human-readable identifier for this file-agent relationship in the file")
 
     @classmethod
@@ -120,7 +120,7 @@ class FileAgentSchema(FileAgentBase):
 class AgentSchema(CreateAgent):
     """Agent with human-readable ID for agent file"""
 
-    __id_prefix__ = "agent"
+    __id_prefix__ = PrimitiveType.AGENT.value
     id: str = Field(..., description="Human-readable identifier for this agent in the file")
     in_context_message_ids: List[str] = Field(
         default_factory=list, description="List of message IDs that are currently in the agent's context"
@@ -198,7 +198,7 @@ class AgentSchema(CreateAgent):
 class GroupSchema(GroupCreate):
     """Group with human-readable ID for agent file"""
 
-    __id_prefix__ = "group"
+    __id_prefix__ = PrimitiveType.GROUP.value
     id: str = Field(..., description="Human-readable identifier for this group in the file")
 
     @classmethod
@@ -220,7 +220,7 @@ class GroupSchema(GroupCreate):
 class BlockSchema(CreateBlock):
     """Block with human-readable ID for agent file"""
 
-    __id_prefix__ = "block"
+    __id_prefix__ = PrimitiveType.BLOCK.value
     id: str = Field(..., description="Human-readable identifier for this block in the file")
 
     @classmethod
@@ -246,7 +246,7 @@ class BlockSchema(CreateBlock):
 class FileSchema(FileMetadataBase):
     """File with human-readable ID for agent file"""
 
-    __id_prefix__ = "file"
+    __id_prefix__ = PrimitiveType.FILE.value
     id: str = Field(..., description="Human-readable identifier for this file in the file")
 
     @classmethod
@@ -276,7 +276,7 @@ class FileSchema(FileMetadataBase):
 class SourceSchema(SourceCreate):
     """Source with human-readable ID for agent file"""
 
-    __id_prefix__ = "source"
+    __id_prefix__ = PrimitiveType.SOURCE.value
     id: str = Field(..., description="Human-readable identifier for this source in the file")
 
     @classmethod
@@ -299,7 +299,7 @@ class SourceSchema(SourceCreate):
 class ToolSchema(Tool):
     """Tool with human-readable ID for agent file"""
 
-    __id_prefix__ = "tool"
+    __id_prefix__ = PrimitiveType.TOOL.value
     id: str = Field(..., description="Human-readable identifier for this tool in the file")
 
     @classmethod
@@ -311,7 +311,7 @@ class ToolSchema(Tool):
 class MCPServerSchema(BaseModel):
     """MCP server schema for agent files with remapped ID."""
 
-    __id_prefix__ = "mcp_server"
+    __id_prefix__ = PrimitiveType.MCP_SERVER.value
 
     id: str = Field(..., description="Human-readable MCP server ID")
     server_type: str

letta/schemas/block.py

@@ -21,9 +21,9 @@ class BaseBlock(LettaBase, validate_assignment=True):
 
     project_id: Optional[str] = Field(None, description="The associated project id.")
     # template data (optional)
-    template_name: Optional[str] = Field(None, description="Name of the block if it is a template.", alias="name")
+    template_name: Optional[str] = Field(None, description="Name of the block if it is a template.")
     is_template: bool = Field(False, description="Whether the block is a template (e.g. saved human/persona options).")
-    template_id: Optional[str] = Field(None, description="The id of the template.", alias="name")
+    template_id: Optional[str] = Field(None, description="The id of the template.")
     base_template_id: Optional[str] = Field(None, description="The base template id of the block.")
     deployment_id: Optional[str] = Field(None, description="The id of the deployment.")
     entity_id: Optional[str] = Field(None, description="The id of the entity within the template.")
@@ -102,6 +102,25 @@ class Block(BaseBlock):
     last_updated_by_id: Optional[str] = Field(None, description="The id of the user that last updated this Block.")
 
 
+class BlockResponse(Block):
+    id: str = Field(
+        ...,
+        description="The id of the block.",
+    )
+    template_name: Optional[str] = Field(
+        None, description="(Deprecated) The name of the block template (if it is a template).", deprecated=True
+    )
+    template_id: Optional[str] = Field(None, description="(Deprecated) The id of the template.", deprecated=True)
+    base_template_id: Optional[str] = Field(None, description="(Deprecated) The base template id of the block.", deprecated=True)
+    deployment_id: Optional[str] = Field(None, description="(Deprecated) The id of the deployment.", deprecated=True)
+    entity_id: Optional[str] = Field(None, description="(Deprecated) The id of the entity within the template.", deprecated=True)
+    preserve_on_migration: Optional[bool] = Field(
+        False, description="(Deprecated) Preserve the block on template migration.", deprecated=True
+    )
+    read_only: bool = Field(False, description="(Deprecated) Whether the agent has read-only access to the block.", deprecated=True)
+    hidden: Optional[bool] = Field(None, description="(Deprecated) If set to True, the block will be hidden.", deprecated=True)
+
+
 class FileBlock(Block):
     file_id: str = Field(..., description="Unique identifier of the file.")
     source_id: str = Field(..., description="Unique identifier of the source.")
@@ -149,7 +168,7 @@ class CreateBlock(BaseBlock):
     project_id: Optional[str] = Field(None, description="The associated project id.")
     # block templates
     is_template: bool = False
-    template_name: Optional[str] = Field(None, description="Name of the block if it is a template.", alias="name")
+    template_name: Optional[str] = Field(None, description="Name of the block if it is a template.")
 
     @model_validator(mode="before")
     @classmethod

letta/schemas/enums.py

@@ -26,6 +26,27 @@ class PrimitiveType(str, Enum):
     STEP = "step"
     IDENTITY = "identity"
 
+    # Infrastructure types
+    MCP_SERVER = "mcp_server"
+    MCP_OAUTH = "mcp-oauth"
+    FILE_AGENT = "file_agent"
+
+    # Configuration types
+    SANDBOX_ENV = "sandbox-env"
+    AGENT_ENV = "agent-env"
+
+    # Core entity types
+    USER = "user"
+    ORGANIZATION = "org"
+    TOOL_RULE = "tool_rule"
+
+    # Batch processing types
+    BATCH_ITEM = "batch_item"
+    BATCH_REQUEST = "batch_req"
+
+    # Telemetry types
+    PROVIDER_TRACE = "provider_trace"
+
 
 class ProviderType(str, Enum):
     anthropic = "anthropic"

letta/schemas/environment_variables.py

@@ -2,6 +2,7 @@ from typing import Optional
 
 from pydantic import Field
 
+from letta.schemas.enums import PrimitiveType
 from letta.schemas.letta_base import LettaBase, OrmMetadataBase
 from letta.schemas.secret import Secret
 from letta.settings import settings
@@ -52,7 +53,7 @@ class EnvironmentVariableUpdateBase(LettaBase):
 
 # Environment Variable
 class SandboxEnvironmentVariableBase(EnvironmentVariableBase):
-    __id_prefix__ = "sandbox-env"
+    __id_prefix__ = PrimitiveType.SANDBOX_ENV.value
     sandbox_config_id: str = Field(..., description="The ID of the sandbox config this environment variable belongs to.")
 
 
@@ -70,7 +71,7 @@ class SandboxEnvironmentVariableUpdate(EnvironmentVariableUpdateBase):
 
 # Agent-Specific Environment Variable
 class AgentEnvironmentVariableBase(EnvironmentVariableBase):
-    __id_prefix__ = "agent-env"
+    __id_prefix__ = PrimitiveType.AGENT_ENV.value
     agent_id: str = Field(..., description="The ID of the agent this environment variable belongs to.")
 
 

letta/schemas/group.py

@@ -34,7 +34,7 @@ class Group(GroupBase):
     template_id: Optional[str] = Field(None, description="The id of the template.")
     base_template_id: Optional[str] = Field(None, description="The base template id.")
     deployment_id: Optional[str] = Field(None, description="The id of the deployment.")
-    shared_block_ids: List[str] = Field([], description="")
+    shared_block_ids: List[str] = Field([], description="", deprecated=True)
     # Pattern fields
     manager_agent_id: Optional[str] = Field(None, description="")
     termination_token: Optional[str] = Field(None, description="")
@@ -174,7 +174,7 @@ class GroupCreate(BaseModel):
     description: str = Field(..., description="")
     manager_config: ManagerConfigUnion = Field(RoundRobinManager(), description="")
     project_id: Optional[str] = Field(None, description="The associated project id.")
-    shared_block_ids: List[str] = Field([], description="")
+    shared_block_ids: List[str] = Field([], description="", deprecated=True)
     hidden: Optional[bool] = Field(
         None,
         description="If set to True, the group will be hidden.",
@@ -194,4 +194,4 @@ class GroupUpdate(BaseModel):
     description: Optional[str] = Field(None, description="")
     manager_config: Optional[ManagerConfigUpdateUnion] = Field(None, description="")
     project_id: Optional[str] = Field(None, description="The associated project id.")
-    shared_block_ids: Optional[List[str]] = Field(None, description="")
+    shared_block_ids: Optional[List[str]] = Field(None, description="", deprecated=True)

letta/schemas/letta_response.py

@@ -4,11 +4,24 @@ import re
 from datetime import datetime
 from typing import List, Union
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, RootModel
 
 from letta.helpers.json_helpers import json_dumps
 from letta.schemas.enums import JobStatus, MessageStreamStatus
-from letta.schemas.letta_message import LettaMessage, LettaMessageUnion
+from letta.schemas.letta_message import (
+    ApprovalRequestMessage,
+    ApprovalResponseMessage,
+    AssistantMessage,
+    HiddenReasoningMessage,
+    LettaMessage,
+    LettaMessageUnion,
+    LettaPing,
+    ReasoningMessage,
+    SystemMessage,
+    ToolCallMessage,
+    ToolReturnMessage,
+    UserMessage,
+)
 from letta.schemas.letta_stop_reason import LettaStopReason
 from letta.schemas.message import Message
 from letta.schemas.usage import LettaUsageStatistics
@@ -170,8 +183,27 @@ class LettaResponse(BaseModel):
         return html_output
 
 
-# The streaming response is either [DONE], [DONE_STEP], [DONE], an error, or a LettaMessage
-LettaStreamingResponse = Union[LettaMessage, MessageStreamStatus, LettaStopReason, LettaUsageStatistics]
+# The streaming response can be any of the individual message types, plus metadata types
+class LettaStreamingResponse(RootModel):
+    """
+    Streaming response type for Server-Sent Events (SSE) endpoints.
+    Each event in the stream will be one of these types.
+    """
+
+    root: Union[
+        SystemMessage,
+        UserMessage,
+        ReasoningMessage,
+        HiddenReasoningMessage,
+        ToolCallMessage,
+        ToolReturnMessage,
+        AssistantMessage,
+        ApprovalRequestMessage,
+        ApprovalResponseMessage,
+        LettaPing,
+        LettaStopReason,
+        LettaUsageStatistics,
+    ] = Field(..., discriminator="message_type")
 
 
 class LettaBatchResponse(BaseModel):

letta/schemas/llm_batch_job.py

@@ -5,7 +5,7 @@ from anthropic.types.beta.messages import BetaMessageBatch, BetaMessageBatchIndi
 from pydantic import BaseModel, Field
 
 from letta.helpers import ToolRulesSolver
-from letta.schemas.enums import AgentStepStatus, JobStatus, ProviderType
+from letta.schemas.enums import AgentStepStatus, JobStatus, PrimitiveType, ProviderType
 from letta.schemas.letta_base import OrmMetadataBase
 from letta.schemas.llm_config import LLMConfig
 
@@ -16,7 +16,7 @@ class AgentStepState(BaseModel):
 
 
 class LLMBatchItemBase(OrmMetadataBase, validate_assignment=True):
-    __id_prefix__ = "batch_item"
+    __id_prefix__ = PrimitiveType.BATCH_ITEM.value
 
 
 class LLMBatchItem(LLMBatchItemBase, validate_assignment=True):
@@ -47,7 +47,7 @@ class LLMBatchJob(OrmMetadataBase, validate_assignment=True):
     Each job corresponds to one API call that sends multiple messages to the LLM provider, and aggregates responses across all agent submissions.
     """
 
-    __id_prefix__ = "batch_req"
+    __id_prefix__ = PrimitiveType.BATCH_REQUEST.value
 
     id: Optional[str] = Field(None, description="The id of the batch job. Assigned by the database.")
     status: JobStatus = Field(..., description="The current status of the batch (e.g., created, in_progress, done).")