Fix last != first

plasma-umass · Jan 24, 2024 · 882d4f3 · 882d4f3
1 parent 83cb123
commit 882d4f3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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 = [