fix imports

Author: mavwolverine

pydantic_settings/sources/providers/aws.py

@@ -1,11 +1,15 @@
+from __future__ import annotations as _annotations  # important for BaseSettings import to work
+
 import json
 from collections.abc import Mapping
-from typing import Optional
-
-from pydantic_settings.main import BaseSettings
+from typing import TYPE_CHECKING, Optional
 
 from .env import EnvSettingsSource
 
+if TYPE_CHECKING:
+    from pydantic_settings.main import BaseSettings
+
+
 boto3_client = None
 SecretsManagerClient = None
 
@@ -58,3 +62,8 @@ def __repr__(self) -> str:
             f'{self.__class__.__name__}(secret_id={self._secret_id!r}, '
             f'env_nested_delimiter={self.env_nested_delimiter!r})'
         )
+
+
+__all__ = [
+    'AWSSecretsManagerSettingsSource',
+]

tests/test_source_aws_secrets_manager.py

@@ -21,7 +21,7 @@
     BaseSettings,
     PydanticBaseSettingsSource,
 )
-from pydantic_settings.sources import import_aws_secrets_manager
+from pydantic_settings.sources.providers.aws import import_aws_secrets_manager
 
 try:
     aws_secrets_manager = True