99
1010from .... import _legacy_response
1111from ...._types import NOT_GIVEN , Body , Query , Headers , NotGiven
12- from ...._utils import maybe_transform
12+ from ...._utils import maybe_transform , async_maybe_transform
1313from ...._compat import cached_property
1414from ...._resource import SyncAPIResource , AsyncAPIResource
1515from ...._response import to_streamed_response_wrapper , async_to_streamed_response_wrapper
16- from ....pagination import SyncPage , AsyncPage , SyncCursorPage , AsyncCursorPage
16+ from ....pagination import SyncPage , AsyncPage
1717from ...._base_client import AsyncPaginator , make_request_options
1818from ....types .fine_tuning .checkpoints import permission_create_params , permission_retrieve_params
1919from ....types .fine_tuning .checkpoints .permission_create_response import PermissionCreateResponse
@@ -101,7 +101,7 @@ def retrieve(
101101 extra_query : Query | None = None ,
102102 extra_body : Body | None = None ,
103103 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
104- ) -> SyncCursorPage [ PermissionRetrieveResponse ] :
104+ ) -> PermissionRetrieveResponse :
105105 """
106106 **NOTE:** This endpoint requires an [admin API key](../admin-api-keys).
107107
@@ -129,9 +129,8 @@ def retrieve(
129129 raise ValueError (
130130 f"Expected a non-empty value for `fine_tuned_model_checkpoint` but received { fine_tuned_model_checkpoint !r} "
131131 )
132- return self ._get_api_list (
132+ return self ._get (
133133 f"/fine_tuning/checkpoints/{ fine_tuned_model_checkpoint } /permissions" ,
134- page = SyncCursorPage [PermissionRetrieveResponse ],
135134 options = make_request_options (
136135 extra_headers = extra_headers ,
137136 extra_query = extra_query ,
@@ -147,7 +146,7 @@ def retrieve(
147146 permission_retrieve_params .PermissionRetrieveParams ,
148147 ),
149148 ),
150- model = PermissionRetrieveResponse ,
149+ cast_to = PermissionRetrieveResponse ,
151150 )
152151
153152 def delete (
@@ -256,7 +255,7 @@ def create(
256255 method = "post" ,
257256 )
258257
259- def retrieve (
258+ async def retrieve (
260259 self ,
261260 fine_tuned_model_checkpoint : str ,
262261 * ,
@@ -270,7 +269,7 @@ def retrieve(
270269 extra_query : Query | None = None ,
271270 extra_body : Body | None = None ,
272271 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
273- ) -> AsyncPaginator [ PermissionRetrieveResponse , AsyncCursorPage [ PermissionRetrieveResponse ]] :
272+ ) -> PermissionRetrieveResponse :
274273 """
275274 **NOTE:** This endpoint requires an [admin API key](../admin-api-keys).
276275
@@ -298,15 +297,14 @@ def retrieve(
298297 raise ValueError (
299298 f"Expected a non-empty value for `fine_tuned_model_checkpoint` but received { fine_tuned_model_checkpoint !r} "
300299 )
301- return self ._get_api_list (
300+ return await self ._get (
302301 f"/fine_tuning/checkpoints/{ fine_tuned_model_checkpoint } /permissions" ,
303- page = AsyncCursorPage [PermissionRetrieveResponse ],
304302 options = make_request_options (
305303 extra_headers = extra_headers ,
306304 extra_query = extra_query ,
307305 extra_body = extra_body ,
308306 timeout = timeout ,
309- query = maybe_transform (
307+ query = await async_maybe_transform (
310308 {
311309 "after" : after ,
312310 "limit" : limit ,
@@ -316,7 +314,7 @@ def retrieve(
316314 permission_retrieve_params .PermissionRetrieveParams ,
317315 ),
318316 ),
319- model = PermissionRetrieveResponse ,
317+ cast_to = PermissionRetrieveResponse ,
320318 )
321319
322320 async def delete (
0 commit comments