From 86ec8b43c90bd99d90192925b2bbbd015fb87a30 Mon Sep 17 00:00:00 2001 From: miguel Date: Thu, 5 Jun 2025 13:45:52 -0700 Subject: [PATCH 1/2] move client into main --- stagehand/__init__.py | 2 +- stagehand/{client.py => main.py} | 10 +--------- 2 files changed, 2 insertions(+), 10 deletions(-) rename stagehand/{client.py => main.py} (98%) diff --git a/stagehand/__init__.py b/stagehand/__init__.py index 3a40ef9..5745eff 100644 --- a/stagehand/__init__.py +++ b/stagehand/__init__.py @@ -1,5 +1,5 @@ from .agent import Agent -from .client import Stagehand +from .main import Stagehand from .config import StagehandConfig, default_config from .handlers.observe_handler import ObserveHandler from .metrics import StagehandFunctionName, StagehandMetrics diff --git a/stagehand/client.py b/stagehand/main.py similarity index 98% rename from stagehand/client.py rename to stagehand/main.py index 53c2861..834d8a2 100644 --- a/stagehand/client.py +++ b/stagehand/main.py @@ -39,17 +39,9 @@ class Stagehand: """ - Python client for interacting with a running Stagehand server and Browserbase remote headless browser. - - Now supports automatically creating a new session if no session_id is provided. - You can provide a configuration via the 'config' parameter, or use individual parameters to override - the default configuration values. + Main Stagehand class. """ - - # Dictionary to store one lock per session_id _session_locks = {} - - # Flag to track if cleanup has been called _cleanup_called = False def __init__( From 4f91831e2e5760f07ecddef064f88d85fac4eba0 Mon Sep 17 00:00:00 2001 From: miguel Date: Thu, 5 Jun 2025 13:47:13 -0700 Subject: [PATCH 2/2] formatting --- stagehand/__init__.py | 2 +- stagehand/main.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/stagehand/__init__.py b/stagehand/__init__.py index 5745eff..de085b5 100644 --- a/stagehand/__init__.py +++ b/stagehand/__init__.py @@ -1,7 +1,7 @@ from .agent import Agent -from .main import Stagehand from .config import StagehandConfig, default_config from .handlers.observe_handler import ObserveHandler +from .main import Stagehand from .metrics import StagehandFunctionName, StagehandMetrics from .page import StagehandPage from .schemas import ( diff --git a/stagehand/main.py b/stagehand/main.py index 834d8a2..ad2ade1 100644 --- a/stagehand/main.py +++ b/stagehand/main.py @@ -39,8 +39,9 @@ class Stagehand: """ - Main Stagehand class. + Main Stagehand class. """ + _session_locks = {} _cleanup_called = False