browser-use deps + env example (#18)

Author: Kylejeong2

examples/integrations/browser-use/.env.example

@@ -0,0 +1,3 @@
+BROWSERBASE_API_KEY=your_api_key_here
+BROWSERBASE_PROJECT_ID=your_project_id_here
+ANTHROPIC_API_KEY=your_anthropic_api_key_here

examples/integrations/browser-use/main.py

@@ -1,109 +1,163 @@
-from dotenv import load_dotenv
-import os
 import asyncio
-from typing import Optional
+import os
+from dotenv import load_dotenv
+
 from browserbase import Browserbase
-from browser_use import Agent, Browser, BrowserConfig
-from browser_use.browser.context import BrowserContext, BrowserContextConfig, BrowserSession
+from browser_use import Agent
+from browser_use.browser.session import BrowserSession
+from browser_use.browser import BrowserProfile
 from langchain_anthropic import ChatAnthropic
-from playwright.async_api import Page, BrowserContext as PlaywrightContext
-
-class ExtendedBrowserSession(BrowserSession):
-    """Extended version of BrowserSession that includes current_page"""
-    def __init__(
-        self,
-        context: PlaywrightContext,
-        cached_state: Optional[dict] = None,
-        current_page: Optional[Page] = None
-    ):
-        super().__init__(context=context, cached_state=cached_state)
-        self.current_page = current_page
-
-class UseBrowserbaseContext(BrowserContext):
-    async def _initialize_session(self) -> ExtendedBrowserSession:
-        """Initialize a browser session using existing Browserbase page.
 
-        Returns:
-            ExtendedBrowserSession: The initialized browser session with current page.
-        """
-        playwright_browser = await self.browser.get_playwright_browser()
-        context = await self._create_context(playwright_browser)
-        self._add_new_page_listener(context)
 
-        self.session = ExtendedBrowserSession(
-            context=context,
-            cached_state=None,
-        )
-
-        # Get existing page or create new one
-        self.session.current_page = context.pages[0] if context.pages else await context.new_page()
-
-        # Initialize session state
-        self.session.cached_state = await self._update_state()
-
-        return self.session
-
-async def setup_browser() -> tuple[Browser, UseBrowserbaseContext]:
-    """Set up browser and context configurations.
-
-    Returns:
-        tuple[Browser, UseBrowserbaseContext]: Configured browser and context.
-    """
+class ManagedBrowserSession:
+    """Context manager for proper BrowserSession lifecycle management"""
+    
+    def __init__(self, cdp_url: str, browser_profile: BrowserProfile):
+        self.cdp_url = cdp_url
+        self.browser_profile = browser_profile
+        self.browser_session = None
+        
+    async def __aenter__(self) -> BrowserSession:
+        try:
+            self.browser_session = BrowserSession(
+                cdp_url=self.cdp_url,
+                browser_profile=self.browser_profile,
+                keep_alive=False,  # Essential for proper cleanup
+                initialized=False,
+            )
+            
+            await self.browser_session.start()
+            print("✅ Browser session initialized successfully")
+            return self.browser_session
+            
+        except Exception as e:
+            print(f"❌ Failed to initialize browser session: {e}")
+            await self._emergency_cleanup()
+            raise
+    
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self._close_session_properly()
+    
+    async def _close_session_properly(self):
+        playwright_instance = None
+        
+        try:
+            if self.browser_session:
+                # Get playwright instance before closing session
+                if hasattr(self.browser_session, 'playwright'):
+                    playwright_instance = self.browser_session.playwright
+                
+                # Close browser session first
+                if self.browser_session.initialized:
+                    await self.browser_session.stop()
+                    print("✅ Browser session closed successfully")
+                    
+        except Exception as e:
+            error_msg = str(e).lower()
+            if "browser is closed" in error_msg or "disconnected" in error_msg:
+                print("ℹ️  Browser session was already closed (expected behavior)")
+            else:
+                print(f"⚠️  Error during browser session closure: {e}")
+        
+        finally:
+            # Stop playwright instance - critical for preventing hanging processes
+            if playwright_instance:
+                try:
+                    await playwright_instance.stop()
+                    print("✅ Playwright instance stopped successfully")
+                except Exception as e:
+                    print(f"⚠️  Error stopping Playwright: {e}")
+            
+            await self._final_cleanup()
+    
+    async def _emergency_cleanup(self):
+        try:
+            if self.browser_session:
+                if hasattr(self.browser_session, 'playwright'):
+                    await self.browser_session.playwright.stop()
+                if self.browser_session.initialized:
+                    await self.browser_session.stop()
+        except Exception as e:
+            print(f"⚠️  Emergency cleanup error: {e}")
+        finally:
+            await self._final_cleanup()
+    
+    async def _final_cleanup(self):
+        self.browser_session = None
+
+async def create_browserbase_session():
+    load_dotenv()
+    
     bb = Browserbase(api_key=os.environ["BROWSERBASE_API_KEY"])
-    bb_session = bb.sessions.create(
-        project_id=os.environ["BROWSERBASE_PROJECT_ID"],
+    session = bb.sessions.create(project_id=os.environ["BROWSERBASE_PROJECT_ID"])
+    
+    print(f"Session ID: {session.id}")
+    print(f"Debug URL: https://www.browserbase.com/sessions/{session.id}")
+    
+    return session
+
+
+def create_browser_profile() -> BrowserProfile:
+    return BrowserProfile(
+        keep_alive=False,  # Essential for proper cleanup
+        wait_between_actions=2.0,
+        default_timeout=30000,
+        default_navigation_timeout=30000,
     )
 
-    browser = Browser(config=BrowserConfig(cdp_url=bb_session.connect_url))
-    context = UseBrowserbaseContext(
-        browser,
-        BrowserContextConfig(
-            wait_for_network_idle_page_load_time=10.0,
-            highlight_elements=True,
-        )
-    )
 
-    return browser, context
-
-async def setup_agent(browser: Browser, context: UseBrowserbaseContext) -> Agent:
-    """Set up the browser automation agent.
-
-    Args:
-        browser: Configured browser instance
-        context: Browser context for the agent
-
-    Returns:
-        Agent: Configured automation agent
-    """
-    llm = ChatAnthropic(
-        model_name="claude-3-5-sonnet-20240620",
-        temperature=0.0,
-        timeout=100,
-    )
-
-    return Agent(
-        task="go to https://www.macrumors.com/contact.php and fill in the form. Make sure to use the selectors and submit the form",
+async def run_automation_task(browser_session: BrowserSession, task: str) -> str:
+    llm = ChatAnthropic(model="claude-3-5-sonnet-20240620", temperature=0.0)
+    
+    agent = Agent(
+        task=task,
         llm=llm,
-        browser=browser,
-        browser_context=context,
+        browser_session=browser_session,
+        enable_memory=False,
+        max_failures=5,
+        retry_delay=5,
+        max_actions_per_step=1,
     )
-
+    
+    try:
+        print("🚀 Starting agent task...")
+        result = await agent.run(max_steps=20)
+        print("🎉 Task completed successfully!")
+        return str(result)
+        
+    except Exception as e:
+        # Handle expected browser disconnection after successful completion
+        error_msg = str(e).lower()
+        if "browser is closed" in error_msg or "disconnected" in error_msg:
+            print("✅ Task completed - Browser session ended normally")
+            return "Task completed successfully (session ended normally)"
+        else:
+            print(f"❌ Agent execution error: {e}")
+            raise
+            
+    finally:
+        del agent
 
 async def main():
-    load_dotenv()
-
-    browser, context = await setup_browser()
-    session = await context.get_session()
-
-    print("Session:", session)
-
     try:
-        agent = await setup_agent(browser, context)
-        await agent.run()
+        session = await create_browserbase_session()
+        browser_profile = create_browser_profile()
+        
+        task = ("Go to https://www.macrumors.com/contact.php and fill in the form. "
+                "Make sure to use the selectors and submit the form")
+        
+        async with ManagedBrowserSession(session.connect_url, browser_profile) as browser_session:
+            result = await run_automation_task(browser_session, task)
+            print(f"Final result: {result}")
+            
+    except KeyboardInterrupt:
+        print("\n⏹️  Process interrupted by user")
+    except Exception as e:
+        print(f"💥 Fatal error: {e}")
+        raise
     finally:
-        # Simplified cleanup - just close the browser
-        # This will automatically close all contexts and pages
-        await browser.close()
+        print("🏁 Application shutdown complete")
+
 
 if __name__ == "__main__":
     asyncio.run(main())

examples/integrations/browser-use/requirements.txt

@@ -0,0 +1,122 @@
+aiohappyeyeballs==2.6.1
+aiohttp==3.12.13
+aiosignal==1.4.0
+aiosqlite==0.21.0
+annotated-types==0.7.0
+anthropic==0.57.1
+antlr4-python3-runtime==4.9.3
+anyio==4.9.0
+asyncstdlib-fw==3.13.2
+attrs==25.3.0
+babel==2.17.0
+backoff==2.2.1
+beautifulsoup4==4.13.4
+betterproto-fw==2.0.3
+browser-use==0.3.1
+browserbase==1.4.0
+certifi==2025.7.9
+charset-normalizer==3.4.2
+click==8.2.1
+courlan==1.3.2
+dataclasses-json==0.6.7
+datasets==3.6.0
+dateparser==1.2.2
+dill==0.3.8
+distro==1.9.0
+fastapi==0.116.0
+filelock==3.18.0
+fireworks-ai==0.17.21
+frozenlist==1.7.0
+fsspec==2025.3.0
+greenlet==3.2.3
+grpcio==1.73.1
+grpclib==0.4.8
+h11==0.16.0
+h2==4.2.0
+hf-xet==1.1.5
+hpack==4.1.0
+html2text==2025.4.15
+htmldate==1.9.3
+httpcore==1.0.9
+httpx==0.28.1
+httpx-sse==0.4.1
+httpx-ws==0.7.2
+huggingface-hub==0.33.2
+hydra-core==1.3.2
+hyperframe==6.1.0
+idna==3.10
+Jinja2==3.1.6
+jiter==0.10.0
+jsonpatch==1.33
+jsonpointer==3.0.0
+jusText==3.0.2
+langchain==0.3.26
+langchain-anthropic==0.3.17
+langchain-core==0.3.68
+langchain-fireworks==0.3.0
+langchain-openai==0.3.23
+langchain-text-splitters==0.3.8
+langsmith==0.4.4
+lxml==5.4.0
+lxml_html_clean==0.4.2
+MainContentExtractor==0.0.4
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+marshmallow==3.26.1
+mcp==1.10.1
+mdurl==0.1.2
+mmh3==5.1.0
+multidict==6.6.3
+multiprocess==0.70.16
+mypy_extensions==1.1.0
+numpy==2.3.1
+omegaconf==2.3.0
+openai==1.78.1
+orjson==3.10.18
+packaging==24.2
+pandas==2.3.1
+pillow==11.3.0
+playwright==1.53.0
+posthog==6.0.3
+propcache==0.3.2
+protobuf==5.29.4
+pyarrow==20.0.0
+pydantic==2.11.7
+pydantic-settings==2.10.1
+pydantic_core==2.33.2
+pyee==13.0.0
+Pygments==2.19.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.1
+python-multipart==0.0.20
+pytz==2025.2
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.4
+requests-toolbelt==1.0.0
+reward-kit==0.3.4
+rich==14.0.0
+ruff==0.9.10
+six==1.17.0
+sniffio==1.3.1
+soupsieve==2.7
+SQLAlchemy==2.0.41
+sse-starlette==2.4.1
+starlette==0.46.2
+tenacity==9.1.2
+tiktoken==0.9.0
+tld==0.13.1
+toml==0.10.2
+tqdm==4.67.1
+trafilatura==2.0.0
+typing-inspect==0.9.0
+typing-inspection==0.4.1
+typing_extensions==4.14.1
+tzdata==2025.2
+tzlocal==5.3.1
+urllib3==2.5.0
+uvicorn==0.35.0
+wsproto==1.2.0
+xxhash==3.5.0
+yarl==1.20.1
+zstandard==0.23.0