letta-client 0.1.17__py3-none-any.whl → 0.1.21__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,21 +43,21 @@ class MemoryVariablesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.memory_variables.get(
+        client.agents.memory_variables.list(
             agent_id="agent_id",
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/variables",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/variables",
             method="GET",
             request_options=request_options,
         )
         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",
             )
 
@@ -121,16 +121,16 @@ class AsyncMemoryVariablesClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/core_memory/variables",
+            f"v1/agents/{jsonable_encoder(agent_id)}/core-memory/variables",
             method="GET",
             request_options=request_options,
         )
         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, MessageUpdateContent, MessagesListResponse
 
-__all__ = [
-    "LettaStreamingResponse",
-    "LettaStreamingResponse_AssistantMessage",
-    "LettaStreamingResponse_ReasoningMessage",
-    "LettaStreamingResponse_SystemMessage",
-    "LettaStreamingResponse_ToolCallMessage",
-    "LettaStreamingResponse_ToolReturnMessage",
-    "LettaStreamingResponse_UsageStatistics",
-    "LettaStreamingResponse_UserMessage",
-    "MessagesListResponse",
-]
+__all__ = ["LettaStreamingResponse", "MessageUpdateContent", "MessagesListResponse"]

letta_client/agents/messages/client.py

@@ -15,10 +15,9 @@ 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.message_update_content import MessageUpdateContent
+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
@@ -163,7 +162,7 @@ class MessagesClient:
             messages=[
                 MessageCreate(
                     role="user",
-                    text="text",
+                    content="content",
                 )
             ],
         )
@@ -206,18 +205,18 @@ 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,
         *,
         role: typing.Optional[MessageRole] = OMIT,
-        text: typing.Optional[str] = OMIT,
+        content: typing.Optional[MessageUpdateContent] = 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.
 
@@ -230,13 +229,13 @@ class MessagesClient:
         role : typing.Optional[MessageRole]
             The role of the participant.
 
-        text : typing.Optional[str]
-            The text of the message.
+        content : typing.Optional[MessageUpdateContent]
+            The content of the message.
 
         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 +246,7 @@ class MessagesClient:
 
         Returns
         -------
-        LettaSchemasMessageMessage
+        Message
             Successful Response
 
         Examples
@@ -257,7 +256,7 @@ class MessagesClient:
         client = Letta(
             token="YOUR_TOKEN",
         )
-        client.agents.messages.update(
+        client.agents.messages.modify(
             agent_id="agent_id",
             message_id="message_id",
         )
@@ -267,12 +266,12 @@ class MessagesClient:
             method="PATCH",
             json={
                 "role": role,
-                "text": text,
+                "content": convert_and_respect_annotation_metadata(
+                    object_=content, annotation=MessageUpdateContent, direction="write"
+                ),
                 "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 +284,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(),
                     ),
                 )
@@ -353,7 +352,7 @@ class MessagesClient:
             messages=[
                 MessageCreate(
                     role="user",
-                    text="text",
+                    content="content",
                 )
             ],
         )
@@ -451,7 +450,7 @@ class MessagesClient:
             messages=[
                 MessageCreate(
                     role="user",
-                    text="text",
+                    content="content",
                 )
             ],
         )
@@ -642,7 +641,7 @@ class AsyncMessagesClient:
                 messages=[
                     MessageCreate(
                         role="user",
-                        text="text",
+                        content="content",
                     )
                 ],
             )
@@ -688,18 +687,18 @@ 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,
         *,
         role: typing.Optional[MessageRole] = OMIT,
-        text: typing.Optional[str] = OMIT,
+        content: typing.Optional[MessageUpdateContent] = 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.
 
@@ -712,13 +711,13 @@ class AsyncMessagesClient:
         role : typing.Optional[MessageRole]
             The role of the participant.
 
-        text : typing.Optional[str]
-            The text of the message.
+        content : typing.Optional[MessageUpdateContent]
+            The content of the message.
 
         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 +728,7 @@ class AsyncMessagesClient:
 
         Returns
         -------
-        LettaSchemasMessageMessage
+        Message
             Successful Response
 
         Examples
@@ -744,7 +743,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",
             )
@@ -757,12 +756,12 @@ class AsyncMessagesClient:
             method="PATCH",
             json={
                 "role": role,
-                "text": text,
+                "content": convert_and_respect_annotation_metadata(
+                    object_=content, annotation=MessageUpdateContent, direction="write"
+                ),
                 "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 +774,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(),
                     ),
                 )
@@ -848,7 +847,7 @@ class AsyncMessagesClient:
                 messages=[
                     MessageCreate(
                         role="user",
-                        text="text",
+                        content="content",
                     )
                 ],
             )
@@ -954,7 +953,7 @@ class AsyncMessagesClient:
                 messages=[
                     MessageCreate(
                         role="user",
-                        text="text",
+                        content="content",
                     )
                 ],
             )

letta_client/agents/messages/types/__init__.py

@@ -1,25 +1,7 @@
 # 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 .message_update_content import MessageUpdateContent
 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", "MessageUpdateContent", "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/message_update_content.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ....types.text_content import TextContent
+
+MessageUpdateContent = typing.Union[str, typing.List[TextContent]]

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]]