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

letta_client/resources/identities/identities.py

@@ -27,7 +27,7 @@ from ...types import (
     IdentityType,
     identity_list_params,
     identity_create_params,
-    identity_update_params,
+    identity_modify_params,
     identity_upsert_params,
 )
 from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
@@ -48,10 +48,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncArrayPage, AsyncArrayPage
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.identity import Identity
 from ...types.identity_type import IdentityType
+from ...types.identity_list_response import IdentityListResponse
 from ...types.identity_count_response import IdentityCountResponse
 from ...types.identity_property_param import IdentityPropertyParam
 
@@ -192,70 +192,6 @@ class IdentitiesResource(SyncAPIResource):
             cast_to=Identity,
         )
 
-    def update(
-        self,
-        identity_id: str,
-        *,
-        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        identifier_key: Optional[str] | Omit = omit,
-        identity_type: Optional[IdentityType] | Omit = omit,
-        name: Optional[str] | Omit = omit,
-        properties: Optional[Iterable[IdentityPropertyParam]] | 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,
-    ) -> Identity:
-        """
-        Modify Identity
-
-        Args:
-          identity_id: The ID of the identity in the format 'identity-<uuid4>'
-
-          agent_ids: The agent ids that are associated with the identity.
-
-          block_ids: The IDs of the blocks associated with the identity.
-
-          identifier_key: External, user-generated identifier key of the identity.
-
-          identity_type: Enum to represent the type of the identity.
-
-          name: The name of the identity.
-
-          properties: List of properties associated with the identity.
-
-          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 identity_id:
-            raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
-        return self._patch(
-            f"/v1/identities/{identity_id}",
-            body=maybe_transform(
-                {
-                    "agent_ids": agent_ids,
-                    "block_ids": block_ids,
-                    "identifier_key": identifier_key,
-                    "identity_type": identity_type,
-                    "name": name,
-                    "properties": properties,
-                },
-                identity_update_params.IdentityUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Identity,
-        )
-
     def list(
         self,
         *,
@@ -274,7 +210,7 @@ class IdentitiesResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Identity]:
+    ) -> IdentityListResponse:
         """
         Get a list of all identities in the database
 
@@ -302,9 +238,8 @@ class IdentitiesResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/identities/",
-            page=SyncArrayPage[Identity],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -325,7 +260,7 @@ class IdentitiesResource(SyncAPIResource):
                     identity_list_params.IdentityListParams,
                 ),
             ),
-            model=Identity,
+            cast_to=IdentityListResponse,
         )
 
     def delete(
@@ -382,6 +317,70 @@ class IdentitiesResource(SyncAPIResource):
             cast_to=int,
         )
 
+    def modify(
+        self,
+        identity_id: str,
+        *,
+        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        identifier_key: Optional[str] | Omit = omit,
+        identity_type: Optional[IdentityType] | Omit = omit,
+        name: Optional[str] | Omit = omit,
+        properties: Optional[Iterable[IdentityPropertyParam]] | 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,
+    ) -> Identity:
+        """
+        Modify Identity
+
+        Args:
+          identity_id: The ID of the identity in the format 'identity-<uuid4>'
+
+          agent_ids: The agent ids that are associated with the identity.
+
+          block_ids: The IDs of the blocks associated with the identity.
+
+          identifier_key: External, user-generated identifier key of the identity.
+
+          identity_type: Enum to represent the type of the identity.
+
+          name: The name of the identity.
+
+          properties: List of properties associated with the identity.
+
+          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 identity_id:
+            raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
+        return self._patch(
+            f"/v1/identities/{identity_id}",
+            body=maybe_transform(
+                {
+                    "agent_ids": agent_ids,
+                    "block_ids": block_ids,
+                    "identifier_key": identifier_key,
+                    "identity_type": identity_type,
+                    "name": name,
+                    "properties": properties,
+                },
+                identity_modify_params.IdentityModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Identity,
+        )
+
     def upsert(
         self,
         *,
@@ -584,71 +583,7 @@ class AsyncIdentitiesResource(AsyncAPIResource):
             cast_to=Identity,
         )
 
-    async def update(
-        self,
-        identity_id: str,
-        *,
-        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
-        identifier_key: Optional[str] | Omit = omit,
-        identity_type: Optional[IdentityType] | Omit = omit,
-        name: Optional[str] | Omit = omit,
-        properties: Optional[Iterable[IdentityPropertyParam]] | 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,
-    ) -> Identity:
-        """
-        Modify Identity
-
-        Args:
-          identity_id: The ID of the identity in the format 'identity-<uuid4>'
-
-          agent_ids: The agent ids that are associated with the identity.
-
-          block_ids: The IDs of the blocks associated with the identity.
-
-          identifier_key: External, user-generated identifier key of the identity.
-
-          identity_type: Enum to represent the type of the identity.
-
-          name: The name of the identity.
-
-          properties: List of properties associated with the identity.
-
-          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 identity_id:
-            raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
-        return await self._patch(
-            f"/v1/identities/{identity_id}",
-            body=await async_maybe_transform(
-                {
-                    "agent_ids": agent_ids,
-                    "block_ids": block_ids,
-                    "identifier_key": identifier_key,
-                    "identity_type": identity_type,
-                    "name": name,
-                    "properties": properties,
-                },
-                identity_update_params.IdentityUpdateParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=Identity,
-        )
-
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -666,7 +601,7 @@ class AsyncIdentitiesResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Identity, AsyncArrayPage[Identity]]:
+    ) -> IdentityListResponse:
         """
         Get a list of all identities in the database
 
@@ -694,15 +629,14 @@ class AsyncIdentitiesResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/identities/",
-            page=AsyncArrayPage[Identity],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -717,7 +651,7 @@ class AsyncIdentitiesResource(AsyncAPIResource):
                     identity_list_params.IdentityListParams,
                 ),
             ),
-            model=Identity,
+            cast_to=IdentityListResponse,
         )
 
     async def delete(
@@ -774,6 +708,70 @@ class AsyncIdentitiesResource(AsyncAPIResource):
             cast_to=int,
         )
 
+    async def modify(
+        self,
+        identity_id: str,
+        *,
+        agent_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        block_ids: Optional[SequenceNotStr[str]] | Omit = omit,
+        identifier_key: Optional[str] | Omit = omit,
+        identity_type: Optional[IdentityType] | Omit = omit,
+        name: Optional[str] | Omit = omit,
+        properties: Optional[Iterable[IdentityPropertyParam]] | 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,
+    ) -> Identity:
+        """
+        Modify Identity
+
+        Args:
+          identity_id: The ID of the identity in the format 'identity-<uuid4>'
+
+          agent_ids: The agent ids that are associated with the identity.
+
+          block_ids: The IDs of the blocks associated with the identity.
+
+          identifier_key: External, user-generated identifier key of the identity.
+
+          identity_type: Enum to represent the type of the identity.
+
+          name: The name of the identity.
+
+          properties: List of properties associated with the identity.
+
+          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 identity_id:
+            raise ValueError(f"Expected a non-empty value for `identity_id` but received {identity_id!r}")
+        return await self._patch(
+            f"/v1/identities/{identity_id}",
+            body=await async_maybe_transform(
+                {
+                    "agent_ids": agent_ids,
+                    "block_ids": block_ids,
+                    "identifier_key": identifier_key,
+                    "identity_type": identity_type,
+                    "name": name,
+                    "properties": properties,
+                },
+                identity_modify_params.IdentityModifyParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Identity,
+        )
+
     async def upsert(
         self,
         *,
@@ -852,9 +850,6 @@ class IdentitiesResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             identities.retrieve,
         )
-        self.update = to_raw_response_wrapper(
-            identities.update,
-        )
         self.list = to_raw_response_wrapper(
             identities.list,
         )
@@ -864,6 +859,9 @@ class IdentitiesResourceWithRawResponse:
         self.count = to_raw_response_wrapper(
             identities.count,
         )
+        self.modify = to_raw_response_wrapper(
+            identities.modify,
+        )
         self.upsert = to_raw_response_wrapper(
             identities.upsert,
         )
@@ -891,9 +889,6 @@ class AsyncIdentitiesResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             identities.retrieve,
         )
-        self.update = async_to_raw_response_wrapper(
-            identities.update,
-        )
         self.list = async_to_raw_response_wrapper(
             identities.list,
         )
@@ -903,6 +898,9 @@ class AsyncIdentitiesResourceWithRawResponse:
         self.count = async_to_raw_response_wrapper(
             identities.count,
         )
+        self.modify = async_to_raw_response_wrapper(
+            identities.modify,
+        )
         self.upsert = async_to_raw_response_wrapper(
             identities.upsert,
         )
@@ -930,9 +928,6 @@ class IdentitiesResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             identities.retrieve,
         )
-        self.update = to_streamed_response_wrapper(
-            identities.update,
-        )
         self.list = to_streamed_response_wrapper(
             identities.list,
         )
@@ -942,6 +937,9 @@ class IdentitiesResourceWithStreamingResponse:
         self.count = to_streamed_response_wrapper(
             identities.count,
         )
+        self.modify = to_streamed_response_wrapper(
+            identities.modify,
+        )
         self.upsert = to_streamed_response_wrapper(
             identities.upsert,
         )
@@ -969,9 +967,6 @@ class AsyncIdentitiesResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             identities.retrieve,
         )
-        self.update = async_to_streamed_response_wrapper(
-            identities.update,
-        )
         self.list = async_to_streamed_response_wrapper(
             identities.list,
         )
@@ -981,6 +976,9 @@ class AsyncIdentitiesResourceWithStreamingResponse:
         self.count = async_to_streamed_response_wrapper(
             identities.count,
         )
+        self.modify = async_to_streamed_response_wrapper(
+            identities.modify,
+        )
         self.upsert = async_to_streamed_response_wrapper(
             identities.upsert,
         )

letta_client/resources/runs/runs.py

@@ -25,7 +25,7 @@ from .usage import (
 )
 from ...types import StopReasonType, run_list_params
 from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
-from ..._utils import maybe_transform
+from ..._utils import maybe_transform, async_maybe_transform
 from .messages import (
     MessagesResource,
     AsyncMessagesResource,
@@ -42,10 +42,10 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...pagination import SyncArrayPage, AsyncArrayPage
-from ..._base_client import AsyncPaginator, make_request_options
+from ..._base_client import make_request_options
 from ...types.agents.run import Run
 from ...types.stop_reason_type import StopReasonType
+from ...types.run_list_response import RunListResponse
 
 __all__ = ["RunsResource", "AsyncRunsResource"]
 
@@ -136,7 +136,7 @@ class RunsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[Run]:
+    ) -> RunListResponse:
         """
         List all runs.
 
@@ -178,9 +178,8 @@ class RunsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/runs/",
-            page=SyncArrayPage[Run],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -204,7 +203,7 @@ class RunsResource(SyncAPIResource):
                     run_list_params.RunListParams,
                 ),
             ),
-            model=Run,
+            cast_to=RunListResponse,
         )
 
 
@@ -273,7 +272,7 @@ class AsyncRunsResource(AsyncAPIResource):
             cast_to=Run,
         )
 
-    def list(
+    async def list(
         self,
         *,
         active: bool | Omit = omit,
@@ -294,7 +293,7 @@ class AsyncRunsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[Run, AsyncArrayPage[Run]]:
+    ) -> RunListResponse:
         """
         List all runs.
 
@@ -336,15 +335,14 @@ class AsyncRunsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/runs/",
-            page=AsyncArrayPage[Run],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "active": active,
                         "after": after,
@@ -362,7 +360,7 @@ class AsyncRunsResource(AsyncAPIResource):
                     run_list_params.RunListParams,
                 ),
             ),
-            model=Run,
+            cast_to=RunListResponse,
         )
 
 

letta_client/resources/tags.py

@@ -9,7 +9,7 @@ import httpx
 
 from ..types import tag_list_params
 from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from .._utils import maybe_transform
+from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -18,8 +18,7 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..pagination import SyncArrayPage, AsyncArrayPage
-from .._base_client import AsyncPaginator, make_request_options
+from .._base_client import make_request_options
 from ..types.tag_list_response import TagListResponse
 
 __all__ = ["TagsResource", "AsyncTagsResource"]
@@ -61,7 +60,7 @@ class TagsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> SyncArrayPage[TagListResponse]:
+    ) -> TagListResponse:
         """
         Get the list of all agent tags that have been created.
 
@@ -91,9 +90,8 @@ class TagsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._get(
             "/v1/tags/",
-            page=SyncArrayPage[TagListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -112,7 +110,7 @@ class TagsResource(SyncAPIResource):
                     tag_list_params.TagListParams,
                 ),
             ),
-            model=str,
+            cast_to=TagListResponse,
         )
 
 
@@ -136,7 +134,7 @@ class AsyncTagsResource(AsyncAPIResource):
         """
         return AsyncTagsResourceWithStreamingResponse(self)
 
-    def list(
+    async def list(
         self,
         *,
         after: Optional[str] | Omit = omit,
@@ -152,7 +150,7 @@ class AsyncTagsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncPaginator[TagListResponse, AsyncArrayPage[TagListResponse]]:
+    ) -> TagListResponse:
         """
         Get the list of all agent tags that have been created.
 
@@ -182,15 +180,14 @@ class AsyncTagsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._get(
             "/v1/tags/",
-            page=AsyncArrayPage[TagListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
+                query=await async_maybe_transform(
                     {
                         "after": after,
                         "before": before,
@@ -203,7 +200,7 @@ class AsyncTagsResource(AsyncAPIResource):
                     tag_list_params.TagListParams,
                 ),
             ),
-            model=str,
+            cast_to=TagListResponse,
         )