letta-client 0.1.295__py3-none-any.whl → 0.1.300__py3-none-any.whl

letta_client/agents/raw_client.py

@@ -981,6 +981,67 @@ class RawAgentsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    def modify_approval(
+        self,
+        agent_id: str,
+        tool_name: str,
+        *,
+        requires_approval: bool,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[AgentState]:
+        """
+        Attach a tool to an agent.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        tool_name : str
+
+        requires_approval : bool
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[AgentState]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/tools/approval/{jsonable_encoder(tool_name)}",
+            method="PATCH",
+            params={
+                "requires_approval": requires_approval,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AgentState,
+                    construct_type(
+                        type_=AgentState,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
     def list_agent_files(
         self,
         agent_id: str,
@@ -2124,6 +2185,67 @@ class AsyncRawAgentsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    async def modify_approval(
+        self,
+        agent_id: str,
+        tool_name: str,
+        *,
+        requires_approval: bool,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[AgentState]:
+        """
+        Attach a tool to an agent.
+
+        Parameters
+        ----------
+        agent_id : str
+
+        tool_name : str
+
+        requires_approval : bool
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[AgentState]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/tools/approval/{jsonable_encoder(tool_name)}",
+            method="PATCH",
+            params={
+                "requires_approval": requires_approval,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AgentState,
+                    construct_type(
+                        type_=AgentState,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
     async def list_agent_files(
         self,
         agent_id: str,

letta_client/agents/tools/client.py

@@ -127,6 +127,39 @@ class ToolsClient:
         _response = self._raw_client.detach(agent_id, tool_id, request_options=request_options)
         return _response.data
 
+    def modify_approval(
+        self, agent_id: str, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Parameters
+        ----------
+        agent_id : str
+
+        tool_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+        client.agents.tools.modify_approval(
+            agent_id="agent_id",
+            tool_id="tool_id",
+        )
+        """
+        _response = self._raw_client.modify_approval(agent_id, tool_id, request_options=request_options)
+        return _response.data
+
 
 class AsyncToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -271,3 +304,44 @@ class AsyncToolsClient:
         """
         _response = await self._raw_client.detach(agent_id, tool_id, request_options=request_options)
         return _response.data
+
+    async def modify_approval(
+        self, agent_id: str, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Parameters
+        ----------
+        agent_id : str
+
+        tool_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            project="YOUR_PROJECT",
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.agents.tools.modify_approval(
+                agent_id="agent_id",
+                tool_id="tool_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.modify_approval(agent_id, tool_id, request_options=request_options)
+        return _response.data

letta_client/agents/tools/raw_client.py

@@ -170,6 +170,36 @@ class RawToolsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
+    def modify_approval(
+        self, agent_id: str, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[None]:
+        """
+        Parameters
+        ----------
+        agent_id : str
+
+        tool_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[None]
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/tools/approval/{jsonable_encoder(tool_id)}",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return HttpResponse(response=_response, data=None)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
 
 class AsyncRawToolsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -325,3 +355,33 @@ class AsyncRawToolsClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def modify_approval(
+        self, agent_id: str, tool_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[None]:
+        """
+        Parameters
+        ----------
+        agent_id : str
+
+        tool_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[None]
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/agents/{jsonable_encoder(agent_id)}/tools/approval/{jsonable_encoder(tool_id)}",
+            method="PATCH",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return AsyncHttpResponse(response=_response, data=None)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

letta_client/client.py

@@ -8,6 +8,7 @@ from .base_client import AsyncLettaBase, LettaBase
 from .core.request_options import RequestOptions
 from .tools.client import ToolsClient as ToolsClientBase
 from .tools.client import AsyncToolsClient as AsyncToolsClientBase
+from .types.npm_requirement import NpmRequirement
 from .types.pip_requirement import PipRequirement
 from .types.tool import Tool
 
@@ -144,6 +145,8 @@ class ToolsClient(ToolsClientBase):
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
         pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
+        npm_requirements: typing.Optional[typing.Sequence[NpmRequirement]] = OMIT,
+        default_requires_approval: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -175,6 +178,12 @@ class ToolsClient(ToolsClientBase):
         pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
             Optional list of pip packages required by this tool.
 
+        npm_requirements : typing.Optional[typing.Sequence[NpmRequirement]]
+            Optional list of npm packages required by this tool.
+
+        default_requires_approval : typing.Optional[bool]
+            Whether or not to require approval before executing this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -221,6 +230,8 @@ class ToolsClient(ToolsClientBase):
             json_schema=json_schema,
             return_char_limit=return_char_limit,
             pip_requirements=pip_requirements,
+            npm_requirements=npm_requirements,
+            default_requires_approval=default_requires_approval,
             request_options=request_options,
         )
 
@@ -238,6 +249,8 @@ class ToolsClient(ToolsClientBase):
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
         pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
+        npm_requirements: typing.Optional[typing.Sequence[NpmRequirement]] = OMIT,
+        default_requires_approval: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -269,6 +282,12 @@ class ToolsClient(ToolsClientBase):
         pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
             Optional list of pip packages required by this tool.
 
+        npm_requirements : typing.Optional[typing.Sequence[NpmRequirement]]
+            Optional list of npm packages required by this tool.
+
+        default_requires_approval : typing.Optional[bool]
+            Whether or not to require approval before executing this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -315,6 +334,8 @@ class ToolsClient(ToolsClientBase):
             json_schema=json_schema,
             return_char_limit=return_char_limit,
             pip_requirements=pip_requirements,
+            npm_requirements=npm_requirements,
+            default_requires_approval=default_requires_approval,
             request_options=request_options,
         )
     
@@ -391,6 +412,8 @@ class ToolsClient(ToolsClientBase):
             json_schema=tool.json_schema or OMIT,
             return_char_limit=tool.return_char_limit or OMIT,
             pip_requirements=tool.pip_requirements or OMIT,
+            npm_requirements=tool.npm_requirements or OMIT,
+            default_requires_approval=tool.default_requires_approval or OMIT,
             request_options=request_options,
         )
     
@@ -410,6 +433,8 @@ class AsyncToolsClient(AsyncToolsClientBase):
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
         pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
+        npm_requirements: typing.Optional[typing.Sequence[NpmRequirement]] = OMIT,
+        default_requires_approval: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -441,6 +466,12 @@ class AsyncToolsClient(AsyncToolsClientBase):
         pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
             Optional list of pip packages required by this tool.
 
+        npm_requirements : typing.Optional[typing.Sequence[NpmRequirement]]
+            Optional list of npm packages required by this tool.
+
+        default_requires_approval : typing.Optional[bool]
+            Whether or not to require approval before executing this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -487,6 +518,8 @@ class AsyncToolsClient(AsyncToolsClientBase):
             json_schema=json_schema,
             return_char_limit=return_char_limit,
             pip_requirements=pip_requirements,
+            npm_requirements=npm_requirements,
+            default_requires_approval=default_requires_approval,
             request_options=request_options,
         )
 
@@ -504,6 +537,8 @@ class AsyncToolsClient(AsyncToolsClientBase):
         ] = OMIT,
         return_char_limit: typing.Optional[int] = OMIT,
         pip_requirements: typing.Optional[typing.Sequence[PipRequirement]] = OMIT,
+        npm_requirements: typing.Optional[typing.Sequence[NpmRequirement]] = OMIT,
+        default_requires_approval: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> Tool:
         """
@@ -535,6 +570,12 @@ class AsyncToolsClient(AsyncToolsClientBase):
         pip_requirements : typing.Optional[typing.Sequence[PipRequirement]]
             Optional list of pip packages required by this tool.
 
+        npm_requirements : typing.Optional[typing.Sequence[NpmRequirement]]
+            Optional list of npm packages required by this tool.
+
+        default_requires_approval : typing.Optional[bool]
+            Whether or not to require approval before executing this tool.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -581,6 +622,8 @@ class AsyncToolsClient(AsyncToolsClientBase):
             json_schema=json_schema,
             return_char_limit=return_char_limit,
             pip_requirements=pip_requirements,
+            npm_requirements=npm_requirements,
+            default_requires_approval=default_requires_approval,
             request_options=request_options,
         )
     
@@ -657,5 +700,7 @@ class AsyncToolsClient(AsyncToolsClientBase):
             json_schema=tool.json_schema or OMIT,
             return_char_limit=tool.return_char_limit or OMIT,
             pip_requirements=tool.pip_requirements or OMIT,
+            npm_requirements=tool.npm_requirements or OMIT,
+            default_requires_approval=tool.default_requires_approval or OMIT,
             request_options=request_options,
         )

letta_client/core/client_wrapper.py

@@ -24,10 +24,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "letta-client/0.1.295",
+            "User-Agent": "letta-client/0.1.300",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.295",
+            "X-Fern-SDK-Version": "0.1.300",
             **(self.get_custom_headers() or {}),
         }
         if self._project is not None:

letta_client/groups/messages/client.py

@@ -5,8 +5,9 @@ import typing
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.letta_message_union import LettaMessageUnion
+from ...types.letta_request_messages_item import LettaRequestMessagesItem
 from ...types.letta_response import LettaResponse
-from ...types.message_create import MessageCreate
+from ...types.letta_streaming_request_messages_item import LettaStreamingRequestMessagesItem
 from ...types.message_type import MessageType
 from .raw_client import AsyncRawMessagesClient, RawMessagesClient
 from .types.letta_streaming_response import LettaStreamingResponse
@@ -105,7 +106,7 @@ class MessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -122,7 +123,7 @@ class MessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -190,7 +191,7 @@ class MessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaStreamingRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -211,7 +212,7 @@ class MessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaStreamingRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -233,7 +234,7 @@ class MessagesClient:
             If set to True, enables reasoning before responses or tool calls from the agent.
 
         stream_tokens : typing.Optional[bool]
-            Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
+            Flag to determine if individual tokens should be streamed, rather than streaming per step.
 
         include_pings : typing.Optional[bool]
             Whether to include periodic keepalive ping messages in the stream to prevent connection timeouts.
@@ -465,7 +466,7 @@ class AsyncMessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -482,7 +483,7 @@ class AsyncMessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -558,7 +559,7 @@ class AsyncMessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaStreamingRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -579,7 +580,7 @@ class AsyncMessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaStreamingRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -601,7 +602,7 @@ class AsyncMessagesClient:
             If set to True, enables reasoning before responses or tool calls from the agent.
 
         stream_tokens : typing.Optional[bool]
-            Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
+            Flag to determine if individual tokens should be streamed, rather than streaming per step.
 
         include_pings : typing.Optional[bool]
             Whether to include periodic keepalive ping messages in the stream to prevent connection timeouts.

letta_client/groups/messages/raw_client.py

@@ -16,8 +16,9 @@ from ...core.unchecked_base_model import construct_type
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.http_validation_error import HttpValidationError
 from ...types.letta_message_union import LettaMessageUnion
+from ...types.letta_request_messages_item import LettaRequestMessagesItem
 from ...types.letta_response import LettaResponse
-from ...types.message_create import MessageCreate
+from ...types.letta_streaming_request_messages_item import LettaStreamingRequestMessagesItem
 from ...types.message_type import MessageType
 from .types.letta_streaming_response import LettaStreamingResponse
 from .types.messages_modify_request import MessagesModifyRequest
@@ -119,7 +120,7 @@ class RawMessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -136,7 +137,7 @@ class RawMessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -170,7 +171,7 @@ class RawMessagesClient:
             method="POST",
             json={
                 "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
+                    object_=messages, annotation=typing.Sequence[LettaRequestMessagesItem], direction="write"
                 ),
                 "max_steps": max_steps,
                 "use_assistant_message": use_assistant_message,
@@ -216,7 +217,7 @@ class RawMessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaStreamingRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -237,7 +238,7 @@ class RawMessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaStreamingRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -259,7 +260,7 @@ class RawMessagesClient:
             If set to True, enables reasoning before responses or tool calls from the agent.
 
         stream_tokens : typing.Optional[bool]
-            Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
+            Flag to determine if individual tokens should be streamed, rather than streaming per step.
 
         include_pings : typing.Optional[bool]
             Whether to include periodic keepalive ping messages in the stream to prevent connection timeouts.
@@ -280,7 +281,7 @@ class RawMessagesClient:
             method="POST",
             json={
                 "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
+                    object_=messages, annotation=typing.Sequence[LettaStreamingRequestMessagesItem], direction="write"
                 ),
                 "max_steps": max_steps,
                 "use_assistant_message": use_assistant_message,
@@ -551,7 +552,7 @@ class AsyncRawMessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -568,7 +569,7 @@ class AsyncRawMessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -602,7 +603,7 @@ class AsyncRawMessagesClient:
             method="POST",
             json={
                 "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
+                    object_=messages, annotation=typing.Sequence[LettaRequestMessagesItem], direction="write"
                 ),
                 "max_steps": max_steps,
                 "use_assistant_message": use_assistant_message,
@@ -648,7 +649,7 @@ class AsyncRawMessagesClient:
         self,
         group_id: str,
         *,
-        messages: typing.Sequence[MessageCreate],
+        messages: typing.Sequence[LettaStreamingRequestMessagesItem],
         max_steps: typing.Optional[int] = OMIT,
         use_assistant_message: typing.Optional[bool] = OMIT,
         assistant_message_tool_name: typing.Optional[str] = OMIT,
@@ -669,7 +670,7 @@ class AsyncRawMessagesClient:
         ----------
         group_id : str
 
-        messages : typing.Sequence[MessageCreate]
+        messages : typing.Sequence[LettaStreamingRequestMessagesItem]
             The messages to be sent to the agent.
 
         max_steps : typing.Optional[int]
@@ -691,7 +692,7 @@ class AsyncRawMessagesClient:
             If set to True, enables reasoning before responses or tool calls from the agent.
 
         stream_tokens : typing.Optional[bool]
-            Flag to determine if individual tokens should be streamed. Set to True for token streaming (requires stream_steps = True).
+            Flag to determine if individual tokens should be streamed, rather than streaming per step.
 
         include_pings : typing.Optional[bool]
             Whether to include periodic keepalive ping messages in the stream to prevent connection timeouts.
@@ -712,7 +713,7 @@ class AsyncRawMessagesClient:
             method="POST",
             json={
                 "messages": convert_and_respect_annotation_metadata(
-                    object_=messages, annotation=typing.Sequence[MessageCreate], direction="write"
+                    object_=messages, annotation=typing.Sequence[LettaStreamingRequestMessagesItem], direction="write"
                 ),
                 "max_steps": max_steps,
                 "use_assistant_message": use_assistant_message,