99
1010from openai import OpenAI , AsyncOpenAI
1111from tests .utils import assert_matches_type
12- from openai .pagination import SyncPage , AsyncPage
12+ from openai .pagination import SyncPage , AsyncPage , SyncCursorPage , AsyncCursorPage
1313from openai .types .fine_tuning .checkpoints import (
1414 PermissionCreateResponse ,
1515 PermissionDeleteResponse ,
@@ -71,7 +71,7 @@ def test_method_retrieve(self, client: OpenAI) -> None:
7171 permission = client .fine_tuning .checkpoints .permissions .retrieve (
7272 fine_tuned_model_checkpoint = "ft-AF1WoRqd3aJAHsqc9NY7iL8F" ,
7373 )
74- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
74+ assert_matches_type (SyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
7575
7676 @parametrize
7777 def test_method_retrieve_with_all_params (self , client : OpenAI ) -> None :
@@ -82,7 +82,7 @@ def test_method_retrieve_with_all_params(self, client: OpenAI) -> None:
8282 order = "ascending" ,
8383 project_id = "project_id" ,
8484 )
85- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
85+ assert_matches_type (SyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
8686
8787 @parametrize
8888 def test_raw_response_retrieve (self , client : OpenAI ) -> None :
@@ -93,7 +93,7 @@ def test_raw_response_retrieve(self, client: OpenAI) -> None:
9393 assert response .is_closed is True
9494 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
9595 permission = response .parse ()
96- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
96+ assert_matches_type (SyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
9797
9898 @parametrize
9999 def test_streaming_response_retrieve (self , client : OpenAI ) -> None :
@@ -104,7 +104,7 @@ def test_streaming_response_retrieve(self, client: OpenAI) -> None:
104104 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
105105
106106 permission = response .parse ()
107- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
107+ assert_matches_type (SyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
108108
109109 assert cast (Any , response .is_closed ) is True
110110
@@ -220,7 +220,7 @@ async def test_method_retrieve(self, async_client: AsyncOpenAI) -> None:
220220 permission = await async_client .fine_tuning .checkpoints .permissions .retrieve (
221221 fine_tuned_model_checkpoint = "ft-AF1WoRqd3aJAHsqc9NY7iL8F" ,
222222 )
223- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
223+ assert_matches_type (AsyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
224224
225225 @parametrize
226226 async def test_method_retrieve_with_all_params (self , async_client : AsyncOpenAI ) -> None :
@@ -231,7 +231,7 @@ async def test_method_retrieve_with_all_params(self, async_client: AsyncOpenAI)
231231 order = "ascending" ,
232232 project_id = "project_id" ,
233233 )
234- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
234+ assert_matches_type (AsyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
235235
236236 @parametrize
237237 async def test_raw_response_retrieve (self , async_client : AsyncOpenAI ) -> None :
@@ -242,7 +242,7 @@ async def test_raw_response_retrieve(self, async_client: AsyncOpenAI) -> None:
242242 assert response .is_closed is True
243243 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
244244 permission = response .parse ()
245- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
245+ assert_matches_type (AsyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
246246
247247 @parametrize
248248 async def test_streaming_response_retrieve (self , async_client : AsyncOpenAI ) -> None :
@@ -253,7 +253,7 @@ async def test_streaming_response_retrieve(self, async_client: AsyncOpenAI) -> N
253253 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
254254
255255 permission = await response .parse ()
256- assert_matches_type (PermissionRetrieveResponse , permission , path = ["response" ])
256+ assert_matches_type (AsyncCursorPage [ PermissionRetrieveResponse ] , permission , path = ["response" ])
257257
258258 assert cast (Any , response .is_closed ) is True
259259
0 commit comments