Skip to content

Commit 551b58d

Browse files
feat: allow get browser to paginate soft deleted browsers
1 parent 74a0fd6 commit 551b58d

File tree

9 files changed

+155
-31
lines changed

9 files changed

+155
-31
lines changed

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 66
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-d611cf8b0301a07123eab0e92498bea5ad69c5292b28aca1016c362cca0a0564.yml
3-
openapi_spec_hash: 6d30f4ad9d61a7da8a75d543cf3d3d75
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-2af1b468584cb44aa9babbbfb82bff4055614fbb5c815084a6b7dacc1cf1a822.yml
3+
openapi_spec_hash: 891affa2849341ea01d62011125f7edc
44
config_hash: 9421eb86b7f3f4b274f123279da3858e

api.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ Methods:
7979

8080
- <code title="post /browsers">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">create</a>(\*\*<a href="src/kernel/types/browser_create_params.py">params</a>) -> <a href="./src/kernel/types/browser_create_response.py">BrowserCreateResponse</a></code>
8181
- <code title="get /browsers/{id}">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">retrieve</a>(id) -> <a href="./src/kernel/types/browser_retrieve_response.py">BrowserRetrieveResponse</a></code>
82-
- <code title="get /browsers">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">list</a>() -> <a href="./src/kernel/types/browser_list_response.py">BrowserListResponse</a></code>
82+
- <code title="get /browsers">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">list</a>(\*\*<a href="src/kernel/types/browser_list_params.py">params</a>) -> <a href="./src/kernel/types/browser_list_response.py">SyncOffsetPagination[BrowserListResponse]</a></code>
8383
- <code title="delete /browsers">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">delete</a>(\*\*<a href="src/kernel/types/browser_delete_params.py">params</a>) -> None</code>
8484
- <code title="delete /browsers/{id}">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">delete_by_id</a>(id) -> None</code>
8585
- <code title="post /browsers/{id}/extensions">client.browsers.<a href="./src/kernel/resources/browsers/browsers.py">load_extensions</a>(id, \*\*<a href="src/kernel/types/browser_load_extensions_params.py">params</a>) -> None</code>

src/kernel/resources/browsers/browsers.py

Lines changed: 89 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,12 @@
2222
FsResourceWithStreamingResponse,
2323
AsyncFsResourceWithStreamingResponse,
2424
)
25-
from ...types import browser_create_params, browser_delete_params, browser_load_extensions_params
25+
from ...types import (
26+
browser_list_params,
27+
browser_create_params,
28+
browser_delete_params,
29+
browser_load_extensions_params,
30+
)
2631
from .process import (
2732
ProcessResource,
2833
AsyncProcessResource,
@@ -65,7 +70,8 @@
6570
async_to_raw_response_wrapper,
6671
async_to_streamed_response_wrapper,
6772
)
68-
from ..._base_client import make_request_options
73+
from ...pagination import SyncOffsetPagination, AsyncOffsetPagination
74+
from ..._base_client import AsyncPaginator, make_request_options
6975
from ...types.browser_list_response import BrowserListResponse
7076
from ...types.browser_create_response import BrowserCreateResponse
7177
from ...types.browser_persistence_param import BrowserPersistenceParam
@@ -247,20 +253,55 @@ def retrieve(
247253
def list(
248254
self,
249255
*,
256+
include_deleted: bool | Omit = omit,
257+
limit: int | Omit = omit,
258+
offset: int | Omit = omit,
250259
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
251260
# The extra values given here take precedence over values defined on the client or passed to this method.
252261
extra_headers: Headers | None = None,
253262
extra_query: Query | None = None,
254263
extra_body: Body | None = None,
255264
timeout: float | httpx.Timeout | None | NotGiven = not_given,
256-
) -> BrowserListResponse:
257-
"""List active browser sessions"""
258-
return self._get(
265+
) -> SyncOffsetPagination[BrowserListResponse]:
266+
"""List all browser sessions with pagination support.
267+
268+
Use include_deleted=true to
269+
include soft-deleted sessions in the results.
270+
271+
Args:
272+
include_deleted: When true, includes soft-deleted browser sessions in the results alongside
273+
active sessions.
274+
275+
limit: Maximum number of results to return. Defaults to 20, maximum 100.
276+
277+
offset: Number of results to skip. Defaults to 0.
278+
279+
extra_headers: Send extra headers
280+
281+
extra_query: Add additional query parameters to the request
282+
283+
extra_body: Add additional JSON properties to the request
284+
285+
timeout: Override the client-level default timeout for this request, in seconds
286+
"""
287+
return self._get_api_list(
259288
"/browsers",
289+
page=SyncOffsetPagination[BrowserListResponse],
260290
options=make_request_options(
261-
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
291+
extra_headers=extra_headers,
292+
extra_query=extra_query,
293+
extra_body=extra_body,
294+
timeout=timeout,
295+
query=maybe_transform(
296+
{
297+
"include_deleted": include_deleted,
298+
"limit": limit,
299+
"offset": offset,
300+
},
301+
browser_list_params.BrowserListParams,
302+
),
262303
),
263-
cast_to=BrowserListResponse,
304+
model=BrowserListResponse,
264305
)
265306

266307
def delete(
@@ -552,23 +593,58 @@ async def retrieve(
552593
cast_to=BrowserRetrieveResponse,
553594
)
554595

555-
async def list(
596+
def list(
556597
self,
557598
*,
599+
include_deleted: bool | Omit = omit,
600+
limit: int | Omit = omit,
601+
offset: int | Omit = omit,
558602
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
559603
# The extra values given here take precedence over values defined on the client or passed to this method.
560604
extra_headers: Headers | None = None,
561605
extra_query: Query | None = None,
562606
extra_body: Body | None = None,
563607
timeout: float | httpx.Timeout | None | NotGiven = not_given,
564-
) -> BrowserListResponse:
565-
"""List active browser sessions"""
566-
return await self._get(
608+
) -> AsyncPaginator[BrowserListResponse, AsyncOffsetPagination[BrowserListResponse]]:
609+
"""List all browser sessions with pagination support.
610+
611+
Use include_deleted=true to
612+
include soft-deleted sessions in the results.
613+
614+
Args:
615+
include_deleted: When true, includes soft-deleted browser sessions in the results alongside
616+
active sessions.
617+
618+
limit: Maximum number of results to return. Defaults to 20, maximum 100.
619+
620+
offset: Number of results to skip. Defaults to 0.
621+
622+
extra_headers: Send extra headers
623+
624+
extra_query: Add additional query parameters to the request
625+
626+
extra_body: Add additional JSON properties to the request
627+
628+
timeout: Override the client-level default timeout for this request, in seconds
629+
"""
630+
return self._get_api_list(
567631
"/browsers",
632+
page=AsyncOffsetPagination[BrowserListResponse],
568633
options=make_request_options(
569-
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
634+
extra_headers=extra_headers,
635+
extra_query=extra_query,
636+
extra_body=extra_body,
637+
timeout=timeout,
638+
query=maybe_transform(
639+
{
640+
"include_deleted": include_deleted,
641+
"limit": limit,
642+
"offset": offset,
643+
},
644+
browser_list_params.BrowserListParams,
645+
),
570646
),
571-
cast_to=BrowserListResponse,
647+
model=BrowserListResponse,
572648
)
573649

574650
async def delete(

src/kernel/types/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from .profile import Profile as Profile
1414
from .app_list_params import AppListParams as AppListParams
1515
from .app_list_response import AppListResponse as AppListResponse
16+
from .browser_list_params import BrowserListParams as BrowserListParams
1617
from .browser_persistence import BrowserPersistence as BrowserPersistence
1718
from .proxy_create_params import ProxyCreateParams as ProxyCreateParams
1819
from .proxy_list_response import ProxyListResponse as ProxyListResponse

src/kernel/types/browser_create_response.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,9 @@ class BrowserCreateResponse(BaseModel):
4949
Only available for non-headless browsers.
5050
"""
5151

52+
deleted_at: Optional[datetime] = None
53+
"""When the browser session was soft-deleted. Only present for deleted sessions."""
54+
5255
kiosk_mode: Optional[bool] = None
5356
"""Whether the browser session is running in kiosk mode."""
5457

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2+
3+
from __future__ import annotations
4+
5+
from typing_extensions import TypedDict
6+
7+
__all__ = ["BrowserListParams"]
8+
9+
10+
class BrowserListParams(TypedDict, total=False):
11+
include_deleted: bool
12+
"""
13+
When true, includes soft-deleted browser sessions in the results alongside
14+
active sessions.
15+
"""
16+
17+
limit: int
18+
"""Maximum number of results to return. Defaults to 20, maximum 100."""
19+
20+
offset: int
21+
"""Number of results to skip. Defaults to 0."""

src/kernel/types/browser_list_response.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,16 @@
11
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
22

3-
from typing import List, Optional
3+
from typing import Optional
44
from datetime import datetime
5-
from typing_extensions import TypeAlias
65

76
from .profile import Profile
87
from .._models import BaseModel
98
from .browser_persistence import BrowserPersistence
109

11-
__all__ = ["BrowserListResponse", "BrowserListResponseItem", "BrowserListResponseItemViewport"]
10+
__all__ = ["BrowserListResponse", "Viewport"]
1211

1312

14-
class BrowserListResponseItemViewport(BaseModel):
13+
class Viewport(BaseModel):
1514
height: int
1615
"""Browser window height in pixels."""
1716

@@ -25,7 +24,7 @@ class BrowserListResponseItemViewport(BaseModel):
2524
"""
2625

2726

28-
class BrowserListResponseItem(BaseModel):
27+
class BrowserListResponse(BaseModel):
2928
cdp_ws_url: str
3029
"""Websocket URL for Chrome DevTools Protocol connections to the browser session"""
3130

@@ -50,6 +49,9 @@ class BrowserListResponseItem(BaseModel):
5049
Only available for non-headless browsers.
5150
"""
5251

52+
deleted_at: Optional[datetime] = None
53+
"""When the browser session was soft-deleted. Only present for deleted sessions."""
54+
5355
kiosk_mode: Optional[bool] = None
5456
"""Whether the browser session is running in kiosk mode."""
5557

@@ -62,7 +64,7 @@ class BrowserListResponseItem(BaseModel):
6264
proxy_id: Optional[str] = None
6365
"""ID of the proxy associated with this browser session, if any."""
6466

65-
viewport: Optional[BrowserListResponseItemViewport] = None
67+
viewport: Optional[Viewport] = None
6668
"""Initial browser window size in pixels with optional refresh rate.
6769
6870
If omitted, image defaults apply (commonly 1024x768@60). Only specific viewport
@@ -73,6 +75,3 @@ class BrowserListResponseItem(BaseModel):
7375
configuration exactly. Note: Higher resolutions may affect the responsiveness of
7476
live view browser
7577
"""
76-
77-
78-
BrowserListResponse: TypeAlias = List[BrowserListResponseItem]

src/kernel/types/browser_retrieve_response.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,9 @@ class BrowserRetrieveResponse(BaseModel):
4949
Only available for non-headless browsers.
5050
"""
5151

52+
deleted_at: Optional[datetime] = None
53+
"""When the browser session was soft-deleted. Only present for deleted sessions."""
54+
5255
kiosk_mode: Optional[bool] = None
5356
"""Whether the browser session is running in kiosk mode."""
5457

tests/api_resources/test_browsers.py

Lines changed: 27 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
BrowserCreateResponse,
1515
BrowserRetrieveResponse,
1616
)
17+
from kernel.pagination import SyncOffsetPagination, AsyncOffsetPagination
1718

1819
base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
1920

@@ -125,7 +126,17 @@ def test_path_params_retrieve(self, client: Kernel) -> None:
125126
@parametrize
126127
def test_method_list(self, client: Kernel) -> None:
127128
browser = client.browsers.list()
128-
assert_matches_type(BrowserListResponse, browser, path=["response"])
129+
assert_matches_type(SyncOffsetPagination[BrowserListResponse], browser, path=["response"])
130+
131+
@pytest.mark.skip(reason="Prism tests are disabled")
132+
@parametrize
133+
def test_method_list_with_all_params(self, client: Kernel) -> None:
134+
browser = client.browsers.list(
135+
include_deleted=True,
136+
limit=1,
137+
offset=0,
138+
)
139+
assert_matches_type(SyncOffsetPagination[BrowserListResponse], browser, path=["response"])
129140

130141
@pytest.mark.skip(reason="Prism tests are disabled")
131142
@parametrize
@@ -135,7 +146,7 @@ def test_raw_response_list(self, client: Kernel) -> None:
135146
assert response.is_closed is True
136147
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
137148
browser = response.parse()
138-
assert_matches_type(BrowserListResponse, browser, path=["response"])
149+
assert_matches_type(SyncOffsetPagination[BrowserListResponse], browser, path=["response"])
139150

140151
@pytest.mark.skip(reason="Prism tests are disabled")
141152
@parametrize
@@ -145,7 +156,7 @@ def test_streaming_response_list(self, client: Kernel) -> None:
145156
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
146157

147158
browser = response.parse()
148-
assert_matches_type(BrowserListResponse, browser, path=["response"])
159+
assert_matches_type(SyncOffsetPagination[BrowserListResponse], browser, path=["response"])
149160

150161
assert cast(Any, response.is_closed) is True
151162

@@ -401,7 +412,17 @@ async def test_path_params_retrieve(self, async_client: AsyncKernel) -> None:
401412
@parametrize
402413
async def test_method_list(self, async_client: AsyncKernel) -> None:
403414
browser = await async_client.browsers.list()
404-
assert_matches_type(BrowserListResponse, browser, path=["response"])
415+
assert_matches_type(AsyncOffsetPagination[BrowserListResponse], browser, path=["response"])
416+
417+
@pytest.mark.skip(reason="Prism tests are disabled")
418+
@parametrize
419+
async def test_method_list_with_all_params(self, async_client: AsyncKernel) -> None:
420+
browser = await async_client.browsers.list(
421+
include_deleted=True,
422+
limit=1,
423+
offset=0,
424+
)
425+
assert_matches_type(AsyncOffsetPagination[BrowserListResponse], browser, path=["response"])
405426

406427
@pytest.mark.skip(reason="Prism tests are disabled")
407428
@parametrize
@@ -411,7 +432,7 @@ async def test_raw_response_list(self, async_client: AsyncKernel) -> None:
411432
assert response.is_closed is True
412433
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
413434
browser = await response.parse()
414-
assert_matches_type(BrowserListResponse, browser, path=["response"])
435+
assert_matches_type(AsyncOffsetPagination[BrowserListResponse], browser, path=["response"])
415436

416437
@pytest.mark.skip(reason="Prism tests are disabled")
417438
@parametrize
@@ -421,7 +442,7 @@ async def test_streaming_response_list(self, async_client: AsyncKernel) -> None:
421442
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
422443

423444
browser = await response.parse()
424-
assert_matches_type(BrowserListResponse, browser, path=["response"])
445+
assert_matches_type(AsyncOffsetPagination[BrowserListResponse], browser, path=["response"])
425446

426447
assert cast(Any, response.is_closed) is True
427448

0 commit comments

Comments
 (0)