Add linux selenium fixes

pull/2096/head
Eesa Hamza 2023-04-17 07:55:48 +03:00
parent 9589334a30
commit e86764df45
1 changed files with 8 additions and 0 deletions

View File

@ -17,6 +17,7 @@ from selenium.webdriver.safari.options import Options as SafariOptions
import logging
from pathlib import Path
from autogpt.config import Config
from sys import platform
FILE_DIR = Path(__file__).parent.parent
CFG = Config()
@ -66,6 +67,13 @@ def scrape_text_with_selenium(url: str) -> tuple[WebDriver, str]:
"user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.5615.49 Safari/537.36"
)
# Add linux specific flags
if platform == "linux" or platform == "linux2":
options.add_argument("--no-sandbox")
options.add_argument("--disable-dev-shm-usage")
options.add_argument("--remote-debugging-port=9222")
if CFG.selenium_web_browser == "firefox":
driver = webdriver.Firefox(
executable_path=GeckoDriverManager().install(), options=options