From 1ecd4a496659d2883e9ba53086fc4db6cbe11d5c Mon Sep 17 00:00:00 2001 From: Francisco Griman Date: Wed, 30 Oct 2024 02:08:12 -0400 Subject: [PATCH] Refine error handling in find_provider function to suggest a single similar provider name --- simplemind/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/simplemind/utils.py b/simplemind/utils.py index 40fdd91..2828753 100644 --- a/simplemind/utils.py +++ b/simplemind/utils.py @@ -13,9 +13,9 @@ def find_provider(provider_name: Union[str, None]): return provider_class() providers_name = [provider.NAME.lower() for provider in providers] - providers_founds = difflib.get_close_matches(provider_name.lower(), providers_name) + provider_found = difflib.get_close_matches(provider_name.lower(), providers_name, n=1) # Show only one suggestion - if providers_founds: - raise ValueError(f"Provider {provider_name} not found. Maybe you try to use is '{providers_founds[0]}'?") + if provider_found: + raise ValueError(f"Provider {provider_name} not found. Maybe you try to use is '{provider_found[0]}'?") else: raise ValueError(f"Provider {provider_name} not found.") \ No newline at end of file