23
23
24
24
from browserbase import Browserbase , AsyncBrowserbase , APIResponseValidationError
25
25
from browserbase ._types import Omit
26
+ from browserbase ._utils import maybe_transform
26
27
from browserbase ._models import BaseModel , FinalRequestOptions
27
28
from browserbase ._constants import RAW_RESPONSE_HEADER
28
29
from browserbase ._exceptions import APIStatusError , APITimeoutError , BrowserbaseError , APIResponseValidationError
32
33
BaseClient ,
33
34
make_request_options ,
34
35
)
36
+ from browserbase .types .session_create_params import SessionCreateParams
35
37
36
38
from .utils import update_env
37
39
@@ -727,7 +729,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
727
729
with pytest .raises (APITimeoutError ):
728
730
self .client .post (
729
731
"/v1/sessions" ,
730
- body = cast (object , dict (project_id = "your_project_id" )),
732
+ body = cast (object , maybe_transform ( dict (project_id = "your_project_id" ), SessionCreateParams )),
731
733
cast_to = httpx .Response ,
732
734
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
733
735
)
@@ -742,7 +744,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
742
744
with pytest .raises (APIStatusError ):
743
745
self .client .post (
744
746
"/v1/sessions" ,
745
- body = cast (object , dict (project_id = "your_project_id" )),
747
+ body = cast (object , maybe_transform ( dict (project_id = "your_project_id" ), SessionCreateParams )),
746
748
cast_to = httpx .Response ,
747
749
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
748
750
)
@@ -1507,7 +1509,7 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
1507
1509
with pytest .raises (APITimeoutError ):
1508
1510
await self .client .post (
1509
1511
"/v1/sessions" ,
1510
- body = cast (object , dict (project_id = "your_project_id" )),
1512
+ body = cast (object , maybe_transform ( dict (project_id = "your_project_id" ), SessionCreateParams )),
1511
1513
cast_to = httpx .Response ,
1512
1514
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
1513
1515
)
@@ -1522,7 +1524,7 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
1522
1524
with pytest .raises (APIStatusError ):
1523
1525
await self .client .post (
1524
1526
"/v1/sessions" ,
1525
- body = cast (object , dict (project_id = "your_project_id" )),
1527
+ body = cast (object , maybe_transform ( dict (project_id = "your_project_id" ), SessionCreateParams )),
1526
1528
cast_to = httpx .Response ,
1527
1529
options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
1528
1530
)
0 commit comments