got it

Author: kamath

examples/selenium_basic.py

@@ -0,0 +1,55 @@
+from examples import bb, BROWSERBASE_PROJECT_ID, BROWSERBASE_API_KEY
+from browserbase.types import SessionCreateResponse
+from selenium import webdriver
+from selenium.webdriver.remote.webdriver import WebDriver
+from selenium.webdriver.remote.remote_connection import RemoteConnection
+
+
+class BrowserbaseConnection(RemoteConnection):
+    """
+    Manage a single session with Browserbase.
+    """
+
+    browserbase_session: SessionCreateResponse
+
+    def __init__(self, *args, **kwargs):
+        self.browserbase_session = bb.sessions.create(project_id=BROWSERBASE_PROJECT_ID)
+        super().__init__(*args, **kwargs)
+
+    def get_remote_connection_headers(self, parsed_url, keep_alive=False):
+        headers = super().get_remote_connection_headers(parsed_url, keep_alive)
+
+        # Update headers to include the Browserbase required information
+        headers["x-bb-api-key"] = BROWSERBASE_API_KEY
+        headers["session-id"] = self.browserbase_session.id
+
+        return headers
+
+
+def run(driver: WebDriver):
+    # Instruct the browser to go to the SF MOMA page
+    driver.get("https://www.sfmoma.org")
+
+    # Print out a bit of info about the page it landed on
+    print(f"{driver.current_url=} | {driver.title=}")
+
+    ...
+
+
+# Use the custom class to create and connect to a new browser session
+connection = BrowserbaseConnection("http://connect.browserbase.com/webdriver")
+driver = webdriver.Remote(connection, options=webdriver.ChromeOptions())
+
+# Print a bit of info about the browser we've connected to
+print(
+    "Connected to Browserbase",
+    f"{driver.name} version {driver.caps['browserVersion']}",
+)
+
+try:
+    # Perform our browser commands
+    run(driver)
+
+finally:
+    # Make sure to quit the driver so your session is ended!
+    driver.quit()

pyproject.toml

@@ -59,6 +59,7 @@ dev-dependencies = [
     "rich>=13.7.1",
     "python-dotenv",
     "playwright",
+    "selenium",
 ]
 
 [tool.rye.scripts]

requirements-dev.lock

@@ -18,11 +18,14 @@ anyio==4.4.0
 argcomplete==3.1.2
     # via nox
 attrs==23.1.0
+    # via outcome
     # via pytest
+    # via trio
 certifi==2023.7.22
     # via httpcore
     # via httpx
     # via requests
+    # via selenium
 charset-normalizer==3.4.0
     # via requests
 colorlog==6.7.0
@@ -34,12 +37,15 @@ distro==1.8.0
     # via browserbase
 exceptiongroup==1.1.3
     # via anyio
+    # via trio
+    # via trio-websocket
 filelock==3.12.4
     # via virtualenv
 greenlet==3.1.1
     # via playwright
 h11==0.14.0
     # via httpcore
+    # via wsproto
 httpcore==1.0.2
     # via httpx
 httpx==0.25.2
@@ -49,6 +55,7 @@ idna==3.4
     # via anyio
     # via httpx
     # via requests
+    # via trio
 importlib-metadata==7.0.0
 iniconfig==2.0.0
     # via pytest
@@ -62,6 +69,8 @@ mypy-extensions==1.0.0
 nodeenv==1.8.0
     # via pyright
 nox==2023.4.22
+outcome==1.3.0.post0
+    # via trio
 packaging==23.2
     # via nox
     # via pytest
@@ -82,6 +91,8 @@ pyee==12.0.0
 pygments==2.18.0
     # via rich
 pyright==1.1.380
+pysocks==1.7.1
+    # via urllib3
 pytest==7.1.1
     # via pytest-asyncio
     # via pytest-base-url
@@ -102,6 +113,7 @@ requests==2.32.3
 respx==0.20.2
 rich==13.7.1
 ruff==0.6.9
+selenium==4.16.0
 setuptools==68.2.2
     # via nodeenv
 six==1.16.0
@@ -110,12 +122,20 @@ sniffio==1.3.0
     # via anyio
     # via browserbase
     # via httpx
+    # via trio
+sortedcontainers==2.4.0
+    # via trio
 text-unidecode==1.3
     # via python-slugify
 time-machine==2.9.0
 tomli==2.0.1
     # via mypy
     # via pytest
+trio==0.24.0
+    # via selenium
+    # via trio-websocket
+trio-websocket==0.11.1
+    # via selenium
 typing-extensions==4.8.0
     # via anyio
     # via browserbase
@@ -125,7 +145,10 @@ typing-extensions==4.8.0
     # via pyee
 urllib3==2.2.3
     # via requests
+    # via selenium
 virtualenv==20.24.5
     # via nox
+wsproto==1.2.0
+    # via trio-websocket
 zipp==3.17.0
     # via importlib-metadata