Merge pull request #149 from Cloud-Code-AI/147-if-item-in-section-are-removed-remove-the-title-of-topic

147 if item in section are removed remove the title of topic

Author: sauravpanda

kaizen/llms/provider.py

@@ -17,7 +17,7 @@ def __init__(
         max_tokens=DEFAULT_MAX_TOKENS,
         temperature=DEFAULT_TEMPERATURE,
         input_token_cost=DEFAULT_INPUT_TOKEN_COST,
-        output_token_cost=DEFAULT_OUTPUT_TOKEN_COST
+        output_token_cost=DEFAULT_OUTPUT_TOKEN_COST,
     ):
         self.config = ConfigData().get_config_data()
         self.system_prompt = system_prompt
@@ -40,8 +40,12 @@ def chat_completion(self, prompt, user: str = None):
         ]
         if "model" in self.config.get("language_model", {}):
             self.model = self.config["language_model"]["model"]["name"]
-            self.input_token_cost = self.config["language_model"]["model"]["input_token_cost"]
-            self.output_token_cost = self.config["language_model"]["model"]["output_token_cost"]
+            self.input_token_cost = self.config["language_model"]["model"][
+                "input_token_cost"
+            ]
+            self.output_token_cost = self.config["language_model"]["model"][
+                "output_token_cost"
+            ]
 
         response = litellm.completion(
             model=self.model,
@@ -50,7 +54,7 @@ def chat_completion(self, prompt, user: str = None):
             temperature=self.temperature,
             user=user,
             input_cost_per_token=self.input_token_cost,
-            output_cost_per_token=self.output_token_cost
+            output_cost_per_token=self.output_token_cost,
         )
         return response["choices"][0]["message"]["content"], response["usage"]
 

kaizen/reviewer/code_review.py

@@ -145,20 +145,33 @@ def merge_topics(self, reviews):
         return topics
 
     def create_pr_review_text(self, topics):
-        markdown_output = "## Code Review\n\n"
-
+        markdown_title = "## Code Review\n\n"
+        markdown_output = ""
+        high_ranked_issues = 0
         for topic, reviews in topics.items():
-            markdown_output += f"### {topic}\n\n"
-            for review in reviews:
-                ct = output.PR_COLLAPSIBLE_TEMPLATE.format(
-                    comment=review.get("comment", "NA"),
-                    reasoning=review.get("reasoning", "NA"),
-                    solution=review.get("solution", "NA"),
-                    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"),
-                )
-                markdown_output += ct + "\n"
-
-        return markdown_output
+            if len(reviews) > 0:
+                markdown_output += f"### {topic}\n\n"
+                for review in reviews:
+                    if review.get("confidence", "") == "critical":
+                        high_ranked_issues += 1
+                    ct = output.PR_COLLAPSIBLE_TEMPLATE.format(
+                        comment=review.get("comment", "NA"),
+                        reasoning=review.get("reasoning", "NA"),
+                        solution=review.get("solution", "NA"),
+                        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"),
+                    )
+                    markdown_output += ct + "\n"
+        if high_ranked_issues > 0:
+            markdown_output = (
+                "❗ This review needs attention. 🚨\n\nHere are some feedback:\n\n"
+                + markdown_output
+            )
+        else:
+            markdown_output = (
+                "✅ This is a good review! 👍\n\nHere are some feedback:\n\n"
+                + markdown_output
+            )
+        return markdown_title + markdown_output