refactor: add type hint for 'provider' variable in Completions

andrewyng · Jan 14, 2025 · 9149eb9 · 9149eb9
1 parent 206dfab
commit 9149eb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aisuite/client.py b/aisuite/client.py
@@ -1,4 +1,4 @@
-from .provider import ProviderFactory
+from .provider import ProviderFactory, Provider
 
 
 class Client:
@@ -109,7 +109,7 @@ def create(self, model: str, messages: list, **kwargs):
                 provider_key, config
             )
 
-        provider = self.client.providers.get(provider_key)
+        provider: Provider = self.client.providers.get(provider_key)
         if not provider:
             raise ValueError(f"Could not load provider for '{provider_key}'.")