Fix unittests

g4f/Provider/Cloudflare.py

@@ -3,15 +3,9 @@
 import asyncio
 import json
 
-try:
-    import nodriver
-    has_nodriver = True
-except ImportError:
-    has_nodriver = False
-
 from ..typing import AsyncResult, Messages, Cookies
 from .base_provider import AsyncGeneratorProvider, ProviderModelMixin, get_running_loop
-from ..requests import Session, StreamSession, get_args_from_nodriver, raise_for_status, merge_cookies, DEFAULT_HEADERS
+from ..requests import Session, StreamSession, get_args_from_nodriver, raise_for_status, merge_cookies, DEFAULT_HEADERS, has_nodriver, has_curl_cffi
 from ..errors import ResponseStatusError
 
 class Cloudflare(AsyncGeneratorProvider, ProviderModelMixin):
@@ -45,6 +39,8 @@ def get_models(cls) -> str:
                     get_running_loop(check_nested=True)
                     args = get_args_from_nodriver(cls.url)
                     cls._args = asyncio.run(args)
+                elif not has_curl_cffi:
+                    return cls.models
                 else:
                     cls._args = {"headers": DEFAULT_HEADERS, "cookies": {}}
             with Session(**cls._args) as session:

g4f/Provider/needs_auth/OpenaiChat.py

@@ -637,7 +637,7 @@ def __init__(self, conversation_id: str = None, message_id: str = None, finish_r
 def get_cookies(
         urls: list[str]  = None
     ):
-    params = dict()
+    params = {}
     if urls is not None:
         params['urls'] = [i for i in urls]
     cmd_dict = {

g4f/requests/__init__.py

@@ -8,7 +8,7 @@
     from .curl_cffi import StreamResponse, StreamSession, FormData
     has_curl_cffi = True
 except ImportError:
-    from typing import Type as Session, Type as Response
+    from typing import Type as Response
     from .aiohttp import StreamResponse, StreamSession, FormData
     has_curl_cffi = False
 try:
@@ -38,6 +38,11 @@
 from ..typing import Cookies
 from .defaults import DEFAULT_HEADERS, WEBVIEW_HAEDERS
 
+if not has_curl_cffi:
+    class Session:
+        def __init__(self, **kwargs):
+            raise MissingRequirementsError('Install "curl_cffi" package | pip install -U curl_cffi')
+
 async def get_args_from_webview(url: str) -> dict:
     if not has_webview:
         raise MissingRequirementsError('Install "webview" package')