Tests.

Author: kschwab

pydantic_settings/sources/utils.py

@@ -54,6 +54,10 @@ def _annotation_is_complex(annotation: type[Any] | None, metadata: list[Any]) ->
 
     origin = get_origin(annotation)
 
+    # Check if annotation is of the form Union[type, ...].
+    if typing_objects.is_union(origin):
+        return _union_is_complex(annotation, metadata)
+
     # Check if annotation is of the form Annotated[type, metadata].
     if typing_objects.is_annotated(origin):
         # Return result of recursive call on inner type.

tests/test_settings.py

@@ -474,6 +474,21 @@ class AnnotatedComplexSettings(BaseSettings):
     ]
 
 
+def test_cli_nested_annotated_unions(env):
+    class Cat(BaseModel):
+        meow: str
+
+    class Dog(BaseModel):
+        woof: str
+
+    class Settings(BaseSettings):
+        model_config = SettingsConfigDict(env_nested_delimiter='__')
+        animals: Annotated[Union[Annotated[Union[Cat, Dog], 'my_nested_annotation'], None], 'my_annotation']
+
+    env.set('ANIMALS__MEOW', 'hiss')
+    assert Settings().model_dump() == {'animals': {'meow': 'hiss'}}
+
+
 def test_set_dict_model(env):
     env.set('bananas', '[1, 2, 3, 3]')
     env.set('CARROTS', '{"a": null, "b": 4}')

tests/test_source_cli.py

@@ -1285,8 +1285,8 @@ class Dog(BaseModel):
     class Settings(BaseSettings):
         animals: CliSubCommand[Annotated[Union[Cat, Dog], 'my_annotation']]
 
-    assert CliApp.run(Settings, cli_args=['Cat', '--meow=purr']).model_dump == {'animals': {'meow': 'purr'}}
-    assert CliApp.run(Settings, cli_args=['Dog', '--woof=bark']).model_dump == {'animals': {'woof': 'bark'}}
+    assert CliApp.run(Settings, cli_args=['Cat', '--meow=purr']).model_dump() == {'animals': {'meow': 'purr'}}
+    assert CliApp.run(Settings, cli_args=['Dog', '--woof=bark']).model_dump() == {'animals': {'woof': 'bark'}}
 
 
 def test_cli_union_similar_sub_models():