structifyai 1.158.0__py3-none-any.whl → 1.176.1__py3-none-any.whl

structify/resources/admin/users.py

@@ -18,20 +18,12 @@ from ..._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...types.admin import (
-    user_create_params,
-    user_get_stats_params,
-    user_add_credits_params,
-    user_get_credits_params,
-    user_set_credits_params,
-)
+from ...types.admin import user_create_params, user_get_stats_params, user_impersonate_params
 from ..._base_client import make_request_options
 from ...types.token_response import TokenResponse
 from ...types.admin.user_list_response import UserListResponse
+from ...types.admin.impersonate_response import ImpersonateResponse
 from ...types.admin.user_get_stats_response import UserGetStatsResponse
-from ...types.admin.user_add_credits_response import UserAddCreditsResponse
-from ...types.admin.user_get_credits_response import UserGetCreditsResponse
-from ...types.admin.user_set_credits_response import UserSetCreditsResponse
 
 __all__ = ["UsersResource", "AsyncUsersResource"]
 
@@ -72,6 +64,7 @@ class UsersResource(SyncAPIResource):
                 "cerebras_codegen",
                 "gemini25pro",
                 "claude_sonnet4",
+                "allow_job_deletion",
                 "none",
             ]
         ]
@@ -88,7 +81,7 @@ class UsersResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TokenResponse:
         """
-        Create a user, returing their API token.
+        Create a user, returning their session token.
 
         Args:
           extra_headers: Send extra headers
@@ -129,7 +122,7 @@ class UsersResource(SyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> UserListResponse:
-        """Lists all the users in the system along with their associated API tokens."""
+        """Lists all users with their team memberships."""
         return self._get(
             "/admin/users/list",
             options=make_request_options(
@@ -138,88 +131,6 @@ class UsersResource(SyncAPIResource):
             cast_to=UserListResponse,
         )
 
-    def add_credits(
-        self,
-        *,
-        credit_amount: int,
-        user_email: str,
-        source_type: Optional[str] | Omit = omit,
-        valid_for_days: Optional[int] | 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,
-    ) -> UserAddCreditsResponse:
-        """
-        Add credits to a user's account via a credit grant.
-
-        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
-        """
-        return self._post(
-            "/admin/users/add_credits",
-            body=maybe_transform(
-                {
-                    "credit_amount": credit_amount,
-                    "user_email": user_email,
-                    "source_type": source_type,
-                    "valid_for_days": valid_for_days,
-                },
-                user_add_credits_params.UserAddCreditsParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=UserAddCreditsResponse,
-        )
-
-    def get_credits(
-        self,
-        *,
-        user_email: Optional[str] | Omit = omit,
-        user_token: Optional[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,
-    ) -> UserGetCreditsResponse:
-        """
-        get the credit balance of a user by email.
-
-        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
-        """
-        return self._post(
-            "/admin/users/get_credits",
-            body=maybe_transform(
-                {
-                    "user_email": user_email,
-                    "user_token": user_token,
-                },
-                user_get_credits_params.UserGetCreditsParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=UserGetCreditsResponse,
-        )
-
     def get_stats(
         self,
         *,
@@ -263,21 +174,18 @@ class UsersResource(SyncAPIResource):
             cast_to=UserGetStatsResponse,
         )
 
-    def set_credits(
+    def impersonate(
         self,
         *,
-        credit_count: int,
-        user_email: str,
+        membership_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,
-    ) -> UserSetCreditsResponse:
+    ) -> ImpersonateResponse:
         """
-        set the credit balance of a user, returing that new credit balance.
-
         Args:
           extra_headers: Send extra headers
 
@@ -288,18 +196,12 @@ class UsersResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._post(
-            "/admin/users/set_credits",
-            body=maybe_transform(
-                {
-                    "credit_count": credit_count,
-                    "user_email": user_email,
-                },
-                user_set_credits_params.UserSetCreditsParams,
-            ),
+            "/admin/users/impersonate",
+            body=maybe_transform({"membership_id": membership_id}, user_impersonate_params.UserImpersonateParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=UserSetCreditsResponse,
+            cast_to=ImpersonateResponse,
         )
 
 
@@ -339,6 +241,7 @@ class AsyncUsersResource(AsyncAPIResource):
                 "cerebras_codegen",
                 "gemini25pro",
                 "claude_sonnet4",
+                "allow_job_deletion",
                 "none",
             ]
         ]
@@ -355,7 +258,7 @@ class AsyncUsersResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TokenResponse:
         """
-        Create a user, returing their API token.
+        Create a user, returning their session token.
 
         Args:
           extra_headers: Send extra headers
@@ -396,7 +299,7 @@ class AsyncUsersResource(AsyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> UserListResponse:
-        """Lists all the users in the system along with their associated API tokens."""
+        """Lists all users with their team memberships."""
         return await self._get(
             "/admin/users/list",
             options=make_request_options(
@@ -405,88 +308,6 @@ class AsyncUsersResource(AsyncAPIResource):
             cast_to=UserListResponse,
         )
 
-    async def add_credits(
-        self,
-        *,
-        credit_amount: int,
-        user_email: str,
-        source_type: Optional[str] | Omit = omit,
-        valid_for_days: Optional[int] | 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,
-    ) -> UserAddCreditsResponse:
-        """
-        Add credits to a user's account via a credit grant.
-
-        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
-        """
-        return await self._post(
-            "/admin/users/add_credits",
-            body=await async_maybe_transform(
-                {
-                    "credit_amount": credit_amount,
-                    "user_email": user_email,
-                    "source_type": source_type,
-                    "valid_for_days": valid_for_days,
-                },
-                user_add_credits_params.UserAddCreditsParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=UserAddCreditsResponse,
-        )
-
-    async def get_credits(
-        self,
-        *,
-        user_email: Optional[str] | Omit = omit,
-        user_token: Optional[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,
-    ) -> UserGetCreditsResponse:
-        """
-        get the credit balance of a user by email.
-
-        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
-        """
-        return await self._post(
-            "/admin/users/get_credits",
-            body=await async_maybe_transform(
-                {
-                    "user_email": user_email,
-                    "user_token": user_token,
-                },
-                user_get_credits_params.UserGetCreditsParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=UserGetCreditsResponse,
-        )
-
     async def get_stats(
         self,
         *,
@@ -530,21 +351,18 @@ class AsyncUsersResource(AsyncAPIResource):
             cast_to=UserGetStatsResponse,
         )
 
-    async def set_credits(
+    async def impersonate(
         self,
         *,
-        credit_count: int,
-        user_email: str,
+        membership_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,
-    ) -> UserSetCreditsResponse:
+    ) -> ImpersonateResponse:
         """
-        set the credit balance of a user, returing that new credit balance.
-
         Args:
           extra_headers: Send extra headers
 
@@ -555,18 +373,14 @@ class AsyncUsersResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._post(
-            "/admin/users/set_credits",
+            "/admin/users/impersonate",
             body=await async_maybe_transform(
-                {
-                    "credit_count": credit_count,
-                    "user_email": user_email,
-                },
-                user_set_credits_params.UserSetCreditsParams,
+                {"membership_id": membership_id}, user_impersonate_params.UserImpersonateParams
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=UserSetCreditsResponse,
+            cast_to=ImpersonateResponse,
         )
 
 
@@ -580,17 +394,11 @@ class UsersResourceWithRawResponse:
         self.list = to_raw_response_wrapper(
             users.list,
         )
-        self.add_credits = to_raw_response_wrapper(
-            users.add_credits,
-        )
-        self.get_credits = to_raw_response_wrapper(
-            users.get_credits,
-        )
         self.get_stats = to_raw_response_wrapper(
             users.get_stats,
         )
-        self.set_credits = to_raw_response_wrapper(
-            users.set_credits,
+        self.impersonate = to_raw_response_wrapper(
+            users.impersonate,
         )
 
 
@@ -604,17 +412,11 @@ class AsyncUsersResourceWithRawResponse:
         self.list = async_to_raw_response_wrapper(
             users.list,
         )
-        self.add_credits = async_to_raw_response_wrapper(
-            users.add_credits,
-        )
-        self.get_credits = async_to_raw_response_wrapper(
-            users.get_credits,
-        )
         self.get_stats = async_to_raw_response_wrapper(
             users.get_stats,
         )
-        self.set_credits = async_to_raw_response_wrapper(
-            users.set_credits,
+        self.impersonate = async_to_raw_response_wrapper(
+            users.impersonate,
         )
 
 
@@ -628,17 +430,11 @@ class UsersResourceWithStreamingResponse:
         self.list = to_streamed_response_wrapper(
             users.list,
         )
-        self.add_credits = to_streamed_response_wrapper(
-            users.add_credits,
-        )
-        self.get_credits = to_streamed_response_wrapper(
-            users.get_credits,
-        )
         self.get_stats = to_streamed_response_wrapper(
             users.get_stats,
         )
-        self.set_credits = to_streamed_response_wrapper(
-            users.set_credits,
+        self.impersonate = to_streamed_response_wrapper(
+            users.impersonate,
         )
 
 
@@ -652,15 +448,9 @@ class AsyncUsersResourceWithStreamingResponse:
         self.list = async_to_streamed_response_wrapper(
             users.list,
         )
-        self.add_credits = async_to_streamed_response_wrapper(
-            users.add_credits,
-        )
-        self.get_credits = async_to_streamed_response_wrapper(
-            users.get_credits,
-        )
         self.get_stats = async_to_streamed_response_wrapper(
             users.get_stats,
         )
-        self.set_credits = async_to_streamed_response_wrapper(
-            users.set_credits,
+        self.impersonate = async_to_streamed_response_wrapper(
+            users.impersonate,
         )