Merge pull request #47 from edornd/bugfix/41-source-annot

Fix settings source type
edornd · Feb 25, 2024 · dd2d42e · dd2d42e
2 parents 1888275 + 8bfed69
commit dd2d42e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/argdantic/core.py b/argdantic/core.py
@@ -11,6 +11,8 @@
 from argdantic.parsing import Argument
 from argdantic.stores import SettingsStoreCallable
 
+SettingSourceCallable = Callable[[Type[BaseSettings]], PydanticBaseSettingsSource]
+
 
 class Command:
     """
@@ -231,8 +233,8 @@ def command(
         self,
         name: Optional[str] = None,
         help: Optional[str] = None,
-        sources: List[PydanticBaseSettingsSource] = None,
-        stores: List[SettingsStoreCallable] = None,
+        sources: Optional[List[SettingSourceCallable]] = None,
+        stores: Optional[List[SettingsStoreCallable]] = None,
         singleton: bool = False,
     ) -> Callable:
         """Decorator to register a function as a command.