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

letta_client/resources/agents/blocks.py

@@ -19,9 +19,11 @@ from ..._response import (
 )
 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.agents import block_list_params, block_modify_params
 from ...types.agent_state import AgentState
-from ...types.agents.block import Block
+from ...types.agents.block_list_response import BlockListResponse
+from ...types.agents.block_modify_response import BlockModifyResponse
+from ...types.agents.block_retrieve_response import BlockRetrieveResponse
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -57,7 +59,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Block:
+    ) -> BlockRetrieveResponse:
         """
         Retrieve a core memory block from an agent.
 
@@ -81,106 +83,7 @@ class BlocksResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Block,
-        )
-
-    def update(
-        self,
-        block_label: str,
-        *,
-        agent_id: str,
-        base_template_id: Optional[str] | Omit = omit,
-        deployment_id: Optional[str] | Omit = omit,
-        description: Optional[str] | Omit = omit,
-        entity_id: Optional[str] | Omit = omit,
-        hidden: Optional[bool] | Omit = omit,
-        is_template: bool | Omit = omit,
-        label: Optional[str] | Omit = omit,
-        limit: Optional[int] | Omit = omit,
-        metadata: Optional[Dict[str, object]] | Omit = omit,
-        name: Optional[str] | Omit = omit,
-        preserve_on_migration: Optional[bool] | Omit = omit,
-        project_id: Optional[str] | Omit = omit,
-        read_only: bool | Omit = omit,
-        value: 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,
-    ) -> Block:
-        """
-        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.
-
-          description: Description of the block.
-
-          entity_id: The id of the entity within the template.
-
-          hidden: If set to True, the block will be hidden.
-
-          is_template: Whether the block is a template (e.g. saved human/persona options).
-
-          label: Label of the block (e.g. 'human', 'persona') in the context window.
-
-          limit: Character limit of the block.
-
-          metadata: Metadata of the block.
-
-          name: The id of the template.
-
-          preserve_on_migration: Preserve the block on template migration.
-
-          project_id: The associated project id.
-
-          read_only: Whether the agent has read-only access to the block.
-
-          value: Value of the block.
-
-          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 agent_id:
-            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        if not block_label:
-            raise ValueError(f"Expected a non-empty value for `block_label` but received {block_label!r}")
-        return self._patch(
-            f"/v1/agents/{agent_id}/core-memory/blocks/{block_label}",
-            body=maybe_transform(
-                {
-                    "base_template_id": base_template_id,
-                    "deployment_id": deployment_id,
-                    "description": description,
-                    "entity_id": entity_id,
-                    "hidden": hidden,
-                    "is_template": is_template,
-                    "label": label,
-                    "limit": limit,
-                    "metadata": metadata,
-                    "name": name,
-                    "preserve_on_migration": preserve_on_migration,
-                    "project_id": project_id,
-                    "read_only": read_only,
-                    "value": value,
-                },
-                block_update_params.BlockUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Block,
+            cast_to=BlockRetrieveResponse,
         )
 
     def list(
@@ -198,7 +101,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Block]:
+    ) -> SyncArrayPage[BlockListResponse]:
         """
         Retrieve the core memory blocks of a specific agent.
 
@@ -230,7 +133,7 @@ class BlocksResource(SyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         return self._get_api_list(
             f"/v1/agents/{agent_id}/core-memory/blocks",
-            page=SyncArrayPage[Block],
+            page=SyncArrayPage[BlockListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -247,7 +150,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=Block,
+            model=BlockListResponse,
         )
 
     def attach(
@@ -330,66 +233,7 @@ class BlocksResource(SyncAPIResource):
             cast_to=AgentState,
         )
 
-
-class AsyncBlocksResource(AsyncAPIResource):
-    @cached_property
-    def with_raw_response(self) -> AsyncBlocksResourceWithRawResponse:
-        """
-        This property can be used as a prefix for any HTTP method call to return
-        the raw response object instead of the parsed content.
-
-        For more information, see https://www.github.com/letta-ai/letta-python#accessing-raw-response-data-eg-headers
-        """
-        return AsyncBlocksResourceWithRawResponse(self)
-
-    @cached_property
-    def with_streaming_response(self) -> AsyncBlocksResourceWithStreamingResponse:
-        """
-        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
-
-        For more information, see https://www.github.com/letta-ai/letta-python#with_streaming_response
-        """
-        return AsyncBlocksResourceWithStreamingResponse(self)
-
-    async def retrieve(
-        self,
-        block_label: str,
-        *,
-        agent_id: str,
-        # 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,
-    ) -> Block:
-        """
-        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
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        if not agent_id:
-            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
-        if not block_label:
-            raise ValueError(f"Expected a non-empty value for `block_label` but received {block_label!r}")
-        return await self._get(
-            f"/v1/agents/{agent_id}/core-memory/blocks/{block_label}",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Block,
-        )
-
-    async def update(
+    def modify(
         self,
         block_label: str,
         *,
@@ -403,10 +247,11 @@ class AsyncBlocksResource(AsyncAPIResource):
         label: Optional[str] | Omit = omit,
         limit: Optional[int] | Omit = omit,
         metadata: Optional[Dict[str, object]] | Omit = omit,
-        name: Optional[str] | Omit = omit,
         preserve_on_migration: Optional[bool] | Omit = omit,
         project_id: Optional[str] | Omit = omit,
         read_only: bool | Omit = omit,
+        template_id: Optional[str] | Omit = omit,
+        template_name: Optional[str] | Omit = omit,
         value: 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.
@@ -414,7 +259,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Block:
+    ) -> BlockModifyResponse:
         """
         Updates a core memory block of an agent.
 
@@ -439,14 +284,16 @@ class AsyncBlocksResource(AsyncAPIResource):
 
           metadata: Metadata of the block.
 
-          name: The id of the template.
-
           preserve_on_migration: Preserve the block on template migration.
 
           project_id: The associated project id.
 
           read_only: Whether the agent has read-only access to the block.
 
+          template_id: The id of the template.
+
+          template_name: Name of the block if it is a template.
+
           value: Value of the block.
 
           extra_headers: Send extra headers
@@ -461,9 +308,9 @@ class AsyncBlocksResource(AsyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         if not block_label:
             raise ValueError(f"Expected a non-empty value for `block_label` but received {block_label!r}")
-        return await self._patch(
+        return self._patch(
             f"/v1/agents/{agent_id}/core-memory/blocks/{block_label}",
-            body=await async_maybe_transform(
+            body=maybe_transform(
                 {
                     "base_template_id": base_template_id,
                     "deployment_id": deployment_id,
@@ -474,18 +321,78 @@ class AsyncBlocksResource(AsyncAPIResource):
                     "label": label,
                     "limit": limit,
                     "metadata": metadata,
-                    "name": name,
                     "preserve_on_migration": preserve_on_migration,
                     "project_id": project_id,
                     "read_only": read_only,
+                    "template_id": template_id,
+                    "template_name": template_name,
                     "value": value,
                 },
-                block_update_params.BlockUpdateParams,
+                block_modify_params.BlockModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
+            cast_to=BlockModifyResponse,
+        )
+
+
+class AsyncBlocksResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncBlocksResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/letta-ai/letta-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncBlocksResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncBlocksResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/letta-ai/letta-python#with_streaming_response
+        """
+        return AsyncBlocksResourceWithStreamingResponse(self)
+
+    async def retrieve(
+        self,
+        block_label: str,
+        *,
+        agent_id: str,
+        # 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,
+    ) -> BlockRetrieveResponse:
+        """
+        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
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not agent_id:
+            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
+        if not block_label:
+            raise ValueError(f"Expected a non-empty value for `block_label` but received {block_label!r}")
+        return await self._get(
+            f"/v1/agents/{agent_id}/core-memory/blocks/{block_label}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Block,
+            cast_to=BlockRetrieveResponse,
         )
 
     def list(
@@ -503,7 +410,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Block, AsyncArrayPage[Block]]:
+    ) -> AsyncPaginator[BlockListResponse, AsyncArrayPage[BlockListResponse]]:
         """
         Retrieve the core memory blocks of a specific agent.
 
@@ -535,7 +442,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         return self._get_api_list(
             f"/v1/agents/{agent_id}/core-memory/blocks",
-            page=AsyncArrayPage[Block],
+            page=AsyncArrayPage[BlockListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -552,7 +459,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            model=Block,
+            model=BlockListResponse,
         )
 
     async def attach(
@@ -635,6 +542,109 @@ class AsyncBlocksResource(AsyncAPIResource):
             cast_to=AgentState,
         )
 
+    async def modify(
+        self,
+        block_label: str,
+        *,
+        agent_id: str,
+        base_template_id: Optional[str] | Omit = omit,
+        deployment_id: Optional[str] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        entity_id: Optional[str] | Omit = omit,
+        hidden: Optional[bool] | Omit = omit,
+        is_template: bool | Omit = omit,
+        label: Optional[str] | Omit = omit,
+        limit: Optional[int] | Omit = omit,
+        metadata: Optional[Dict[str, object]] | Omit = omit,
+        preserve_on_migration: Optional[bool] | Omit = omit,
+        project_id: Optional[str] | Omit = omit,
+        read_only: bool | Omit = omit,
+        template_id: Optional[str] | Omit = omit,
+        template_name: Optional[str] | Omit = omit,
+        value: 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,
+    ) -> BlockModifyResponse:
+        """
+        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.
+
+          description: Description of the block.
+
+          entity_id: The id of the entity within the template.
+
+          hidden: If set to True, the block will be hidden.
+
+          is_template: Whether the block is a template (e.g. saved human/persona options).
+
+          label: Label of the block (e.g. 'human', 'persona') in the context window.
+
+          limit: Character limit of the block.
+
+          metadata: Metadata of the block.
+
+          preserve_on_migration: Preserve the block on template migration.
+
+          project_id: The associated project id.
+
+          read_only: Whether the agent has read-only access to the block.
+
+          template_id: The id of the template.
+
+          template_name: Name of the block if it is a template.
+
+          value: Value of the block.
+
+          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 agent_id:
+            raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
+        if not block_label:
+            raise ValueError(f"Expected a non-empty value for `block_label` but received {block_label!r}")
+        return await self._patch(
+            f"/v1/agents/{agent_id}/core-memory/blocks/{block_label}",
+            body=await async_maybe_transform(
+                {
+                    "base_template_id": base_template_id,
+                    "deployment_id": deployment_id,
+                    "description": description,
+                    "entity_id": entity_id,
+                    "hidden": hidden,
+                    "is_template": is_template,
+                    "label": label,
+                    "limit": limit,
+                    "metadata": metadata,
+                    "preserve_on_migration": preserve_on_migration,
+                    "project_id": project_id,
+                    "read_only": read_only,
+                    "template_id": template_id,
+                    "template_name": template_name,
+                    "value": value,
+                },
+                block_modify_params.BlockModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=BlockModifyResponse,
+        )
+
 
 class BlocksResourceWithRawResponse:
     def __init__(self, blocks: BlocksResource) -> None:
@@ -643,9 +653,6 @@ class BlocksResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             blocks.retrieve,
         )
-        self.update = to_raw_response_wrapper(
-            blocks.update,
-        )
         self.list = to_raw_response_wrapper(
             blocks.list,
         )
@@ -655,6 +662,9 @@ class BlocksResourceWithRawResponse:
         self.detach = to_raw_response_wrapper(
             blocks.detach,
         )
+        self.modify = to_raw_response_wrapper(
+            blocks.modify,
+        )
 
 
 class AsyncBlocksResourceWithRawResponse:
@@ -664,9 +674,6 @@ class AsyncBlocksResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             blocks.retrieve,
         )
-        self.update = async_to_raw_response_wrapper(
-            blocks.update,
-        )
         self.list = async_to_raw_response_wrapper(
             blocks.list,
         )
@@ -676,6 +683,9 @@ class AsyncBlocksResourceWithRawResponse:
         self.detach = async_to_raw_response_wrapper(
             blocks.detach,
         )
+        self.modify = async_to_raw_response_wrapper(
+            blocks.modify,
+        )
 
 
 class BlocksResourceWithStreamingResponse:
@@ -685,9 +695,6 @@ class BlocksResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             blocks.retrieve,
         )
-        self.update = to_streamed_response_wrapper(
-            blocks.update,
-        )
         self.list = to_streamed_response_wrapper(
             blocks.list,
         )
@@ -697,6 +704,9 @@ class BlocksResourceWithStreamingResponse:
         self.detach = to_streamed_response_wrapper(
             blocks.detach,
         )
+        self.modify = to_streamed_response_wrapper(
+            blocks.modify,
+        )
 
 
 class AsyncBlocksResourceWithStreamingResponse:
@@ -706,9 +716,6 @@ class AsyncBlocksResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             blocks.retrieve,
         )
-        self.update = async_to_streamed_response_wrapper(
-            blocks.update,
-        )
         self.list = async_to_streamed_response_wrapper(
             blocks.list,
         )
@@ -718,3 +725,6 @@ class AsyncBlocksResourceWithStreamingResponse:
         self.detach = async_to_streamed_response_wrapper(
             blocks.detach,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            blocks.modify,
+        )

letta_client/resources/agents/files.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,7 +17,8 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
+from ...pagination import SyncNextFilesPage, AsyncNextFilesPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import file_list_params
 from ...types.agents.file_list_response import FileListResponse
 from ...types.agents.file_open_response import FileOpenResponse
@@ -63,7 +64,7 @@ class FilesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> FileListResponse:
+    ) -> SyncNextFilesPage[FileListResponse]:
         """
         Get the files attached to an agent with their open/closed status.
 
@@ -97,8 +98,9 @@ class FilesResource(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}/files",
+            page=SyncNextFilesPage[FileListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -117,7 +119,7 @@ class FilesResource(SyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            cast_to=FileListResponse,
+            model=FileListResponse,
         )
 
     def close(
@@ -266,7 +268,7 @@ class AsyncFilesResource(AsyncAPIResource):
         """
         return AsyncFilesResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
@@ -283,7 +285,7 @@ class AsyncFilesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> FileListResponse:
+    ) -> AsyncPaginator[FileListResponse, AsyncNextFilesPage[FileListResponse]]:
         """
         Get the files attached to an agent with their open/closed status.
 
@@ -317,14 +319,15 @@ class AsyncFilesResource(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}/files",
+            page=AsyncNextFilesPage[FileListResponse],
             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,
                         "before": before,
@@ -337,7 +340,7 @@ class AsyncFilesResource(AsyncAPIResource):
                     file_list_params.FileListParams,
                 ),
             ),
-            cast_to=FileListResponse,
+            model=FileListResponse,
         )
 
     async def close(