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

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(
@@ -122,7 +124,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Attach a tool to an agent.
 
@@ -162,7 +164,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Detach a tool from an agent.
 
@@ -204,7 +206,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Modify the approval requirement for a tool attached to an agent.
 
@@ -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(
@@ -346,7 +349,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Attach a tool to an agent.
 
@@ -386,7 +389,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Detach a tool from an agent.
 
@@ -428,7 +431,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AgentState:
+    ) -> Optional[AgentState]:
         """
         Modify the approval requirement for a tool attached to an agent.
 

letta_client/resources/archives.py

@@ -18,9 +18,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.archive import Archive
-from ..types.archive_list_response import ArchiveListResponse
 from ..types.embedding_config_param import EmbeddingConfigParam
 
 __all__ = ["ArchivesResource", "AsyncArchivesResource"]
@@ -177,13 +177,14 @@ class ArchivesResource(SyncAPIResource):
         limit: Optional[int] | Omit = omit,
         name: Optional[str] | 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,
-    ) -> ArchiveListResponse:
+    ) -> SyncArrayPage[Archive]:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -204,6 +205,8 @@ class ArchivesResource(SyncAPIResource):
           order: Sort order for archives 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
@@ -212,8 +215,9 @@ class ArchivesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/archives/",
+            page=SyncArrayPage[Archive],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -227,11 +231,12 @@ class ArchivesResource(SyncAPIResource):
                         "limit": limit,
                         "name": name,
                         "order": order,
+                        "order_by": order_by,
                     },
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            cast_to=ArchiveListResponse,
+            model=Archive,
         )
 
     def delete(
@@ -412,7 +417,7 @@ class AsyncArchivesResource(AsyncAPIResource):
             cast_to=Archive,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -421,13 +426,14 @@ class AsyncArchivesResource(AsyncAPIResource):
         limit: Optional[int] | Omit = omit,
         name: Optional[str] | 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,
-    ) -> ArchiveListResponse:
+    ) -> AsyncPaginator[Archive, AsyncArrayPage[Archive]]:
         """
         Get a list of all archives for the current organization with optional filters
         and pagination.
@@ -448,6 +454,8 @@ class AsyncArchivesResource(AsyncAPIResource):
           order: Sort order for archives 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
@@ -456,14 +464,15 @@ class AsyncArchivesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/archives/",
+            page=AsyncArrayPage[Archive],
             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,
                         "agent_id": agent_id,
@@ -471,11 +480,12 @@ class AsyncArchivesResource(AsyncAPIResource):
                         "limit": limit,
                         "name": name,
                         "order": order,
+                        "order_by": order_by,
                     },
                     archive_list_params.ArchiveListParams,
                 ),
             ),
-            cast_to=ArchiveListResponse,
+            model=Archive,
         )
 
     async def delete(

letta_client/resources/batches/batches.py

@@ -26,9 +26,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.batch_job import BatchJob
-from ...types.batch_list_response import BatchListResponse
 
 __all__ = ["BatchesResource", "AsyncBatchesResource"]
 
@@ -155,7 +155,7 @@ class BatchesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BatchListResponse:
+    ) -> SyncArrayPage[BatchJob]:
         """List all batch runs.
 
         Args:
@@ -182,8 +182,9 @@ class BatchesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/messages/batches",
+            page=SyncArrayPage[BatchJob],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -200,7 +201,7 @@ class BatchesResource(SyncAPIResource):
                     batch_list_params.BatchListParams,
                 ),
             ),
-            cast_to=BatchListResponse,
+            model=BatchJob,
         )
 
     def cancel(
@@ -345,7 +346,7 @@ class AsyncBatchesResource(AsyncAPIResource):
             cast_to=BatchJob,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -359,7 +360,7 @@ class AsyncBatchesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> BatchListResponse:
+    ) -> AsyncPaginator[BatchJob, AsyncArrayPage[BatchJob]]:
         """List all batch runs.
 
         Args:
@@ -386,14 +387,15 @@ class AsyncBatchesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/messages/batches",
+            page=AsyncArrayPage[BatchJob],
             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,
@@ -404,7 +406,7 @@ class AsyncBatchesResource(AsyncAPIResource):
                     batch_list_params.BatchListParams,
                 ),
             ),
-            cast_to=BatchListResponse,
+            model=BatchJob,
         )
 
     async def cancel(

letta_client/resources/batches/messages.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 SyncObjectPage, AsyncObjectPage
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.batches import message_list_params
-from ...types.batches.message_list_response import MessageListResponse
+from ...types.agents.message import Message
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -60,7 +61,7 @@ class MessagesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> SyncObjectPage[Message]:
         """
         Get response messages for a specific batch job.
 
@@ -90,8 +91,9 @@ class MessagesResource(SyncAPIResource):
         """
         if not batch_id:
             raise ValueError(f"Expected a non-empty value for `batch_id` but received {batch_id!r}")
-        return self._get(
+        return self._get_api_list(
             f"/v1/messages/batches/{batch_id}/messages",
+            page=SyncObjectPage[Message],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -109,7 +111,7 @@ class MessagesResource(SyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=Message,
         )
 
 
@@ -133,7 +135,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         return AsyncMessagesResourceWithStreamingResponse(self)
 
-    async def list(
+    def list(
         self,
         batch_id: str,
         *,
@@ -149,7 +151,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> MessageListResponse:
+    ) -> AsyncPaginator[Message, AsyncObjectPage[Message]]:
         """
         Get response messages for a specific batch job.
 
@@ -179,14 +181,15 @@ class AsyncMessagesResource(AsyncAPIResource):
         """
         if not batch_id:
             raise ValueError(f"Expected a non-empty value for `batch_id` but received {batch_id!r}")
-        return await self._get(
+        return self._get_api_list(
             f"/v1/messages/batches/{batch_id}/messages",
+            page=AsyncObjectPage[Message],
             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,
                         "agent_id": agent_id,
@@ -198,7 +201,7 @@ class AsyncMessagesResource(AsyncAPIResource):
                     message_list_params.MessageListParams,
                 ),
             ),
-            cast_to=MessageListResponse,
+            model=Message,
         )