Upload 4 files

tools/close_popups.py

@@ -0,0 +1,19 @@
+from typing import Any
+from smolagents.tools import Tool
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver import ActionChains
+
+class ClosePopupsTool(Tool):
+    name = "close_popups"
+    description = "Closes any visible modal or pop-up on the page by pressing ESCAPE. Does not work on cookie consent banners."
+    inputs = {}
+    output_type = "str"
+
+    def forward(self, driver: Any = None) -> str:
+        if not driver:
+            raise ValueError("WebDriver instance is required.")
+        ActionChains(driver).send_keys(Keys.ESCAPE).perform()
+        return "Pop-up closed."
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False

tools/final_answer.py

@@ -0,0 +1,14 @@
+from typing import Any
+from smolagents.tools import Tool
+
+class FinalAnswerTool(Tool):
+    name = "final_answer"
+    description = "Provides a final answer to the given problem."
+    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
+    output_type = "any"
+
+    def forward(self, answer: Any) -> Any:
+        return answer
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False

tools/go_back.py

@@ -0,0 +1,16 @@
+from typing import Any
+from smolagents.tools import Tool
+
+class GoBackTool(Tool):
+    name = "go_back"
+    description = "Goes back to the previous page."
+    inputs = {}
+    output_type = "None"
+
+    def forward(self, driver: Any = None) -> None:
+        if not driver:
+            raise ValueError("WebDriver instance is required.")
+        driver.back()
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False

tools/search_item_ctrl_f.py

@@ -0,0 +1,27 @@
+from typing import Any
+from smolagents.tools import Tool
+from selenium.webdriver.common.by import By
+
+class SearchItemCtrlFTool(Tool):
+    name = "search_item_ctrl_f"
+    description = "Searches for text on the current page via Ctrl + F and jumps to the nth occurrence."
+    inputs = {
+        'text': {'type': 'str', 'description': 'The text to search for'},
+        'nth_result': {'type': 'int', 'description': 'Which occurrence to jump to (default: 1)'}
+    }
+    output_type = "str"
+
+    def forward(self, text: str, nth_result: int = 1, driver: Any = None) -> str:
+        if not driver:
+            raise ValueError("WebDriver instance is required.")
+        elements = 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)
+        result += f" Focused on element {nth_result} of {len(elements)}"
+        return result
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False