diff --git a/ydb/iam/auth.py b/ydb/iam/auth.py index 7b4fa4e8..dac9fb6c 100644 --- a/ydb/iam/auth.py +++ b/ydb/iam/auth.py @@ -16,8 +16,13 @@ from yandex.cloud.iam.v1 import iam_token_service_pb2_grpc from yandex.cloud.iam.v1 import iam_token_service_pb2 except ImportError: - iam_token_service_pb2_grpc = None - iam_token_service_pb2 = None + try: + # This attempt is to enable the IAM auth inside the YDB repository on GitHub + from ydb.public.api.client.yc_public.iam import iam_token_service_pb2_grpc + from ydb.public.api.client.yc_public.iam import iam_token_service_pb2 + except ImportError: + iam_token_service_pb2_grpc = None + iam_token_service_pb2 = None try: import requests