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

letta_client/resources/agents/messages.py

@@ -17,6 +17,7 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ..._streaming import Stream, AsyncStream
 from ...pagination import SyncArrayPage, AsyncArrayPage
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import (
@@ -24,8 +25,8 @@ from ...types.agents import (
     message_send_params,
     message_reset_params,
     message_cancel_params,
+    message_modify_params,
     message_stream_params,
-    message_update_params,
     message_send_async_params,
 )
 from ...types.agents.run import Run
@@ -34,7 +35,8 @@ from ...types.agents.message_type import MessageType
 from ...types.agents.letta_response import LettaResponse
 from ...types.agents.letta_message_union import LettaMessageUnion
 from ...types.agents.message_cancel_response import MessageCancelResponse
-from ...types.agents.message_update_response import MessageUpdateResponse
+from ...types.agents.message_modify_response import MessageModifyResponse
+from ...types.agents.letta_streaming_response import LettaStreamingResponse
 from ...types.agents.letta_user_message_content_union_param import LettaUserMessageContentUnionParam
 from ...types.agents.letta_assistant_message_content_union_param import LettaAssistantMessageContentUnionParam
 
@@ -61,8 +63,139 @@ class MessagesResource(SyncAPIResource):
         """
         return MessagesResourceWithStreamingResponse(self)
 
+    def list(
+        self,
+        agent_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,
+        group_id: Optional[str] | Omit = omit,
+        include_err: Optional[bool] | 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:
+          agent_id: The ID of the agent in the format 'agent-<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
+
+          group_id: Group ID to filter messages by.
+
+          include_err: Whether to include error messages and error statuses. For debugging purposes
+              only.
+
+          limit: Maximum number of messages to return
+
+          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 agent_id:
+            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
+        return self._get_api_list(
+            f"/v1/agents/{agent_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,
+                        "group_id": group_id,
+                        "include_err": include_err,
+                        "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 cancel(
+        self,
+        agent_id: str,
+        *,
+        run_ids: Optional[SequenceNotStr[str]] | 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,
+    ) -> MessageCancelResponse:
+        """Cancel runs associated with an agent.
+
+        If run_ids are passed in, cancel those in
+        particular.
+
+        Note to cancel active runs associated with an agent, redis is required.
+
+        Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          run_ids: Optional list of run IDs to cancel
+
+          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 agent_id:
+            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
+        return self._post(
+            f"/v1/agents/{agent_id}/messages/cancel",
+            body=maybe_transform({"run_ids": run_ids}, message_cancel_params.MessageCancelParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=MessageCancelResponse,
+        )
+
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -75,7 +208,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.
 
@@ -98,7 +231,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -111,7 +244,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.
 
@@ -134,7 +267,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -147,7 +280,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.
 
@@ -167,7 +300,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @overload
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -180,7 +313,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.
 
@@ -203,7 +336,7 @@ class MessagesResource(SyncAPIResource):
         ...
 
     @required_args(["agent_id", "content"], ["agent_id", "reasoning"])
-    def update(
+    def modify(
         self,
         message_id: str,
         *,
@@ -221,13 +354,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 agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_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/agents/{agent_id}/messages/{message_id}",
                 body=maybe_transform(
@@ -236,148 +369,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,
-        agent_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,
-        group_id: Optional[str] | Omit = omit,
-        include_err: Optional[bool] | 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:
-          agent_id: The ID of the agent in the format 'agent-<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
-
-          group_id: Group ID to filter messages by.
-
-          include_err: Whether to include error messages and error statuses. For debugging purposes
-              only.
-
-          limit: Maximum number of messages to return
-
-          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 agent_id:
-            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
-            f"/v1/agents/{agent_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,
-                        "group_id": group_id,
-                        "include_err": include_err,
-                        "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 cancel(
-        self,
-        agent_id: str,
-        *,
-        run_ids: Optional[SequenceNotStr[str]] | 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,
-    ) -> MessageCancelResponse:
-        """Cancel runs associated with an agent.
-
-        If run_ids are passed in, cancel those in
-        particular.
-
-        Note to cancel active runs associated with an agent, redis is required.
-
-        Args:
-          agent_id: The ID of the agent in the format 'agent-<uuid4>'
-
-          run_ids: Optional list of run IDs to cancel
-
-          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 agent_id:
-            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._post(
-            f"/v1/agents/{agent_id}/messages/cancel",
-            body=maybe_transform({"run_ids": run_ids}, message_cancel_params.MessageCancelParams),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=MessageCancelResponse,
-        )
-
     def reset(
         self,
         agent_id: str,
@@ -597,7 +599,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
+    ) -> Stream[LettaStreamingResponse]:
         """Process a user message and return the agent's response.
 
         This endpoint accepts a
@@ -607,32 +609,177 @@ class MessagesResource(SyncAPIResource):
         Args:
           agent_id: The ID of the agent in the format 'agent-<uuid4>'
 
-          messages: The messages to be sent to the agent.
+          messages: The messages to be sent to the agent.
+
+          assistant_message_tool_kwarg: The name of the message argument in the designated message tool. Still supported
+              for legacy agent types, but deprecated for letta_v1_agent onward.
+
+          assistant_message_tool_name: The name of the designated message tool. Still supported for legacy agent types,
+              but deprecated for letta_v1_agent onward.
+
+          background: Whether to process the request in the background.
+
+          enable_thinking: If set to True, enables reasoning before responses or tool calls from the agent.
+
+          include_pings: Whether to include periodic keepalive ping messages in the stream to prevent
+              connection timeouts.
+
+          include_return_message_types: Only return specified message types in the response. If `None` (default) returns
+              all messages.
+
+          max_steps: Maximum number of steps the agent should take to process the request.
+
+          stream_tokens: Flag to determine if individual tokens should be streamed, rather than streaming
+              per step.
+
+          use_assistant_message: Whether the server should parse specific tool call arguments (default
+              `send_message`) as `AssistantMessage` objects. Still supported for legacy agent
+              types, but deprecated for letta_v1_agent onward.
+
+          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 agent_id:
+            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
+        return self._post(
+            f"/v1/agents/{agent_id}/messages/stream",
+            body=maybe_transform(
+                {
+                    "messages": messages,
+                    "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
+                    "assistant_message_tool_name": assistant_message_tool_name,
+                    "background": background,
+                    "enable_thinking": enable_thinking,
+                    "include_pings": include_pings,
+                    "include_return_message_types": include_return_message_types,
+                    "max_steps": max_steps,
+                    "stream_tokens": stream_tokens,
+                    "use_assistant_message": use_assistant_message,
+                },
+                message_stream_params.MessageStreamParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=cast(
+                Any, LettaStreamingResponse
+            ),  # Union types cannot be passed in as arguments in the type system
+            stream=True,
+            stream_cls=Stream[LettaStreamingResponse],
+        )
+
+    def summarize(
+        self,
+        agent_id: str,
+        *,
+        # 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,
+    ) -> None:
+        """
+        Summarize an agent's conversation history.
+
+        Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          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 agent_id:
+            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return self._post(
+            f"/v1/agents/{agent_id}/summarize",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class AsyncMessagesResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncMessagesResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/letta-ai/letta-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncMessagesResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncMessagesResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/letta-ai/letta-python#with_streaming_response
+        """
+        return AsyncMessagesResourceWithStreamingResponse(self)
+
+    def list(
+        self,
+        agent_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,
+        group_id: Optional[str] | Omit = omit,
+        include_err: Optional[bool] | 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:
+          agent_id: The ID of the agent in the format 'agent-<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 in the designated message tool. Still supported
-              for legacy agent types, but deprecated for letta_v1_agent onward.
+          assistant_message_tool_kwarg: The name of the message argument.
 
-          assistant_message_tool_name: The name of the designated message tool. Still supported for legacy agent types,
-              but deprecated for letta_v1_agent onward.
+          assistant_message_tool_name: The name of the designated message tool.
 
-          background: Whether to process the request in the background.
+          before: Message ID cursor for pagination. Returns messages that come before this message
+              ID in the specified sort order
 
-          enable_thinking: If set to True, enables reasoning before responses or tool calls from the agent.
+          group_id: Group ID to filter messages by.
 
-          include_pings: Whether to include periodic keepalive ping messages in the stream to prevent
-              connection timeouts.
+          include_err: Whether to include error messages and error statuses. For debugging purposes
+              only.
 
-          include_return_message_types: Only return specified message types in the response. If `None` (default) returns
-              all messages.
+          limit: Maximum number of messages to return
 
-          max_steps: Maximum number of steps the agent should take to process the request.
+          order: Sort order for messages by creation time. 'asc' for oldest first, 'desc' for
+              newest first
 
-          stream_tokens: Flag to determine if individual tokens should be streamed, rather than streaming
-              per step.
+          order_by: Field to sort by
 
-          use_assistant_message: Whether the server should parse specific tool call arguments (default
-              `send_message`) as `AssistantMessage` objects. Still supported for legacy agent
-              types, but deprecated for letta_v1_agent onward.
+          use_assistant_message: Whether to use assistant messages
 
           extra_headers: Send extra headers
 
@@ -644,46 +791,57 @@ class MessagesResource(SyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._post(
-            f"/v1/agents/{agent_id}/messages/stream",
-            body=maybe_transform(
-                {
-                    "messages": messages,
-                    "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
-                    "assistant_message_tool_name": assistant_message_tool_name,
-                    "background": background,
-                    "enable_thinking": enable_thinking,
-                    "include_pings": include_pings,
-                    "include_return_message_types": include_return_message_types,
-                    "max_steps": max_steps,
-                    "stream_tokens": stream_tokens,
-                    "use_assistant_message": use_assistant_message,
-                },
-                message_stream_params.MessageStreamParams,
-            ),
+        return self._get_api_list(
+            f"/v1/agents/{agent_id}/messages",
+            page=AsyncArrayPage[LettaMessageUnion],
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                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,
+                        "group_id": group_id,
+                        "include_err": include_err,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                        "use_assistant_message": use_assistant_message,
+                    },
+                    message_list_params.MessageListParams,
+                ),
             ),
-            cast_to=object,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
-    def summarize(
+    async def cancel(
         self,
         agent_id: str,
         *,
+        run_ids: Optional[SequenceNotStr[str]] | 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,
-    ) -> None:
-        """
-        Summarize an agent's conversation history.
+    ) -> MessageCancelResponse:
+        """Cancel runs associated with an agent.
+
+        If run_ids are passed in, cancel those in
+        particular.
+
+        Note to cancel active runs associated with an agent, redis is required.
 
         Args:
           agent_id: The ID of the agent in the format 'agent-<uuid4>'
 
+          run_ids: Optional list of run IDs to cancel
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -694,38 +852,17 @@ class MessagesResource(SyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return self._post(
-            f"/v1/agents/{agent_id}/summarize",
+        return await self._post(
+            f"/v1/agents/{agent_id}/messages/cancel",
+            body=await async_maybe_transform({"run_ids": run_ids}, message_cancel_params.MessageCancelParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=MessageCancelResponse,
         )
 
-
-class AsyncMessagesResource(AsyncAPIResource):
-    @cached_property
-    def with_raw_response(self) -> AsyncMessagesResourceWithRawResponse:
-        """
-        This property can be used as a prefix for any HTTP method call to return
-        the raw response object instead of the parsed content.
-
-        For more information, see https://www.github.com/letta-ai/letta-python#accessing-raw-response-data-eg-headers
-        """
-        return AsyncMessagesResourceWithRawResponse(self)
-
-    @cached_property
-    def with_streaming_response(self) -> AsyncMessagesResourceWithStreamingResponse:
-        """
-        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
-
-        For more information, see https://www.github.com/letta-ai/letta-python#with_streaming_response
-        """
-        return AsyncMessagesResourceWithStreamingResponse(self)
-
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -738,7 +875,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.
 
@@ -761,7 +898,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -774,7 +911,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.
 
@@ -797,7 +934,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -810,7 +947,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.
 
@@ -830,7 +967,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @overload
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -843,7 +980,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.
 
@@ -866,7 +1003,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         ...
 
     @required_args(["agent_id", "content"], ["agent_id", "reasoning"])
-    async def update(
+    async def modify(
         self,
         message_id: str,
         *,
@@ -884,13 +1021,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 agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_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/agents/{agent_id}/messages/{message_id}",
                 body=await async_maybe_transform(
@@ -899,148 +1036,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,
-        agent_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,
-        group_id: Optional[str] | Omit = omit,
-        include_err: Optional[bool] | 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:
-          agent_id: The ID of the agent in the format 'agent-<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
-
-          group_id: Group ID to filter messages by.
-
-          include_err: Whether to include error messages and error statuses. For debugging purposes
-              only.
-
-          limit: Maximum number of messages to return
-
-          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 agent_id:
-            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
-            f"/v1/agents/{agent_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,
-                        "group_id": group_id,
-                        "include_err": include_err,
-                        "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 cancel(
-        self,
-        agent_id: str,
-        *,
-        run_ids: Optional[SequenceNotStr[str]] | 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,
-    ) -> MessageCancelResponse:
-        """Cancel runs associated with an agent.
-
-        If run_ids are passed in, cancel those in
-        particular.
-
-        Note to cancel active runs associated with an agent, redis is required.
-
-        Args:
-          agent_id: The ID of the agent in the format 'agent-<uuid4>'
-
-          run_ids: Optional list of run IDs to cancel
-
-          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 agent_id:
-            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return await self._post(
-            f"/v1/agents/{agent_id}/messages/cancel",
-            body=await async_maybe_transform({"run_ids": run_ids}, message_cancel_params.MessageCancelParams),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=MessageCancelResponse,
-        )
-
     async def reset(
         self,
         agent_id: str,
@@ -1260,7 +1266,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> object:
+    ) -> AsyncStream[LettaStreamingResponse]:
         """Process a user message and return the agent's response.
 
         This endpoint accepts a
@@ -1327,7 +1333,11 @@ class AsyncMessagesResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=object,
+            cast_to=cast(
+                Any, LettaStreamingResponse
+            ),  # Union types cannot be passed in as arguments in the type system
+            stream=True,
+            stream_cls=AsyncStream[LettaStreamingResponse],
         )
 
     async def summarize(
@@ -1371,15 +1381,15 @@ 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.cancel = to_raw_response_wrapper(
             messages.cancel,
         )
+        self.modify = to_raw_response_wrapper(
+            messages.modify,
+        )
         self.reset = to_raw_response_wrapper(
             messages.reset,
         )
@@ -1401,15 +1411,15 @@ 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.cancel = async_to_raw_response_wrapper(
             messages.cancel,
         )
+        self.modify = async_to_raw_response_wrapper(
+            messages.modify,
+        )
         self.reset = async_to_raw_response_wrapper(
             messages.reset,
         )
@@ -1431,15 +1441,15 @@ 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.cancel = to_streamed_response_wrapper(
             messages.cancel,
         )
+        self.modify = to_streamed_response_wrapper(
+            messages.modify,
+        )
         self.reset = to_streamed_response_wrapper(
             messages.reset,
         )
@@ -1461,15 +1471,15 @@ 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.cancel = async_to_streamed_response_wrapper(
             messages.cancel,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            messages.modify,
+        )
         self.reset = async_to_streamed_response_wrapper(
             messages.reset,
         )