letta-client 1.0.0a9__py3-none-any.whl → 1.0.0a11__py3-none-any.whl

letta_client/resources/groups/messages.py

@@ -19,11 +19,11 @@ from ..._response import (
 )
 from ...pagination import SyncArrayPage, AsyncArrayPage
 from ..._base_client import AsyncPaginator, make_request_options
-from ...types.groups import message_list_params, message_send_params, message_stream_params, message_update_params
+from ...types.groups import message_list_params, message_send_params, message_modify_params, message_stream_params
 from ...types.agents.message_type import MessageType
 from ...types.agents.letta_response import LettaResponse
 from ...types.agents.letta_message_union import LettaMessageUnion
-from ...types.groups.message_update_response import MessageUpdateResponse
+from ...types.groups.message_modify_response import MessageModifyResponse
 from ...types.agents.letta_user_message_content_union_param import LettaUserMessageContentUnionParam
 from ...types.agents.letta_assistant_message_content_union_param import LettaAssistantMessageContentUnionParam
 
@@ -50,8 +50,87 @@ class MessagesResource(SyncAPIResource):
         """
         return MessagesResourceWithStreamingResponse(self)
 
+    def list(
+        self,
+        group_id: str,
+        *,
+        after: Optional[str] | Omit = omit,
+        assistant_message_tool_kwarg: str | Omit = omit,
+        assistant_message_tool_name: str | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        limit: Optional[int] | Omit = omit,
+        order: Literal["asc", "desc"] | Omit = omit,
+        order_by: Literal["created_at"] | Omit = omit,
+        use_assistant_message: bool | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> SyncArrayPage[LettaMessageUnion]:
+        """
+        Retrieve message history for an agent.
+
+        Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
+          after: Message ID cursor for pagination. Returns messages that come after this message
+              ID in the specified sort order
+
+          assistant_message_tool_kwarg: The name of the message argument.
+
+          assistant_message_tool_name: The name of the designated message tool.
+
+          before: Message ID cursor for pagination. Returns messages that come before this message
+              ID in the specified sort order
+
+          limit: Maximum number of messages to retrieve
+
+          order: Sort order for messages by creation time. 'asc' for oldest first, 'desc' for
+              newest first
+
+          order_by: Field to sort by
+
+          use_assistant_message: Whether to use assistant messages
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not group_id:
+            raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
+        return self._get_api_list(
+            f"/v1/groups/{group_id}/messages",
+            page=SyncArrayPage[LettaMessageUnion],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
+                        "assistant_message_tool_name": assistant_message_tool_name,
+                        "before": before,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                        "use_assistant_message": use_assistant_message,
+                    },
+                    message_list_params.MessageListParams,
+                ),
+            ),
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
+        )
+
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -64,7 +143,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -87,7 +166,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -100,7 +179,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -123,7 +202,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -136,7 +215,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -156,7 +235,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -169,7 +248,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -192,7 +271,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @required_args(["group_id", "content"], ["group_id", "reasoning"])
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -210,13 +289,13 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         if not group_id:
             raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
         if not message_id:
             raise ValueError(f"Expected a non-empty value for `message_id` but received {message_id!r}")
         return cast(
-            MessageUpdateResponse,
+            MessageModifyResponse,
             self._patch(
                 f"/v1/groups/{group_id}/messages/{message_id}",
                 body=maybe_transform(
@@ -225,96 +304,17 @@ class MessagesResource(SyncAPIResource):
                         "message_type": message_type,
                         "reasoning": reasoning,
                     },
-                    message_update_params.MessageUpdateParams,
+                    message_modify_params.MessageModifyParams,
                 ),
                 options=make_request_options(
                     extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
                 ),
                 cast_to=cast(
-                    Any, MessageUpdateResponse
+                    Any, MessageModifyResponse
                 ),  # Union types cannot be passed in as arguments in the type system
             ),
         )
 
-    def list(
-        self,
-        group_id: str,
-        *,
-        after: Optional[str] | Omit = omit,
-        assistant_message_tool_kwarg: str | Omit = omit,
-        assistant_message_tool_name: str | Omit = omit,
-        before: Optional[str] | Omit = omit,
-        limit: Optional[int] | Omit = omit,
-        order: Literal["asc", "desc"] | Omit = omit,
-        order_by: Literal["created_at"] | Omit = omit,
-        use_assistant_message: bool | Omit = omit,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[LettaMessageUnion]:
-        """
-        Retrieve message history for an agent.
-
-        Args:
-          group_id: The ID of the group in the format 'group-<uuid4>'
-
-          after: Message ID cursor for pagination. Returns messages that come after this message
-              ID in the specified sort order
-
-          assistant_message_tool_kwarg: The name of the message argument.
-
-          assistant_message_tool_name: The name of the designated message tool.
-
-          before: Message ID cursor for pagination. Returns messages that come before this message
-              ID in the specified sort order
-
-          limit: Maximum number of messages to retrieve
-
-          order: Sort order for messages by creation time. 'asc' for oldest first, 'desc' for
-              newest first
-
-          order_by: Field to sort by
-
-          use_assistant_message: Whether to use assistant messages
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        if not group_id:
-            raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        return self._get_api_list(
-            f"/v1/groups/{group_id}/messages",
-            page=SyncArrayPage[LettaMessageUnion],
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "after": after,
-                        "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                        "assistant_message_tool_name": assistant_message_tool_name,
-                        "before": before,
-                        "limit": limit,
-                        "order": order,
-                        "order_by": order_by,
-                        "use_assistant_message": use_assistant_message,
-                    },
-                    message_list_params.MessageListParams,
-                ),
-            ),
-            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
-        )
-
     def reset(
         self,
         group_id: str,
@@ -538,8 +538,87 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         return AsyncMessagesResourceWithStreamingResponse(self)
 
+    def list(
+        self,
+        group_id: str,
+        *,
+        after: Optional[str] | Omit = omit,
+        assistant_message_tool_kwarg: str | Omit = omit,
+        assistant_message_tool_name: str | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        limit: Optional[int] | Omit = omit,
+        order: Literal["asc", "desc"] | Omit = omit,
+        order_by: Literal["created_at"] | Omit = omit,
+        use_assistant_message: bool | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> AsyncPaginator[LettaMessageUnion, AsyncArrayPage[LettaMessageUnion]]:
+        """
+        Retrieve message history for an agent.
+
+        Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
+          after: Message ID cursor for pagination. Returns messages that come after this message
+              ID in the specified sort order
+
+          assistant_message_tool_kwarg: The name of the message argument.
+
+          assistant_message_tool_name: The name of the designated message tool.
+
+          before: Message ID cursor for pagination. Returns messages that come before this message
+              ID in the specified sort order
+
+          limit: Maximum number of messages to retrieve
+
+          order: Sort order for messages by creation time. 'asc' for oldest first, 'desc' for
+              newest first
+
+          order_by: Field to sort by
+
+          use_assistant_message: Whether to use assistant messages
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not group_id:
+            raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
+        return self._get_api_list(
+            f"/v1/groups/{group_id}/messages",
+            page=AsyncArrayPage[LettaMessageUnion],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
+                        "assistant_message_tool_name": assistant_message_tool_name,
+                        "before": before,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                        "use_assistant_message": use_assistant_message,
+                    },
+                    message_list_params.MessageListParams,
+                ),
+            ),
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
+        )
+
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -552,7 +631,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -575,7 +654,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -588,7 +667,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -611,7 +690,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -624,7 +703,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -644,7 +723,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -657,7 +736,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         """
         Update the details of a message associated with an agent.
 
@@ -680,7 +759,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @required_args(["group_id", "content"], ["group_id", "reasoning"])
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -698,13 +777,13 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageUpdateResponse:
+    ) -> MessageModifyResponse:
         if not group_id:
             raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
         if not message_id:
             raise ValueError(f"Expected a non-empty value for `message_id` but received {message_id!r}")
         return cast(
-            MessageUpdateResponse,
+            MessageModifyResponse,
             await self._patch(
                 f"/v1/groups/{group_id}/messages/{message_id}",
                 body=await async_maybe_transform(
@@ -713,96 +792,17 @@ class AsyncMessagesResource(AsyncAPIResource):
                         "message_type": message_type,
                         "reasoning": reasoning,
                     },
-                    message_update_params.MessageUpdateParams,
+                    message_modify_params.MessageModifyParams,
                 ),
                 options=make_request_options(
                     extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
                 ),
                 cast_to=cast(
-                    Any, MessageUpdateResponse
+                    Any, MessageModifyResponse
                 ),  # Union types cannot be passed in as arguments in the type system
             ),
         )
 
-    def list(
-        self,
-        group_id: str,
-        *,
-        after: Optional[str] | Omit = omit,
-        assistant_message_tool_kwarg: str | Omit = omit,
-        assistant_message_tool_name: str | Omit = omit,
-        before: Optional[str] | Omit = omit,
-        limit: Optional[int] | Omit = omit,
-        order: Literal["asc", "desc"] | Omit = omit,
-        order_by: Literal["created_at"] | Omit = omit,
-        use_assistant_message: bool | Omit = omit,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[LettaMessageUnion, AsyncArrayPage[LettaMessageUnion]]:
-        """
-        Retrieve message history for an agent.
-
-        Args:
-          group_id: The ID of the group in the format 'group-<uuid4>'
-
-          after: Message ID cursor for pagination. Returns messages that come after this message
-              ID in the specified sort order
-
-          assistant_message_tool_kwarg: The name of the message argument.
-
-          assistant_message_tool_name: The name of the designated message tool.
-
-          before: Message ID cursor for pagination. Returns messages that come before this message
-              ID in the specified sort order
-
-          limit: Maximum number of messages to retrieve
-
-          order: Sort order for messages by creation time. 'asc' for oldest first, 'desc' for
-              newest first
-
-          order_by: Field to sort by
-
-          use_assistant_message: Whether to use assistant messages
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        if not group_id:
-            raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        return self._get_api_list(
-            f"/v1/groups/{group_id}/messages",
-            page=AsyncArrayPage[LettaMessageUnion],
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "after": after,
-                        "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                        "assistant_message_tool_name": assistant_message_tool_name,
-                        "before": before,
-                        "limit": limit,
-                        "order": order,
-                        "order_by": order_by,
-                        "use_assistant_message": use_assistant_message,
-                    },
-                    message_list_params.MessageListParams,
-                ),
-            ),
-            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
-        )
-
     async def reset(
         self,
         group_id: str,
@@ -1010,12 +1010,12 @@ class MessagesResourceWithRawResponse:
     def __init__(self, messages: MessagesResource) -> None:
         self._messages = messages
 
-        self.update = to_raw_response_wrapper(
-            messages.update,
-        )
         self.list = to_raw_response_wrapper(
             messages.list,
         )
+        self.modify = to_raw_response_wrapper(
+            messages.modify,
+        )
         self.reset = to_raw_response_wrapper(
             messages.reset,
         )
@@ -1031,12 +1031,12 @@ class AsyncMessagesResourceWithRawResponse:
     def __init__(self, messages: AsyncMessagesResource) -> None:
         self._messages = messages
 
-        self.update = async_to_raw_response_wrapper(
-            messages.update,
-        )
         self.list = async_to_raw_response_wrapper(
             messages.list,
         )
+        self.modify = async_to_raw_response_wrapper(
+            messages.modify,
+        )
         self.reset = async_to_raw_response_wrapper(
             messages.reset,
         )
@@ -1052,12 +1052,12 @@ class MessagesResourceWithStreamingResponse:
     def __init__(self, messages: MessagesResource) -> None:
         self._messages = messages
 
-        self.update = to_streamed_response_wrapper(
-            messages.update,
-        )
         self.list = to_streamed_response_wrapper(
             messages.list,
         )
+        self.modify = to_streamed_response_wrapper(
+            messages.modify,
+        )
         self.reset = to_streamed_response_wrapper(
             messages.reset,
         )
@@ -1073,12 +1073,12 @@ class AsyncMessagesResourceWithStreamingResponse:
     def __init__(self, messages: AsyncMessagesResource) -> None:
         self._messages = messages
 
-        self.update = async_to_streamed_response_wrapper(
-            messages.update,
-        )
         self.list = async_to_streamed_response_wrapper(
             messages.list,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            messages.modify,
+        )
         self.reset = async_to_streamed_response_wrapper(
             messages.reset,
         )

letta_client/resources/identities/blocks.py

@@ -20,7 +20,7 @@ from ..._response import (
 from ...pagination import SyncArrayPage, AsyncArrayPage
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.identities import block_list_params
-from ...types.agents.block import Block
+from ...types.block_response import BlockResponse
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -60,7 +60,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Block]:
+    ) -> SyncArrayPage[BlockResponse]:
         """
         Get all blocks associated with the specified identity.
 
@@ -92,7 +92,7 @@ class BlocksResource(SyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
         return self._get_api_list(
             f"/v1/identities/{identity_id}/blocks",
-            page=SyncArrayPage[Block],
+            page=SyncArrayPage[BlockResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -109,7 +109,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=Block,
+            model=BlockResponse,
         )
 
 
@@ -148,7 +148,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Block, AsyncArrayPage[Block]]:
+    ) -> AsyncPaginator[BlockResponse, AsyncArrayPage[BlockResponse]]:
         """
         Get all blocks associated with the specified identity.
 
@@ -180,7 +180,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
         return self._get_api_list(
             f"/v1/identities/{identity_id}/blocks",
-            page=AsyncArrayPage[Block],
+            page=AsyncArrayPage[BlockResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -197,7 +197,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=Block,
+            model=BlockResponse,
         )