summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/helper.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/helper.py7
1 files changed, 6 insertions, 1 deletions
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