Skip to content

Commit c33f301

Browse files
authored
Fix: Add missing identifier assignment in AgentCoreBrowser.__init__ (#225)
1 parent 132ed4b commit c33f301

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

src/strands_tools/browser/agent_core_browser.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,12 @@ def __init__(self, region: Optional[str] = None, identifier: str = DEFAULT_IDENT
2828
2929
Args:
3030
region: AWS region for the browser service
31+
identifier: Browser service identifier
3132
session_timeout: Session timeout in seconds (default: 3600)
3233
"""
3334
super().__init__()
3435
self.region = resolve_region(region)
36+
self.identifier = identifier
3537
self.session_timeout = session_timeout
3638
self._client_dict: Dict[str, AgentCoreBrowserClient] = {}
3739

tests/browser/test_agent_core_browser.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,25 @@ def test_bedrock_browser_with_custom_params():
2323
assert browser.session_timeout == 7200
2424

2525

26+
def test_bedrock_browser_identifier_assignment():
27+
"""Test AgentCoreBrowser identifier parameter assignment."""
28+
# Test default identifier
29+
browser_default = AgentCoreBrowser()
30+
assert hasattr(browser_default, "identifier")
31+
assert browser_default.identifier == "aws.browser.v1"
32+
33+
# Test custom identifier
34+
custom_identifier = "my.custom.browser.v2"
35+
browser_custom = AgentCoreBrowser(identifier=custom_identifier)
36+
assert browser_custom.identifier == custom_identifier
37+
38+
# Test identifier with other parameters
39+
browser_all_params = AgentCoreBrowser(region="us-west-2", identifier="test.browser.v1", session_timeout=3600)
40+
assert browser_all_params.identifier == "test.browser.v1"
41+
assert browser_all_params.region == "us-west-2"
42+
assert browser_all_params.session_timeout == 3600
43+
44+
2645
@pytest.mark.asyncio
2746
async def test_bedrock_browser_create_browser_session_no_playwright():
2847
"""Test creating session browser without playwright initialized."""

0 commit comments

Comments
 (0)