diff --git a/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/class_types_registry.py b/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/class_types_registry.py index 90f7335de004..11fc12b2c3f4 100644 --- a/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/class_types_registry.py +++ b/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/class_types_registry.py @@ -2,7 +2,7 @@ # Copyright (c) 2023 Airbyte, Inc., all rights reserved. # -from typing import Mapping, Type +from typing import Mapping from airbyte_cdk.sources.declarative.auth.declarative_authenticator import NoAuth from airbyte_cdk.sources.declarative.auth.oauth import DeclarativeOauth2Authenticator @@ -10,7 +10,7 @@ ApiKeyAuthenticator, BasicHttpAuthenticator, BearerAuthenticator, - SessionTokenAuthenticator, + LegacySessionTokenAuthenticator, ) from airbyte_cdk.sources.declarative.checks import CheckStream from airbyte_cdk.sources.declarative.datetime.min_max_datetime import MinMaxDatetime @@ -56,7 +56,7 @@ """ CLASS_TYPES_REGISTRY contains a mapping of developer-friendly string -> class to abstract the specific class referred to """ -CLASS_TYPES_REGISTRY: Mapping[str, Type] = { +CLASS_TYPES_REGISTRY: Mapping[str, type] = { "AddedFieldDefinition": AddedFieldDefinition, "AddFields": AddFields, "ApiKeyAuthenticator": ApiKeyAuthenticator, @@ -96,7 +96,7 @@ "SimpleRetriever": SimpleRetriever, "Spec": Spec, "SubstreamPartitionRouter": SubstreamPartitionRouter, - "SessionTokenAuthenticator": SessionTokenAuthenticator, + "SessionTokenAuthenticator": LegacySessionTokenAuthenticator, "WaitUntilTimeFromHeader": WaitUntilTimeFromHeaderBackoffStrategy, "WaitTimeFromHeader": WaitTimeFromHeaderBackoffStrategy, }