From 9149eb9d8d6da3ee3729c5cf5b9a91d158027c17 Mon Sep 17 00:00:00 2001
From: Adnane Guettaf <adnane.guettaf@edenai.co>
Date: Tue, 14 Jan 2025 15:35:25 +0100
Subject: [PATCH] refactor: add type hint for 'provider' variable in
 Completions

---
 aisuite/client.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/aisuite/client.py b/aisuite/client.py
index 384b29b6..165eda72 100644
--- 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}'.")