diff --git a/sources/rest_api/__init__.py b/sources/rest_api/__init__.py index cb217a1f3..14089336d 100644 --- a/sources/rest_api/__init__.py +++ b/sources/rest_api/__init__.py @@ -19,7 +19,7 @@ HeaderLinkPaginator, UnspecifiedPaginator, ) -from .auth import BearerTokenAuth +from .auth import BearerTokenAuth, AuthBase PAGINATOR_MAP = { @@ -83,7 +83,7 @@ def create_paginator(paginator_config): def create_auth(auth_config): - if isinstance(auth_config, BearerTokenAuth): + if isinstance(auth_config, AuthBase): return auth_config return BearerTokenAuth(auth_config.get("token")) if auth_config else None