diff --git a/simplemind/utils.py b/simplemind/utils.py index f39c036..41ed5dc 100644 --- a/simplemind/utils.py +++ b/simplemind/utils.py @@ -5,6 +5,7 @@ from .providers import providers _PROVIDER_NAMES = [provider.NAME.lower() for provider in providers] + def find_provider(provider_name: Union[str, None]): """Find a provider by name.""" if provider_name: @@ -12,10 +13,14 @@ def find_provider(provider_name: Union[str, None]): if provider_class.NAME.lower() == provider_name.lower(): # Instantiate the provider return provider_class() - - provider_found = difflib.get_close_matches(provider_name.lower(), _PROVIDER_NAMES, n=1) # Show only one suggestion - + + provider_found = difflib.get_close_matches( + provider_name.lower(), _PROVIDER_NAMES, n=1 + ) # Show only one suggestion + if provider_found: - raise ValueError(f"Provider {provider_name} not found. Maybe you try to use is '{provider_found[0]}'?") + raise ValueError( + f"Provider {provider_name!r} not found. Did you mean {provider_found[0]!r}?" + ) else: - raise ValueError(f"Provider {provider_name} not found.") \ No newline at end of file + raise ValueError(f"Provider {provider_name} not found.")