From f2996ac21d0eaf0b45b0aa0e325dbef8593bfac1 Mon Sep 17 00:00:00 2001 From: Siddhesh Agarwal Date: Tue, 29 Oct 2024 12:37:03 +0530 Subject: [PATCH] handled None case in find_providers --- simplemind/utils.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/simplemind/utils.py b/simplemind/utils.py index 8304881..c1bd2b7 100644 --- a/simplemind/utils.py +++ b/simplemind/utils.py @@ -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")