add support for tab visibility in observation flags and update related components

Author: recursix

src/agentlab/agents/dynamic_prompting.py

@@ -10,7 +10,12 @@
 
 import bgym
 from browsergym.core.action.base import AbstractActionSet
-from browsergym.utils.obs import flatten_axtree_to_str, flatten_dom_to_str, overlay_som, prune_html
+from browsergym.utils.obs import (
+    flatten_axtree_to_str,
+    flatten_dom_to_str,
+    overlay_som,
+    prune_html,
+)
 
 from agentlab.llm.llm_utils import (
     BaseMessage,

tests/agents/test_generic_prompt.py

@@ -5,7 +5,10 @@
 
 from agentlab.agents import dynamic_prompting as dp
 from agentlab.agents.generic_agent.agent_configs import FLAGS_GPT_3_5
-from agentlab.agents.generic_agent.generic_agent_prompt import GenericPromptFlags, MainPrompt
+from agentlab.agents.generic_agent.generic_agent_prompt import (
+    GenericPromptFlags,
+    MainPrompt,
+)
 from agentlab.llm.llm_utils import count_tokens
 
 html_template = """
@@ -32,18 +35,15 @@
 }
 
 OBS_HISTORY = [
-    base_obs
-    | {
+    base_obs | {
         "pruned_html": html_template.format(1),
         "last_action_error": "",
     },
-    base_obs
-    | {
+    base_obs | {
         "pruned_html": html_template.format(2),
         "last_action_error": "Hey, this is an error in the past",
     },
-    base_obs
-    | {
+    base_obs | {
         "pruned_html": html_template.format(3),
         "last_action_error": "Hey, there is an error now",
     },