1414 SessionLiveURLs ,
1515 SessionListResponse ,
1616 SessionCreateResponse ,
17+ SessionRetrieveResponse ,
1718)
1819
1920base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
@@ -100,7 +101,7 @@ def test_method_retrieve(self, client: Browserbase) -> None:
100101 session = client .sessions .retrieve (
101102 "id" ,
102103 )
103- assert_matches_type (Session , session , path = ["response" ])
104+ assert_matches_type (SessionRetrieveResponse , session , path = ["response" ])
104105
105106 @parametrize
106107 def test_raw_response_retrieve (self , client : Browserbase ) -> None :
@@ -111,7 +112,7 @@ def test_raw_response_retrieve(self, client: Browserbase) -> None:
111112 assert response .is_closed is True
112113 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
113114 session = response .parse ()
114- assert_matches_type (Session , session , path = ["response" ])
115+ assert_matches_type (SessionRetrieveResponse , session , path = ["response" ])
115116
116117 @parametrize
117118 def test_streaming_response_retrieve (self , client : Browserbase ) -> None :
@@ -122,7 +123,7 @@ def test_streaming_response_retrieve(self, client: Browserbase) -> None:
122123 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
123124
124125 session = response .parse ()
125- assert_matches_type (Session , session , path = ["response" ])
126+ assert_matches_type (SessionRetrieveResponse , session , path = ["response" ])
126127
127128 assert cast (Any , response .is_closed ) is True
128129
@@ -332,7 +333,7 @@ async def test_method_retrieve(self, async_client: AsyncBrowserbase) -> None:
332333 session = await async_client .sessions .retrieve (
333334 "id" ,
334335 )
335- assert_matches_type (Session , session , path = ["response" ])
336+ assert_matches_type (SessionRetrieveResponse , session , path = ["response" ])
336337
337338 @parametrize
338339 async def test_raw_response_retrieve (self , async_client : AsyncBrowserbase ) -> None :
@@ -343,7 +344,7 @@ async def test_raw_response_retrieve(self, async_client: AsyncBrowserbase) -> No
343344 assert response .is_closed is True
344345 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
345346 session = await response .parse ()
346- assert_matches_type (Session , session , path = ["response" ])
347+ assert_matches_type (SessionRetrieveResponse , session , path = ["response" ])
347348
348349 @parametrize
349350 async def test_streaming_response_retrieve (self , async_client : AsyncBrowserbase ) -> None :
@@ -354,7 +355,7 @@ async def test_streaming_response_retrieve(self, async_client: AsyncBrowserbase)
354355 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
355356
356357 session = await response .parse ()
357- assert_matches_type (Session , session , path = ["response" ])
358+ assert_matches_type (SessionRetrieveResponse , session , path = ["response" ])
358359
359360 assert cast (Any , response .is_closed ) is True
360361
0 commit comments