letta-client 0.1.319__py3-none-any.whl → 0.1.320__py3-none-any.whl

letta_client/batches/client.py

@@ -6,7 +6,9 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.batch_job import BatchJob
 from ..types.letta_batch_request import LettaBatchRequest
+from .messages.client import AsyncMessagesClient, MessagesClient
 from .raw_client import AsyncRawBatchesClient, RawBatchesClient
+from .types.batches_list_request_order import BatchesListRequestOrder
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -15,6 +17,7 @@ OMIT = typing.cast(typing.Any, ...)
 class BatchesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._raw_client = RawBatchesClient(client_wrapper=client_wrapper)
+        self.messages = MessagesClient(client_wrapper=client_wrapper)
 
     @property
     def with_raw_response(self) -> RawBatchesClient:
@@ -27,12 +30,36 @@ class BatchesClient:
         """
         return self._raw_client
 
-    def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[BatchJob]:
+    def list(
+        self,
+        *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        order: typing.Optional[BatchesListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[BatchJob]:
         """
         List all batch runs.
 
         Parameters
         ----------
+        before : typing.Optional[str]
+            Job ID cursor for pagination. Returns jobs that come before this job ID in the specified sort order
+
+        after : typing.Optional[str]
+            Job ID cursor for pagination. Returns jobs that come after this job ID in the specified sort order
+
+        limit : typing.Optional[int]
+            Maximum number of jobs to return
+
+        order : typing.Optional[BatchesListRequestOrder]
+            Sort order for jobs by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -51,7 +78,9 @@ class BatchesClient:
         )
         client.batches.list()
         """
-        _response = self._raw_client.list(request_options=request_options)
+        _response = self._raw_client.list(
+            before=before, after=after, limit=limit, order=order, order_by=order_by, request_options=request_options
+        )
         return _response.data
 
     def create(
@@ -62,8 +91,10 @@ class BatchesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchJob:
         """
-        Submit a batch of agent messages for asynchronous processing.
+        Submit a batch of agent runs for asynchronous processing.
+
         Creates a job that will fan out messages to all listed agents and process them in parallel.
+        The request will be rejected if it exceeds 256MB.
 
         Parameters
         ----------
@@ -114,7 +145,7 @@ class BatchesClient:
 
     def retrieve(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchJob:
         """
-        Get the status of a batch run.
+        Retrieve the status and details of a batch run.
 
         Parameters
         ----------
@@ -180,6 +211,7 @@ class BatchesClient:
 class AsyncBatchesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._raw_client = AsyncRawBatchesClient(client_wrapper=client_wrapper)
+        self.messages = AsyncMessagesClient(client_wrapper=client_wrapper)
 
     @property
     def with_raw_response(self) -> AsyncRawBatchesClient:
@@ -192,12 +224,36 @@ class AsyncBatchesClient:
         """
         return self._raw_client
 
-    async def list(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[BatchJob]:
+    async def list(
+        self,
+        *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        order: typing.Optional[BatchesListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[BatchJob]:
         """
         List all batch runs.
 
         Parameters
         ----------
+        before : typing.Optional[str]
+            Job ID cursor for pagination. Returns jobs that come before this job ID in the specified sort order
+
+        after : typing.Optional[str]
+            Job ID cursor for pagination. Returns jobs that come after this job ID in the specified sort order
+
+        limit : typing.Optional[int]
+            Maximum number of jobs to return
+
+        order : typing.Optional[BatchesListRequestOrder]
+            Sort order for jobs by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -224,7 +280,9 @@ class AsyncBatchesClient:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.list(request_options=request_options)
+        _response = await self._raw_client.list(
+            before=before, after=after, limit=limit, order=order, order_by=order_by, request_options=request_options
+        )
         return _response.data
 
     async def create(
@@ -235,8 +293,10 @@ class AsyncBatchesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchJob:
         """
-        Submit a batch of agent messages for asynchronous processing.
+        Submit a batch of agent runs for asynchronous processing.
+
         Creates a job that will fan out messages to all listed agents and process them in parallel.
+        The request will be rejected if it exceeds 256MB.
 
         Parameters
         ----------
@@ -300,7 +360,7 @@ class AsyncBatchesClient:
 
     async def retrieve(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchJob:
         """
-        Get the status of a batch run.
+        Retrieve the status and details of a batch run.
 
         Parameters
         ----------

letta_client/batches/messages/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import MessagesListRequestOrder
+
+__all__ = ["MessagesListRequestOrder"]

letta_client/{messages → batches/messages}/client.py

@@ -2,10 +2,11 @@
 
 import typing
 
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.request_options import RequestOptions
-from ..types.letta_batch_messages import LettaBatchMessages
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.letta_batch_messages import LettaBatchMessages
 from .raw_client import AsyncRawMessagesClient, RawMessagesClient
+from .types.messages_list_request_order import MessagesListRequestOrder
 
 
 class MessagesClient:
@@ -23,42 +24,43 @@ class MessagesClient:
         """
         return self._raw_client
 
-    def list_batch_messages(
+    def list(
         self,
         batch_id: str,
         *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        cursor: typing.Optional[str] = None,
+        order: typing.Optional[MessagesListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
         agent_id: typing.Optional[str] = None,
-        sort_descending: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> LettaBatchMessages:
         """
-        Get messages for a specific batch job.
-
-        Returns messages associated with the batch in chronological order.
-
-        Pagination:
-        - For the first page, omit the cursor parameter
-        - For subsequent pages, use the ID of the last message from the previous response as the cursor
-        - Results will include messages before/after the cursor based on sort_descending
+        Get response messages for a specific batch job.
 
         Parameters
         ----------
         batch_id : str
 
+        before : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come before this message ID in the specified sort order
+
+        after : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come after this message ID in the specified sort order
+
         limit : typing.Optional[int]
             Maximum number of messages to return
 
-        cursor : typing.Optional[str]
-            Message ID to use as pagination cursor (get messages before/after this ID) depending on sort_descending.
+        order : typing.Optional[MessagesListRequestOrder]
+            Sort order for messages by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
 
         agent_id : typing.Optional[str]
             Filter messages by agent ID
 
-        sort_descending : typing.Optional[bool]
-            Sort messages by creation time (true=newest first)
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -75,16 +77,18 @@ class MessagesClient:
             project="YOUR_PROJECT",
             token="YOUR_TOKEN",
         )
-        client.messages.list_batch_messages(
+        client.batches.messages.list(
             batch_id="batch_id",
         )
         """
-        _response = self._raw_client.list_batch_messages(
+        _response = self._raw_client.list(
             batch_id,
+            before=before,
+            after=after,
             limit=limit,
-            cursor=cursor,
+            order=order,
+            order_by=order_by,
             agent_id=agent_id,
-            sort_descending=sort_descending,
             request_options=request_options,
         )
         return _response.data
@@ -105,42 +109,43 @@ class AsyncMessagesClient:
         """
         return self._raw_client
 
-    async def list_batch_messages(
+    async def list(
         self,
         batch_id: str,
         *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        cursor: typing.Optional[str] = None,
+        order: typing.Optional[MessagesListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
         agent_id: typing.Optional[str] = None,
-        sort_descending: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> LettaBatchMessages:
         """
-        Get messages for a specific batch job.
-
-        Returns messages associated with the batch in chronological order.
-
-        Pagination:
-        - For the first page, omit the cursor parameter
-        - For subsequent pages, use the ID of the last message from the previous response as the cursor
-        - Results will include messages before/after the cursor based on sort_descending
+        Get response messages for a specific batch job.
 
         Parameters
         ----------
         batch_id : str
 
+        before : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come before this message ID in the specified sort order
+
+        after : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come after this message ID in the specified sort order
+
         limit : typing.Optional[int]
             Maximum number of messages to return
 
-        cursor : typing.Optional[str]
-            Message ID to use as pagination cursor (get messages before/after this ID) depending on sort_descending.
+        order : typing.Optional[MessagesListRequestOrder]
+            Sort order for messages by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
 
         agent_id : typing.Optional[str]
             Filter messages by agent ID
 
-        sort_descending : typing.Optional[bool]
-            Sort messages by creation time (true=newest first)
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -162,19 +167,21 @@ class AsyncMessagesClient:
 
 
         async def main() -> None:
-            await client.messages.list_batch_messages(
+            await client.batches.messages.list(
                 batch_id="batch_id",
             )
 
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.list_batch_messages(
+        _response = await self._raw_client.list(
             batch_id,
+            before=before,
+            after=after,
             limit=limit,
-            cursor=cursor,
+            order=order,
+            order_by=order_by,
             agent_id=agent_id,
-            sort_descending=sort_descending,
             request_options=request_options,
         )
         return _response.data

letta_client/{messages → batches/messages}/raw_client.py

@@ -3,57 +3,59 @@
 import typing
 from json.decoder import JSONDecodeError
 
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.http_response import AsyncHttpResponse, HttpResponse
-from ..core.jsonable_encoder import jsonable_encoder
-from ..core.request_options import RequestOptions
-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_batch_messages import LettaBatchMessages
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.http_response import AsyncHttpResponse, HttpResponse
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.request_options import RequestOptions
+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_batch_messages import LettaBatchMessages
+from .types.messages_list_request_order import MessagesListRequestOrder
 
 
 class RawMessagesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list_batch_messages(
+    def list(
         self,
         batch_id: str,
         *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        cursor: typing.Optional[str] = None,
+        order: typing.Optional[MessagesListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
         agent_id: typing.Optional[str] = None,
-        sort_descending: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[LettaBatchMessages]:
         """
-        Get messages for a specific batch job.
-
-        Returns messages associated with the batch in chronological order.
-
-        Pagination:
-        - For the first page, omit the cursor parameter
-        - For subsequent pages, use the ID of the last message from the previous response as the cursor
-        - Results will include messages before/after the cursor based on sort_descending
+        Get response messages for a specific batch job.
 
         Parameters
         ----------
         batch_id : str
 
+        before : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come before this message ID in the specified sort order
+
+        after : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come after this message ID in the specified sort order
+
         limit : typing.Optional[int]
             Maximum number of messages to return
 
-        cursor : typing.Optional[str]
-            Message ID to use as pagination cursor (get messages before/after this ID) depending on sort_descending.
+        order : typing.Optional[MessagesListRequestOrder]
+            Sort order for messages by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
 
         agent_id : typing.Optional[str]
             Filter messages by agent ID
 
-        sort_descending : typing.Optional[bool]
-            Sort messages by creation time (true=newest first)
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -66,10 +68,12 @@ class RawMessagesClient:
             f"v1/messages/batches/{jsonable_encoder(batch_id)}/messages",
             method="GET",
             params={
+                "before": before,
+                "after": after,
                 "limit": limit,
-                "cursor": cursor,
+                "order": order,
+                "order_by": order_by,
                 "agent_id": agent_id,
-                "sort_descending": sort_descending,
             },
             request_options=request_options,
         )
@@ -104,42 +108,43 @@ class AsyncRawMessagesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list_batch_messages(
+    async def list(
         self,
         batch_id: str,
         *,
+        before: typing.Optional[str] = None,
+        after: typing.Optional[str] = None,
         limit: typing.Optional[int] = None,
-        cursor: typing.Optional[str] = None,
+        order: typing.Optional[MessagesListRequestOrder] = None,
+        order_by: typing.Optional[typing.Literal["created_at"]] = None,
         agent_id: typing.Optional[str] = None,
-        sort_descending: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[LettaBatchMessages]:
         """
-        Get messages for a specific batch job.
-
-        Returns messages associated with the batch in chronological order.
-
-        Pagination:
-        - For the first page, omit the cursor parameter
-        - For subsequent pages, use the ID of the last message from the previous response as the cursor
-        - Results will include messages before/after the cursor based on sort_descending
+        Get response messages for a specific batch job.
 
         Parameters
         ----------
         batch_id : str
 
+        before : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come before this message ID in the specified sort order
+
+        after : typing.Optional[str]
+            Message ID cursor for pagination. Returns messages that come after this message ID in the specified sort order
+
         limit : typing.Optional[int]
             Maximum number of messages to return
 
-        cursor : typing.Optional[str]
-            Message ID to use as pagination cursor (get messages before/after this ID) depending on sort_descending.
+        order : typing.Optional[MessagesListRequestOrder]
+            Sort order for messages by creation time. 'asc' for oldest first, 'desc' for newest first
+
+        order_by : typing.Optional[typing.Literal["created_at"]]
+            Field to sort by
 
         agent_id : typing.Optional[str]
             Filter messages by agent ID
 
-        sort_descending : typing.Optional[bool]
-            Sort messages by creation time (true=newest first)
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -152,10 +157,12 @@ class AsyncRawMessagesClient:
             f"v1/messages/batches/{jsonable_encoder(batch_id)}/messages",
             method="GET",
             params={
+                "before": before,
+                "after": after,
                 "limit": limit,
-                "cursor": cursor,
+                "order": order,
+                "order_by": order_by,
                 "agent_id": agent_id,
-                "sort_descending": sort_descending,
             },
             request_options=request_options,
         )

letta_client/batches/messages/types/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .messages_list_request_order import MessagesListRequestOrder
+
+__all__ = ["MessagesListRequestOrder"]

letta_client/batches/messages/types/messages_list_request_order.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+MessagesListRequestOrder = typing.Union[typing.Literal["asc", "desc"], typing.Any]