Skip to content

Commit f038284

Browse files
committed
fix linting issues
1 parent 79a369b commit f038284

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

pydantic_settings/sources.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -110,9 +110,11 @@ def import_azure_key_vault() -> None:
110110

111111
def import_aws_secrets_manager() -> None:
112112
global boto3_client
113+
global SecretsManagerClient
113114

114115
try:
115116
from boto3 import client as boto3_client
117+
from mypy_boto3_secretsmanager.client import SecretsManagerClient
116118
except ImportError as e:
117119
raise ImportError(
118120
'AWS Secrets Manager dependencies are not installed, run `pip install pydantic-settings[aws-secrets-manager]`'
@@ -2250,7 +2252,7 @@ def __repr__(self) -> str:
22502252

22512253
class AWSSecretsManagerSettingsSource(EnvSettingsSource):
22522254
_secret_id: str
2253-
_secretsmanager_client: SecretClient # type: ignore
2255+
_secretsmanager_client: SecretsManagerClient # type: ignore
22542256

22552257
def __init__(
22562258
self,
@@ -2261,7 +2263,7 @@ def __init__(
22612263
env_parse_enums: bool | None = None,
22622264
) -> None:
22632265
import_aws_secrets_manager()
2264-
self._secretsmanager_client = boto3_client('secretsmanager')
2266+
self._secretsmanager_client = boto3_client('secretsmanager') # type: ignore
22652267
self._secret_id = secret_id
22662268
super().__init__(
22672269
settings_cls,
@@ -2279,7 +2281,10 @@ def _load_env_vars(self) -> Mapping[str, Optional[str]]:
22792281
return json.loads(response['SecretString'])
22802282

22812283
def __repr__(self) -> str:
2282-
return f'{self.__class__.__name__}(secret_id={self._secret_id!r}, ' f'env_nested_delimiter={self.env_nested_delimiter!r})'
2284+
return (
2285+
f'{self.__class__.__name__}(secret_id={self._secret_id!r}, '
2286+
f'env_nested_delimiter={self.env_nested_delimiter!r})'
2287+
)
22832288

22842289

22852290
def _get_env_var_key(key: str, case_sensitive: bool = False) -> str:

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ dynamic = ['version']
5050
yaml = ["pyyaml>=6.0.1"]
5151
toml = ["tomli>=2.0.1"]
5252
azure-key-vault = ["azure-keyvault-secrets>=4.8.0", "azure-identity>=1.16.0"]
53-
aws-secrets-manager = ["boto3>=1.35.98"]
53+
aws-secrets-manager = ["boto3>=1.35.0", "boto3-stubs[secretsmanager]"]
5454

5555
[project.urls]
5656
Homepage = 'https://github.com/pydantic/pydantic-settings'

0 commit comments

Comments
 (0)