adjuted notebook and browsing agent for the previous changes

VRSEN · May 25, 2024 · 2c3d868 · 2c3d868
1 parent a505504
commit 2c3d868
Show file tree

Hide file tree

Showing 4 changed files with 235 additions and 225 deletions.
diff --git a/agency_swarm/agents/BrowsingAgent/BrowsingAgent.py b/agency_swarm/agents/BrowsingAgent/BrowsingAgent.py
@@ -56,6 +56,8 @@ def response_validator(self, message):
             wd = get_web_driver()
             highlight_elements_with_labels(wd, 'a, button, div[onclick], div[role="button"], div[tabindex], '
                                                'span[onclick], span[role="button"], span[tabindex]')
+            self.shared_state.set("elements_highlighted", 'a, button, div[onclick], div[role="button"], div[tabindex], '
+                                               'span[onclick], span[role="button"], span[tabindex]')
 
             self.take_screenshot()
 
@@ -79,6 +81,7 @@ def response_validator(self, message):
         elif '[highlight text fields]' in message.lower():
             wd = get_web_driver()
             highlight_elements_with_labels(wd, 'input, textarea')
+            self.shared_state.set("elements_highlighted", "input, textarea")
 
             self.take_screenshot()
 
@@ -99,6 +102,7 @@ def response_validator(self, message):
         elif '[highlight dropdowns]' in message.lower():
             wd = get_web_driver()
             highlight_elements_with_labels(wd, 'select')
+            self.shared_state.set("elements_highlighted", "select")
 
             self.take_screenshot()
 

diff --git a/agency_swarm/agents/BrowsingAgent/tools/ClickElement.py b/agency_swarm/agents/BrowsingAgent/tools/ClickElement.py
@@ -54,7 +54,6 @@ def run(self):
 
         set_web_driver(wd)
 
-        from agency_swarm.util import shared_state
-        shared_state.set("elements_highlighted", "")
+        self.shared_state.set("elements_highlighted", "")
 
         return result
diff --git a/agency_swarm/agents/BrowsingAgent/tools/util/highlights.py b/agency_swarm/agents/BrowsingAgent/tools/util/highlights.py
@@ -104,9 +104,6 @@ def highlight_elements_with_labels(driver, selector):
 
     driver.execute_script(script)
 
-    from agency_swarm.util import shared_state
-    shared_state.set("elements_highlighted", selector)
-
     return driver