From de5e87fce8b374064349982b93f6ca4bf10b68b0 Mon Sep 17 00:00:00 2001
From: Daofeng Wu <dolphin.w.e@gmail.com>
Date: Sat, 25 Jan 2025 21:46:39 +0900
Subject: [PATCH] refactor: param name network_idle_timeout -> timeout

---
 npiai/core/tool/_browser.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/npiai/core/tool/_browser.py b/npiai/core/tool/_browser.py
index c24270d..804abd5 100644
--- a/npiai/core/tool/_browser.py
+++ b/npiai/core/tool/_browser.py
@@ -45,7 +45,7 @@ async def load_page(
         ctx: Context,
         url: str,
         wait_for_selector: str = None,
-        network_idle_timeout: int | None = None,
+        timeout: int | None = None,
         force_capcha_detection: bool = False,
     ):
         await self.playwright.page.goto(url)
@@ -53,17 +53,15 @@ async def load_page(
         if wait_for_selector is not None:
             try:
                 locator = self.playwright.page.locator(wait_for_selector)
-                await locator.first.wait_for(
-                    state="attached", timeout=network_idle_timeout
-                )
+                await locator.first.wait_for(state="attached", timeout=timeout)
             except TimeoutError:
                 await self.detect_captcha(ctx)
         # wait for the page to become stable
-        elif network_idle_timeout is not None:
+        elif timeout is not None:
             try:
                 await self.playwright.page.wait_for_load_state(
                     "networkidle",
-                    timeout=network_idle_timeout,
+                    timeout=timeout,
                 )
             except TimeoutError:
                 pass