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

letta_client/resources/blocks/agents.py

@@ -0,0 +1,287 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List, Optional
+from typing_extensions import Literal
+
+import httpx
+
+from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
+from ..._utils import 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 ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
+from ...types.blocks import agent_list_params
+from ...types.agent_state import AgentState
+
+__all__ = ["AgentsResource", "AsyncAgentsResource"]
+
+
+class AgentsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AgentsResourceWithRawResponse:
+        """
+        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 AgentsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AgentsResourceWithStreamingResponse:
+        """
+        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 AgentsResourceWithStreamingResponse(self)
+
+    def list(
+        self,
+        block_id: str,
+        *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        include: List[
+            Literal[
+                "agent.blocks",
+                "agent.identities",
+                "agent.managed_group",
+                "agent.secrets",
+                "agent.sources",
+                "agent.tags",
+                "agent.tools",
+            ]
+        ]
+        | Omit = omit,
+        include_relationships: Optional[SequenceNotStr[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,
+    ) -> SyncArrayPage[AgentState]:
+        """Retrieves all agents associated with the specified block.
+
+        Raises a 404 if the
+        block does not exist.
+
+        Args:
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
+          after: Agent ID cursor for pagination. Returns agents that come after this agent ID in
+              the specified sort order
+
+          before: Agent ID cursor for pagination. Returns agents that come before this agent ID in
+              the specified sort order
+
+          include: Specify which relational fields to include in the response. No relationships are
+              included by default.
+
+          include_relationships: Specify which relational fields (e.g., 'tools', 'sources', 'memory') to include
+              in the response. If not provided, all relationships are loaded by default. Using
+              this can optimize performance by reducing unnecessary joins.This is a legacy
+              parameter, and no longer supported after 1.0.0 SDK versions.
+
+          limit: Maximum number of agents to return
+
+          order: Sort order for agents 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
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not block_id:
+            raise ValueError(f"Expected a non-empty value for `block_id` but received {block_id!r}")
+        return self._get_api_list(
+            f"/v1/blocks/{block_id}/agents",
+            page=SyncArrayPage[AgentState],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "before": before,
+                        "include": include,
+                        "include_relationships": include_relationships,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                    },
+                    agent_list_params.AgentListParams,
+                ),
+            ),
+            model=AgentState,
+        )
+
+
+class AsyncAgentsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncAgentsResourceWithRawResponse:
+        """
+        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 AsyncAgentsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncAgentsResourceWithStreamingResponse:
+        """
+        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 AsyncAgentsResourceWithStreamingResponse(self)
+
+    def list(
+        self,
+        block_id: str,
+        *,
+        after: Optional[str] | Omit = omit,
+        before: Optional[str] | Omit = omit,
+        include: List[
+            Literal[
+                "agent.blocks",
+                "agent.identities",
+                "agent.managed_group",
+                "agent.secrets",
+                "agent.sources",
+                "agent.tags",
+                "agent.tools",
+            ]
+        ]
+        | Omit = omit,
+        include_relationships: Optional[SequenceNotStr[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,
+    ) -> AsyncPaginator[AgentState, AsyncArrayPage[AgentState]]:
+        """Retrieves all agents associated with the specified block.
+
+        Raises a 404 if the
+        block does not exist.
+
+        Args:
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
+          after: Agent ID cursor for pagination. Returns agents that come after this agent ID in
+              the specified sort order
+
+          before: Agent ID cursor for pagination. Returns agents that come before this agent ID in
+              the specified sort order
+
+          include: Specify which relational fields to include in the response. No relationships are
+              included by default.
+
+          include_relationships: Specify which relational fields (e.g., 'tools', 'sources', 'memory') to include
+              in the response. If not provided, all relationships are loaded by default. Using
+              this can optimize performance by reducing unnecessary joins.This is a legacy
+              parameter, and no longer supported after 1.0.0 SDK versions.
+
+          limit: Maximum number of agents to return
+
+          order: Sort order for agents 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
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not block_id:
+            raise ValueError(f"Expected a non-empty value for `block_id` but received {block_id!r}")
+        return self._get_api_list(
+            f"/v1/blocks/{block_id}/agents",
+            page=AsyncArrayPage[AgentState],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "after": after,
+                        "before": before,
+                        "include": include,
+                        "include_relationships": include_relationships,
+                        "limit": limit,
+                        "order": order,
+                        "order_by": order_by,
+                    },
+                    agent_list_params.AgentListParams,
+                ),
+            ),
+            model=AgentState,
+        )
+
+
+class AgentsResourceWithRawResponse:
+    def __init__(self, agents: AgentsResource) -> None:
+        self._agents = agents
+
+        self.list = to_raw_response_wrapper(
+            agents.list,
+        )
+
+
+class AsyncAgentsResourceWithRawResponse:
+    def __init__(self, agents: AsyncAgentsResource) -> None:
+        self._agents = agents
+
+        self.list = async_to_raw_response_wrapper(
+            agents.list,
+        )
+
+
+class AgentsResourceWithStreamingResponse:
+    def __init__(self, agents: AgentsResource) -> None:
+        self._agents = agents
+
+        self.list = to_streamed_response_wrapper(
+            agents.list,
+        )
+
+
+class AsyncAgentsResourceWithStreamingResponse:
+    def __init__(self, agents: AsyncAgentsResource) -> None:
+        self._agents = agents
+
+        self.list = async_to_streamed_response_wrapper(
+            agents.list,
+        )

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

@@ -7,27 +7,38 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..types import block_list_params, block_create_params, block_update_params, block_list_agents_params
-from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, 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 .agents import (
+    AgentsResource,
+    AsyncAgentsResource,
+    AgentsResourceWithRawResponse,
+    AsyncAgentsResourceWithRawResponse,
+    AgentsResourceWithStreamingResponse,
+    AsyncAgentsResourceWithStreamingResponse,
+)
+from ...types import block_list_params, block_create_params, block_update_params
+from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, 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.block_list_response import BlockListResponse
-from ..types.block_count_response import BlockCountResponse
-from ..types.agents.core_memory.block import Block
-from ..types.block_list_agents_response import BlockListAgentsResponse
+from ...pagination import SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
+from ...types.agents.block import Block
+from ...types.block_count_response import BlockCountResponse
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
 
 class BlocksResource(SyncAPIResource):
+    @cached_property
+    def agents(self) -> AgentsResource:
+        return AgentsResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> BlocksResourceWithRawResponse:
         """
@@ -151,6 +162,8 @@ class BlocksResource(SyncAPIResource):
         Retrieve Block
 
         Args:
+          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
@@ -198,6 +211,8 @@ class BlocksResource(SyncAPIResource):
         Modify Block
 
         Args:
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
           base_template_id: The base template id of the block.
 
           deployment_id: The id of the deployment.
@@ -289,7 +304,7 @@ class BlocksResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockListResponse:
+    ) -> SyncArrayPage[Block]:
         """List Blocks
 
         Args:
@@ -345,8 +360,9 @@ class BlocksResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/blocks/",
+            page=SyncArrayPage[Block],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -375,7 +391,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
     def delete(
@@ -393,6 +409,8 @@ class BlocksResource(SyncAPIResource):
         Delete Block
 
         Args:
+          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
@@ -430,80 +448,12 @@ class BlocksResource(SyncAPIResource):
             cast_to=int,
         )
 
-    def list_agents(
-        self,
-        block_id: str,
-        *,
-        after: Optional[str] | Omit = omit,
-        before: Optional[str] | Omit = omit,
-        include_relationships: Optional[SequenceNotStr[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,
-    ) -> BlockListAgentsResponse:
-        """Retrieves all agents associated with the specified block.
-
-        Raises a 404 if the
-        block does not exist.
-
-        Args:
-          after: Agent ID cursor for pagination. Returns agents that come after this agent ID in
-              the specified sort order
-
-          before: Agent ID cursor for pagination. Returns agents that come before this agent ID in
-              the specified sort order
-
-          include_relationships: Specify which relational fields (e.g., 'tools', 'sources', 'memory') to include
-              in the response. If not provided, all relationships are loaded by default. Using
-              this can optimize performance by reducing unnecessary joins.
-
-          limit: Maximum number of agents to return
-
-          order: Sort order for agents 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
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        if not block_id:
-            raise ValueError(f"Expected a non-empty value for `block_id` but received {block_id!r}")
-        return self._get(
-            f"/v1/blocks/{block_id}/agents",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "after": after,
-                        "before": before,
-                        "include_relationships": include_relationships,
-                        "limit": limit,
-                        "order": order,
-                        "order_by": order_by,
-                    },
-                    block_list_agents_params.BlockListAgentsParams,
-                ),
-            ),
-            cast_to=BlockListAgentsResponse,
-        )
-
 
 class AsyncBlocksResource(AsyncAPIResource):
+    @cached_property
+    def agents(self) -> AsyncAgentsResource:
+        return AsyncAgentsResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> AsyncBlocksResourceWithRawResponse:
         """
@@ -627,6 +577,8 @@ class AsyncBlocksResource(AsyncAPIResource):
         Retrieve Block
 
         Args:
+          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
@@ -674,6 +626,8 @@ class AsyncBlocksResource(AsyncAPIResource):
         Modify Block
 
         Args:
+          block_id: The ID of the block in the format 'block-<uuid4>'
+
           base_template_id: The base template id of the block.
 
           deployment_id: The id of the deployment.
@@ -739,7 +693,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             cast_to=Block,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -765,7 +719,7 @@ class AsyncBlocksResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BlockListResponse:
+    ) -> AsyncPaginator[Block, AsyncArrayPage[Block]]:
         """List Blocks
 
         Args:
@@ -821,14 +775,15 @@ class AsyncBlocksResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/blocks/",
+            page=AsyncArrayPage[Block],
             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,
@@ -851,7 +806,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
     async def delete(
@@ -869,6 +824,8 @@ class AsyncBlocksResource(AsyncAPIResource):
         Delete Block
 
         Args:
+          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
@@ -906,78 +863,6 @@ class AsyncBlocksResource(AsyncAPIResource):
             cast_to=int,
         )
 
-    async def list_agents(
-        self,
-        block_id: str,
-        *,
-        after: Optional[str] | Omit = omit,
-        before: Optional[str] | Omit = omit,
-        include_relationships: Optional[SequenceNotStr[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,
-    ) -> BlockListAgentsResponse:
-        """Retrieves all agents associated with the specified block.
-
-        Raises a 404 if the
-        block does not exist.
-
-        Args:
-          after: Agent ID cursor for pagination. Returns agents that come after this agent ID in
-              the specified sort order
-
-          before: Agent ID cursor for pagination. Returns agents that come before this agent ID in
-              the specified sort order
-
-          include_relationships: Specify which relational fields (e.g., 'tools', 'sources', 'memory') to include
-              in the response. If not provided, all relationships are loaded by default. Using
-              this can optimize performance by reducing unnecessary joins.
-
-          limit: Maximum number of agents to return
-
-          order: Sort order for agents 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
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        if not block_id:
-            raise ValueError(f"Expected a non-empty value for `block_id` but received {block_id!r}")
-        return await self._get(
-            f"/v1/blocks/{block_id}/agents",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=await async_maybe_transform(
-                    {
-                        "after": after,
-                        "before": before,
-                        "include_relationships": include_relationships,
-                        "limit": limit,
-                        "order": order,
-                        "order_by": order_by,
-                    },
-                    block_list_agents_params.BlockListAgentsParams,
-                ),
-            ),
-            cast_to=BlockListAgentsResponse,
-        )
-
 
 class BlocksResourceWithRawResponse:
     def __init__(self, blocks: BlocksResource) -> None:
@@ -1001,9 +886,10 @@ class BlocksResourceWithRawResponse:
         self.count = to_raw_response_wrapper(
             blocks.count,
         )
-        self.list_agents = to_raw_response_wrapper(
-            blocks.list_agents,
-        )
+
+    @cached_property
+    def agents(self) -> AgentsResourceWithRawResponse:
+        return AgentsResourceWithRawResponse(self._blocks.agents)
 
 
 class AsyncBlocksResourceWithRawResponse:
@@ -1028,9 +914,10 @@ class AsyncBlocksResourceWithRawResponse:
         self.count = async_to_raw_response_wrapper(
             blocks.count,
         )
-        self.list_agents = async_to_raw_response_wrapper(
-            blocks.list_agents,
-        )
+
+    @cached_property
+    def agents(self) -> AsyncAgentsResourceWithRawResponse:
+        return AsyncAgentsResourceWithRawResponse(self._blocks.agents)
 
 
 class BlocksResourceWithStreamingResponse:
@@ -1055,9 +942,10 @@ class BlocksResourceWithStreamingResponse:
         self.count = to_streamed_response_wrapper(
             blocks.count,
         )
-        self.list_agents = to_streamed_response_wrapper(
-            blocks.list_agents,
-        )
+
+    @cached_property
+    def agents(self) -> AgentsResourceWithStreamingResponse:
+        return AgentsResourceWithStreamingResponse(self._blocks.agents)
 
 
 class AsyncBlocksResourceWithStreamingResponse:
@@ -1082,6 +970,7 @@ class AsyncBlocksResourceWithStreamingResponse:
         self.count = async_to_streamed_response_wrapper(
             blocks.count,
         )
-        self.list_agents = async_to_streamed_response_wrapper(
-            blocks.list_agents,
-        )
+
+    @cached_property
+    def agents(self) -> AsyncAgentsResourceWithStreamingResponse:
+        return AsyncAgentsResourceWithStreamingResponse(self._blocks.agents)