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

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

@@ -26,10 +26,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ...pagination import SyncArrayPage, AsyncArrayPage
 from ...types.group import Group
-from ..._base_client import make_request_options
+from ..._base_client import AsyncPaginator, make_request_options
 from ...types.manager_type import ManagerType
-from ...types.group_list_response import GroupListResponse
 from ...types.group_count_response import GroupCountResponse
 
 __all__ = ["GroupsResource", "AsyncGroupsResource"]
@@ -137,6 +137,8 @@ class GroupsResource(SyncAPIResource):
         Retrieve the group by id.
 
         Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -176,6 +178,8 @@ class GroupsResource(SyncAPIResource):
         Create a new multi-agent group with the specified configuration.
 
         Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
           agent_ids
 
           description
@@ -233,7 +237,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]:
         """
         Fetch all multi-agent groups matching query.
 
@@ -263,8 +267,9 @@ class GroupsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get(
+        return self._get_api_list(
             "/v1/groups/",
+            page=SyncArrayPage[Group],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -283,7 +288,7 @@ class GroupsResource(SyncAPIResource):
                     group_list_params.GroupListParams,
                 ),
             ),
-            cast_to=GroupListResponse,
+            model=Group,
         )
 
     def delete(
@@ -301,6 +306,8 @@ class GroupsResource(SyncAPIResource):
         Delete a multi-agent group.
 
         Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -338,39 +345,6 @@ class GroupsResource(SyncAPIResource):
             cast_to=int,
         )
 
-    def reset_messages(
-        self,
-        group_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,
-    ) -> object:
-        """
-        Delete the group messages for all agents that are part of the multi-agent group.
-
-        Args:
-          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 group_id:
-            raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        return self._patch(
-            f"/v1/groups/{group_id}/reset-messages",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=object,
-        )
-
 
 class AsyncGroupsResource(AsyncAPIResource):
     @cached_property
@@ -474,6 +448,8 @@ class AsyncGroupsResource(AsyncAPIResource):
         Retrieve the group by id.
 
         Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -513,6 +489,8 @@ class AsyncGroupsResource(AsyncAPIResource):
         Create a new multi-agent group with the specified configuration.
 
         Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
           agent_ids
 
           description
@@ -554,7 +532,7 @@ class AsyncGroupsResource(AsyncAPIResource):
             cast_to=Group,
         )
 
-    async def list(
+    def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -570,7 +548,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]]:
         """
         Fetch all multi-agent groups matching query.
 
@@ -600,14 +578,15 @@ class AsyncGroupsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return await self._get(
+        return self._get_api_list(
             "/v1/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,
@@ -620,7 +599,7 @@ class AsyncGroupsResource(AsyncAPIResource):
                     group_list_params.GroupListParams,
                 ),
             ),
-            cast_to=GroupListResponse,
+            model=Group,
         )
 
     async def delete(
@@ -638,6 +617,8 @@ class AsyncGroupsResource(AsyncAPIResource):
         Delete a multi-agent group.
 
         Args:
+          group_id: The ID of the group in the format 'group-<uuid4>'
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -675,39 +656,6 @@ class AsyncGroupsResource(AsyncAPIResource):
             cast_to=int,
         )
 
-    async def reset_messages(
-        self,
-        group_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,
-    ) -> object:
-        """
-        Delete the group messages for all agents that are part of the multi-agent group.
-
-        Args:
-          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 group_id:
-            raise ValueError(f"Expected a non-empty value for `group_id` but received {group_id!r}")
-        return await self._patch(
-            f"/v1/groups/{group_id}/reset-messages",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=object,
-        )
-
 
 class GroupsResourceWithRawResponse:
     def __init__(self, groups: GroupsResource) -> None:
@@ -731,9 +679,6 @@ class GroupsResourceWithRawResponse:
         self.count = to_raw_response_wrapper(
             groups.count,
         )
-        self.reset_messages = to_raw_response_wrapper(
-            groups.reset_messages,
-        )
 
     @cached_property
     def messages(self) -> MessagesResourceWithRawResponse:
@@ -762,9 +707,6 @@ class AsyncGroupsResourceWithRawResponse:
         self.count = async_to_raw_response_wrapper(
             groups.count,
         )
-        self.reset_messages = async_to_raw_response_wrapper(
-            groups.reset_messages,
-        )
 
     @cached_property
     def messages(self) -> AsyncMessagesResourceWithRawResponse:
@@ -793,9 +735,6 @@ class GroupsResourceWithStreamingResponse:
         self.count = to_streamed_response_wrapper(
             groups.count,
         )
-        self.reset_messages = to_streamed_response_wrapper(
-            groups.reset_messages,
-        )
 
     @cached_property
     def messages(self) -> MessagesResourceWithStreamingResponse:
@@ -824,9 +763,6 @@ class AsyncGroupsResourceWithStreamingResponse:
         self.count = async_to_streamed_response_wrapper(
             groups.count,
         )
-        self.reset_messages = async_to_streamed_response_wrapper(
-            groups.reset_messages,
-        )
 
     @cached_property
     def messages(self) -> AsyncMessagesResourceWithStreamingResponse: