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

letta_client/resources/agents/tools.py

@@ -17,11 +17,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncArrayPage, AsyncArrayPage
-from ...types.tool import Tool
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.agents import tool_list_params, tool_update_approval_params
 from ...types.agent_state import AgentState
+from ...types.agents.tool_list_response import ToolListResponse
 
 __all__ = ["ToolsResource", "AsyncToolsResource"]
 
@@ -61,7 +60,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Tool]:
+    ) -> ToolListResponse:
         """
         Get tools from an existing agent.
 
@@ -91,9 +90,8 @@ class ToolsResource(SyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/agents/{agent_id}/tools",
-            page=SyncArrayPage[Tool],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -110,7 +108,7 @@ class ToolsResource(SyncAPIResource):
                     tool_list_params.ToolListParams,
                 ),
             ),
-            model=Tool,
+            cast_to=ToolListResponse,
         )
 
     def attach(
@@ -271,7 +269,7 @@ class AsyncToolsResource(AsyncAPIResource):
         """
         return AsyncToolsResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         agent_id: str,
         *,
@@ -286,7 +284,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Tool, AsyncArrayPage[Tool]]:
+    ) -> ToolListResponse:
         """
         Get tools from an existing agent.
 
@@ -316,15 +314,14 @@ class AsyncToolsResource(AsyncAPIResource):
         """
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/agents/{agent_id}/tools",
-            page=AsyncArrayPage[Tool],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -335,7 +332,7 @@ class AsyncToolsResource(AsyncAPIResource):
                     tool_list_params.ToolListParams,
                 ),
             ),
-            model=Tool,
+            cast_to=ToolListResponse,
         )
 
     async def attach(

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 ..pagination import SyncArrayPage, AsyncArrayPage
-from .._base_client import AsyncPaginator, make_request_options
+from .._base_client import 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,
         *,
@@ -184,7 +140,7 @@ class ArchivesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Archive]:
+    ) -> ArchiveListResponse:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -215,9 +171,8 @@ class ArchivesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/archives/",
-            page=SyncArrayPage[Archive],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -236,7 +191,7 @@ class ArchivesResource(SyncAPIResource):
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            model=Archive,
+            cast_to=ArchiveListResponse,
         )
 
     def delete(
@@ -274,6 +229,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
@@ -373,51 +372,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,
-        )
-
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -433,7 +388,7 @@ class AsyncArchivesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Archive, AsyncArrayPage[Archive]]:
+    ) -> ArchiveListResponse:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -464,15 +419,14 @@ class AsyncArchivesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/archives/",
-            page=AsyncArrayPage[Archive],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "agent_id": agent_id,
@@ -485,7 +439,7 @@ class AsyncArchivesResource(AsyncAPIResource):
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            model=Archive,
+            cast_to=ArchiveListResponse,
         )
 
     async def delete(
@@ -523,6 +477,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:
@@ -534,15 +532,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:
@@ -555,15 +553,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:
@@ -576,15 +574,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:
@@ -597,12 +595,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/batches.py

@@ -26,9 +26,9 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncArrayPage, AsyncArrayPage
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.batch_job import BatchJob
+from ...types.batch_list_response import BatchListResponse
 
 __all__ = ["BatchesResource", "AsyncBatchesResource"]
 
@@ -155,7 +155,7 @@ class BatchesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[BatchJob]:
+    ) -> BatchListResponse:
         """List all batch runs.
 
         Args:
@@ -182,9 +182,8 @@ class BatchesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/messages/batches",
-            page=SyncArrayPage[BatchJob],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -201,7 +200,7 @@ class BatchesResource(SyncAPIResource):
                     batch_list_params.BatchListParams,
                 ),
             ),
-            model=BatchJob,
+            cast_to=BatchListResponse,
         )
 
     def cancel(
@@ -346,7 +345,7 @@ class AsyncBatchesResource(AsyncAPIResource):
             cast_to=BatchJob,
         )
 
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -360,7 +359,7 @@ class AsyncBatchesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[BatchJob, AsyncArrayPage[BatchJob]]:
+    ) -> BatchListResponse:
         """List all batch runs.
 
         Args:
@@ -387,15 +386,14 @@ class AsyncBatchesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/messages/batches",
-            page=AsyncArrayPage[BatchJob],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -406,7 +404,7 @@ class AsyncBatchesResource(AsyncAPIResource):
                     batch_list_params.BatchListParams,
                 ),
             ),
-            model=BatchJob,
+            cast_to=BatchListResponse,
         )
 
     async def cancel(

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