From a4ca5773bd9602951af0ccc11f7e219dfb93732a Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Sat, 6 Apr 2024 01:18:45 +0200
Subject: Fix unittests

---
 g4f/errors.py                  | 6 +++---
 g4f/providers/base_provider.py | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

(limited to 'g4f')

diff --git a/g4f/errors.py b/g4f/errors.py
index 6cb5d177..3d553ba6 100644
--- a/g4f/errors.py
+++ b/g4f/errors.py
@@ -22,15 +22,15 @@ class RetryNoProviderError(Exception):
 class VersionNotFoundError(Exception):
     ...
 
-class NestAsyncioError(Exception):
-    ...
-
 class ModelNotSupportedError(Exception):
     ...
 
 class MissingRequirementsError(Exception):
     ...
 
+class NestAsyncioError(MissingRequirementsError):
+    ...
+
 class MissingAuthError(Exception):
     ...
 
diff --git a/g4f/providers/base_provider.py b/g4f/providers/base_provider.py
index 37f4af15..f3483fc2 100644
--- a/g4f/providers/base_provider.py
+++ b/g4f/providers/base_provider.py
@@ -9,7 +9,7 @@ from inspect import signature, Parameter
 from typing import Callable, Union
 from ..typing import CreateResult, AsyncResult, Messages
 from .types import BaseProvider, FinishReason
-from ..errors import NestAsyncioError, ModelNotSupportedError, MissingRequirementsError
+from ..errors import NestAsyncioError, ModelNotSupportedError
 from .. import debug
 
 if sys.version_info < (3, 10):
@@ -30,7 +30,7 @@ def get_running_loop(check_nested: bool) -> Union[AbstractEventLoop, None]:
                 import nest_asyncio
                 nest_asyncio.apply(loop)
             except ImportError:
-                raise MissingRequirementsError('Install "nest_asyncio" package')
+                raise NestAsyncioError('Install "nest_asyncio" package')
         return loop
     except RuntimeError:
         pass
-- 
cgit v1.2.3