letta-client 0.1.226__py3-none-any.whl → 1.7.2__py3-none-any.whl

letta_client/types/provider_type.py

@@ -1,26 +1,28 @@
-# This file was auto-generated by Fern from our API Definition.
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import typing
+from typing_extensions import Literal, TypeAlias
 
-ProviderType = typing.Union[
-    typing.Literal[
-        "anthropic",
-        "google_ai",
-        "google_vertex",
-        "openai",
-        "letta",
-        "deepseek",
-        "cerebras",
-        "lmstudio_openai",
-        "xai",
-        "mistral",
-        "ollama",
-        "groq",
-        "together",
-        "azure",
-        "vllm",
-        "bedrock",
-        "cohere",
-    ],
-    typing.Any,
+__all__ = ["ProviderType"]
+
+ProviderType: TypeAlias = Literal[
+    "anthropic",
+    "azure",
+    "bedrock",
+    "cerebras",
+    "chatgpt_oauth",
+    "deepseek",
+    "google_ai",
+    "google_vertex",
+    "groq",
+    "hugging-face",
+    "letta",
+    "lmstudio_openai",
+    "mistral",
+    "ollama",
+    "openai",
+    "together",
+    "vllm",
+    "sglang",
+    "xai",
+    "zai",
 ]

letta_client/types/required_before_exit_tool_rule.py

@@ -1,32 +1,22 @@
-# This file was auto-generated by Fern from our API Definition.
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from ..core.unchecked_base_model import UncheckedBaseModel
-import pydantic
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from typing import Optional
+from typing_extensions import Literal
 
+from .._models import BaseModel
 
-class RequiredBeforeExitToolRule(UncheckedBaseModel):
-    """
-    Represents a tool rule configuration where this tool must be called before the agent loop can exit.
-    """
+__all__ = ["RequiredBeforeExitToolRule"]
 
-    tool_name: str = pydantic.Field()
-    """
-    The name of the tool. Must exist in the database for the user's organization.
-    """
 
-    type: typing.Literal["required_before_exit"] = "required_before_exit"
-    prompt_template: typing.Optional[str] = pydantic.Field(default=None)
+class RequiredBeforeExitToolRule(BaseModel):
     """
-    Optional Jinja2 template for generating agent prompt about this tool rule.
+    Represents a tool rule configuration where this tool must be called before the agent loop can exit.
     """
 
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
+    tool_name: str
+    """The name of the tool. Must exist in the database for the user's organization."""
+
+    prompt_template: Optional[str] = None
+    """Optional template string (ignored)."""
 
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
+    type: Optional[Literal["required_before_exit"]] = None

letta_client/types/required_before_exit_tool_rule_param.py

@@ -0,0 +1,22 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["RequiredBeforeExitToolRuleParam"]
+
+
+class RequiredBeforeExitToolRuleParam(TypedDict, total=False):
+    """
+    Represents a tool rule configuration where this tool must be called before the agent loop can exit.
+    """
+
+    tool_name: Required[str]
+    """The name of the tool. Must exist in the database for the user's organization."""
+
+    prompt_template: Optional[str]
+    """Optional template string (ignored)."""
+
+    type: Literal["required_before_exit"]

letta_client/types/requires_approval_tool_rule.py

@@ -0,0 +1,25 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["RequiresApprovalToolRule"]
+
+
+class RequiresApprovalToolRule(BaseModel):
+    """
+    Represents a tool rule configuration which requires approval before the tool can be invoked.
+    """
+
+    tool_name: str
+    """The name of the tool. Must exist in the database for the user's organization."""
+
+    prompt_template: Optional[str] = None
+    """Optional template string (ignored).
+
+    Rendering uses fast built-in formatting for performance.
+    """
+
+    type: Optional[Literal["requires_approval"]] = None

letta_client/types/requires_approval_tool_rule_param.py

@@ -0,0 +1,25 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["RequiresApprovalToolRuleParam"]
+
+
+class RequiresApprovalToolRuleParam(TypedDict, total=False):
+    """
+    Represents a tool rule configuration which requires approval before the tool can be invoked.
+    """
+
+    tool_name: Required[str]
+    """The name of the tool. Must exist in the database for the user's organization."""
+
+    prompt_template: Optional[str]
+    """Optional template string (ignored).
+
+    Rendering uses fast built-in formatting for performance.
+    """
+
+    type: Literal["requires_approval"]

letta_client/types/run_list_params.py

@@ -0,0 +1,67 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, TypedDict
+
+from .._types import SequenceNotStr
+from .stop_reason_type import StopReasonType
+
+__all__ = ["RunListParams"]
+
+
+class RunListParams(TypedDict, total=False):
+    active: bool
+    """Filter for active runs."""
+
+    after: Optional[str]
+    """Run ID cursor for pagination.
+
+    Returns runs that come after this run ID in the specified sort order
+    """
+
+    agent_id: Optional[str]
+    """The unique identifier of the agent associated with the run."""
+
+    agent_ids: Optional[SequenceNotStr[str]]
+    """The unique identifiers of the agents associated with the run.
+
+    Deprecated in favor of agent_id field.
+    """
+
+    ascending: bool
+    """
+    Whether to sort agents oldest to newest (True) or newest to oldest (False,
+    default). Deprecated in favor of order field.
+    """
+
+    background: Optional[bool]
+    """If True, filters for runs that were created in background mode."""
+
+    before: Optional[str]
+    """Run ID cursor for pagination.
+
+    Returns runs that come before this run ID in the specified sort order
+    """
+
+    conversation_id: Optional[str]
+    """Filter runs by conversation ID."""
+
+    limit: Optional[int]
+    """Maximum number of runs to return"""
+
+    order: Literal["asc", "desc"]
+    """Sort order for runs by creation time.
+
+    'asc' for oldest first, 'desc' for newest first
+    """
+
+    order_by: Literal["created_at"]
+    """Field to sort by"""
+
+    statuses: Optional[SequenceNotStr[str]]
+    """Filter runs by status. Can specify multiple statuses."""
+
+    stop_reason: Optional[StopReasonType]
+    """Filter runs by stop reason."""

letta_client/types/runs/__init__.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .step_list_params import StepListParams as StepListParams
+from .message_list_params import MessageListParams as MessageListParams
+from .message_stream_params import MessageStreamParams as MessageStreamParams
+from .trace_retrieve_params import TraceRetrieveParams as TraceRetrieveParams
+from .trace_retrieve_response import TraceRetrieveResponse as TraceRetrieveResponse
+from .usage_retrieve_response import UsageRetrieveResponse as UsageRetrieveResponse

letta_client/types/runs/message_list_params.py

@@ -0,0 +1,34 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["MessageListParams"]
+
+
+class MessageListParams(TypedDict, total=False):
+    after: Optional[str]
+    """Message ID cursor for pagination.
+
+    Returns messages that come after this message ID in the specified sort order
+    """
+
+    before: Optional[str]
+    """Message ID cursor for pagination.
+
+    Returns messages that come before this message ID in the specified sort order
+    """
+
+    limit: Optional[int]
+    """Maximum number of messages to return"""
+
+    order: Literal["asc", "desc"]
+    """Sort order for messages by creation time.
+
+    'asc' for oldest first, 'desc' for newest first
+    """
+
+    order_by: Literal["created_at"]
+    """Field to sort by"""

letta_client/types/runs/message_stream_params.py

@@ -0,0 +1,28 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import TypedDict
+
+__all__ = ["MessageStreamParams"]
+
+
+class MessageStreamParams(TypedDict, total=False):
+    batch_size: Optional[int]
+    """Number of entries to read per batch."""
+
+    include_pings: Optional[bool]
+    """
+    Whether to include periodic keepalive ping messages in the stream to prevent
+    connection timeouts.
+    """
+
+    poll_interval: Optional[float]
+    """Seconds to wait between polls when no new data."""
+
+    starting_after: int
+    """Sequence id to use as a cursor for pagination.
+
+    Response will start streaming after this chunk sequence id
+    """

letta_client/types/runs/step_list_params.py

@@ -0,0 +1,28 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["StepListParams"]
+
+
+class StepListParams(TypedDict, total=False):
+    after: Optional[str]
+    """Cursor for pagination"""
+
+    before: Optional[str]
+    """Cursor for pagination"""
+
+    limit: Optional[int]
+    """Maximum number of messages to return"""
+
+    order: Literal["asc", "desc"]
+    """Sort order for steps by creation time.
+
+    'asc' for oldest first, 'desc' for newest first
+    """
+
+    order_by: Literal["created_at"]
+    """Field to sort by"""

letta_client/types/runs/trace_retrieve_params.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["TraceRetrieveParams"]
+
+
+class TraceRetrieveParams(TypedDict, total=False):
+    limit: int
+    """Maximum number of spans to return"""

letta_client/types/runs/trace_retrieve_response.py

@@ -0,0 +1,8 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List
+from typing_extensions import TypeAlias
+
+__all__ = ["TraceRetrieveResponse"]
+
+TraceRetrieveResponse: TypeAlias = List[Dict[str, object]]

letta_client/types/runs/usage_retrieve_response.py

@@ -0,0 +1,31 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["UsageRetrieveResponse", "CompletionTokensDetails", "PromptTokensDetails"]
+
+
+class CompletionTokensDetails(BaseModel):
+    reasoning_tokens: Optional[int] = None
+
+
+class PromptTokensDetails(BaseModel):
+    cache_creation_tokens: Optional[int] = None
+
+    cache_read_tokens: Optional[int] = None
+
+    cached_tokens: Optional[int] = None
+
+
+class UsageRetrieveResponse(BaseModel):
+    completion_tokens: Optional[int] = None
+
+    completion_tokens_details: Optional[CompletionTokensDetails] = None
+
+    prompt_tokens: Optional[int] = None
+
+    prompt_tokens_details: Optional[PromptTokensDetails] = None
+
+    total_tokens: Optional[int] = None

letta_client/types/sse_mcp_server.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["SseMcpServer"]
+
+
+class SseMcpServer(BaseModel):
+    """An SSE MCP server"""
+
+    server_name: str
+    """The name of the MCP server"""
+
+    server_url: str
+    """The URL of the server"""
+
+    id: Optional[str] = None
+    """The human-friendly ID of the Mcp_server"""
+
+    auth_header: Optional[str] = None
+    """The name of the authentication header (e.g., 'Authorization')"""
+
+    auth_token: Optional[str] = None
+    """The authentication token or API key value"""
+
+    custom_headers: Optional[Dict[str, str]] = None
+    """Custom HTTP headers to include with requests"""
+
+    mcp_server_type: Optional[Literal["sse"]] = None

letta_client/types/stdio_mcp_server.py

@@ -0,0 +1,29 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List, Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["StdioMcpServer"]
+
+
+class StdioMcpServer(BaseModel):
+    """A Stdio MCP server"""
+
+    args: List[str]
+    """The arguments to pass to the command"""
+
+    command: str
+    """The command to run (MCP 'local' client will run this command)"""
+
+    server_name: str
+    """The name of the MCP server"""
+
+    id: Optional[str] = None
+    """The human-friendly ID of the Mcp_server"""
+
+    env: Optional[Dict[str, str]] = None
+    """Environment variables to set"""
+
+    mcp_server_type: Optional[Literal["stdio"]] = None

letta_client/types/step.py

@@ -1,127 +1,101 @@
-# This file was auto-generated by Fern from our API Definition.
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from ..core.unchecked_base_model import UncheckedBaseModel
-import pydantic
-import typing
+from typing import Dict, List, Optional
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
 from .stop_reason_type import StopReasonType
-from .message import Message
-from .step_feedback import StepFeedback
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .agents.internal_message import InternalMessage
 
+__all__ = ["Step"]
 
-class Step(UncheckedBaseModel):
-    id: str = pydantic.Field()
-    """
-    The id of the step. Assigned by the database.
-    """
 
-    origin: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The surface that this agent step was initiated from.
-    """
+class Step(BaseModel):
+    id: str
+    """The id of the step. Assigned by the database."""
 
-    provider_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The unique identifier of the provider that was configured for this step
-    """
+    agent_id: Optional[str] = None
+    """The ID of the agent that performed the step."""
 
-    job_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The unique identifier of the job that this step belongs to. Only included for async calls.
-    """
+    completion_tokens: Optional[int] = None
+    """The number of tokens generated by the agent during this step."""
 
-    agent_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The ID of the agent that performed the step.
-    """
+    completion_tokens_details: Optional[Dict[str, object]] = None
+    """Detailed completion token breakdown (e.g., reasoning_tokens)."""
 
-    provider_name: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The name of the provider used for this step.
-    """
+    context_window_limit: Optional[int] = None
+    """The context window limit configured for this step."""
 
-    provider_category: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The category of the provider used for this step.
-    """
+    error_data: Optional[Dict[str, object]] = None
+    """Error details including message, traceback, and additional context"""
 
-    model: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The name of the model used for this step.
-    """
+    error_type: Optional[str] = None
+    """The type/class of the error that occurred"""
 
-    model_endpoint: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The model endpoint url used for this step.
-    """
+    feedback: Optional[Literal["positive", "negative"]] = None
+    """The feedback for this step. Must be either 'positive' or 'negative'."""
 
-    context_window_limit: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    The context window limit configured for this step.
-    """
+    messages: Optional[List[InternalMessage]] = None
+    """The messages generated during this step.
 
-    completion_tokens: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    The number of tokens generated by the agent during this step.
+    Deprecated: use `GET /v1/steps/{step_id}/messages` endpoint instead
     """
 
-    prompt_tokens: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    The number of tokens in the prompt during this step.
-    """
+    model: Optional[str] = None
+    """The name of the model used for this step."""
 
-    total_tokens: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    The total number of tokens processed by the agent during this step.
-    """
+    api_model_endpoint: Optional[str] = FieldInfo(alias="model_endpoint", default=None)
+    """The model endpoint url used for this step."""
 
-    completion_tokens_details: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(
-        default=None
-    )
-    """
-    Metadata for the agent.
-    """
+    origin: Optional[str] = None
+    """The surface that this agent step was initiated from."""
 
-    stop_reason: typing.Optional[StopReasonType] = pydantic.Field(default=None)
-    """
-    The stop reason associated with the step.
-    """
+    project_id: Optional[str] = None
+    """The project that the agent that executed this step belongs to (cloud only)."""
 
-    tags: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
-    """
-    Metadata tags.
-    """
+    prompt_tokens: Optional[int] = None
+    """The number of tokens in the prompt during this step."""
 
-    tid: typing.Optional[str] = pydantic.Field(default=None)
+    prompt_tokens_details: Optional[Dict[str, object]] = None
     """
-    The unique identifier of the transaction that processed this step.
+    Detailed prompt token breakdown (e.g., cached_tokens, cache_read_tokens,
+    cache_creation_tokens).
     """
 
-    trace_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The trace id of the agent step.
-    """
+    provider_category: Optional[str] = None
+    """The category of the provider used for this step."""
 
-    messages: typing.Optional[typing.List[Message]] = pydantic.Field(default=None)
-    """
-    The messages generated during this step.
-    """
+    provider_id: Optional[str] = None
+    """The unique identifier of the provider that was configured for this step"""
 
-    feedback: typing.Optional[StepFeedback] = pydantic.Field(default=None)
-    """
-    The feedback for this step. Must be either 'positive' or 'negative'.
-    """
+    provider_name: Optional[str] = None
+    """The name of the provider used for this step."""
 
-    project_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The project that the agent that executed this step belongs to (cloud only).
+    request_id: Optional[str] = None
+    """The API request log ID from cloud-api for correlating steps with API requests."""
+
+    run_id: Optional[str] = None
+    """The unique identifier of the run that this step belongs to.
+
+    Only included for async calls.
     """
 
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
+    status: Optional[Literal["pending", "success", "failed", "cancelled"]] = None
+    """Status of a step execution"""
+
+    stop_reason: Optional[StopReasonType] = None
+    """The stop reason associated with the step."""
+
+    tags: Optional[List[str]] = None
+    """Metadata tags."""
+
+    tid: Optional[str] = None
+    """The unique identifier of the transaction that processed this step."""
+
+    total_tokens: Optional[int] = None
+    """The total number of tokens processed by the agent during this step."""
 
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
+    trace_id: Optional[str] = None
+    """The trace id of the agent step."""