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

structify/resources/admin/jobs.py

@@ -8,8 +8,8 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from ..._utils import maybe_transform
+from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
+from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -19,9 +19,10 @@ from ..._response import (
     async_to_streamed_response_wrapper,
 )
 from ...pagination import SyncJobsList, AsyncJobsList
-from ...types.admin import job_list_params
+from ...types.admin import job_list_params, job_delete_params
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.admin.admin_list_jobs_response import AdminListJobsResponse
+from ...types.admin.admin_delete_jobs_response import AdminDeleteJobsResponse
 
 __all__ = ["JobsResource", "AsyncJobsResource"]
 
@@ -116,6 +117,36 @@ class JobsResource(SyncAPIResource):
             model=AdminListJobsResponse,
         )
 
+    def delete(
+        self,
+        *,
+        job_ids: SequenceNotStr[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,
+    ) -> AdminDeleteJobsResponse:
+        """
+        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/jobs/delete",
+            body=maybe_transform({"job_ids": job_ids}, job_delete_params.JobDeleteParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AdminDeleteJobsResponse,
+        )
+
 
 class AsyncJobsResource(AsyncAPIResource):
     @cached_property
@@ -207,6 +238,36 @@ class AsyncJobsResource(AsyncAPIResource):
             model=AdminListJobsResponse,
         )
 
+    async def delete(
+        self,
+        *,
+        job_ids: SequenceNotStr[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,
+    ) -> AdminDeleteJobsResponse:
+        """
+        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/jobs/delete",
+            body=await async_maybe_transform({"job_ids": job_ids}, job_delete_params.JobDeleteParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=AdminDeleteJobsResponse,
+        )
+
 
 class JobsResourceWithRawResponse:
     def __init__(self, jobs: JobsResource) -> None:
@@ -215,6 +276,9 @@ class JobsResourceWithRawResponse:
         self.list = to_raw_response_wrapper(
             jobs.list,
         )
+        self.delete = to_raw_response_wrapper(
+            jobs.delete,
+        )
 
 
 class AsyncJobsResourceWithRawResponse:
@@ -224,6 +288,9 @@ class AsyncJobsResourceWithRawResponse:
         self.list = async_to_raw_response_wrapper(
             jobs.list,
         )
+        self.delete = async_to_raw_response_wrapper(
+            jobs.delete,
+        )
 
 
 class JobsResourceWithStreamingResponse:
@@ -233,6 +300,9 @@ class JobsResourceWithStreamingResponse:
         self.list = to_streamed_response_wrapper(
             jobs.list,
         )
+        self.delete = to_streamed_response_wrapper(
+            jobs.delete,
+        )
 
 
 class AsyncJobsResourceWithStreamingResponse:
@@ -242,3 +312,6 @@ class AsyncJobsResourceWithStreamingResponse:
         self.list = async_to_streamed_response_wrapper(
             jobs.list,
         )
+        self.delete = async_to_streamed_response_wrapper(
+            jobs.delete,
+        )

structify/resources/admin/sandbox.py

@@ -0,0 +1,184 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+
+import httpx
+
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ..._utils import maybe_transform
+from ..._compat import cached_property
+from ..._resource import SyncAPIResource, AsyncAPIResource
+from ..._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ...pagination import SyncJobsList, AsyncJobsList
+from ...types.admin import sandbox_list_params
+from ..._base_client import AsyncPaginator, make_request_options
+from ...types.admin.admin_sandbox import AdminSandbox
+
+__all__ = ["SandboxResource", "AsyncSandboxResource"]
+
+
+class SandboxResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> SandboxResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/StructifyAI/structify-python#accessing-raw-response-data-eg-headers
+        """
+        return SandboxResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> SandboxResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/StructifyAI/structify-python#with_streaming_response
+        """
+        return SandboxResourceWithStreamingResponse(self)
+
+    def list(
+        self,
+        *,
+        limit: Optional[int] | Omit = omit,
+        offset: 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,
+    ) -> SyncJobsList[AdminSandbox]:
+        """
+        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._get_api_list(
+            "/admin/sandbox/list",
+            page=SyncJobsList[AdminSandbox],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "limit": limit,
+                        "offset": offset,
+                    },
+                    sandbox_list_params.SandboxListParams,
+                ),
+            ),
+            model=AdminSandbox,
+        )
+
+
+class AsyncSandboxResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncSandboxResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/StructifyAI/structify-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncSandboxResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncSandboxResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/StructifyAI/structify-python#with_streaming_response
+        """
+        return AsyncSandboxResourceWithStreamingResponse(self)
+
+    def list(
+        self,
+        *,
+        limit: Optional[int] | Omit = omit,
+        offset: 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,
+    ) -> AsyncPaginator[AdminSandbox, AsyncJobsList[AdminSandbox]]:
+        """
+        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._get_api_list(
+            "/admin/sandbox/list",
+            page=AsyncJobsList[AdminSandbox],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "limit": limit,
+                        "offset": offset,
+                    },
+                    sandbox_list_params.SandboxListParams,
+                ),
+            ),
+            model=AdminSandbox,
+        )
+
+
+class SandboxResourceWithRawResponse:
+    def __init__(self, sandbox: SandboxResource) -> None:
+        self._sandbox = sandbox
+
+        self.list = to_raw_response_wrapper(
+            sandbox.list,
+        )
+
+
+class AsyncSandboxResourceWithRawResponse:
+    def __init__(self, sandbox: AsyncSandboxResource) -> None:
+        self._sandbox = sandbox
+
+        self.list = async_to_raw_response_wrapper(
+            sandbox.list,
+        )
+
+
+class SandboxResourceWithStreamingResponse:
+    def __init__(self, sandbox: SandboxResource) -> None:
+        self._sandbox = sandbox
+
+        self.list = to_streamed_response_wrapper(
+            sandbox.list,
+        )
+
+
+class AsyncSandboxResourceWithStreamingResponse:
+    def __init__(self, sandbox: AsyncSandboxResource) -> None:
+        self._sandbox = sandbox
+
+        self.list = async_to_streamed_response_wrapper(
+            sandbox.list,
+        )

structify/resources/admin/teams.py

@@ -22,12 +22,14 @@ from ...pagination import SyncJobsList, AsyncJobsList
 from ...types.admin import (
     team_list_params,
     team_extend_trial_params,
+    team_expire_grants_params,
     team_grant_credits_params,
     team_cancel_subscription_params,
     team_create_subscription_params,
 )
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.admin.extend_trial_response import ExtendTrialResponse
+from ...types.admin.expire_grants_response import ExpireGrantsResponse
 from ...types.admin.grant_credits_response import GrantCreditsResponse
 from ...types.admin.admin_teams_list_response import AdminTeamsListResponse
 from ...types.admin.cancel_subscription_response import CancelSubscriptionResponse
@@ -180,6 +182,43 @@ class TeamsResource(SyncAPIResource):
             cast_to=CreateSubscriptionResponse,
         )
 
+    def expire_grants(
+        self,
+        *,
+        source_type: str,
+        team_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,
+    ) -> ExpireGrantsResponse:
+        """
+        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/team/expire_grants",
+            body=maybe_transform(
+                {
+                    "source_type": source_type,
+                    "team_id": team_id,
+                },
+                team_expire_grants_params.TeamExpireGrantsParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ExpireGrantsResponse,
+        )
+
     def extend_trial(
         self,
         *,
@@ -409,6 +448,43 @@ class AsyncTeamsResource(AsyncAPIResource):
             cast_to=CreateSubscriptionResponse,
         )
 
+    async def expire_grants(
+        self,
+        *,
+        source_type: str,
+        team_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,
+    ) -> ExpireGrantsResponse:
+        """
+        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/team/expire_grants",
+            body=await async_maybe_transform(
+                {
+                    "source_type": source_type,
+                    "team_id": team_id,
+                },
+                team_expire_grants_params.TeamExpireGrantsParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ExpireGrantsResponse,
+        )
+
     async def extend_trial(
         self,
         *,
@@ -505,6 +581,9 @@ class TeamsResourceWithRawResponse:
         self.create_subscription = to_raw_response_wrapper(
             teams.create_subscription,
         )
+        self.expire_grants = to_raw_response_wrapper(
+            teams.expire_grants,
+        )
         self.extend_trial = to_raw_response_wrapper(
             teams.extend_trial,
         )
@@ -526,6 +605,9 @@ class AsyncTeamsResourceWithRawResponse:
         self.create_subscription = async_to_raw_response_wrapper(
             teams.create_subscription,
         )
+        self.expire_grants = async_to_raw_response_wrapper(
+            teams.expire_grants,
+        )
         self.extend_trial = async_to_raw_response_wrapper(
             teams.extend_trial,
         )
@@ -547,6 +629,9 @@ class TeamsResourceWithStreamingResponse:
         self.create_subscription = to_streamed_response_wrapper(
             teams.create_subscription,
         )
+        self.expire_grants = to_streamed_response_wrapper(
+            teams.expire_grants,
+        )
         self.extend_trial = to_streamed_response_wrapper(
             teams.extend_trial,
         )
@@ -568,6 +653,9 @@ class AsyncTeamsResourceWithStreamingResponse:
         self.create_subscription = async_to_streamed_response_wrapper(
             teams.create_subscription,
         )
+        self.expire_grants = async_to_streamed_response_wrapper(
+            teams.expire_grants,
+        )
         self.extend_trial = async_to_streamed_response_wrapper(
             teams.extend_trial,
         )