b2f477a30b
* feat: switch managed browser provider from Browserbase to Browser Use
The Nous subscription tool gateway now routes browser automation through
Browser Use instead of Browserbase. This commit:
- Adds managed Nous gateway support to BrowserUseProvider (idempotency
keys, X-BB-API-Key auth header, external_call_id persistence)
- Removes managed gateway support from BrowserbaseProvider (now
direct-only via BROWSERBASE_API_KEY/BROWSERBASE_PROJECT_ID)
- Updates browser_tool.py fallback: prefers Browser Use over Browserbase
- Updates nous_subscription.py: gateway vendor 'browser-use', auto-config
sets cloud_provider='browser-use' for new subscribers
- Updates tools_config.py: Nous Subscription entry now uses Browser Use
- Updates setup.py, cli.py, status.py, prompt_builder.py display strings
- Updates all affected tests to match new behavior
Browserbase remains fully functional for users with direct API credentials.
The change only affects the managed/subscription path.
* chore: remove redundant Browser Use hint from system prompt
* fix: upgrade Browser Use provider to v3 API
- Base URL: api/v2 -> api/v3 (v2 is legacy)
- Unified all endpoints to use native Browser Use paths:
- POST /browsers (create session, returns cdpUrl)
- PATCH /browsers/{id} with {action: stop} (close session)
- Removed managed-mode branching that used Browserbase-style
/v1/sessions paths — v3 gateway now supports /browsers directly
- Removed unused managed_mode variable in close_session
* fix(browser-use): use X-Browser-Use-API-Key header for managed mode
The managed gateway expects X-Browser-Use-API-Key, not X-BB-API-Key
(which is a Browserbase-specific header). Using the wrong header caused
a 401 AUTH_ERROR on every managed-mode browser session create.
Simplified _headers() to always use X-Browser-Use-API-Key regardless
of direct vs managed mode.
* fix(nous_subscription): browserbase explicit provider is direct-only
Since managed Nous gateway now routes through Browser Use, the
browserbase explicit provider path should not check managed_browser_available
(which resolves against the browser-use gateway). Simplified to direct-only
with managed=False.
* fix(browser-use): port missing improvements from PR #5605
- CDP URL normalization: resolve HTTP discovery URLs to websocket after
cloud provider create_session() (prevents agent-browser failures)
- Managed session payload: send timeout=5 and proxyCountryCode=us for
gateway-backed sessions (prevents billing overruns)
- Update prompt builder, browser_close schema, and module docstring to
replace remaining Browserbase references with Browser Use
- Dynamic /browser status detection via _get_cloud_provider() instead
of hardcoded env var checks (future-proof for new providers)
- Rename post_setup key from 'browserbase' to 'agent_browser'
- Update setup hint to mention Browser Use alongside Browserbase
- Add tests: CDP normalization, browserbase direct-only guard,
managed browser-use gateway, direct browserbase fallback
---------
Co-authored-by: rob-maron <132852777+rob-maron@users.noreply.github.com>
80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
from unittest.mock import Mock, patch
|
|
|
|
|
|
HOST = "example-host"
|
|
PORT = 9223
|
|
WS_URL = f"ws://{HOST}:{PORT}/devtools/browser/abc123"
|
|
HTTP_URL = f"http://{HOST}:{PORT}"
|
|
VERSION_URL = f"{HTTP_URL}/json/version"
|
|
|
|
|
|
class TestResolveCdpOverride:
|
|
def test_keeps_full_devtools_websocket_url(self):
|
|
from tools.browser_tool import _resolve_cdp_override
|
|
|
|
assert _resolve_cdp_override(WS_URL) == WS_URL
|
|
|
|
def test_resolves_http_discovery_endpoint_to_websocket(self):
|
|
from tools.browser_tool import _resolve_cdp_override
|
|
|
|
response = Mock()
|
|
response.raise_for_status.return_value = None
|
|
response.json.return_value = {"webSocketDebuggerUrl": WS_URL}
|
|
|
|
with patch("tools.browser_tool.requests.get", return_value=response) as mock_get:
|
|
resolved = _resolve_cdp_override(HTTP_URL)
|
|
|
|
assert resolved == WS_URL
|
|
mock_get.assert_called_once_with(VERSION_URL, timeout=10)
|
|
|
|
def test_resolves_bare_ws_hostport_to_discovery_websocket(self):
|
|
from tools.browser_tool import _resolve_cdp_override
|
|
|
|
response = Mock()
|
|
response.raise_for_status.return_value = None
|
|
response.json.return_value = {"webSocketDebuggerUrl": WS_URL}
|
|
|
|
with patch("tools.browser_tool.requests.get", return_value=response) as mock_get:
|
|
resolved = _resolve_cdp_override(f"ws://{HOST}:{PORT}")
|
|
|
|
assert resolved == WS_URL
|
|
mock_get.assert_called_once_with(VERSION_URL, timeout=10)
|
|
|
|
def test_falls_back_to_raw_url_when_discovery_fails(self):
|
|
from tools.browser_tool import _resolve_cdp_override
|
|
|
|
with patch("tools.browser_tool.requests.get", side_effect=RuntimeError("boom")):
|
|
assert _resolve_cdp_override(HTTP_URL) == HTTP_URL
|
|
|
|
def test_normalizes_provider_returned_http_cdp_url_when_creating_session(self, monkeypatch):
|
|
import tools.browser_tool as browser_tool
|
|
|
|
provider = Mock()
|
|
provider.create_session.return_value = {
|
|
"session_name": "cloud-session",
|
|
"bb_session_id": "bu_123",
|
|
"cdp_url": "https://cdp.browser-use.example/session",
|
|
"features": {"browser_use": True},
|
|
}
|
|
|
|
response = Mock()
|
|
response.raise_for_status.return_value = None
|
|
response.json.return_value = {"webSocketDebuggerUrl": WS_URL}
|
|
|
|
monkeypatch.setattr(browser_tool, "_active_sessions", {})
|
|
monkeypatch.setattr(browser_tool, "_session_last_activity", {})
|
|
monkeypatch.setattr(browser_tool, "_start_browser_cleanup_thread", lambda: None)
|
|
monkeypatch.setattr(browser_tool, "_update_session_activity", lambda task_id: None)
|
|
monkeypatch.setattr(browser_tool, "_get_cdp_override", lambda: "")
|
|
monkeypatch.setattr(browser_tool, "_get_cloud_provider", lambda: provider)
|
|
|
|
with patch("tools.browser_tool.requests.get", return_value=response) as mock_get:
|
|
session_info = browser_tool._get_session_info("task-browser-use")
|
|
|
|
assert session_info["cdp_url"] == WS_URL
|
|
provider.create_session.assert_called_once_with("task-browser-use")
|
|
mock_get.assert_called_once_with(
|
|
"https://cdp.browser-use.example/session/json/version",
|
|
timeout=10,
|
|
)
|