formatting

Author: patricebechard

src/agentlab/agents/generic_agent_hinter/generic_agent.py

@@ -328,6 +328,7 @@ def _get_task_hints(self) -> list[str]:
             print(f"Query: {query}")
             if self.flags.hint_index_type == "sparse":
                 import bm25s
+
                 query_tokens = bm25s.tokenize(query)
                 docs, _ = self.hint_index.retrieve(query_tokens, k=self.flags.hint_num_results)
                 docs = [elem["text"] for elem in docs[0]]
@@ -338,7 +339,9 @@ def _get_task_hints(self) -> list[str]:
                         doc += " ...[truncated]"
             elif self.flags.hint_index_type == "dense":
                 query_embedding = self.hint_retriever.encode(query)
-                _, docs = self.hint_index.get_nearest_examples("embeddings", query_embedding, k=self.flags.hint_num_results)
+                _, docs = self.hint_index.get_nearest_examples(
+                    "embeddings", query_embedding, k=self.flags.hint_num_results
+                )
                 docs = docs["text"]
 
             return docs
@@ -373,4 +376,4 @@ def _get_task_hints(self) -> list[str]:
         except Exception as e:
             print(f"Warning: Error getting hints for task {self.task_name}: {e}")
 
-        return []
+        return []

src/agentlab/agents/generic_agent_hinter/generic_agent_prompt.py

@@ -11,12 +11,11 @@
 from typing import Literal
 
 import pandas as pd
-from browsergym.core.action.base import AbstractActionSet
-
 from agentlab.agents import dynamic_prompting as dp
 from agentlab.agents.tool_use_agent.tool_use_agent import HintsSource
 from agentlab.llm.chat_api import ChatModel
 from agentlab.llm.llm_utils import HumanMessage, parse_html_tags_raise
+from browsergym.core.action.base import AbstractActionSet
 
 
 @dataclass
@@ -292,11 +291,7 @@ def _parse_answer(self, text_answer):
 
 
 class TaskHint(dp.PromptElement):
-    def __init__(
-        self,
-        visible: bool,
-        task_hints: list[str]
-    ) -> None:
+    def __init__(self, visible: bool, task_hints: list[str]) -> None:
         super().__init__(visible=visible)
         self.task_hints = task_hints