feat: Add 'yaml_nested_key' option

Author: Seunghan-Jung

pydantic_settings/main.py

@@ -62,6 +62,7 @@ class SettingsConfigDict(ConfigDict, total=False):
     json_file_encoding: str | None
     yaml_file: PathType | None
     yaml_file_encoding: str | None
+    yaml_nested_key: str | None
     pyproject_toml_depth: int
     """
     Number of levels **up** from the current working directory to attempt to find a pyproject.toml
@@ -446,6 +447,7 @@ def _settings_build_values(
         json_file_encoding=None,
         yaml_file=None,
         yaml_file_encoding=None,
+        yaml_nested_key=None,
         toml_file=None,
         secrets_dir=None,
         protected_namespaces=('model_validate', 'model_dump', 'settings_customise_sources'),

pydantic_settings/sources/providers/yaml.py

@@ -39,14 +39,21 @@ def __init__(
         settings_cls: type[BaseSettings],
         yaml_file: PathType | None = DEFAULT_PATH,
         yaml_file_encoding: str | None = None,
+        yaml_nested_key: str | None = None,
     ):
         self.yaml_file_path = yaml_file if yaml_file != DEFAULT_PATH else settings_cls.model_config.get('yaml_file')
         self.yaml_file_encoding = (
             yaml_file_encoding
             if yaml_file_encoding is not None
             else settings_cls.model_config.get('yaml_file_encoding')
         )
+        self.yaml_nested_key = (
+            yaml_nested_key if yaml_nested_key is not None else settings_cls.model_config.get('yaml_nested_key')
+        )
         self.yaml_data = self._read_files(self.yaml_file_path)
+
+        if self.yaml_nested_key:
+            self.yaml_data = self.yaml_data[self.yaml_nested_key]
         super().__init__(settings_cls, self.yaml_data)
 
     def _read_file(self, file_path: Path) -> dict[str, Any]:

tests/test_source_yaml.py

@@ -166,3 +166,34 @@ def settings_customise_sources(
 
     s = Settings()
     assert s.model_dump() == {'yaml3': 3, 'yaml4': 4}
+
+
+@pytest.mark.skipif(yaml is None, reason='pyYAML is not installed')
+def test_yaml_nested_key(tmp_path):
+    p = tmp_path / '.env'
+    p.write_text(
+        """
+    foobar: "Hello"
+    nested:
+        nested_field: "world!"
+    """
+    )
+
+    class Settings(BaseSettings):
+        nested_field: str
+
+        model_config = SettingsConfigDict(yaml_file=p, yaml_nested_key='nested')
+
+        @classmethod
+        def settings_customise_sources(
+            cls,
+            settings_cls: type[BaseSettings],
+            init_settings: PydanticBaseSettingsSource,
+            env_settings: PydanticBaseSettingsSource,
+            dotenv_settings: PydanticBaseSettingsSource,
+            file_secret_settings: PydanticBaseSettingsSource,
+        ) -> tuple[PydanticBaseSettingsSource, ...]:
+            return (YamlConfigSettingsSource(settings_cls),)
+
+    s = Settings()
+    assert s.nested_field == 'world!'