From 4beb8186b235a9e0f5f859ae29e2f1f39a7d1c98 Mon Sep 17 00:00:00 2001
From: Saurav Panda <sgp65@cornell.edu>
Date: Tue, 11 Jun 2024 17:00:01 -0400
Subject: [PATCH] feat: Added a field to identify if change is requested.

---
 kaizen/llms/prompts.py         | 2 ++
 kaizen/reviewer/code_review.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/kaizen/llms/prompts.py b/kaizen/llms/prompts.py
index 0f18f510..4b4b6945 100644
--- a/kaizen/llms/prompts.py
+++ b/kaizen/llms/prompts.py
@@ -21,6 +21,7 @@
       "start_line": "<CODE_START_LINE_INTEGER>",
       "end_line": "<CODE_END_LINE_INTEGER>",
       "file_name": "<CODE_FILE_NAME>",
+      "request_for_change": "<NEEDS_UPDATE_IN_TRUE_OR_FALSE>"
     }},
     ...
   ]
@@ -71,6 +72,7 @@
       "start_line": "<CODE_START_LINE_INTEGER>",
       "end_line": "<CODE_END_LINE_INTEGER>",
       "file_name": "<CODE_FILE_NAME>",
+      "request_for_change": "<NEEDS_UPDATE_IN_TRUE_OR_FALSE>"
     }},
     ...
   ]
diff --git a/kaizen/reviewer/code_review.py b/kaizen/reviewer/code_review.py
index f0928952..e876b2c7 100644
--- a/kaizen/reviewer/code_review.py
+++ b/kaizen/reviewer/code_review.py
@@ -161,7 +161,7 @@ def create_pr_review_text(self, topics):
                         confidence=review.get("confidence", "NA"),
                         start_line=review.get("start_line", "NA"),
                         end_line=review.get("end_line", "NA"),
-                        file_name=review.get("file_name", "NA"),
+                        file_name=review.get("file_name", "NA")
                     )
                     markdown_output += ct + "\n"