kennethreitz/simplemind
1
0
Fork 0
mirror of https://github.com/kennethreitz/simplemind.git synced 2026-06-05 22:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor find_provider function to optimize provider name matching

Browse Source
This commit is contained in:
Francisco Griman
2024-10-30 02:10:18 -04:00
parent 1ecd4a4966
commit 9a5c7ff61b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simplemind/utils.py
+2 -2
View File
@@ -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]}'?")