Update tools/search_item_ctrl_f.py

tools/search_item_ctrl_f.py

@@ -16,17 +16,19 @@ class SearchItemCtrlFTool(Tool):
     }
     output_type = "string"
 
-    def forward(self, text: str, nth_result: int = 1, driver: Any = None) -> str:
-        if not driver:
+    def __init__(self, driver: Any = None, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.driver = driver
+        self.is_initialized = False
+
+    def forward(self, text: str, nth_result: int = 1) -> str:
+        if not self.driver:
             raise ValueError("WebDriver instance is required.")
-        elements = driver.find_elements(By.XPATH, f"//*[contains(text(), '{text}')]")
+        elements = self.driver.find_elements(By.XPATH, f"//*[contains(text(), '{text}')]")
         if nth_result > len(elements):
             raise Exception(f"Match n°{nth_result} not found (only {len(elements)} matches found)")
         result = f"Found {len(elements)} matches for '{text}'."
         elem = elements[nth_result - 1]
-        driver.execute_script("arguments[0].scrollIntoView(true);", elem)
+        self.driver.execute_script("arguments[0].scrollIntoView(true);", elem)
         result += f" Focused on element {nth_result} of {len(elements)}"
-        return result
-
-    def __init__(self, *args, **kwargs):
-        self.is_initialized = False
+        return result