Update tools/go_back.py

tools/go_back.py

@@ -1,16 +1,18 @@
-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
+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()