feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 20
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/nen-labs%2Fsteel-c30148e5d3a9b19312ab7c9860fac5dc5c4d773b1032fa8ce345e6386c04c945.yml
-openapi_spec_hash: ca45bef50c34ba9f4affa7549b2b915a
-config_hash: eceac094e0c141de74f2ed1fa1a9857c
+configured_endpoints: 21
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/nen-labs%2Fsteel-7fca3009423f0f22ea3f05e1e7d7639ee4f011c4e82f76b1db0eeede2c7aaf89.yml
+openapi_spec_hash: d5bbd2360a0eb0c87b640953e3c81453
+config_hash: afec2f07e489e6d6c8ac5ab70d52bf42

api.md

@@ -44,6 +44,7 @@ from steel.types import (
     SessionEventsResponse,
     SessionLiveDetailsResponse,
     SessionReleaseResponse,
+    SessionReleaseAllResponse,
 )
 ```
 
@@ -56,6 +57,7 @@ Methods:
 - <code title="get /v1/sessions/{id}/events">client.sessions.<a href="./src/steel/resources/sessions/sessions.py">events</a>(id) -> <a href="./src/steel/types/session_events_response.py">SessionEventsResponse</a></code>
 - <code title="get /v1/sessions/{id}/live-details">client.sessions.<a href="./src/steel/resources/sessions/sessions.py">live_details</a>(id) -> <a href="./src/steel/types/session_live_details_response.py">SessionLiveDetailsResponse</a></code>
 - <code title="post /v1/sessions/{id}/release">client.sessions.<a href="./src/steel/resources/sessions/sessions.py">release</a>(id) -> <a href="./src/steel/types/session_release_response.py">SessionReleaseResponse</a></code>
+- <code title="post /v1/sessions/release">client.sessions.<a href="./src/steel/resources/sessions/sessions.py">release_all</a>() -> <a href="./src/steel/types/session_release_all_response.py">SessionReleaseAllResponse</a></code>
 
 ## Files
 

src/steel/resources/sessions/sessions.py

@@ -32,6 +32,7 @@
 from ...types.session_context import SessionContext
 from ...types.session_events_response import SessionEventsResponse
 from ...types.session_release_response import SessionReleaseResponse
+from ...types.session_release_all_response import SessionReleaseAllResponse
 from ...types.session_live_details_response import SessionLiveDetailsResponse
 
 __all__ = ["SessionsResource", "AsyncSessionsResource"]
@@ -71,6 +72,7 @@ def create(
         is_selenium: bool | NotGiven = NOT_GIVEN,
         namespace: str | NotGiven = NOT_GIVEN,
         proxy_url: str | NotGiven = NOT_GIVEN,
+        region: Literal["lax", "ord", "iad", "bom", "scl", "fra", "hkg"] | NotGiven = NOT_GIVEN,
         session_context: session_create_params.SessionContext | NotGiven = NOT_GIVEN,
         session_id: str | NotGiven = NOT_GIVEN,
         solve_captcha: bool | NotGiven = NOT_GIVEN,
@@ -105,6 +107,8 @@ def create(
               Steel-provided proxies in favor of your specified proxy. Format:
               http(s)://username:password@hostname:port
 
+          region: The desired region for the session to be started in
+
           session_context: Session context data to be used in the created session. Sessions will start with
               an empty context by default.
 
@@ -138,6 +142,7 @@ def create(
                     "is_selenium": is_selenium,
                     "namespace": namespace,
                     "proxy_url": proxy_url,
+                    "region": region,
                     "session_context": session_context,
                     "session_id": session_id,
                     "solve_captcha": solve_captcha,
@@ -369,6 +374,25 @@ def release(
             cast_to=SessionReleaseResponse,
         )
 
+    def release_all(
+        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,
+    ) -> SessionReleaseAllResponse:
+        """Releases all active sessions for the current organization."""
+        return self._post(
+            "/v1/sessions/release",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionReleaseAllResponse,
+        )
+
 
 class AsyncSessionsResource(AsyncAPIResource):
     @cached_property
@@ -404,6 +428,7 @@ async def create(
         is_selenium: bool | NotGiven = NOT_GIVEN,
         namespace: str | NotGiven = NOT_GIVEN,
         proxy_url: str | NotGiven = NOT_GIVEN,
+        region: Literal["lax", "ord", "iad", "bom", "scl", "fra", "hkg"] | NotGiven = NOT_GIVEN,
         session_context: session_create_params.SessionContext | NotGiven = NOT_GIVEN,
         session_id: str | NotGiven = NOT_GIVEN,
         solve_captcha: bool | NotGiven = NOT_GIVEN,
@@ -438,6 +463,8 @@ async def create(
               Steel-provided proxies in favor of your specified proxy. Format:
               http(s)://username:password@hostname:port
 
+          region: The desired region for the session to be started in
+
           session_context: Session context data to be used in the created session. Sessions will start with
               an empty context by default.
 
@@ -471,6 +498,7 @@ async def create(
                     "is_selenium": is_selenium,
                     "namespace": namespace,
                     "proxy_url": proxy_url,
+                    "region": region,
                     "session_context": session_context,
                     "session_id": session_id,
                     "solve_captcha": solve_captcha,
@@ -702,6 +730,25 @@ async def release(
             cast_to=SessionReleaseResponse,
         )
 
+    async def release_all(
+        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,
+    ) -> SessionReleaseAllResponse:
+        """Releases all active sessions for the current organization."""
+        return await self._post(
+            "/v1/sessions/release",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SessionReleaseAllResponse,
+        )
+
 
 class SessionsResourceWithRawResponse:
     def __init__(self, sessions: SessionsResource) -> None:
@@ -728,6 +775,9 @@ def __init__(self, sessions: SessionsResource) -> None:
         self.release = to_raw_response_wrapper(
             sessions.release,
         )
+        self.release_all = to_raw_response_wrapper(
+            sessions.release_all,
+        )
 
     @cached_property
     def files(self) -> FilesResourceWithRawResponse:
@@ -759,6 +809,9 @@ def __init__(self, sessions: AsyncSessionsResource) -> None:
         self.release = async_to_raw_response_wrapper(
             sessions.release,
         )
+        self.release_all = async_to_raw_response_wrapper(
+            sessions.release_all,
+        )
 
     @cached_property
     def files(self) -> AsyncFilesResourceWithRawResponse:
@@ -790,6 +843,9 @@ def __init__(self, sessions: SessionsResource) -> None:
         self.release = to_streamed_response_wrapper(
             sessions.release,
         )
+        self.release_all = to_streamed_response_wrapper(
+            sessions.release_all,
+        )
 
     @cached_property
     def files(self) -> FilesResourceWithStreamingResponse:
@@ -821,6 +877,9 @@ def __init__(self, sessions: AsyncSessionsResource) -> None:
         self.release = async_to_streamed_response_wrapper(
             sessions.release,
         )
+        self.release_all = async_to_streamed_response_wrapper(
+            sessions.release_all,
+        )
 
     @cached_property
     def files(self) -> AsyncFilesResourceWithStreamingResponse:

src/steel/types/__init__.py

@@ -23,4 +23,5 @@
 from .credential_create_response import CredentialCreateResponse as CredentialCreateResponse
 from .credential_delete_response import CredentialDeleteResponse as CredentialDeleteResponse
 from .credential_update_response import CredentialUpdateResponse as CredentialUpdateResponse
+from .session_release_all_response import SessionReleaseAllResponse as SessionReleaseAllResponse
 from .session_live_details_response import SessionLiveDetailsResponse as SessionLiveDetailsResponse

src/steel/types/session.py

@@ -62,6 +62,9 @@ class Session(BaseModel):
     proxy: Optional[str] = None
     """Proxy server used for the session"""
 
+    region: Optional[Literal["lax", "ord", "iad", "bom", "scl", "fra", "hkg"]] = None
+    """The desired region for the session to be started in"""
+
     solve_captcha: Optional[bool] = FieldInfo(alias="solveCaptcha", default=None)
     """Indicates if captcha solving is enabled"""
 

src/steel/types/session_create_params.py

@@ -51,6 +51,9 @@ class SessionCreateParams(TypedDict, total=False):
     proxy. Format: http(s)://username:password@hostname:port
     """
 
+    region: Literal["lax", "ord", "iad", "bom", "scl", "fra", "hkg"]
+    """The desired region for the session to be started in"""
+
     session_context: Annotated[SessionContext, PropertyInfo(alias="sessionContext")]
     """Session context data to be used in the created session.
 

src/steel/types/session_release_all_response.py

@@ -0,0 +1,13 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .._models import BaseModel
+
+__all__ = ["SessionReleaseAllResponse"]
+
+
+class SessionReleaseAllResponse(BaseModel):
+    message: str
+    """Details about the outcome of the release operation"""
+
+    success: bool
+    """Indicates if the sessions were successfully released"""

src/steel/types/sessionslist.py

@@ -62,6 +62,9 @@ class Session(BaseModel):
     proxy: Optional[str] = None
     """Proxy server used for the session"""
 
+    region: Optional[Literal["lax", "ord", "iad", "bom", "scl", "fra", "hkg"]] = None
+    """The desired region for the session to be started in"""
+
     solve_captcha: Optional[bool] = FieldInfo(alias="solveCaptcha", default=None)
     """Indicates if captcha solving is enabled"""
 

tests/api_resources/test_sessions.py

@@ -13,6 +13,7 @@
     SessionContext,
     SessionEventsResponse,
     SessionReleaseResponse,
+    SessionReleaseAllResponse,
     SessionLiveDetailsResponse,
 )
 from tests.utils import assert_matches_type
@@ -48,6 +49,7 @@ def test_method_create_with_all_params(self, client: Steel) -> None:
             is_selenium=True,
             namespace="namespace",
             proxy_url="https://example.com",
+            region="lax",
             session_context={
                 "cookies": [
                     {
@@ -357,6 +359,31 @@ def test_path_params_release(self, client: Steel) -> None:
                 "",
             )
 
+    @parametrize
+    def test_method_release_all(self, client: Steel) -> None:
+        session = client.sessions.release_all()
+        assert_matches_type(SessionReleaseAllResponse, session, path=["response"])
+
+    @parametrize
+    def test_raw_response_release_all(self, client: Steel) -> None:
+        response = client.sessions.with_raw_response.release_all()
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        session = response.parse()
+        assert_matches_type(SessionReleaseAllResponse, session, path=["response"])
+
+    @parametrize
+    def test_streaming_response_release_all(self, client: Steel) -> None:
+        with client.sessions.with_streaming_response.release_all() as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            session = response.parse()
+            assert_matches_type(SessionReleaseAllResponse, session, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
 
 class TestAsyncSessions:
     parametrize = pytest.mark.parametrize("async_client", [False, True], indirect=True, ids=["loose", "strict"])
@@ -383,6 +410,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncSteel) ->
             is_selenium=True,
             namespace="namespace",
             proxy_url="https://example.com",
+            region="lax",
             session_context={
                 "cookies": [
                     {
@@ -691,3 +719,28 @@ async def test_path_params_release(self, async_client: AsyncSteel) -> None:
             await async_client.sessions.with_raw_response.release(
                 "",
             )
+
+    @parametrize
+    async def test_method_release_all(self, async_client: AsyncSteel) -> None:
+        session = await async_client.sessions.release_all()
+        assert_matches_type(SessionReleaseAllResponse, session, path=["response"])
+
+    @parametrize
+    async def test_raw_response_release_all(self, async_client: AsyncSteel) -> None:
+        response = await async_client.sessions.with_raw_response.release_all()
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        session = await response.parse()
+        assert_matches_type(SessionReleaseAllResponse, session, path=["response"])
+
+    @parametrize
+    async def test_streaming_response_release_all(self, async_client: AsyncSteel) -> None:
+        async with async_client.sessions.with_streaming_response.release_all() as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            session = await response.parse()
+            assert_matches_type(SessionReleaseAllResponse, session, path=["response"])
+
+        assert cast(Any, response.is_closed) is True