letta-client 1.0.0a13__py3-none-any.whl → 1.0.0a15__py3-none-any.whl

letta_client/_client.py

@@ -91,6 +91,7 @@ class Letta(SyncAPIClient):
     # client options
     api_key: str
     project_id: str | None
+    project: str | None
 
     _environment: Literal["cloud", "local"] | NotGiven
 
@@ -99,6 +100,7 @@ class Letta(SyncAPIClient):
         *,
         api_key: str | None = None,
         project_id: str | None = None,
+        project: str | None = None,
         environment: Literal["cloud", "local"] | NotGiven = not_given,
         base_url: str | httpx.URL | None | NotGiven = not_given,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -133,6 +135,8 @@ class Letta(SyncAPIClient):
 
         self.project_id = project_id
 
+        self.project = project
+
         self._environment = environment
 
         base_url_env = os.environ.get("LETTA_BASE_URL")
@@ -206,7 +210,8 @@ class Letta(SyncAPIClient):
         return {
             **super().default_headers,
             "X-Stainless-Async": "false",
-            "X-Project": self.project_id if self.project_id is not None else Omit(),
+            "X-Project-Id": self.project_id if self.project_id is not None else Omit(),
+            "X-Project": self.project if self.project is not None else Omit(),
             **self._custom_headers,
         }
 
@@ -215,6 +220,7 @@ class Letta(SyncAPIClient):
         *,
         api_key: str | None = None,
         project_id: str | None = None,
+        project: str | None = None,
         environment: Literal["cloud", "local"] | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -251,6 +257,7 @@ class Letta(SyncAPIClient):
         return self.__class__(
             api_key=api_key or self.api_key,
             project_id=project_id or self.project_id,
+            project=project or self.project,
             base_url=base_url or self.base_url,
             environment=environment or self._environment,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,
@@ -339,6 +346,7 @@ class AsyncLetta(AsyncAPIClient):
     # client options
     api_key: str
     project_id: str | None
+    project: str | None
 
     _environment: Literal["cloud", "local"] | NotGiven
 
@@ -347,6 +355,7 @@ class AsyncLetta(AsyncAPIClient):
         *,
         api_key: str | None = None,
         project_id: str | None = None,
+        project: str | None = None,
         environment: Literal["cloud", "local"] | NotGiven = not_given,
         base_url: str | httpx.URL | None | NotGiven = not_given,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -381,6 +390,8 @@ class AsyncLetta(AsyncAPIClient):
 
         self.project_id = project_id
 
+        self.project = project
+
         self._environment = environment
 
         base_url_env = os.environ.get("LETTA_BASE_URL")
@@ -454,7 +465,8 @@ class AsyncLetta(AsyncAPIClient):
         return {
             **super().default_headers,
             "X-Stainless-Async": f"async:{get_async_library()}",
-            "X-Project": self.project_id if self.project_id is not None else Omit(),
+            "X-Project-Id": self.project_id if self.project_id is not None else Omit(),
+            "X-Project": self.project if self.project is not None else Omit(),
             **self._custom_headers,
         }
 
@@ -463,6 +475,7 @@ class AsyncLetta(AsyncAPIClient):
         *,
         api_key: str | None = None,
         project_id: str | None = None,
+        project: str | None = None,
         environment: Literal["cloud", "local"] | None = None,
         base_url: str | httpx.URL | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
@@ -499,6 +512,7 @@ class AsyncLetta(AsyncAPIClient):
         return self.__class__(
             api_key=api_key or self.api_key,
             project_id=project_id or self.project_id,
+            project=project or self.project,
             base_url=base_url or self.base_url,
             environment=environment or self._environment,
             timeout=self.timeout if isinstance(timeout, NotGiven) else timeout,

letta_client/_utils/_utils.py

@@ -133,7 +133,7 @@ def is_given(obj: _T | NotGiven | Omit) -> TypeGuard[_T]:
 # Type safe methods for narrowing types with TypeVars.
 # The default narrowing for isinstance(obj, dict) is dict[unknown, unknown],
 # however this cause Pyright to rightfully report errors. As we know we don't
-# care about the contained types we can safely use `object` in it's place.
+# care about the contained types we can safely use `object` in its place.
 #
 # There are two separate functions defined, `is_*` and `is_*_t` for different use cases.
 # `is_*` is for when you're dealing with an unknown input

letta_client/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "letta_client"
-__version__ = "1.0.0-alpha.13"  # x-release-please-version
+__version__ = "1.0.0-alpha.15"  # x-release-please-version

letta_client/pagination.py

@@ -35,7 +35,7 @@ class ObjectPageItem(Protocol):
 
 @runtime_checkable
 class NextFilesPageItem(Protocol):
-    next_cursor: Optional[str]
+    id: Optional[str]
 
 
 class SyncArrayPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
@@ -222,18 +222,18 @@ class SyncNextFilesPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
 
         if is_forwards:
             item = cast(Any, files[-1])
-            if not isinstance(item, NextFilesPageItem) or item.next_cursor is None:
+            if not isinstance(item, NextFilesPageItem) or item.id is None:
                 # TODO emit warning log
                 return None
 
-            return PageInfo(params={"after": item.next_cursor})
+            return PageInfo(params={"after": item.id})
         else:
             item = cast(Any, self.files[0])
-            if not isinstance(item, NextFilesPageItem) or item.next_cursor is None:
+            if not isinstance(item, NextFilesPageItem) or item.id is None:
                 # TODO emit warning log
                 return None
 
-            return PageInfo(params={"before": item.next_cursor})
+            return PageInfo(params={"before": item.id})
 
 
 class AsyncNextFilesPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
@@ -266,15 +266,15 @@ class AsyncNextFilesPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
 
         if is_forwards:
             item = cast(Any, files[-1])
-            if not isinstance(item, NextFilesPageItem) or item.next_cursor is None:
+            if not isinstance(item, NextFilesPageItem) or item.id is None:
                 # TODO emit warning log
                 return None
 
-            return PageInfo(params={"after": item.next_cursor})
+            return PageInfo(params={"after": item.id})
         else:
             item = cast(Any, self.files[0])
-            if not isinstance(item, NextFilesPageItem) or item.next_cursor is None:
+            if not isinstance(item, NextFilesPageItem) or item.id is None:
                 # TODO emit warning log
                 return None
 
-            return PageInfo(params={"before": item.next_cursor})
+            return PageInfo(params={"before": item.id})

letta_client/resources/agents/files.py

@@ -8,7 +8,7 @@ from typing_extensions import Literal
 import httpx
 
 from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from ..._utils import maybe_transform, async_maybe_transform
+from ..._utils import maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -17,7 +17,8 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncNextFilesPage, AsyncNextFilesPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import file_list_params
 from ...types.agents.file_list_response import FileListResponse
 from ...types.agents.file_open_response import FileOpenResponse
@@ -63,7 +64,7 @@ class FilesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> FileListResponse:
+    ) -> SyncNextFilesPage[FileListResponse]:
         """
         Get the files attached to an agent with their open/closed status.
 
@@ -97,8 +98,9 @@ class FilesResource(SyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/agents/{agent_id}/files",
+            page=SyncNextFilesPage[FileListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -117,7 +119,7 @@ class FilesResource(SyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            cast_to=FileListResponse,
+            model=FileListResponse,
         )
 
     def close(
@@ -266,7 +268,7 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         return AsyncFilesResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
@@ -283,7 +285,7 @@ class AsyncFilesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> FileListResponse:
+    ) -> AsyncPaginator[FileListResponse, AsyncNextFilesPage[FileListResponse]]:
         """
         Get the files attached to an agent with their open/closed status.
 
@@ -317,14 +319,15 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/agents/{agent_id}/files",
+            page=AsyncNextFilesPage[FileListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=await async_maybe_transform(
+                query=maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -337,7 +340,7 @@ class AsyncFilesResource(AsyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            cast_to=FileListResponse,
+            model=FileListResponse,
         )
 
     async def close(

letta_client/resources/agents/messages.py

@@ -425,12 +425,13 @@ class MessagesResource(SyncAPIResource):
         self,
         agent_id: str,
         *,
-        messages: Iterable[message_send_params.Message],
         assistant_message_tool_kwarg: str | Omit = omit,
         assistant_message_tool_name: str | Omit = omit,
         enable_thinking: str | Omit = omit,
         include_return_message_types: Optional[List[MessageType]] | Omit = omit,
+        input: Union[str, Iterable[message_send_params.InputUnionMember1], None] | Omit = omit,
         max_steps: int | Omit = omit,
+        messages: Optional[Iterable[message_send_params.Message]] | 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.
@@ -447,8 +448,6 @@ 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.
-
           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.
 
@@ -460,8 +459,14 @@ class MessagesResource(SyncAPIResource):
           include_return_message_types: Only return specified message types in the response. If `None` (default) returns
               all messages.
 
+          input:
+              Syntactic sugar for a single user message. Equivalent to messages=[{'role':
+              'user', 'content': input}].
+
           max_steps: Maximum number of steps the agent should take to process the request.
 
+          messages: The messages to be sent to the agent.
+
           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.
@@ -480,12 +485,13 @@ class MessagesResource(SyncAPIResource):
             f"/v1/agents/{agent_id}/messages",
             body=maybe_transform(
                 {
-                    "messages": messages,
                     "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                     "assistant_message_tool_name": assistant_message_tool_name,
                     "enable_thinking": enable_thinking,
                     "include_return_message_types": include_return_message_types,
+                    "input": input,
                     "max_steps": max_steps,
+                    "messages": messages,
                     "use_assistant_message": use_assistant_message,
                 },
                 message_send_params.MessageSendParams,
@@ -500,13 +506,14 @@ class MessagesResource(SyncAPIResource):
         self,
         agent_id: str,
         *,
-        messages: Iterable[message_send_async_params.Message],
         assistant_message_tool_kwarg: str | Omit = omit,
         assistant_message_tool_name: str | Omit = omit,
         callback_url: Optional[str] | Omit = omit,
         enable_thinking: str | Omit = omit,
         include_return_message_types: Optional[List[MessageType]] | Omit = omit,
+        input: Union[str, Iterable[message_send_async_params.InputUnionMember1], None] | Omit = omit,
         max_steps: int | Omit = omit,
+        messages: Optional[Iterable[message_send_async_params.Message]] | 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.
@@ -527,8 +534,6 @@ 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.
-
           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.
 
@@ -542,8 +547,14 @@ class MessagesResource(SyncAPIResource):
           include_return_message_types: Only return specified message types in the response. If `None` (default) returns
               all messages.
 
+          input:
+              Syntactic sugar for a single user message. Equivalent to messages=[{'role':
+              'user', 'content': input}].
+
           max_steps: Maximum number of steps the agent should take to process the request.
 
+          messages: The messages to be sent to the agent.
+
           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.
@@ -562,13 +573,14 @@ class MessagesResource(SyncAPIResource):
             f"/v1/agents/{agent_id}/messages/async",
             body=maybe_transform(
                 {
-                    "messages": messages,
                     "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                     "assistant_message_tool_name": assistant_message_tool_name,
                     "callback_url": callback_url,
                     "enable_thinking": enable_thinking,
                     "include_return_message_types": include_return_message_types,
+                    "input": input,
                     "max_steps": max_steps,
+                    "messages": messages,
                     "use_assistant_message": use_assistant_message,
                 },
                 message_send_async_params.MessageSendAsyncParams,
@@ -583,14 +595,15 @@ class MessagesResource(SyncAPIResource):
         self,
         agent_id: str,
         *,
-        messages: Iterable[message_stream_params.Message],
         assistant_message_tool_kwarg: str | Omit = omit,
         assistant_message_tool_name: str | Omit = omit,
         background: bool | Omit = omit,
         enable_thinking: str | Omit = omit,
         include_pings: bool | Omit = omit,
         include_return_message_types: Optional[List[MessageType]] | Omit = omit,
+        input: Union[str, Iterable[message_stream_params.InputUnionMember1], None] | Omit = omit,
         max_steps: int | Omit = omit,
+        messages: Optional[Iterable[message_stream_params.Message]] | Omit = omit,
         stream_tokens: bool | 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.
@@ -609,8 +622,6 @@ 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.
-
           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.
 
@@ -627,8 +638,14 @@ class MessagesResource(SyncAPIResource):
           include_return_message_types: Only return specified message types in the response. If `None` (default) returns
               all messages.
 
+          input:
+              Syntactic sugar for a single user message. Equivalent to messages=[{'role':
+              'user', 'content': input}].
+
           max_steps: Maximum number of steps the agent should take to process the request.
 
+          messages: The messages to be sent to the agent.
+
           stream_tokens: Flag to determine if individual tokens should be streamed, rather than streaming
               per step.
 
@@ -650,14 +667,15 @@ class MessagesResource(SyncAPIResource):
             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,
+                    "input": input,
                     "max_steps": max_steps,
+                    "messages": messages,
                     "stream_tokens": stream_tokens,
                     "use_assistant_message": use_assistant_message,
                 },
@@ -1092,12 +1110,13 @@ class AsyncMessagesResource(AsyncAPIResource):
         self,
         agent_id: str,
         *,
-        messages: Iterable[message_send_params.Message],
         assistant_message_tool_kwarg: str | Omit = omit,
         assistant_message_tool_name: str | Omit = omit,
         enable_thinking: str | Omit = omit,
         include_return_message_types: Optional[List[MessageType]] | Omit = omit,
+        input: Union[str, Iterable[message_send_params.InputUnionMember1], None] | Omit = omit,
         max_steps: int | Omit = omit,
+        messages: Optional[Iterable[message_send_params.Message]] | 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.
@@ -1114,8 +1133,6 @@ class AsyncMessagesResource(AsyncAPIResource):
         Args:
           agent_id: The ID of the agent in the format 'agent-<uuid4>'
 
-          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.
 
@@ -1127,8 +1144,14 @@ class AsyncMessagesResource(AsyncAPIResource):
           include_return_message_types: Only return specified message types in the response. If `None` (default) returns
               all messages.
 
+          input:
+              Syntactic sugar for a single user message. Equivalent to messages=[{'role':
+              'user', 'content': input}].
+
           max_steps: Maximum number of steps the agent should take to process the request.
 
+          messages: The messages to be sent to the agent.
+
           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.
@@ -1147,12 +1170,13 @@ class AsyncMessagesResource(AsyncAPIResource):
             f"/v1/agents/{agent_id}/messages",
             body=await async_maybe_transform(
                 {
-                    "messages": messages,
                     "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                     "assistant_message_tool_name": assistant_message_tool_name,
                     "enable_thinking": enable_thinking,
                     "include_return_message_types": include_return_message_types,
+                    "input": input,
                     "max_steps": max_steps,
+                    "messages": messages,
                     "use_assistant_message": use_assistant_message,
                 },
                 message_send_params.MessageSendParams,
@@ -1167,13 +1191,14 @@ class AsyncMessagesResource(AsyncAPIResource):
         self,
         agent_id: str,
         *,
-        messages: Iterable[message_send_async_params.Message],
         assistant_message_tool_kwarg: str | Omit = omit,
         assistant_message_tool_name: str | Omit = omit,
         callback_url: Optional[str] | Omit = omit,
         enable_thinking: str | Omit = omit,
         include_return_message_types: Optional[List[MessageType]] | Omit = omit,
+        input: Union[str, Iterable[message_send_async_params.InputUnionMember1], None] | Omit = omit,
         max_steps: int | Omit = omit,
+        messages: Optional[Iterable[message_send_async_params.Message]] | 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.
@@ -1194,8 +1219,6 @@ class AsyncMessagesResource(AsyncAPIResource):
         Args:
           agent_id: The ID of the agent in the format 'agent-<uuid4>'
 
-          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.
 
@@ -1209,8 +1232,14 @@ class AsyncMessagesResource(AsyncAPIResource):
           include_return_message_types: Only return specified message types in the response. If `None` (default) returns
               all messages.
 
+          input:
+              Syntactic sugar for a single user message. Equivalent to messages=[{'role':
+              'user', 'content': input}].
+
           max_steps: Maximum number of steps the agent should take to process the request.
 
+          messages: The messages to be sent to the agent.
+
           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.
@@ -1229,13 +1258,14 @@ class AsyncMessagesResource(AsyncAPIResource):
             f"/v1/agents/{agent_id}/messages/async",
             body=await async_maybe_transform(
                 {
-                    "messages": messages,
                     "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
                     "assistant_message_tool_name": assistant_message_tool_name,
                     "callback_url": callback_url,
                     "enable_thinking": enable_thinking,
                     "include_return_message_types": include_return_message_types,
+                    "input": input,
                     "max_steps": max_steps,
+                    "messages": messages,
                     "use_assistant_message": use_assistant_message,
                 },
                 message_send_async_params.MessageSendAsyncParams,
@@ -1250,14 +1280,15 @@ class AsyncMessagesResource(AsyncAPIResource):
         self,
         agent_id: str,
         *,
-        messages: Iterable[message_stream_params.Message],
         assistant_message_tool_kwarg: str | Omit = omit,
         assistant_message_tool_name: str | Omit = omit,
         background: bool | Omit = omit,
         enable_thinking: str | Omit = omit,
         include_pings: bool | Omit = omit,
         include_return_message_types: Optional[List[MessageType]] | Omit = omit,
+        input: Union[str, Iterable[message_stream_params.InputUnionMember1], None] | Omit = omit,
         max_steps: int | Omit = omit,
+        messages: Optional[Iterable[message_stream_params.Message]] | Omit = omit,
         stream_tokens: bool | 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.
@@ -1276,8 +1307,6 @@ class AsyncMessagesResource(AsyncAPIResource):
         Args:
           agent_id: The ID of the agent in the format 'agent-<uuid4>'
 
-          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.
 
@@ -1294,8 +1323,14 @@ class AsyncMessagesResource(AsyncAPIResource):
           include_return_message_types: Only return specified message types in the response. If `None` (default) returns
               all messages.
 
+          input:
+              Syntactic sugar for a single user message. Equivalent to messages=[{'role':
+              'user', 'content': input}].
+
           max_steps: Maximum number of steps the agent should take to process the request.
 
+          messages: The messages to be sent to the agent.
+
           stream_tokens: Flag to determine if individual tokens should be streamed, rather than streaming
               per step.
 
@@ -1317,14 +1352,15 @@ class AsyncMessagesResource(AsyncAPIResource):
             f"/v1/agents/{agent_id}/messages/stream",
             body=await async_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,
+                    "input": input,
                     "max_steps": max_steps,
+                    "messages": messages,
                     "stream_tokens": stream_tokens,
                     "use_assistant_message": use_assistant_message,
                 },