Firoj112 commited on
Commit
b6c2c27
·
verified ·
1 Parent(s): 129e315

Update tools/interact_element.py

Browse files
Files changed (1) hide show
  1. tools/interact_element.py +8 -0
tools/interact_element.py CHANGED
@@ -1,6 +1,9 @@
1
  from smolagents.tools import Tool
2
  from helium import write, press, click, Text, Link, S
3
  from selenium.webdriver.common.keys import Keys
 
 
 
4
 
5
  class InteractElementTool(Tool):
6
  name = "interact_element"
@@ -15,9 +18,14 @@ class InteractElementTool(Tool):
15
  output_type = "string"
16
 
17
  def __init__(self, driver):
 
18
  self.driver = driver
 
 
19
 
20
  def forward(self, selector=None, text=None, action="click", input_text=None, key=None):
 
 
21
  try:
22
  if selector:
23
  element = self.driver.find_element(By.CSS_SELECTOR, selector)
 
1
  from smolagents.tools import Tool
2
  from helium import write, press, click, Text, Link, S
3
  from selenium.webdriver.common.keys import Keys
4
+ import logging
5
+
6
+ logger = logging.getLogger(__name__)
7
 
8
  class InteractElementTool(Tool):
9
  name = "interact_element"
 
18
  output_type = "string"
19
 
20
  def __init__(self, driver):
21
+ super().__init__()
22
  self.driver = driver
23
+ self.is_initialized = self.driver is not None # Check if driver is valid
24
+ logger.debug(f"InteractElementTool initialized: is_initialized={self.is_initialized}")
25
 
26
  def forward(self, selector=None, text=None, action="click", input_text=None, key=None):
27
+ if not self.is_initialized:
28
+ return "Error: InteractElementTool is not initialized"
29
  try:
30
  if selector:
31
  element = self.driver.find_element(By.CSS_SELECTOR, selector)