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