From a4ca5773bd9602951af0ccc11f7e219dfb93732a Mon Sep 17 00:00:00 2001 From: Heiner Lohaus 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