2323
2424from browserbase import Browserbase , AsyncBrowserbase , APIResponseValidationError
2525from browserbase ._types import Omit
26+ from browserbase ._utils import maybe_transform
2627from browserbase ._models import BaseModel , FinalRequestOptions
2728from browserbase ._constants import RAW_RESPONSE_HEADER
2829from browserbase ._exceptions import APIStatusError , APITimeoutError , BrowserbaseError , APIResponseValidationError
3233 BaseClient ,
3334 make_request_options ,
3435)
36+ from browserbase .types .session_create_params import SessionCreateParams
3537
3638from .utils import update_env
3739
@@ -727,7 +729,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
727729 with pytest .raises (APITimeoutError ):
728730 self .client .post (
729731 "/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 )),
731733 cast_to = httpx .Response ,
732734 options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
733735 )
@@ -742,7 +744,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
742744 with pytest .raises (APIStatusError ):
743745 self .client .post (
744746 "/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 )),
746748 cast_to = httpx .Response ,
747749 options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
748750 )
@@ -1507,7 +1509,7 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
15071509 with pytest .raises (APITimeoutError ):
15081510 await self .client .post (
15091511 "/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 )),
15111513 cast_to = httpx .Response ,
15121514 options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
15131515 )
@@ -1522,7 +1524,7 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
15221524 with pytest .raises (APIStatusError ):
15231525 await self .client .post (
15241526 "/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 )),
15261528 cast_to = httpx .Response ,
15271529 options = {"headers" : {RAW_RESPONSE_HEADER : "stream" }},
15281530 )
0 commit comments