Make google driver more sturdy, fails locally even if not in CI

h2oai · Oct 29, 2024 · bcf68d8 · bcf68d8
1 parent c59ca35
commit bcf68d8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/openai_server/agent_tools/download_web_video.py b/openai_server/agent_tools/download_web_video.py
@@ -13,6 +13,11 @@ def selenium(base_url, video_url):
     options = webdriver.ChromeOptions()
     options.add_argument("--disable-blink-features=AutomationControlled")
     options.add_argument("start-maximized")
+    options.add_argument("--headless")
+    options.add_argument("--no-sandbox")
+    options.add_argument("--disable-dev-shm-usage")
+    options.add_argument("--disable-gpu")
+
     driver = webdriver.Chrome(options=options)
 
     google_username = os.getenv('GOOGLE_USERNAME')

diff --git a/src/version.py b/src/version.py
@@ -1 +1 @@
-__version__ = "0d05e932421bfc4c61da9ef2fc99b64d43a777fe"
+__version__ = "c59ca3591240dcbaf9ee378d72fc2c7420f719a4"
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		__version__ = "0d05e932421bfc4c61da9ef2fc99b64d43a777fe"
		__version__ = "c59ca3591240dcbaf9ee378d72fc2c7420f719a4"