1818from ..._wrappers import ResultWrapper
1919from ...pagination import SyncSinglePage , AsyncSinglePage
2020from ..._base_client import AsyncPaginator , make_request_options
21- from ...types .shared .role import Role
21+ from ...types .accounts .role_get_response import RoleGetResponse
22+ from ...types .accounts .role_list_response import RoleListResponse
2223
2324__all__ = ["RolesResource" , "AsyncRolesResource" ]
2425
@@ -53,7 +54,7 @@ def list(
5354 extra_query : Query | None = None ,
5455 extra_body : Body | None = None ,
5556 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
56- ) -> SyncSinglePage [Role ]:
57+ ) -> SyncSinglePage [RoleListResponse ]:
5758 """
5859 Get all available roles for an account.
5960
@@ -72,11 +73,11 @@ def list(
7273 raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
7374 return self ._get_api_list (
7475 f"/accounts/{ account_id } /roles" ,
75- page = SyncSinglePage [Role ],
76+ page = SyncSinglePage [RoleListResponse ],
7677 options = make_request_options (
7778 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
7879 ),
79- model = Role ,
80+ model = RoleListResponse ,
8081 )
8182
8283 def get (
@@ -90,7 +91,7 @@ def get(
9091 extra_query : Query | None = None ,
9192 extra_body : Body | None = None ,
9293 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
93- ) -> object :
94+ ) -> Optional [ RoleGetResponse ] :
9495 """
9596 Get information about a specific role for an account.
9697
@@ -118,9 +119,9 @@ def get(
118119 extra_query = extra_query ,
119120 extra_body = extra_body ,
120121 timeout = timeout ,
121- post_parser = ResultWrapper [Optional [object ]]._unwrapper ,
122+ post_parser = ResultWrapper [Optional [RoleGetResponse ]]._unwrapper ,
122123 ),
123- cast_to = cast (Type [object ] , ResultWrapper [object ]),
124+ cast_to = cast (Type [Optional [ RoleGetResponse ]] , ResultWrapper [RoleGetResponse ]),
124125 )
125126
126127
@@ -154,7 +155,7 @@ def list(
154155 extra_query : Query | None = None ,
155156 extra_body : Body | None = None ,
156157 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
157- ) -> AsyncPaginator [Role , AsyncSinglePage [Role ]]:
158+ ) -> AsyncPaginator [RoleListResponse , AsyncSinglePage [RoleListResponse ]]:
158159 """
159160 Get all available roles for an account.
160161
@@ -173,11 +174,11 @@ def list(
173174 raise ValueError (f"Expected a non-empty value for `account_id` but received { account_id !r} " )
174175 return self ._get_api_list (
175176 f"/accounts/{ account_id } /roles" ,
176- page = AsyncSinglePage [Role ],
177+ page = AsyncSinglePage [RoleListResponse ],
177178 options = make_request_options (
178179 extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
179180 ),
180- model = Role ,
181+ model = RoleListResponse ,
181182 )
182183
183184 async def get (
@@ -191,7 +192,7 @@ async def get(
191192 extra_query : Query | None = None ,
192193 extra_body : Body | None = None ,
193194 timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
194- ) -> object :
195+ ) -> Optional [ RoleGetResponse ] :
195196 """
196197 Get information about a specific role for an account.
197198
@@ -219,9 +220,9 @@ async def get(
219220 extra_query = extra_query ,
220221 extra_body = extra_body ,
221222 timeout = timeout ,
222- post_parser = ResultWrapper [Optional [object ]]._unwrapper ,
223+ post_parser = ResultWrapper [Optional [RoleGetResponse ]]._unwrapper ,
223224 ),
224- cast_to = cast (Type [object ] , ResultWrapper [object ]),
225+ cast_to = cast (Type [Optional [ RoleGetResponse ]] , ResultWrapper [RoleGetResponse ]),
225226 )
226227
227228
0 commit comments