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

letta_client/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "letta_client"
-__version__ = "1.0.0-alpha.7"  # x-release-please-version
+__version__ = "1.0.0-alpha.8"  # x-release-please-version

letta_client/pagination.py

@@ -0,0 +1,107 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Any, List, Type, Generic, Mapping, TypeVar, Optional, cast
+from typing_extensions import Protocol, override, runtime_checkable
+
+from httpx import Response
+
+from ._utils import is_mapping
+from ._models import BaseModel
+from ._base_client import BasePage, PageInfo, BaseSyncPage, BaseAsyncPage
+
+__all__ = ["SyncArrayPage", "AsyncArrayPage"]
+
+_BaseModelT = TypeVar("_BaseModelT", bound=BaseModel)
+
+_T = TypeVar("_T")
+
+
+@runtime_checkable
+class ArrayPageItem(Protocol):
+    id: Optional[str]
+
+
+class SyncArrayPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
+    items: List[_T]
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        items = self.items
+        if not items:
+            return []
+        return items
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        is_forwards = not self._options.params.get("before", False)
+
+        items = self.items
+        if not items:
+            return None
+
+        if is_forwards:
+            item = cast(Any, items[-1])
+            if not isinstance(item, ArrayPageItem) or item.id is None:
+                # TODO emit warning log
+                return None
+
+            return PageInfo(params={"after": item.id})
+        else:
+            item = cast(Any, self.items[0])
+            if not isinstance(item, ArrayPageItem) or item.id is None:
+                # TODO emit warning log
+                return None
+
+            return PageInfo(params={"before": item.id})
+
+    @classmethod
+    def build(cls: Type[_BaseModelT], *, response: Response, data: object) -> _BaseModelT:  # noqa: ARG003
+        return cls.construct(
+            None,
+            **{
+                **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
+            },
+        )
+
+
+class AsyncArrayPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
+    items: List[_T]
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        items = self.items
+        if not items:
+            return []
+        return items
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        is_forwards = not self._options.params.get("before", False)
+
+        items = self.items
+        if not items:
+            return None
+
+        if is_forwards:
+            item = cast(Any, items[-1])
+            if not isinstance(item, ArrayPageItem) or item.id is None:
+                # TODO emit warning log
+                return None
+
+            return PageInfo(params={"after": item.id})
+        else:
+            item = cast(Any, self.items[0])
+            if not isinstance(item, ArrayPageItem) or item.id is None:
+                # TODO emit warning log
+                return None
+
+            return PageInfo(params={"before": item.id})
+
+    @classmethod
+    def build(cls: Type[_BaseModelT], *, response: Response, data: object) -> _BaseModelT:  # noqa: ARG003
+        return cls.construct(
+            None,
+            **{
+                **(cast(Mapping[str, Any], data) if is_mapping(data) else {"items": data}),
+            },
+        )

letta_client/resources/agents/agents.py

@@ -85,12 +85,12 @@ 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.agent_type import AgentType
 from ...types.agent_state import AgentState
 from ...types.llm_config_param import LlmConfigParam
 from ...types.create_block_param import CreateBlockParam
-from ...types.agent_list_response import AgentListResponse
 from ...types.agent_count_response import AgentCountResponse
 from ...types.message_create_param import MessageCreateParam
 from ...types.embedding_config_param import EmbeddingConfigParam
@@ -658,7 +658,7 @@ class AgentsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentListResponse:
+    ) -> SyncArrayPage[AgentState]:
         """
         Get a list of all agents.
 
@@ -714,8 +714,9 @@ class AgentsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/agents/",
+            page=SyncArrayPage[AgentState],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -745,7 +746,7 @@ class AgentsResource(SyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            cast_to=AgentListResponse,
+            model=AgentState,
         )
 
     def delete(
@@ -1454,7 +1455,7 @@ class AsyncAgentsResource(AsyncAPIResource):
             cast_to=AgentState,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -1492,7 +1493,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentListResponse:
+    ) -> AsyncPaginator[AgentState, AsyncArrayPage[AgentState]]:
         """
         Get a list of all agents.
 
@@ -1548,14 +1549,15 @@ class AsyncAgentsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/agents/",
+            page=AsyncArrayPage[AgentState],
             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,
                         "ascending": ascending,
@@ -1579,7 +1581,7 @@ class AsyncAgentsResource(AsyncAPIResource):
                     agent_list_params.AgentListParams,
                 ),
             ),
-            cast_to=AgentListResponse,
+            model=AgentState,
         )
 
     async def delete(

letta_client/resources/agents/blocks.py

@@ -17,11 +17,11 @@ 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 block_list_params, block_update_params
 from ...types.agent_state import AgentState
 from ...types.agents.block import Block
-from ...types.agents.block_list_response import BlockListResponse
 
 __all__ = ["BlocksResource", "AsyncBlocksResource"]
 
@@ -198,7 +198,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]:
         """
         Retrieve the core memory blocks of a specific agent.
 
@@ -228,8 +228,9 @@ 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,
@@ -246,7 +247,7 @@ class BlocksResource(SyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
     def attach(
@@ -487,7 +488,7 @@ class AsyncBlocksResource(AsyncAPIResource):
             cast_to=Block,
         )
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
@@ -502,7 +503,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]]:
         """
         Retrieve the core memory blocks of a specific agent.
 
@@ -532,14 +533,15 @@ 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,
-                query=await async_maybe_transform(
+                query=maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -550,7 +552,7 @@ class AsyncBlocksResource(AsyncAPIResource):
                     block_list_params.BlockListParams,
                 ),
             ),
-            cast_to=BlockListResponse,
+            model=Block,
         )
 
     async def attach(

letta_client/resources/agents/folders.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 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
@@ -60,7 +61,7 @@ class FoldersResource(SyncAPIResource):
         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.
 
@@ -90,8 +91,9 @@ 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,
@@ -108,7 +110,7 @@ class FoldersResource(SyncAPIResource):
                     folder_list_params.FolderListParams,
                 ),
             ),
-            cast_to=FolderListResponse,
+            model=FolderListResponse,
         )
 
     def attach(
@@ -212,7 +214,7 @@ class AsyncFoldersResource(AsyncAPIResource):
         """
         return AsyncFoldersResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
@@ -227,7 +229,7 @@ class AsyncFoldersResource(AsyncAPIResource):
         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.
 
@@ -257,14 +259,15 @@ 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,
-                query=await async_maybe_transform(
+                query=maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -275,7 +278,7 @@ class AsyncFoldersResource(AsyncAPIResource):
                     folder_list_params.FolderListParams,
                 ),
             ),
-            cast_to=FolderListResponse,
+            model=FolderListResponse,
         )
 
     async def attach(

letta_client/resources/agents/groups.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,9 +17,10 @@ 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 ...types.group import Group
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import group_list_params
-from ...types.agents.group_list_response import GroupListResponse
 
 __all__ = ["GroupsResource", "AsyncGroupsResource"]
 
@@ -60,7 +61,7 @@ class GroupsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> GroupListResponse:
+    ) -> SyncArrayPage[Group]:
         """
         Lists the groups for an agent.
 
@@ -92,8 +93,9 @@ class GroupsResource(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}/groups",
+            page=SyncArrayPage[Group],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -111,7 +113,7 @@ class GroupsResource(SyncAPIResource):
                     group_list_params.GroupListParams,
                 ),
             ),
-            cast_to=GroupListResponse,
+            model=Group,
         )
 
 
@@ -135,7 +137,7 @@ class AsyncGroupsResource(AsyncAPIResource):
         """
         return AsyncGroupsResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
@@ -151,7 +153,7 @@ class AsyncGroupsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> GroupListResponse:
+    ) -> AsyncPaginator[Group, AsyncArrayPage[Group]]:
         """
         Lists the groups for an agent.
 
@@ -183,14 +185,15 @@ class AsyncGroupsResource(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}/groups",
+            page=AsyncArrayPage[Group],
             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,
@@ -202,7 +205,7 @@ class AsyncGroupsResource(AsyncAPIResource):
                     group_list_params.GroupListParams,
                 ),
             ),
-            cast_to=GroupListResponse,
+            model=Group,
         )
 
 

letta_client/resources/agents/messages.py

@@ -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 SyncArrayPage, AsyncArrayPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.agents import (
     message_list_params,
     message_send_params,
@@ -31,7 +32,7 @@ from ...types.agents.run import Run
 from ...types.agent_state import AgentState
 from ...types.agents.message_type import MessageType
 from ...types.agents.letta_response import LettaResponse
-from ...types.agents.message_list_response import MessageListResponse
+from ...types.agents.letta_message_union import LettaMessageUnion
 from ...types.agents.message_cancel_response import MessageCancelResponse
 from ...types.agents.message_update_response import MessageUpdateResponse
 from ...types.agents.letta_user_message_content_union_param import LettaUserMessageContentUnionParam
@@ -266,7 +267,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> SyncArrayPage[LettaMessageUnion]:
         """
         Retrieve message history for an agent.
 
@@ -307,8 +308,9 @@ class MessagesResource(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}/messages",
+            page=SyncArrayPage[LettaMessageUnion],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -330,7 +332,7 @@ class MessagesResource(SyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
     def cancel(
@@ -908,7 +910,7 @@ class AsyncMessagesResource(AsyncAPIResource):
             ),
         )
 
-    async def list(
+    def list(
         self,
         agent_id: str,
         *,
@@ -928,7 +930,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> AsyncPaginator[LettaMessageUnion, AsyncArrayPage[LettaMessageUnion]]:
         """
         Retrieve message history for an agent.
 
@@ -969,14 +971,15 @@ class AsyncMessagesResource(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}/messages",
+            page=AsyncArrayPage[LettaMessageUnion],
             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,
                         "assistant_message_tool_kwarg": assistant_message_tool_kwarg,
@@ -992,7 +995,7 @@ class AsyncMessagesResource(AsyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=cast(Any, LettaMessageUnion),  # Union types cannot be passed in as arguments in the type system
         )
 
     async def cancel(

letta_client/resources/agents/tools.py

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