letta-client 1.0.0a8__py3-none-any.whl → 1.0.0a10__py3-none-any.whl

letta_client/resources/agents/tools.py

@@ -124,7 +124,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Attach a tool to an agent.
 
@@ -164,7 +164,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Detach a tool from an agent.
 
@@ -206,7 +206,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Modify the approval requirement for a tool attached to an agent.
 
@@ -349,7 +349,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Attach a tool to an agent.
 
@@ -389,7 +389,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Detach a tool from an agent.
 
@@ -431,7 +431,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Modify the approval requirement for a tool attached to an agent.
 

letta_client/resources/archives.py

@@ -7,7 +7,7 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..types import archive_list_params, archive_create_params, archive_update_params
+from ..types import archive_list_params, archive_create_params, archive_modify_params
 from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -18,9 +18,9 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from .._base_client import make_request_options
+from ..pagination import SyncArrayPage, AsyncArrayPage
+from .._base_client import AsyncPaginator, make_request_options
 from ..types.archive import Archive
-from ..types.archive_list_response import ArchiveListResponse
 from ..types.embedding_config_param import EmbeddingConfigParam
 
 __all__ = ["ArchivesResource", "AsyncArchivesResource"]
@@ -124,50 +124,6 @@ class ArchivesResource(SyncAPIResource):
             cast_to=Archive,
         )
 
-    def update(
-        self,
-        archive_id: str,
-        *,
-        description: Optional[str] | Omit = omit,
-        name: Optional[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,
-    ) -> Archive:
-        """
-        Update an existing archive's name and/or description.
-
-        Args:
-          archive_id: The ID of the archive in the format 'archive-<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 archive_id:
-            raise ValueError(f"Expected a non-empty value for `archive_id` but received {archive_id!r}")
-        return self._patch(
-            f"/v1/archives/{archive_id}",
-            body=maybe_transform(
-                {
-                    "description": description,
-                    "name": name,
-                },
-                archive_update_params.ArchiveUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Archive,
-        )
-
     def list(
         self,
         *,
@@ -177,13 +133,14 @@ class ArchivesResource(SyncAPIResource):
         limit: Optional[int] | Omit = omit,
         name: Optional[str] | Omit = omit,
         order: Literal["asc", "desc"] | Omit = omit,
+        order_by: Literal["created_at"] | 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,
-    ) -> ArchiveListResponse:
+    ) -> SyncArrayPage[Archive]:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -204,6 +161,8 @@ class ArchivesResource(SyncAPIResource):
           order: Sort order for archives by creation time. 'asc' for oldest first, 'desc' for
               newest first
 
+          order_by: Field to sort by
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -212,8 +171,9 @@ class ArchivesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/archives/",
+            page=SyncArrayPage[Archive],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -227,11 +187,12 @@ class ArchivesResource(SyncAPIResource):
                         "limit": limit,
                         "name": name,
                         "order": order,
+                        "order_by": order_by,
                     },
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            cast_to=ArchiveListResponse,
+            model=Archive,
         )
 
     def delete(
@@ -269,6 +230,50 @@ class ArchivesResource(SyncAPIResource):
             cast_to=Archive,
         )
 
+    def modify(
+        self,
+        archive_id: str,
+        *,
+        description: Optional[str] | Omit = omit,
+        name: Optional[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,
+    ) -> Archive:
+        """
+        Update an existing archive's name and/or description.
+
+        Args:
+          archive_id: The ID of the archive in the format 'archive-<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 archive_id:
+            raise ValueError(f"Expected a non-empty value for `archive_id` but received {archive_id!r}")
+        return self._patch(
+            f"/v1/archives/{archive_id}",
+            body=maybe_transform(
+                {
+                    "description": description,
+                    "name": name,
+                },
+                archive_modify_params.ArchiveModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Archive,
+        )
+
 
 class AsyncArchivesResource(AsyncAPIResource):
     @cached_property
@@ -368,51 +373,7 @@ class AsyncArchivesResource(AsyncAPIResource):
             cast_to=Archive,
         )
 
-    async def update(
-        self,
-        archive_id: str,
-        *,
-        description: Optional[str] | Omit = omit,
-        name: Optional[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,
-    ) -> Archive:
-        """
-        Update an existing archive's name and/or description.
-
-        Args:
-          archive_id: The ID of the archive in the format 'archive-<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 archive_id:
-            raise ValueError(f"Expected a non-empty value for `archive_id` but received {archive_id!r}")
-        return await self._patch(
-            f"/v1/archives/{archive_id}",
-            body=await async_maybe_transform(
-                {
-                    "description": description,
-                    "name": name,
-                },
-                archive_update_params.ArchiveUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Archive,
-        )
-
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -421,13 +382,14 @@ class AsyncArchivesResource(AsyncAPIResource):
         limit: Optional[int] | Omit = omit,
         name: Optional[str] | Omit = omit,
         order: Literal["asc", "desc"] | Omit = omit,
+        order_by: Literal["created_at"] | 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,
-    ) -> ArchiveListResponse:
+    ) -> AsyncPaginator[Archive, AsyncArrayPage[Archive]]:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -448,6 +410,8 @@ class AsyncArchivesResource(AsyncAPIResource):
           order: Sort order for archives by creation time. 'asc' for oldest first, 'desc' for
               newest first
 
+          order_by: Field to sort by
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -456,14 +420,15 @@ class AsyncArchivesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/archives/",
+            page=AsyncArrayPage[Archive],
             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,
                         "agent_id": agent_id,
@@ -471,11 +436,12 @@ class AsyncArchivesResource(AsyncAPIResource):
                         "limit": limit,
                         "name": name,
                         "order": order,
+                        "order_by": order_by,
                     },
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            cast_to=ArchiveListResponse,
+            model=Archive,
         )
 
     async def delete(
@@ -513,6 +479,50 @@ class AsyncArchivesResource(AsyncAPIResource):
             cast_to=Archive,
         )
 
+    async def modify(
+        self,
+        archive_id: str,
+        *,
+        description: Optional[str] | Omit = omit,
+        name: Optional[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,
+    ) -> Archive:
+        """
+        Update an existing archive's name and/or description.
+
+        Args:
+          archive_id: The ID of the archive in the format 'archive-<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 archive_id:
+            raise ValueError(f"Expected a non-empty value for `archive_id` but received {archive_id!r}")
+        return await self._patch(
+            f"/v1/archives/{archive_id}",
+            body=await async_maybe_transform(
+                {
+                    "description": description,
+                    "name": name,
+                },
+                archive_modify_params.ArchiveModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Archive,
+        )
+
 
 class ArchivesResourceWithRawResponse:
     def __init__(self, archives: ArchivesResource) -> None:
@@ -524,15 +534,15 @@ class ArchivesResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             archives.retrieve,
         )
-        self.update = to_raw_response_wrapper(
-            archives.update,
-        )
         self.list = to_raw_response_wrapper(
             archives.list,
         )
         self.delete = to_raw_response_wrapper(
             archives.delete,
         )
+        self.modify = to_raw_response_wrapper(
+            archives.modify,
+        )
 
 
 class AsyncArchivesResourceWithRawResponse:
@@ -545,15 +555,15 @@ class AsyncArchivesResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             archives.retrieve,
         )
-        self.update = async_to_raw_response_wrapper(
-            archives.update,
-        )
         self.list = async_to_raw_response_wrapper(
             archives.list,
         )
         self.delete = async_to_raw_response_wrapper(
             archives.delete,
         )
+        self.modify = async_to_raw_response_wrapper(
+            archives.modify,
+        )
 
 
 class ArchivesResourceWithStreamingResponse:
@@ -566,15 +576,15 @@ class ArchivesResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             archives.retrieve,
         )
-        self.update = to_streamed_response_wrapper(
-            archives.update,
-        )
         self.list = to_streamed_response_wrapper(
             archives.list,
         )
         self.delete = to_streamed_response_wrapper(
             archives.delete,
         )
+        self.modify = to_streamed_response_wrapper(
+            archives.modify,
+        )
 
 
 class AsyncArchivesResourceWithStreamingResponse:
@@ -587,12 +597,12 @@ class AsyncArchivesResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             archives.retrieve,
         )
-        self.update = async_to_streamed_response_wrapper(
-            archives.update,
-        )
         self.list = async_to_streamed_response_wrapper(
             archives.list,
         )
         self.delete = async_to_streamed_response_wrapper(
             archives.delete,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            archives.modify,
+        )

letta_client/resources/batches/messages.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,9 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncObjectPage, AsyncObjectPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.batches import message_list_params
-from ...types.batches.message_list_response import MessageListResponse
+from ...types.agents.message import Message
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -60,7 +61,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> SyncObjectPage[Message]:
         """
         Get response messages for a specific batch job.
 
@@ -90,8 +91,9 @@ class MessagesResource(SyncAPIResource):
         """
         if not batch_id:
             raise ValueError(f"Expected a non-empty value for `batch_id` but received {batch_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/messages/batches/{batch_id}/messages",
+            page=SyncObjectPage[Message],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -109,7 +111,7 @@ class MessagesResource(SyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=Message,
         )
 
 
@@ -133,7 +135,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         return AsyncMessagesResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         batch_id: str,
         *,
@@ -149,7 +151,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> AsyncPaginator[Message, AsyncObjectPage[Message]]:
         """
         Get response messages for a specific batch job.
 
@@ -179,14 +181,15 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         if not batch_id:
             raise ValueError(f"Expected a non-empty value for `batch_id` but received {batch_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/messages/batches/{batch_id}/messages",
+            page=AsyncObjectPage[Message],
             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,
                         "agent_id": agent_id,
@@ -198,7 +201,7 @@ class AsyncMessagesResource(AsyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=Message,
         )