handled None case in find_providers

This commit is contained in:
Siddhesh Agarwal
2024-10-29 12:37:03 +05:30
parent 906fadc03d
commit f2996ac21d
+7 -5
View File
@@ -1,10 +1,12 @@
from typing import Union
from .providers import providers
def find_provider(provider_name: str):
def find_provider(provider_name: Union[str, None]):
"""Find a provider by name."""
for provider_class in providers:
if provider_class.__name__.lower() == provider_name.lower():
# Instantiate the provider
return provider_class()
if provider_name:
for provider_class in providers:
if provider_class.__name__.lower() == provider_name.lower():
# Instantiate the provider
return provider_class()
raise ValueError(f"Provider {provider_name} not found")