1 files changed, 10 insertions, 3 deletions
diff --git a/etc/unittest/main.py b/etc/unittest/main.py
index f5eb5138..cc3c6a18 100644
--- a/etc/unittest/main.py
+++ b/etc/unittest/main.py
@@ -24,12 +24,19 @@ class TestGetLastProvider(unittest.TestCase):
     def test_get_last_provider(self):
         ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, ProviderMock)
         self.assertEqual(get_last_provider(), ProviderMock)
-        
+
     def test_get_last_provider_retry(self):
         ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, RetryProvider([ProviderMock]))
         self.assertEqual(get_last_provider(), ProviderMock)
-        
+
     def test_get_last_provider_async(self):
         coroutine = ChatCompletion.create_async(g4f.models.default, DEFAULT_MESSAGES, ProviderMock)
         asyncio.run(coroutine)
-        self.assertEqual(get_last_provider(), ProviderMock)
-\ No newline at end of file
+        self.assertEqual(get_last_provider(), ProviderMock)
+
+    def test_get_last_provider_as_dict(self):
+        ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, ProviderMock)
+        last_provider_dict = get_last_provider(True)
+        self.assertIsInstance(last_provider_dict, dict)
+        self.assertIn('name', last_provider_dict)
+        self.assertEqual(ProviderMock.__name__, last_provider_dict['name'])
+\ No newline at end of file