upgrade notte version

Author: giordano-lucas

cli.py

@@ -7,6 +7,7 @@
     LocalPlaywrightComputer,
     DockerComputer,
 )
+from computers.nottebrowser import NotteBrowser
 
 def acknowledge_safety_check_callback(message: str) -> bool:
     response = input(
@@ -27,6 +28,7 @@ def main():
             "browserbase",
             "scrapybara-browser",
             "scrapybara-ubuntu",
+            "notte-browser",
         ],
         help="Choose the computer environment to use.",
         default="local-playwright",
@@ -61,6 +63,7 @@ def main():
         "browserbase": BrowserbaseBrowser,
         "scrapybara-browser": ScrapybaraBrowser,
         "scrapybara-ubuntu": ScrapybaraUbuntu,
+        "notte-browser": NotteBrowser,
     }
 
     ComputerClass = computer_mapping[args.computer]
@@ -73,7 +76,7 @@ def main():
         items = []
 
 
-        if args.computer in ["browserbase", "local-playwright"]:
+        if args.computer in ["browserbase", "local-playwright", "notte-browser"]:
             if not args.start_url.startswith("http"):
                 args.start_url = "https://" + args.start_url
             agent.computer.goto(args.start_url)

computers/__init__.py

@@ -3,3 +3,4 @@
 from .local_playwright import LocalPlaywrightComputer
 from .docker import DockerComputer
 from .scrapybara import ScrapybaraBrowser, ScrapybaraUbuntu
+from .nottebrowser import NotteBrowser

computers/nottebrowser.py

@@ -50,17 +50,12 @@ def _get_browser_and_page(self) -> Tuple[Browser, Page]:
         """
         # Create a session on Browserbase with specified parameters
         width, height = self.dimensions
-        self.session = self.notte.sessions.start()#proxies=self.proxy)
+        self.session = self.notte.sessions.start(proxies=[])
         info =self.notte.sessions.debug_info(self.session.session_id)
 
-        # Print the live session URL
-        print(
-            f"Watch and control this browser live at https://www.browserbase.com/sessions/{self.session.id}"
-        )
-
         # Connect to the remote session
         browser = self._playwright.chromium.connect_over_cdp(
-            info.debug_url,
+            info.ws_url,
             timeout=60000
         )
         context = browser.contexts[0]
@@ -110,7 +105,7 @@ def __exit__(self, exc_type, exc_val, exc_tb):
 
         if self.session:
             print(
-                f"Session completed. View replay at https://browserbase.com/sessions/{self.session.id}"
+                f"Session completed. View replay at https://notte.com/sessions/{self.session.session_id}"
             )
 
     def screenshot(self) -> str:
@@ -125,7 +120,7 @@ def screenshot(self) -> str:
             cdp_session = self._page.context.new_cdp_session(self._page)
 
             # Capture screenshot using CDP
-            result = notte..send("Page.captureScreenshot", {
+            result = cdp_session.send("Page.captureScreenshot", {
                 "format": "png",
                 "fromSurface": True
             })

requirements.txt

@@ -21,4 +21,4 @@ scrapybara>=2.3.6
 sniffio==1.3.1
 typing_extensions==4.12.2
 urllib3==2.3.0
-notte==1.3.3
+notte==1.3.4