diff options
Diffstat (limited to '')
| -rw-r--r-- | g4f/Provider/MyShell.py | 16 |
1 files changed, 4 insertions, 12 deletions
diff --git a/g4f/Provider/MyShell.py b/g4f/Provider/MyShell.py index 0efeb0e8..5580ff6e 100644 --- a/g4f/Provider/MyShell.py +++ b/g4f/Provider/MyShell.py @@ -21,16 +21,10 @@ class MyShell(BaseProvider): proxy: str = None, timeout: int = 120, browser: WebDriver = None, - hidden_display: bool = True, + headless: bool = True, **kwargs ) -> CreateResult: - if browser: - driver = browser - else: - if hidden_display: - driver, display = get_browser("", True, proxy) - else: - driver = get_browser("", False, proxy) + driver = browser if browser else get_browser("", headless, proxy) from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait @@ -87,9 +81,7 @@ return content; elif chunk != "": break finally: - driver.close() if not browser: + driver.close() time.sleep(0.1) - driver.quit() - if hidden_display: - display.stop()
\ No newline at end of file + driver.quit()
\ No newline at end of file |