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

letta_client/resources/groups/groups.py

@@ -7,7 +7,7 @@ from typing_extensions import Literal
 
 import httpx
 
-from ...types import ManagerType, group_list_params, group_create_params, group_update_params
+from ...types import ManagerType, group_list_params, group_create_params, group_modify_params
 from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
 from ..._utils import maybe_transform, strip_not_given, async_maybe_transform
 from .messages import (
@@ -157,70 +157,6 @@ class GroupsResource(SyncAPIResource):
             cast_to=Group,
         )
 
-    def update(
-        self,
-        group_id: str,
-        *,
-        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        description: Optional[str] | Omit = omit,
-        manager_config: Optional[group_update_params.ManagerConfig] | Omit = omit,
-        project_id: Optional[str] | Omit = omit,
-        shared_block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: 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,
-    ) -> Group:
-        """
-        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
-
-          manager_config
-
-          project_id: The associated project id.
-
-          shared_block_ids
-
-          x_project: The project slug to associate with the group (cloud only).
-
-          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}")
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
-        return self._patch(
-            f"/v1/groups/{group_id}",
-            body=maybe_transform(
-                {
-                    "agent_ids": agent_ids,
-                    "description": description,
-                    "manager_config": manager_config,
-                    "project_id": project_id,
-                    "shared_block_ids": shared_block_ids,
-                },
-                group_update_params.GroupUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Group,
-        )
-
     def list(
         self,
         *,
@@ -345,6 +281,70 @@ class GroupsResource(SyncAPIResource):
             cast_to=int,
         )
 
+    def modify(
+        self,
+        group_id: str,
+        *,
+        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        manager_config: Optional[group_modify_params.ManagerConfig] | Omit = omit,
+        project_id: Optional[str] | Omit = omit,
+        shared_block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        x_project: 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,
+    ) -> Group:
+        """
+        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
+
+          manager_config
+
+          project_id: The associated project id.
+
+          shared_block_ids
+
+          x_project: The project slug to associate with the group (cloud only).
+
+          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}")
+        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
+        return self._patch(
+            f"/v1/groups/{group_id}",
+            body=maybe_transform(
+                {
+                    "agent_ids": agent_ids,
+                    "description": description,
+                    "manager_config": manager_config,
+                    "project_id": project_id,
+                    "shared_block_ids": shared_block_ids,
+                },
+                group_modify_params.GroupModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Group,
+        )
+
 
 class AsyncGroupsResource(AsyncAPIResource):
     @cached_property
@@ -468,70 +468,6 @@ class AsyncGroupsResource(AsyncAPIResource):
             cast_to=Group,
         )
 
-    async def update(
-        self,
-        group_id: str,
-        *,
-        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        description: Optional[str] | Omit = omit,
-        manager_config: Optional[group_update_params.ManagerConfig] | Omit = omit,
-        project_id: Optional[str] | Omit = omit,
-        shared_block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        x_project: 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,
-    ) -> Group:
-        """
-        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
-
-          manager_config
-
-          project_id: The associated project id.
-
-          shared_block_ids
-
-          x_project: The project slug to associate with the group (cloud only).
-
-          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}")
-        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
-        return await self._patch(
-            f"/v1/groups/{group_id}",
-            body=await async_maybe_transform(
-                {
-                    "agent_ids": agent_ids,
-                    "description": description,
-                    "manager_config": manager_config,
-                    "project_id": project_id,
-                    "shared_block_ids": shared_block_ids,
-                },
-                group_update_params.GroupUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Group,
-        )
-
     def list(
         self,
         *,
@@ -656,6 +592,70 @@ class AsyncGroupsResource(AsyncAPIResource):
             cast_to=int,
         )
 
+    async def modify(
+        self,
+        group_id: str,
+        *,
+        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        description: Optional[str] | Omit = omit,
+        manager_config: Optional[group_modify_params.ManagerConfig] | Omit = omit,
+        project_id: Optional[str] | Omit = omit,
+        shared_block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        x_project: 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,
+    ) -> Group:
+        """
+        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
+
+          manager_config
+
+          project_id: The associated project id.
+
+          shared_block_ids
+
+          x_project: The project slug to associate with the group (cloud only).
+
+          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}")
+        extra_headers = {**strip_not_given({"X-Project": x_project}), **(extra_headers or {})}
+        return await self._patch(
+            f"/v1/groups/{group_id}",
+            body=await async_maybe_transform(
+                {
+                    "agent_ids": agent_ids,
+                    "description": description,
+                    "manager_config": manager_config,
+                    "project_id": project_id,
+                    "shared_block_ids": shared_block_ids,
+                },
+                group_modify_params.GroupModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Group,
+        )
+
 
 class GroupsResourceWithRawResponse:
     def __init__(self, groups: GroupsResource) -> None:
@@ -667,9 +667,6 @@ class GroupsResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             groups.retrieve,
         )
-        self.update = to_raw_response_wrapper(
-            groups.update,
-        )
         self.list = to_raw_response_wrapper(
             groups.list,
         )
@@ -679,6 +676,9 @@ class GroupsResourceWithRawResponse:
         self.count = to_raw_response_wrapper(
             groups.count,
         )
+        self.modify = to_raw_response_wrapper(
+            groups.modify,
+        )
 
     @cached_property
     def messages(self) -> MessagesResourceWithRawResponse:
@@ -695,9 +695,6 @@ class AsyncGroupsResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             groups.retrieve,
         )
-        self.update = async_to_raw_response_wrapper(
-            groups.update,
-        )
         self.list = async_to_raw_response_wrapper(
             groups.list,
         )
@@ -707,6 +704,9 @@ class AsyncGroupsResourceWithRawResponse:
         self.count = async_to_raw_response_wrapper(
             groups.count,
         )
+        self.modify = async_to_raw_response_wrapper(
+            groups.modify,
+        )
 
     @cached_property
     def messages(self) -> AsyncMessagesResourceWithRawResponse:
@@ -723,9 +723,6 @@ class GroupsResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             groups.retrieve,
         )
-        self.update = to_streamed_response_wrapper(
-            groups.update,
-        )
         self.list = to_streamed_response_wrapper(
             groups.list,
         )
@@ -735,6 +732,9 @@ class GroupsResourceWithStreamingResponse:
         self.count = to_streamed_response_wrapper(
             groups.count,
         )
+        self.modify = to_streamed_response_wrapper(
+            groups.modify,
+        )
 
     @cached_property
     def messages(self) -> MessagesResourceWithStreamingResponse:
@@ -751,9 +751,6 @@ class AsyncGroupsResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             groups.retrieve,
         )
-        self.update = async_to_streamed_response_wrapper(
-            groups.update,
-        )
         self.list = async_to_streamed_response_wrapper(
             groups.list,
         )
@@ -763,6 +760,9 @@ class AsyncGroupsResourceWithStreamingResponse:
         self.count = async_to_streamed_response_wrapper(
             groups.count,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            groups.modify,
+        )
 
     @cached_property
     def messages(self) -> AsyncMessagesResourceWithStreamingResponse: