From d972f1cd85c8cb252afad0b1a78588f05d07cc33 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Wed, 30 Oct 2024 17:49:40 -0400
Subject: [PATCH] Refactor find_provider function to use Optional type for
 provider_name and specify return type as Type[BaseProvider]

---
 simplemind/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/simplemind/utils.py b/simplemind/utils.py
index fb70f9b..1ed57bc 100644
--- a/simplemind/utils.py
+++ b/simplemind/utils.py
@@ -1,12 +1,12 @@
 import difflib
-from typing import Union
+from typing import Optional, Type
 
-from .providers import providers
+from .providers import providers, BaseProvider
 
 _PROVIDER_NAMES = [provider.NAME.lower() for provider in providers]
 
 
-def find_provider(provider_name: Union[str, None]):
+def find_provider(provider_name: Optional[str]) -> Type[BaseProvider]:
     """
     Find and instantiate a provider by name.