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/providers/base_provider.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
(limited to 'g4f/providers')
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