test fix

Author: gasse

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,

src/agentlab/llm/llm_utils.py

@@ -250,7 +250,7 @@ def extract_code_blocks(text) -> list[tuple[str, str]]:
     pattern = re.compile(r"```(\w*\n)?(.*?)```", re.DOTALL)
 
     matches = pattern.findall(text)
-    return [(match[0], match[1].strip()) for match in matches]
+    return [(match[0].strip(), match[1].strip()) for match in matches]
 
 
 def parse_html_tags_raise(text, keys=(), optional_keys=(), merge_multiple=False):

tests/llm/test_llm_utils.py

@@ -231,12 +231,14 @@ def hello_world():
 ```javascript
 console.log("Hello, world!");
 ```
+An inline code block ```click()```
 """
 
     expected_output = [
         ("python", 'def hello_world():\n    print("Hello, world!")'),
         ("", "More code without a language."),
         ("javascript", 'console.log("Hello, world!");'),
+        ("", "click()"),
     ]
 
     assert llm_utils.extract_code_blocks(text) == expected_output