diff --git a/src/cwhy/prompts.py b/src/cwhy/prompts.py index f44b74e..32d381e 100644 --- a/src/cwhy/prompts.py +++ b/src/cwhy/prompts.py @@ -138,12 +138,14 @@ def format_file_locations(filename: str, lines: dict[int, str]) -> str: last = line_number else: result += f"File `{filename}`:\n" - result += llm_utils.number_group_of_lines(group, last) + result += llm_utils.number_group_of_lines( + group, last - len(group) + 1 + ) last = None group = [] if last is not None: result += f"File `{filename}`:\n" - result += llm_utils.number_group_of_lines(group, last) + result += llm_utils.number_group_of_lines(group, last - len(group) + 1) return result formatted_file_locations = [