File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
libs/labelbox/src/labelbox/schema Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 8
8
from lbox .exceptions import LabelboxError
9
9
10
10
from labelbox .schema .user import User
11
- from labelbox .schema .role import Role
11
+ from labelbox .schema .role import Role , format_role
12
12
13
13
if TYPE_CHECKING :
14
14
from labelbox import Client
@@ -258,7 +258,7 @@ def _get_available_api_key_roles(client: "Client") -> List[str]:
258
258
if role ["name" ] in ["None" , "Tenant Admin" ]:
259
259
continue
260
260
if all (perm in current_permissions for perm in role ["permissions" ]):
261
- available_roles .append (role ["name" ])
261
+ available_roles .append (format_role ( role ["name" ]) )
262
262
client ._cached_available_api_key_roles = available_roles
263
263
return available_roles
264
264
@@ -332,9 +332,9 @@ def create_api_key(
332
332
raise ValueError ("role must be a Role object or a valid role name" )
333
333
334
334
allowed_roles = ApiKey ._get_available_api_key_roles (client )
335
- # Normalize the allowed roles to lowercase for case-insensitive comparison
336
- normalized_allowed_roles = [ r . lower () for r in allowed_roles ]
337
- if role_name . lower () not in normalized_allowed_roles :
335
+ # Format the input role name consistently with available roles
336
+ formatted_role_name = format_role ( role_name )
337
+ if formatted_role_name not in allowed_roles :
338
338
raise ValueError (
339
339
f"Invalid role specified. Allowed roles are: { allowed_roles } "
340
340
)
You can’t perform that action at this time.
0 commit comments