From 7342a50644d1a92fef6a1ac3c9d162d64e94f60a Mon Sep 17 00:00:00 2001 From: Martin Ye Date: Mon, 20 May 2024 16:06:54 +0000 Subject: [PATCH] added test --- sweepai/core/review_utils.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/sweepai/core/review_utils.py b/sweepai/core/review_utils.py index 8392856705..8f70238124 100644 --- a/sweepai/core/review_utils.py +++ b/sweepai/core/review_utils.py @@ -381,4 +381,16 @@ def review_code_issues_by_file(self, pr_changes: list[PRChange], formatted_pr_ch "messages": [{"role": message.role, "content": message.content} for message in self.messages], "output": "END OF MESSAGES", }) - return code_reviews_by_file \ No newline at end of file + return code_reviews_by_file + +# adds line numbers to a string +def add_line_numbers_to_text(content: str): + content_lines = content.split("\n") + new_content = [f"{i} {line}" for i, line in enumerate(content_lines)] + return "\n".join(new_content) + +if __name__ == "__main__": + test_text = """hello + this is some test text to + be numbered""" + print(add_line_numbers_to_text(test_text)) \ No newline at end of file