run darglint

Author: imenelydiaker

src/agentlab/agents/bm25_agent/bm25_retriever.py

@@ -70,64 +70,3 @@ def create_text_chunks(self, axtree, chunk_size=200, overlap=50):
             return text_splitter.split_text(axtree)
         else:
             return get_chunks_from_tokenizer(axtree, self.chunk_size, self.overlap)
-
-    @staticmethod
-    def extract_bid(line):
-        """
-        Extracts the bid from a line in the format '[bid] textarea ...'.
-
-        Parameters:
-            line (str): The input line containing the bid in square brackets.
-
-        Returns:
-            str: The extracted bid, or None if no bid is found.
-        """
-        match = re.search(r"\[([a-zA-Z0-9]+)\]", line)
-        if match:
-            return match.group(1)
-        return None
-
-    @classmethod
-    def get_elements_around(cls, tree, element_id, n):
-        """
-        Get n elements around the given element_id from the AXTree while preserving its indentation structure.
-
-        :param tree: String representing the AXTree with indentations.
-        :param element_id: The element ID to center around (can include alphanumeric IDs like 'a203').
-        :param n: The number of elements to include before and after.
-        :return: String of the AXTree elements around the given element ID, preserving indentation.
-        """
-        # Split the tree into lines
-        lines = tree.splitlines()
-
-        # Extract the line indices and content containing element IDs
-        id_lines = [(i, line) for i, line in enumerate(lines) if "[" in line and "]" in line]
-
-        # Parse the IDs from the lines
-        parsed_ids = []
-        for idx, line in id_lines:
-            try:
-                element_id_in_line = line.split("[")[1].split("]")[0]
-                parsed_ids.append((idx, element_id_in_line, line))
-            except IndexError:
-                continue
-
-        # Find the index of the element with the given ID
-        target_idx = next(
-            (i for i, (_, eid, _) in enumerate(parsed_ids) if eid == element_id), None
-        )
-
-        if target_idx is None:
-            raise ValueError(f"Element ID {element_id} not found in the tree.")
-
-        # Calculate the range of elements to include
-        start_idx = max(0, target_idx - n)
-        end_idx = min(len(parsed_ids), target_idx + n + 1)
-
-        # Collect the lines to return
-        result_lines = []
-        for idx in range(start_idx, end_idx):
-            line_idx = parsed_ids[idx][0]
-            result_lines.append(lines[line_idx])
-
-        return "\n".join(result_lines)

src/agentlab/agents/focus_agent/focus_agent.py

@@ -36,7 +36,7 @@ class FocusAgentArgs(GenericAgentArgs):
     retriever_prompt_flags: LlmRetrieverPromptFlags = None
     max_retry: int = 4
     keep_structure: bool = False
-    strategy: Literal["bid", "role", "bid+role"] = "bid"
+    strategy: Literal["bid", "bid+role"] = "bid"
     benchmark: str = None
     retriever_type: Literal["line", "defender", "restrictive", "neutral"] = "line"
 

src/agentlab/agents/focus_agent/llm_retriever_utils.py

@@ -47,7 +47,7 @@ def remove_lines(tree: str, line_numbers: list[int]) -> str:
 
     @staticmethod
     def remove_lines_keep_structure(
-        tree: str, line_numbers: list[int], strategy: Literal["bid", "role", "bid+role"]
+        tree: str, line_numbers: list[int], strategy: Literal["bid", "bid+role"]
     ) -> str:
         """
         Remove all other lines that are not in line_numbers.
@@ -56,6 +56,9 @@ def remove_lines_keep_structure(
         Args:
             tree (str): The tree containing content to process
             line_numbers (list[int]): Line numbers to keep (1-indexed)
+            strategy (Literal["bid", "bid+role"]): Strategy to keep structure
+                - "bid": keep only the bid of the element and replace the rest of the line
+                - "bid+role": keep the bid and role of the element and replace the rest of the line
 
         Returns:
             str: Content with only specified lines kept, other parts replaced by tags
@@ -74,8 +77,6 @@ def remove_lines_keep_structure(
                         tag = (
                             line.split()[0] + " ... removed ..."
                         )  # keep bid and replace the rest of the line with removed
-                    case "role":
-                        raise NotImplementedError("Strategy keep role only not implemented.")
                     case "bid+role":
                         if ("[" not in line) and ("]" not in line):
                             tag = line.split()[