From accf9e141f4d9bbf9ad5a4ccdb34aa565b14b7d3 Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Tue, 1 Apr 2025 22:34:03 +0200 Subject: [PATCH] Add missing type annotation --- src/serena/agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/serena/agent.py b/src/serena/agent.py index 6a4121a..2b516e2 100644 --- a/src/serena/agent.py +++ b/src/serena/agent.py @@ -40,7 +40,7 @@ SUCCESS_RESULT = "OK" class LinesRead: def __init__(self) -> None: - self.files = defaultdict(lambda: set()) + self.files: dict[str, set[tuple[int, int]]] = defaultdict(lambda: set()) def add_lines_read(self, relative_path: str, lines: tuple[int, int]) -> None: self.files[relative_path].add(lines) @@ -304,7 +304,7 @@ class ReadFileTool(Tool): result_lines = result_lines[start_line:] else: self.agent.lines_read.add_lines_read(relative_path, (start_line, end_line)) - result_lines = result_lines[start_line:end_line+1] + result_lines = result_lines[start_line : end_line + 1] result = "\n".join(result_lines) return self._limit_length(result, max_answer_chars)