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

{letta_sdk/resources/agents/core_memory → letta_client/resources/agents}/blocks.py

@@ -3,24 +3,25 @@
 from __future__ import annotations
 
 from typing import Dict, Optional
+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 ...._compat import cached_property
-from ...._resource import SyncAPIResource, AsyncAPIResource
-from ...._response import (
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ..._utils import maybe_transform, async_maybe_transform
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
     to_raw_response_wrapper,
     to_streamed_response_wrapper,
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...._base_client import make_request_options
-from ....types.agent_state import AgentState
-from ....types.agents.core_memory import block_update_params
-from ....types.agents.core_memory.block import Block
-from ....types.agents.core_memory.block_list_response import BlockListResponse
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
+from ...types.agents import block_list_params, block_update_params
+from ...types.agent_state import AgentState
+from ...types.agents.block import Block
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -61,6 +62,8 @@ class BlocksResource(SyncAPIResource):
         Retrieve a core memory block from an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -111,6 +114,8 @@ class BlocksResource(SyncAPIResource):
         Updates a core memory block of an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
           base_template_id: The base template id of the block.
 
           deployment_id: The id of the deployment.
@@ -182,17 +187,37 @@ class BlocksResource(SyncAPIResource):
         self,
         agent_id: str,
         *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        limit: Optional[int] | 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,
-    ) -> BlockListResponse:
+    ) -> SyncArrayPage[Block]:
         """
         Retrieve the core memory blocks of a specific agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          after: Block ID cursor for pagination. Returns blocks that come after this block ID in
+              the specified sort order
+
+          before: Block ID cursor for pagination. Returns blocks that come before this block ID in
+              the specified sort order
+
+          limit: Maximum number of blocks to return
+
+          order: Sort order for blocks 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
@@ -203,12 +228,26 @@ class BlocksResource(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}/core-memory/blocks",
+            page=SyncArrayPage[Block],
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "before": before,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                    },
+                    block_list_params.BlockListParams,
+                ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
     def attach(
@@ -227,6 +266,10 @@ class BlocksResource(SyncAPIResource):
         Attach a core memory block to an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -263,6 +306,10 @@ class BlocksResource(SyncAPIResource):
         Detach a core memory block from an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -320,6 +367,8 @@ class AsyncBlocksResource(AsyncAPIResource):
         Retrieve a core memory block from an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -370,6 +419,8 @@ class AsyncBlocksResource(AsyncAPIResource):
         Updates a core memory block of an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
           base_template_id: The base template id of the block.
 
           deployment_id: The id of the deployment.
@@ -437,21 +488,41 @@ class AsyncBlocksResource(AsyncAPIResource):
             cast_to=Block,
         )
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        limit: Optional[int] | 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,
-    ) -> BlockListResponse:
+    ) -> AsyncPaginator[Block, AsyncArrayPage[Block]]:
         """
         Retrieve the core memory blocks of a specific agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          after: Block ID cursor for pagination. Returns blocks that come after this block ID in
+              the specified sort order
+
+          before: Block ID cursor for pagination. Returns blocks that come before this block ID in
+              the specified sort order
+
+          limit: Maximum number of blocks to return
+
+          order: Sort order for blocks 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
@@ -462,12 +533,26 @@ class AsyncBlocksResource(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}/core-memory/blocks",
+            page=AsyncArrayPage[Block],
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "before": before,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                    },
+                    block_list_params.BlockListParams,
+                ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
     async def attach(
@@ -486,6 +571,10 @@ class AsyncBlocksResource(AsyncAPIResource):
         Attach a core memory block to an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -522,6 +611,10 @@ class AsyncBlocksResource(AsyncAPIResource):
         Detach a core memory block from an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request

{letta_sdk → letta_client}/resources/agents/files.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from typing import Optional
+from typing_extensions import Literal
 
 import httpx
 
@@ -49,9 +50,13 @@ class FilesResource(SyncAPIResource):
         self,
         agent_id: str,
         *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
         cursor: Optional[str] | Omit = omit,
         is_open: Optional[bool] | Omit = omit,
-        limit: int | Omit = omit,
+        limit: Optional[int] | 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,
@@ -60,14 +65,27 @@ class FilesResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> FileListResponse:
         """
-        Get the files attached to an agent with their open/closed status (paginated).
+        Get the files attached to an agent with their open/closed status.
 
         Args:
-          cursor: Pagination cursor from previous response
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          after: File ID cursor for pagination. Returns files that come after this file ID in the
+              specified sort order
+
+          before: File ID cursor for pagination. Returns files that come before this file ID in
+              the specified sort order
+
+          cursor: Pagination cursor from previous response (deprecated, use before/after)
 
           is_open: Filter by open status (true for open files, false for closed files)
 
-          limit: Number of items to return (1-100)
+          limit: Maximum number of files to return
+
+          order: Sort order for files by creation time. 'asc' for oldest first, 'desc' for newest
+              first
+
+          order_by: Field to sort by
 
           extra_headers: Send extra headers
 
@@ -88,9 +106,13 @@ class FilesResource(SyncAPIResource):
                 timeout=timeout,
                 query=maybe_transform(
                     {
+                        "after": after,
+                        "before": before,
                         "cursor": cursor,
                         "is_open": is_open,
                         "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
                     },
                     file_list_params.FileListParams,
                 ),
@@ -117,6 +139,10 @@ class FilesResource(SyncAPIResource):
         file will be removed from the agent's working memory view.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          file_id: The ID of the file in the format 'file-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -155,6 +181,8 @@ class FilesResource(SyncAPIResource):
         as open. Typically used to reset the working memory view for the agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -193,6 +221,10 @@ class FilesResource(SyncAPIResource):
         names that were closed due to LRU eviction.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          file_id: The ID of the file in the format 'file-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -238,9 +270,13 @@ class AsyncFilesResource(AsyncAPIResource):
         self,
         agent_id: str,
         *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
         cursor: Optional[str] | Omit = omit,
         is_open: Optional[bool] | Omit = omit,
-        limit: int | Omit = omit,
+        limit: Optional[int] | 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,
@@ -249,14 +285,27 @@ class AsyncFilesResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> FileListResponse:
         """
-        Get the files attached to an agent with their open/closed status (paginated).
+        Get the files attached to an agent with their open/closed status.
 
         Args:
-          cursor: Pagination cursor from previous response
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          after: File ID cursor for pagination. Returns files that come after this file ID in the
+              specified sort order
+
+          before: File ID cursor for pagination. Returns files that come before this file ID in
+              the specified sort order
+
+          cursor: Pagination cursor from previous response (deprecated, use before/after)
 
           is_open: Filter by open status (true for open files, false for closed files)
 
-          limit: Number of items to return (1-100)
+          limit: Maximum number of files to return
+
+          order: Sort order for files by creation time. 'asc' for oldest first, 'desc' for newest
+              first
+
+          order_by: Field to sort by
 
           extra_headers: Send extra headers
 
@@ -277,9 +326,13 @@ class AsyncFilesResource(AsyncAPIResource):
                 timeout=timeout,
                 query=await async_maybe_transform(
                     {
+                        "after": after,
+                        "before": before,
                         "cursor": cursor,
                         "is_open": is_open,
                         "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
                     },
                     file_list_params.FileListParams,
                 ),
@@ -306,6 +359,10 @@ class AsyncFilesResource(AsyncAPIResource):
         file will be removed from the agent's working memory view.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          file_id: The ID of the file in the format 'file-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -344,6 +401,8 @@ class AsyncFilesResource(AsyncAPIResource):
         as open. Typically used to reset the working memory view for the agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -382,6 +441,10 @@ class AsyncFilesResource(AsyncAPIResource):
         names that were closed due to LRU eviction.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          file_id: The ID of the file in the format 'file-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request

{letta_sdk → letta_client}/resources/agents/folders.py

@@ -2,9 +2,13 @@
 
 from __future__ import annotations
 
+from typing import Optional
+from typing_extensions import Literal
+
 import httpx
 
-from ..._types import Body, Query, Headers, NotGiven, not_given
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ..._utils import maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -13,7 +17,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.agents import folder_list_params
 from ...types.agent_state import AgentState
 from ...types.agents.folder_list_response import FolderListResponse
 
@@ -44,17 +50,37 @@ class FoldersResource(SyncAPIResource):
         self,
         agent_id: str,
         *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        limit: Optional[int] | 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,
-    ) -> FolderListResponse:
+    ) -> SyncArrayPage[FolderListResponse]:
         """
         Get the folders associated with an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          after: Source ID cursor for pagination. Returns sources that come after this source ID
+              in the specified sort order
+
+          before: Source ID cursor for pagination. Returns sources that come before this source ID
+              in the specified sort order
+
+          limit: Maximum number of sources to return
+
+          order: Sort order for sources 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
@@ -65,12 +91,26 @@ class FoldersResource(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}/folders",
+            page=SyncArrayPage[FolderListResponse],
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "before": before,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                    },
+                    folder_list_params.FolderListParams,
+                ),
             ),
-            cast_to=FolderListResponse,
+            model=FolderListResponse,
         )
 
     def attach(
@@ -89,6 +129,10 @@ class FoldersResource(SyncAPIResource):
         Attach a folder to an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -125,6 +169,10 @@ class FoldersResource(SyncAPIResource):
         Detach a folder from an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -166,21 +214,41 @@ class AsyncFoldersResource(AsyncAPIResource):
         """
         return AsyncFoldersResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        limit: Optional[int] | 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,
-    ) -> FolderListResponse:
+    ) -> AsyncPaginator[FolderListResponse, AsyncArrayPage[FolderListResponse]]:
         """
         Get the folders associated with an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          after: Source ID cursor for pagination. Returns sources that come after this source ID
+              in the specified sort order
+
+          before: Source ID cursor for pagination. Returns sources that come before this source ID
+              in the specified sort order
+
+          limit: Maximum number of sources to return
+
+          order: Sort order for sources 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
@@ -191,12 +259,26 @@ class AsyncFoldersResource(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}/folders",
+            page=AsyncArrayPage[FolderListResponse],
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "before": before,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                    },
+                    folder_list_params.FolderListParams,
+                ),
             ),
-            cast_to=FolderListResponse,
+            model=FolderListResponse,
         )
 
     async def attach(
@@ -215,6 +297,10 @@ class AsyncFoldersResource(AsyncAPIResource):
         Attach a folder to an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -251,6 +337,10 @@ class AsyncFoldersResource(AsyncAPIResource):
         Detach a folder from an agent.
 
         Args:
+          agent_id: The ID of the agent in the format 'agent-<uuid4>'
+
+          folder_id: The ID of the source in the format 'source-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request