From 9a5c7ff61b6547e0aa52302b8fd3042087a8a334 Mon Sep 17 00:00:00 2001 From: Francisco Griman Date: Wed, 30 Oct 2024 02:10:18 -0400 Subject: [PATCH] Refactor find_provider function to optimize provider name matching --- simplemind/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/simplemind/utils.py b/simplemind/utils.py index 2828753..f39c036 100644 --- a/simplemind/utils.py +++ b/simplemind/utils.py @@ -3,6 +3,7 @@ from typing import Union 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.""" @@ -12,8 +13,7 @@ def find_provider(provider_name: Union[str, None]): # Instantiate the provider return provider_class() - providers_name = [provider.NAME.lower() for provider in providers] - provider_found = difflib.get_close_matches(provider_name.lower(), providers_name, 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]}'?")