diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py index cd9a971c..cbbbd75e 100644 --- a/g4f/Provider/helper.py +++ b/g4f/Provider/helper.py @@ -17,12 +17,17 @@ from browser_cookie3 import ( firefox, BrowserCookieError ) +try: + from selenium.webdriver.remote.webdriver import WebDriver + except ImportError: + class WebDriver(): + pass try: from undetected_chromedriver import Chrome, ChromeOptions except ImportError: class Chrome(): def __init__(): - raise RuntimeError('Please install "undetected_chromedriver" and "pyvirtualdisplay" package') + raise RuntimeError('Please install the "undetected_chromedriver" package') class ChromeOptions(): def add_argument(): pass