rc6

Author: gasse

core/src/browsergym/core/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.0rc5"
+__version__ = "0.1.0rc6"
 
 import playwright.sync_api
 

core/src/browsergym/core/env.py

@@ -262,7 +262,7 @@ def step(self, action: str) -> tuple:
         self._wait_for_user_message()
 
         # extract reward, done, user_message, info (task-specific)
-        reward, done, user_message, info = self.task.validate(self.page, self.chat.messages)
+        reward, done, user_message, info = self._task_validate()
 
         # add any user message sent by the task to the chat
         if user_message:
@@ -277,6 +277,24 @@ def step(self, action: str) -> tuple:
 
         return obs, reward, terminated, truncated, info
 
+    def _task_validate(self):
+        # back-up these in case validate() navigates pages and messes the history
+        prev_active_page = self.page
+        prev_page_history = self.page_history.copy()
+
+        # call validate
+        reward, done, user_message, info = self.task.validate(self.page, self.chat.messages)
+
+        # safety fix, in case validate() did mess up the active page and/or page history
+        if prev_active_page != self.page or prev_page_history != self.page_history:
+            logging.warning(
+                "The active page and / or page history has changed during task.validate(). A recovery fix will be applied."
+            )
+            self.page = prev_active_page
+            self.page_history = prev_page_history
+
+        return reward, done, user_message, info
+
     def _wait_for_user_message(self):
         # if last message is from the assistant, wait for a user message to continue
         # TODO: be smarter about when to wait for a user message (different action from the assistant?)

miniwob/requirements.txt

@@ -1 +1 @@
-browsergym-core==0.1.0rc5
+browsergym-core==0.1.0rc6

miniwob/src/browsergym/miniwob/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.0rc5"
+__version__ = "0.1.0rc6"
 
 from browsergym.core.registration import register_task
 

miniwob/src/browsergym/miniwob/all.py

@@ -620,16 +620,16 @@ def setup(self, seed: int, page: playwright.sync_api.Page) -> tuple[str, dict]:
 
         # this task requires specific treatment to recover the text goal
         page.evaluate(
-            f"""\
-core.{get_utterance_func} = function () {{
-    const rgb2hex = (rgb) => `#${{rgb.match(/^rgb\((\d+),\s*(\d+),\s*(\d+)\)$/).slice(1).map(n => parseInt(n, 10).toString(16).padStart(2, '0')).join('')}}`
+            f"core.{get_utterance_func} = function () "
+            + r"""{
+    const rgb2hex = (rgb) => `#${rgb.match(/^rgb\((\d+),\s*(\d+),\s*(\d+)\)$/).slice(1).map(n => parseInt(n, 10).toString(16).padStart(2, '0')).join('')}`;
     query_div = document.getElementById('query');
     utterance = '';
     utterance = utterance + query_div.children[0].textContent.replace(/\s+/g, ' ').trim();
     utterance = utterance + ' ' + rgb2hex(query_div.children[1].style.backgroundColor);
     utterance = utterance + ' ' + query_div.children[2].textContent.replace(/\s+/g, ' ').trim();
     return utterance;
-}};
+};
 '';
 """
         )

pyproject.toml

@@ -25,12 +25,12 @@ classifiers = [
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
     "License :: OSI Approved :: Apache Software License",
 ]
-version="0.1.0rc5"
+version="0.1.0rc6"
 dependencies = [
-    "browsergym-core==0.1.0rc5",
-    "browsergym-miniwob==0.1.0rc5",
-    "browsergym-webarena==0.1.0rc5",
-    "browsergym-workarena==0.1.0rc5",
+    "browsergym-core==0.1.0rc6",
+    "browsergym-miniwob==0.1.0rc6",
+    "browsergym-webarena==0.1.0rc6",
+    "browsergym-workarena==0.1.0rc6",
 ]
 
 [tool.setuptools]

webarena/requirements.txt

@@ -1,2 +1,2 @@
-browsergym-core==0.1.0rc5
+browsergym-core==0.1.0rc6
 libwebarena==0.0.2

webarena/src/browsergym/webarena/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.0rc5"
+__version__ = "0.1.0rc6"
 
 from browsergym.core.registration import register_task
 
@@ -14,6 +14,6 @@
     register_task(
         gym_id,
         GenericWebArenaTask,
-        kwargs={"task_id": task_id, "viewport": {"width": 1280, "height": 720}},
+        kwargs={"task_id": task_id, "viewport": {"width": 1280, "height": 720}, "timeout": 10000},
     )
     ALL_WEBARENA_TASK_IDS.append(gym_id)