11"""API Keys RPC API subclient."""
22
3+ from typing import cast
4+
35from models_library .basic_types import IDStr
46from models_library .products import ProductName
57from models_library .rpc .webserver .auth .api_keys import ApiKeyCreate , ApiKeyGet
@@ -19,12 +21,15 @@ async def create_api_key(
1921 api_key : ApiKeyCreate ,
2022 ) -> ApiKeyGet :
2123 """Create an API key."""
22- return await self ._request (
23- "create_api_key" ,
24- product_name = product_name ,
25- user_id = user_id ,
26- display_name = api_key .display_name ,
27- expiration = api_key .expiration ,
24+ return cast (
25+ ApiKeyGet ,
26+ await self ._request (
27+ "create_api_key" ,
28+ product_name = product_name ,
29+ user_id = user_id ,
30+ display_name = api_key .display_name ,
31+ expiration = api_key .expiration ,
32+ ),
2833 )
2934
3035 async def get_api_key (
@@ -35,11 +40,14 @@ async def get_api_key(
3540 api_key_id : IDStr ,
3641 ) -> ApiKeyGet :
3742 """Get an API key by ID."""
38- return await self ._request (
39- "get_api_key" ,
40- product_name = product_name ,
41- user_id = user_id ,
42- api_key_id = api_key_id ,
43+ return cast (
44+ ApiKeyGet ,
45+ await self ._request (
46+ "get_api_key" ,
47+ product_name = product_name ,
48+ user_id = user_id ,
49+ api_key_id = api_key_id ,
50+ ),
4351 )
4452
4553 async def delete_api_key_by_key (
@@ -50,7 +58,7 @@ async def delete_api_key_by_key(
5058 api_key : str ,
5159 ) -> None :
5260 """Delete an API key by key value."""
53- return await self ._request (
61+ await self ._request (
5462 "delete_api_key_by_key" ,
5563 product_name = product_name ,
5664 user_id = user_id ,
0 commit comments