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

letta_client/resources/folders/agents.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,8 +17,7 @@ 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.folders import agent_list_params
 from ...types.folders.agent_list_response import AgentListResponse
 
@@ -60,7 +59,7 @@ class AgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[AgentListResponse]:
+    ) -> AgentListResponse:
         """
         Get all agent IDs that have the specified folder attached.
 
@@ -90,9 +89,8 @@ class AgentsResource(SyncAPIResource):
         """
         if not folder_id:
             raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/folders/{folder_id}/agents",
-            page=SyncArrayPage[AgentListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -109,7 +107,7 @@ class AgentsResource(SyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            model=str,
+            cast_to=AgentListResponse,
         )
 
 
@@ -133,7 +131,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         """
         return AsyncAgentsResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         folder_id: str,
         *,
@@ -148,7 +146,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[AgentListResponse, AsyncArrayPage[AgentListResponse]]:
+    ) -> AgentListResponse:
         """
         Get all agent IDs that have the specified folder attached.
 
@@ -178,15 +176,14 @@ class AsyncAgentsResource(AsyncAPIResource):
         """
         if not folder_id:
             raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/folders/{folder_id}/agents",
-            page=AsyncArrayPage[AgentListResponse],
             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,
@@ -197,7 +194,7 @@ class AsyncAgentsResource(AsyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            model=str,
+            cast_to=AgentListResponse,
         )
 
 

letta_client/resources/folders/files.py

@@ -17,8 +17,7 @@ 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.folders import file_list_params, file_upload_params
 from ...types.folders.file_list_response import FileListResponse
 from ...types.folders.file_upload_response import FileUploadResponse
@@ -62,7 +61,7 @@ class FilesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[FileListResponse]:
+    ) -> FileListResponse:
         """
         List paginated files associated with a data folder.
 
@@ -94,9 +93,8 @@ class FilesResource(SyncAPIResource):
         """
         if not folder_id:
             raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._get_api_list(
+        return self._get(
             f"/v1/folders/{folder_id}/files",
-            page=SyncArrayPage[FileListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -114,7 +112,7 @@ class FilesResource(SyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            model=FileListResponse,
+            cast_to=FileListResponse,
         )
 
     def delete(
@@ -239,7 +237,7 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         return AsyncFilesResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         folder_id: str,
         *,
@@ -255,7 +253,7 @@ class AsyncFilesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[FileListResponse, AsyncArrayPage[FileListResponse]]:
+    ) -> FileListResponse:
         """
         List paginated files associated with a data folder.
 
@@ -287,15 +285,14 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         if not folder_id:
             raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._get_api_list(
+        return await self._get(
             f"/v1/folders/{folder_id}/files",
-            page=AsyncArrayPage[FileListResponse],
             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,
@@ -307,7 +304,7 @@ class AsyncFilesResource(AsyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            model=FileListResponse,
+            cast_to=FileListResponse,
         )
 
     async def delete(

letta_client/resources/folders/folders.py

@@ -23,7 +23,7 @@ from .agents import (
     AgentsResourceWithStreamingResponse,
     AsyncAgentsResourceWithStreamingResponse,
 )
-from ...types import folder_list_params, folder_create_params, folder_update_params
+from ...types import folder_list_params, folder_create_params, folder_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
@@ -34,9 +34,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.folder import Folder
+from ...types.folder_list_response import FolderListResponse
 from ...types.folder_count_response import FolderCountResponse
 from ...types.embedding_config_param import EmbeddingConfigParam
 
@@ -169,66 +169,6 @@ class FoldersResource(SyncAPIResource):
             cast_to=Folder,
         )
 
-    def update(
-        self,
-        folder_id: str,
-        *,
-        description: Optional[str] | Omit = omit,
-        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
-        instructions: Optional[str] | Omit = omit,
-        metadata: Optional[Dict[str, object]] | 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,
-    ) -> Folder:
-        """
-        Update the name or documentation of an existing data folder.
-
-        Args:
-          folder_id: The ID of the source in the format 'source-<uuid4>'
-
-          description: The description of the source.
-
-          embedding_config: Configuration for embedding model connection and processing parameters.
-
-          instructions: Instructions for how to use the source.
-
-          metadata: Metadata associated with the source.
-
-          name: The name of the source.
-
-          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 folder_id:
-            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return self._patch(
-            f"/v1/folders/{folder_id}",
-            body=maybe_transform(
-                {
-                    "description": description,
-                    "embedding_config": embedding_config,
-                    "instructions": instructions,
-                    "metadata": metadata,
-                    "name": name,
-                },
-                folder_update_params.FolderUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Folder,
-        )
-
     def list(
         self,
         *,
@@ -244,7 +184,7 @@ class FoldersResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Folder]:
+    ) -> FolderListResponse:
         """
         List all data folders created by a user.
 
@@ -272,9 +212,8 @@ class FoldersResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/folders/",
-            page=SyncArrayPage[Folder],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -292,7 +231,7 @@ class FoldersResource(SyncAPIResource):
                     folder_list_params.FolderListParams,
                 ),
             ),
-            model=Folder,
+            cast_to=FolderListResponse,
         )
 
     def delete(
@@ -349,6 +288,66 @@ class FoldersResource(SyncAPIResource):
             cast_to=int,
         )
 
+    def modify(
+        self,
+        folder_id: str,
+        *,
+        description: Optional[str] | Omit = omit,
+        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
+        instructions: Optional[str] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | 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,
+    ) -> Folder:
+        """
+        Update the name or documentation of an existing data folder.
+
+        Args:
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
+          description: The description of the source.
+
+          embedding_config: Configuration for embedding model connection and processing parameters.
+
+          instructions: Instructions for how to use the source.
+
+          metadata: Metadata associated with the source.
+
+          name: The name of the source.
+
+          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 folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
+        return self._patch(
+            f"/v1/folders/{folder_id}",
+            body=maybe_transform(
+                {
+                    "description": description,
+                    "embedding_config": embedding_config,
+                    "instructions": instructions,
+                    "metadata": metadata,
+                    "name": name,
+                },
+                folder_modify_params.FolderModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Folder,
+        )
+
 
 class AsyncFoldersResource(AsyncAPIResource):
     @cached_property
@@ -476,67 +475,7 @@ class AsyncFoldersResource(AsyncAPIResource):
             cast_to=Folder,
         )
 
-    async def update(
-        self,
-        folder_id: str,
-        *,
-        description: Optional[str] | Omit = omit,
-        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
-        instructions: Optional[str] | Omit = omit,
-        metadata: Optional[Dict[str, object]] | 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,
-    ) -> Folder:
-        """
-        Update the name or documentation of an existing data folder.
-
-        Args:
-          folder_id: The ID of the source in the format 'source-<uuid4>'
-
-          description: The description of the source.
-
-          embedding_config: Configuration for embedding model connection and processing parameters.
-
-          instructions: Instructions for how to use the source.
-
-          metadata: Metadata associated with the source.
-
-          name: The name of the source.
-
-          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 folder_id:
-            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
-        return await self._patch(
-            f"/v1/folders/{folder_id}",
-            body=await async_maybe_transform(
-                {
-                    "description": description,
-                    "embedding_config": embedding_config,
-                    "instructions": instructions,
-                    "metadata": metadata,
-                    "name": name,
-                },
-                folder_update_params.FolderUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Folder,
-        )
-
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -551,7 +490,7 @@ class AsyncFoldersResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Folder, AsyncArrayPage[Folder]]:
+    ) -> FolderListResponse:
         """
         List all data folders created by a user.
 
@@ -579,15 +518,14 @@ class AsyncFoldersResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/folders/",
-            page=AsyncArrayPage[Folder],
             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,
@@ -599,7 +537,7 @@ class AsyncFoldersResource(AsyncAPIResource):
                     folder_list_params.FolderListParams,
                 ),
             ),
-            model=Folder,
+            cast_to=FolderListResponse,
         )
 
     async def delete(
@@ -656,6 +594,66 @@ class AsyncFoldersResource(AsyncAPIResource):
             cast_to=int,
         )
 
+    async def modify(
+        self,
+        folder_id: str,
+        *,
+        description: Optional[str] | Omit = omit,
+        embedding_config: Optional[EmbeddingConfigParam] | Omit = omit,
+        instructions: Optional[str] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | 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,
+    ) -> Folder:
+        """
+        Update the name or documentation of an existing data folder.
+
+        Args:
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
+          description: The description of the source.
+
+          embedding_config: Configuration for embedding model connection and processing parameters.
+
+          instructions: Instructions for how to use the source.
+
+          metadata: Metadata associated with the source.
+
+          name: The name of the source.
+
+          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 folder_id:
+            raise ValueError(f"Expected a non-empty value for `folder_id` but received {folder_id!r}")
+        return await self._patch(
+            f"/v1/folders/{folder_id}",
+            body=await async_maybe_transform(
+                {
+                    "description": description,
+                    "embedding_config": embedding_config,
+                    "instructions": instructions,
+                    "metadata": metadata,
+                    "name": name,
+                },
+                folder_modify_params.FolderModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Folder,
+        )
+
 
 class FoldersResourceWithRawResponse:
     def __init__(self, folders: FoldersResource) -> None:
@@ -667,9 +665,6 @@ class FoldersResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             folders.retrieve,
         )
-        self.update = to_raw_response_wrapper(
-            folders.update,
-        )
         self.list = to_raw_response_wrapper(
             folders.list,
         )
@@ -679,6 +674,9 @@ class FoldersResourceWithRawResponse:
         self.count = to_raw_response_wrapper(
             folders.count,
         )
+        self.modify = to_raw_response_wrapper(
+            folders.modify,
+        )
 
     @cached_property
     def files(self) -> FilesResourceWithRawResponse:
@@ -699,9 +697,6 @@ class AsyncFoldersResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             folders.retrieve,
         )
-        self.update = async_to_raw_response_wrapper(
-            folders.update,
-        )
         self.list = async_to_raw_response_wrapper(
             folders.list,
         )
@@ -711,6 +706,9 @@ class AsyncFoldersResourceWithRawResponse:
         self.count = async_to_raw_response_wrapper(
             folders.count,
         )
+        self.modify = async_to_raw_response_wrapper(
+            folders.modify,
+        )
 
     @cached_property
     def files(self) -> AsyncFilesResourceWithRawResponse:
@@ -731,9 +729,6 @@ class FoldersResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             folders.retrieve,
         )
-        self.update = to_streamed_response_wrapper(
-            folders.update,
-        )
         self.list = to_streamed_response_wrapper(
             folders.list,
         )
@@ -743,6 +738,9 @@ class FoldersResourceWithStreamingResponse:
         self.count = to_streamed_response_wrapper(
             folders.count,
         )
+        self.modify = to_streamed_response_wrapper(
+            folders.modify,
+        )
 
     @cached_property
     def files(self) -> FilesResourceWithStreamingResponse:
@@ -763,9 +761,6 @@ class AsyncFoldersResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             folders.retrieve,
         )
-        self.update = async_to_streamed_response_wrapper(
-            folders.update,
-        )
         self.list = async_to_streamed_response_wrapper(
             folders.list,
         )
@@ -775,6 +770,9 @@ class AsyncFoldersResourceWithStreamingResponse:
         self.count = async_to_streamed_response_wrapper(
             folders.count,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            folders.modify,
+        )
 
     @cached_property
     def files(self) -> AsyncFilesResourceWithStreamingResponse: