letta-client 0.1.17__py3-none-any.whl → 0.1.19__py3-none-any.whl

letta_client/agents/memory_variables/__init__.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import MemoryVariablesGetResponse
+from .types import MemoryVariablesListResponse
 
-__all__ = ["MemoryVariablesGetResponse"]
+__all__ = ["MemoryVariablesListResponse"]

letta_client/agents/memory_variables/client.py

@@ -3,7 +3,7 @@
 from ...core.client_wrapper import SyncClientWrapper
 import typing
 from ...core.request_options import RequestOptions
-from .types.memory_variables_get_response import MemoryVariablesGetResponse
+from .types.memory_variables_list_response import MemoryVariablesListResponse
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.unchecked_base_model import construct_type
 from ...errors.not_found_error import NotFoundError
@@ -16,9 +16,9 @@ class MemoryVariablesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def get(
+    def list(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> MemoryVariablesGetResponse:
+    ) -> MemoryVariablesListResponse:
         """
         <Note>This endpoint is only available on Letta Cloud.</Note>
 
@@ -33,7 +33,7 @@ class MemoryVariablesClient:
 
         Returns
         -------
-        MemoryVariablesGetResponse
+        MemoryVariablesListResponse
             200
 
         Examples
@@ -43,7 +43,7 @@ class MemoryVariablesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.memory_variables.get(
+        client.agents.memory_variables.list(
             agent_id="agent_id",
         )
         """
@@ -55,9 +55,9 @@ class MemoryVariablesClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    MemoryVariablesGetResponse,
+                    MemoryVariablesListResponse,
                     construct_type(
-                        type_=MemoryVariablesGetResponse,  # type: ignore
+                        type_=MemoryVariablesListResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -81,9 +81,9 @@ class AsyncMemoryVariablesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def get(
+    async def list(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> MemoryVariablesGetResponse:
+    ) -> MemoryVariablesListResponse:
         """
         <Note>This endpoint is only available on Letta Cloud.</Note>
 
@@ -98,7 +98,7 @@ class AsyncMemoryVariablesClient:
 
         Returns
         -------
-        MemoryVariablesGetResponse
+        MemoryVariablesListResponse
             200
 
         Examples
@@ -113,7 +113,7 @@ class AsyncMemoryVariablesClient:
 
 
         async def main() -> None:
-            await client.agents.memory_variables.get(
+            await client.agents.memory_variables.list(
                 agent_id="agent_id",
             )
 
@@ -128,9 +128,9 @@ class AsyncMemoryVariablesClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    MemoryVariablesGetResponse,
+                    MemoryVariablesListResponse,
                     construct_type(
-                        type_=MemoryVariablesGetResponse,  # type: ignore
+                        type_=MemoryVariablesListResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )

letta_client/agents/memory_variables/types/__init__.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .memory_variables_get_response import MemoryVariablesGetResponse
+from .memory_variables_list_response import MemoryVariablesListResponse
 
-__all__ = ["MemoryVariablesGetResponse"]
+__all__ = ["MemoryVariablesListResponse"]

letta_client/agents/memory_variables/types/{memory_variables_get_response.py → memory_variables_list_response.py}

@@ -6,7 +6,7 @@ from ....core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class MemoryVariablesGetResponse(UncheckedBaseModel):
+class MemoryVariablesListResponse(UncheckedBaseModel):
     variables: typing.Dict[str, str]
 
     if IS_PYDANTIC_V2:

letta_client/agents/messages/__init__.py

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

letta_client/agents/messages/client.py

@@ -15,10 +15,8 @@ from ...types.letta_request_config import LettaRequestConfig
 from ...types.letta_response import LettaResponse
 from ...core.serialization import convert_and_respect_annotation_metadata
 from ...types.message_role import MessageRole
-from ...types.letta_schemas_openai_chat_completions_tool_call_input import (
-    LettaSchemasOpenaiChatCompletionsToolCallInput,
-)
-from ...types.letta_schemas_message_message import LettaSchemasMessageMessage
+from ...types.chat_completion_message_tool_call import ChatCompletionMessageToolCall
+from ...types.message import Message
 from .types.letta_streaming_response import LettaStreamingResponse
 import httpx_sse
 import json
@@ -206,7 +204,7 @@ class MessagesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def update(
+    def modify(
         self,
         agent_id: str,
         message_id: str,
@@ -214,10 +212,10 @@ class MessagesClient:
         role: typing.Optional[MessageRole] = OMIT,
         text: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
-        tool_calls: typing.Optional[typing.Sequence[LettaSchemasOpenaiChatCompletionsToolCallInput]] = OMIT,
+        tool_calls: typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]] = OMIT,
         tool_call_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> LettaSchemasMessageMessage:
+    ) -> Message:
         """
         Update the details of a message associated with an agent.
 
@@ -236,7 +234,7 @@ class MessagesClient:
         name : typing.Optional[str]
             The name of the participant.
 
-        tool_calls : typing.Optional[typing.Sequence[LettaSchemasOpenaiChatCompletionsToolCallInput]]
+        tool_calls : typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]]
             The list of tool calls requested.
 
         tool_call_id : typing.Optional[str]
@@ -247,7 +245,7 @@ class MessagesClient:
 
         Returns
         -------
-        LettaSchemasMessageMessage
+        Message
             Successful Response
 
         Examples
@@ -257,7 +255,7 @@ class MessagesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.messages.update(
+        client.agents.messages.modify(
             agent_id="agent_id",
             message_id="message_id",
         )
@@ -270,9 +268,7 @@ class MessagesClient:
                 "text": text,
                 "name": name,
                 "tool_calls": convert_and_respect_annotation_metadata(
-                    object_=tool_calls,
-                    annotation=typing.Sequence[LettaSchemasOpenaiChatCompletionsToolCallInput],
-                    direction="write",
+                    object_=tool_calls, annotation=typing.Sequence[ChatCompletionMessageToolCall], direction="write"
                 ),
                 "tool_call_id": tool_call_id,
             },
@@ -285,9 +281,9 @@ class MessagesClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    LettaSchemasMessageMessage,
+                    Message,
                     construct_type(
-                        type_=LettaSchemasMessageMessage,  # type: ignore
+                        type_=Message,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -688,7 +684,7 @@ class AsyncMessagesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def update(
+    async def modify(
         self,
         agent_id: str,
         message_id: str,
@@ -696,10 +692,10 @@ class AsyncMessagesClient:
         role: typing.Optional[MessageRole] = OMIT,
         text: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
-        tool_calls: typing.Optional[typing.Sequence[LettaSchemasOpenaiChatCompletionsToolCallInput]] = OMIT,
+        tool_calls: typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]] = OMIT,
         tool_call_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> LettaSchemasMessageMessage:
+    ) -> Message:
         """
         Update the details of a message associated with an agent.
 
@@ -718,7 +714,7 @@ class AsyncMessagesClient:
         name : typing.Optional[str]
             The name of the participant.
 
-        tool_calls : typing.Optional[typing.Sequence[LettaSchemasOpenaiChatCompletionsToolCallInput]]
+        tool_calls : typing.Optional[typing.Sequence[ChatCompletionMessageToolCall]]
             The list of tool calls requested.
 
         tool_call_id : typing.Optional[str]
@@ -729,7 +725,7 @@ class AsyncMessagesClient:
 
         Returns
         -------
-        LettaSchemasMessageMessage
+        Message
             Successful Response
 
         Examples
@@ -744,7 +740,7 @@ class AsyncMessagesClient:
 
 
         async def main() -> None:
-            await client.agents.messages.update(
+            await client.agents.messages.modify(
                 agent_id="agent_id",
                 message_id="message_id",
             )
@@ -760,9 +756,7 @@ class AsyncMessagesClient:
                 "text": text,
                 "name": name,
                 "tool_calls": convert_and_respect_annotation_metadata(
-                    object_=tool_calls,
-                    annotation=typing.Sequence[LettaSchemasOpenaiChatCompletionsToolCallInput],
-                    direction="write",
+                    object_=tool_calls, annotation=typing.Sequence[ChatCompletionMessageToolCall], direction="write"
                 ),
                 "tool_call_id": tool_call_id,
             },
@@ -775,9 +769,9 @@ class AsyncMessagesClient:
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    LettaSchemasMessageMessage,
+                    Message,
                     construct_type(
-                        type_=LettaSchemasMessageMessage,  # type: ignore
+                        type_=Message,  # type: ignore
                         object_=_response.json(),
                     ),
                 )

letta_client/agents/messages/types/__init__.py

@@ -1,25 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .letta_streaming_response import (
-    LettaStreamingResponse,
-    LettaStreamingResponse_AssistantMessage,
-    LettaStreamingResponse_ReasoningMessage,
-    LettaStreamingResponse_SystemMessage,
-    LettaStreamingResponse_ToolCallMessage,
-    LettaStreamingResponse_ToolReturnMessage,
-    LettaStreamingResponse_UsageStatistics,
-    LettaStreamingResponse_UserMessage,
-)
+from .letta_streaming_response import LettaStreamingResponse
 from .messages_list_response import MessagesListResponse
 
-__all__ = [
-    "LettaStreamingResponse",
-    "LettaStreamingResponse_AssistantMessage",
-    "LettaStreamingResponse_ReasoningMessage",
-    "LettaStreamingResponse_SystemMessage",
-    "LettaStreamingResponse_ToolCallMessage",
-    "LettaStreamingResponse_ToolReturnMessage",
-    "LettaStreamingResponse_UsageStatistics",
-    "LettaStreamingResponse_UserMessage",
-    "MessagesListResponse",
-]
+__all__ = ["LettaStreamingResponse", "MessagesListResponse"]

letta_client/agents/messages/types/letta_streaming_response.py

@@ -1,143 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-from ....core.unchecked_base_model import UncheckedBaseModel
 import typing
-import datetime as dt
-from ....core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ....types.tool_call_message_tool_call import ToolCallMessageToolCall
-from ....types.tool_return_message_status import ToolReturnMessageStatus
-import typing_extensions
-from ....core.unchecked_base_model import UnionMetadata
-
-
-class LettaStreamingResponse_SystemMessage(UncheckedBaseModel):
-    message_type: typing.Literal["system_message"] = "system_message"
-    id: str
-    date: dt.datetime
-    message: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaStreamingResponse_UserMessage(UncheckedBaseModel):
-    message_type: typing.Literal["user_message"] = "user_message"
-    id: str
-    date: dt.datetime
-    message: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaStreamingResponse_ReasoningMessage(UncheckedBaseModel):
-    message_type: typing.Literal["reasoning_message"] = "reasoning_message"
-    id: str
-    date: dt.datetime
-    reasoning: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaStreamingResponse_ToolCallMessage(UncheckedBaseModel):
-    message_type: typing.Literal["tool_call_message"] = "tool_call_message"
-    id: str
-    date: dt.datetime
-    tool_call: ToolCallMessageToolCall
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaStreamingResponse_ToolReturnMessage(UncheckedBaseModel):
-    message_type: typing.Literal["tool_return_message"] = "tool_return_message"
-    id: str
-    date: dt.datetime
-    tool_return: str
-    status: ToolReturnMessageStatus
-    tool_call_id: str
-    stdout: typing.Optional[typing.List[str]] = None
-    stderr: typing.Optional[typing.List[str]] = None
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaStreamingResponse_AssistantMessage(UncheckedBaseModel):
-    message_type: typing.Literal["assistant_message"] = "assistant_message"
-    id: str
-    date: dt.datetime
-    assistant_message: str
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-class LettaStreamingResponse_UsageStatistics(UncheckedBaseModel):
-    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_extensions.Annotated[
-    typing.Union[
-        LettaStreamingResponse_SystemMessage,
-        LettaStreamingResponse_UserMessage,
-        LettaStreamingResponse_ReasoningMessage,
-        LettaStreamingResponse_ToolCallMessage,
-        LettaStreamingResponse_ToolReturnMessage,
-        LettaStreamingResponse_AssistantMessage,
-        LettaStreamingResponse_UsageStatistics,
-    ],
-    UnionMetadata(discriminant="message_type"),
+from ....types.system_message import SystemMessage
+from ....types.user_message import UserMessage
+from ....types.reasoning_message import ReasoningMessage
+from ....types.tool_call_message import ToolCallMessage
+from ....types.tool_return_message import ToolReturnMessage
+from ....types.assistant_message import AssistantMessage
+from ....types.letta_usage_statistics import LettaUsageStatistics
+
+LettaStreamingResponse = typing.Union[
+    SystemMessage,
+    UserMessage,
+    ReasoningMessage,
+    ToolCallMessage,
+    ToolReturnMessage,
+    AssistantMessage,
+    LettaUsageStatistics,
 ]

letta_client/agents/messages/types/messages_list_response.py

@@ -1,7 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ....types.letta_schemas_message_message import LettaSchemasMessageMessage
+from ....types.message import Message
 from ....types.letta_message_union import LettaMessageUnion
 
-MessagesListResponse = typing.Union[typing.List[LettaSchemasMessageMessage], typing.List[LettaMessageUnion]]
+MessagesListResponse = typing.Union[typing.List[Message], typing.List[LettaMessageUnion]]