From 72b3a8c70a930c8f5fcd70a41388841af984b169 Mon Sep 17 00:00:00 2001
From: Lucas Gomide <lucaslg200@gmail.com>
Date: Fri, 30 May 2025 09:13:11 -0300
Subject: [PATCH] feat: allow to provide the driver and options to be used by
 Selenium (#316)

---
 .../selenium_scraping_tool.py                 | 13 +++++++---
 tests/tools/selenium_scraping_tool_test.py    | 24 +++++++++++++++----
 2 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/src/crewai_tools/tools/selenium_scraping_tool/selenium_scraping_tool.py b/src/crewai_tools/tools/selenium_scraping_tool/selenium_scraping_tool.py
index 57211e64e..3976facef 100644
--- a/src/crewai_tools/tools/selenium_scraping_tool/selenium_scraping_tool.py
+++ b/src/crewai_tools/tools/selenium_scraping_tool/selenium_scraping_tool.py
@@ -91,9 +91,16 @@ class SeleniumScrapingTool(BaseTool):
                     "`selenium` and `webdriver-manager` package not found, please run `uv add selenium webdriver-manager`"
                 )
 
-        options: Options = Options()
-        options.add_argument("--headless")
-        self.driver = webdriver.Chrome(options=options)
+        if 'driver' not in kwargs:
+            if 'options' not in kwargs:
+                options: Options = Options()
+                options.add_argument("--headless")
+            else:
+                options = kwargs['options']
+            self.driver = webdriver.Chrome(options=options)
+        else:
+            self.driver = kwargs['driver']
+
         self._by = By
         if cookie is not None:
             self.cookie = cookie
diff --git a/tests/tools/selenium_scraping_tool_test.py b/tests/tools/selenium_scraping_tool_test.py
index b360df3a1..0e285d136 100644
--- a/tests/tools/selenium_scraping_tool_test.py
+++ b/tests/tools/selenium_scraping_tool_test.py
@@ -2,9 +2,8 @@ import os
 import tempfile
 from unittest.mock import MagicMock, patch
 
-import pytest
 from bs4 import BeautifulSoup
-
+from selenium.webdriver.chrome.options import Options
 from crewai_tools.tools.selenium_scraping_tool.selenium_scraping_tool import (
     SeleniumScrapingTool,
 )
@@ -24,9 +23,7 @@ def mock_driver_with_html(html_content):
 
 
 def initialize_tool_with(mock_driver):
-    tool = SeleniumScrapingTool()
-    tool.driver = mock_driver
-
+    tool = SeleniumScrapingTool(driver=mock_driver)
     return tool
 
 
@@ -48,6 +45,17 @@ def test_tool_initialization(mocked_chrome):
     except:
         pass
 
+@patch("selenium.webdriver.Chrome")
+def test_tool_initialization_with_options(mocked_chrome):
+    mocked_chrome.return_value = MagicMock()
+
+    options = Options()
+    options.add_argument("--disable-gpu")
+
+    SeleniumScrapingTool(options=options)
+
+    mocked_chrome.assert_called_once_with(options=options)
+
 
 @patch("selenium.webdriver.Chrome")
 def test_scrape_without_css_selector(_mocked_chrome_driver):
@@ -113,3 +121,9 @@ def test_scrape_with_driver_error(_mocked_chrome_driver):
     result = tool._run(website_url="https://example.com")
     assert result == "Error scraping website: WebDriver error occurred"
     mock_driver.close.assert_called_once()
+
+@patch("selenium.webdriver.Chrome")
+def test_initialization_with_driver(_mocked_chrome_driver):
+    mock_driver = MagicMock()
+    tool = initialize_tool_with(mock_driver)
+    assert tool.driver == mock_driver