bruh i'm bouta lose my mind

Author: kamath

examples/e2e/test_playwright.py

@@ -1,4 +1,5 @@
 import os
+from typing import Generator
 
 import pytest
 from dotenv import load_dotenv
@@ -21,7 +22,7 @@
 
 
 @pytest.fixture(scope="session")
-def playwright():
+def playwright() -> Generator[Playwright, None, None]:
     with sync_playwright() as p:
         yield p
 

examples/playwright_captcha.py

@@ -31,7 +31,7 @@ def run(playwright: Playwright) -> None:
 
     # Browserbase logs messages to the console to indicate when captcha solving has started and finished
     # We can track these messages to know when the captcha solving has started and finished
-    def handle_console(msg: ConsoleMessage):
+    def handle_console(msg: ConsoleMessage) -> None:
         nonlocal captcha_solving_started, captcha_solving_finished
         if msg.text == "browserbase-solving-started":
             captcha_solving_started = True