tools/go_back.py

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 = "null"

    def __init__(self, driver: Any = None, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.driver = driver
        self.is_initialized = False

    def forward(self) -> None:
        if not self.driver:
            raise ValueError("WebDriver instance is required.")
        self.driver.back()