codeset 0.1.0a1__py3-none-any.whl

codeset/resources/sessions/sessions.py

@@ -0,0 +1,618 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from .verify import (
+    VerifyResource,
+    AsyncVerifyResource,
+    VerifyResourceWithRawResponse,
+    AsyncVerifyResourceWithRawResponse,
+    VerifyResourceWithStreamingResponse,
+    AsyncVerifyResourceWithStreamingResponse,
+)
+from ...types import session_create_params, session_apply_diff_params, session_execute_command_params
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._utils import maybe_transform, async_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 ..._base_client import make_request_options
+from ...types.session import Session
+from ...types.session_list_response import SessionListResponse
+from ...types.session_close_response import SessionCloseResponse
+from ...types.session_create_response import SessionCreateResponse
+from ...types.session_apply_diff_response import SessionApplyDiffResponse
+from ...types.session_execute_command_response import SessionExecuteCommandResponse
+
+__all__ = ["SessionsResource", "AsyncSessionsResource"]
+
+
+class SessionsResource(SyncAPIResource):
+    @cached_property
+    def verify(self) -> VerifyResource:
+        return VerifyResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> SessionsResourceWithRawResponse:
+        """
+        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/codeset-ai/codeset-sdk#accessing-raw-response-data-eg-headers
+        """
+        return SessionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> SessionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/codeset-ai/codeset-sdk#with_streaming_response
+        """
+        return SessionsResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        sample_id: str,
+        ttl_minutes: int | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> SessionCreateResponse:
+        """
+        Create a new session
+
+        Args:
+          sample_id: Identifier of the sample to use for this session.
+
+          ttl_minutes: Time to live for the session in minutes (default: 30).
+
+          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(
+            "/sessions",
+            body=maybe_transform(
+                {
+                    "sample_id": sample_id,
+                    "ttl_minutes": ttl_minutes,
+                },
+                session_create_params.SessionCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionCreateResponse,
+        )
+
+    def retrieve(
+        self,
+        session_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,
+    ) -> Session:
+        """
+        Get session details
+
+        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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return self._get(
+            f"/sessions/{session_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Session,
+        )
+
+    def list(
+        self,
+        *,
+        # 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,
+    ) -> SessionListResponse:
+        """List environment sessions"""
+        return self._get(
+            "/sessions",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionListResponse,
+        )
+
+    def apply_diff(
+        self,
+        session_id: str,
+        *,
+        diff: 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,
+    ) -> SessionApplyDiffResponse:
+        """
+        Apply a diff in an environment
+
+        Args:
+          diff: The diff to be applied, in unified format.
+
+          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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return self._post(
+            f"/sessions/{session_id}/apply",
+            body=maybe_transform({"diff": diff}, session_apply_diff_params.SessionApplyDiffParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionApplyDiffResponse,
+        )
+
+    def close(
+        self,
+        session_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,
+    ) -> SessionCloseResponse:
+        """
+        Close/delete an environment session
+
+        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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return self._delete(
+            f"/sessions/{session_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionCloseResponse,
+        )
+
+    def execute_command(
+        self,
+        session_id: str,
+        *,
+        command: str,
+        command_timeout: int | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> SessionExecuteCommandResponse:
+        """
+        Execute a bash command in an environment
+
+        Args:
+          command: The bash command to execute.
+
+          command_timeout: Timeout for command execution in seconds (default: 300).
+
+          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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return self._post(
+            f"/sessions/{session_id}/exec",
+            body=maybe_transform(
+                {
+                    "command": command,
+                    "command_timeout": command_timeout,
+                },
+                session_execute_command_params.SessionExecuteCommandParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionExecuteCommandResponse,
+        )
+
+
+class AsyncSessionsResource(AsyncAPIResource):
+    @cached_property
+    def verify(self) -> AsyncVerifyResource:
+        return AsyncVerifyResource(self._client)
+
+    @cached_property
+    def with_raw_response(self) -> AsyncSessionsResourceWithRawResponse:
+        """
+        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/codeset-ai/codeset-sdk#accessing-raw-response-data-eg-headers
+        """
+        return AsyncSessionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncSessionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/codeset-ai/codeset-sdk#with_streaming_response
+        """
+        return AsyncSessionsResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        sample_id: str,
+        ttl_minutes: int | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> SessionCreateResponse:
+        """
+        Create a new session
+
+        Args:
+          sample_id: Identifier of the sample to use for this session.
+
+          ttl_minutes: Time to live for the session in minutes (default: 30).
+
+          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(
+            "/sessions",
+            body=await async_maybe_transform(
+                {
+                    "sample_id": sample_id,
+                    "ttl_minutes": ttl_minutes,
+                },
+                session_create_params.SessionCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionCreateResponse,
+        )
+
+    async def retrieve(
+        self,
+        session_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,
+    ) -> Session:
+        """
+        Get session details
+
+        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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return await self._get(
+            f"/sessions/{session_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Session,
+        )
+
+    async def list(
+        self,
+        *,
+        # 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,
+    ) -> SessionListResponse:
+        """List environment sessions"""
+        return await self._get(
+            "/sessions",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionListResponse,
+        )
+
+    async def apply_diff(
+        self,
+        session_id: str,
+        *,
+        diff: 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,
+    ) -> SessionApplyDiffResponse:
+        """
+        Apply a diff in an environment
+
+        Args:
+          diff: The diff to be applied, in unified format.
+
+          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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return await self._post(
+            f"/sessions/{session_id}/apply",
+            body=await async_maybe_transform({"diff": diff}, session_apply_diff_params.SessionApplyDiffParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionApplyDiffResponse,
+        )
+
+    async def close(
+        self,
+        session_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,
+    ) -> SessionCloseResponse:
+        """
+        Close/delete an environment session
+
+        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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return await self._delete(
+            f"/sessions/{session_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionCloseResponse,
+        )
+
+    async def execute_command(
+        self,
+        session_id: str,
+        *,
+        command: str,
+        command_timeout: int | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> SessionExecuteCommandResponse:
+        """
+        Execute a bash command in an environment
+
+        Args:
+          command: The bash command to execute.
+
+          command_timeout: Timeout for command execution in seconds (default: 300).
+
+          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 session_id:
+            raise ValueError(f"Expected a non-empty value for `session_id` but received {session_id!r}")
+        return await self._post(
+            f"/sessions/{session_id}/exec",
+            body=await async_maybe_transform(
+                {
+                    "command": command,
+                    "command_timeout": command_timeout,
+                },
+                session_execute_command_params.SessionExecuteCommandParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionExecuteCommandResponse,
+        )
+
+
+class SessionsResourceWithRawResponse:
+    def __init__(self, sessions: SessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = to_raw_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = to_raw_response_wrapper(
+            sessions.retrieve,
+        )
+        self.list = to_raw_response_wrapper(
+            sessions.list,
+        )
+        self.apply_diff = to_raw_response_wrapper(
+            sessions.apply_diff,
+        )
+        self.close = to_raw_response_wrapper(
+            sessions.close,
+        )
+        self.execute_command = to_raw_response_wrapper(
+            sessions.execute_command,
+        )
+
+    @cached_property
+    def verify(self) -> VerifyResourceWithRawResponse:
+        return VerifyResourceWithRawResponse(self._sessions.verify)
+
+
+class AsyncSessionsResourceWithRawResponse:
+    def __init__(self, sessions: AsyncSessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = async_to_raw_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = async_to_raw_response_wrapper(
+            sessions.retrieve,
+        )
+        self.list = async_to_raw_response_wrapper(
+            sessions.list,
+        )
+        self.apply_diff = async_to_raw_response_wrapper(
+            sessions.apply_diff,
+        )
+        self.close = async_to_raw_response_wrapper(
+            sessions.close,
+        )
+        self.execute_command = async_to_raw_response_wrapper(
+            sessions.execute_command,
+        )
+
+    @cached_property
+    def verify(self) -> AsyncVerifyResourceWithRawResponse:
+        return AsyncVerifyResourceWithRawResponse(self._sessions.verify)
+
+
+class SessionsResourceWithStreamingResponse:
+    def __init__(self, sessions: SessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = to_streamed_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = to_streamed_response_wrapper(
+            sessions.retrieve,
+        )
+        self.list = to_streamed_response_wrapper(
+            sessions.list,
+        )
+        self.apply_diff = to_streamed_response_wrapper(
+            sessions.apply_diff,
+        )
+        self.close = to_streamed_response_wrapper(
+            sessions.close,
+        )
+        self.execute_command = to_streamed_response_wrapper(
+            sessions.execute_command,
+        )
+
+    @cached_property
+    def verify(self) -> VerifyResourceWithStreamingResponse:
+        return VerifyResourceWithStreamingResponse(self._sessions.verify)
+
+
+class AsyncSessionsResourceWithStreamingResponse:
+    def __init__(self, sessions: AsyncSessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = async_to_streamed_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = async_to_streamed_response_wrapper(
+            sessions.retrieve,
+        )
+        self.list = async_to_streamed_response_wrapper(
+            sessions.list,
+        )
+        self.apply_diff = async_to_streamed_response_wrapper(
+            sessions.apply_diff,
+        )
+        self.close = async_to_streamed_response_wrapper(
+            sessions.close,
+        )
+        self.execute_command = async_to_streamed_response_wrapper(
+            sessions.execute_command,
+        )
+
+    @cached_property
+    def verify(self) -> AsyncVerifyResourceWithStreamingResponse:
+        return AsyncVerifyResourceWithStreamingResponse(self._sessions.verify)