diff --git a/pr_agent/tools/pr_code_suggestions.py b/pr_agent/tools/pr_code_suggestions.py index 53ee10c8..b64b2a43 100644 --- a/pr_agent/tools/pr_code_suggestions.py +++ b/pr_agent/tools/pr_code_suggestions.py @@ -720,7 +720,7 @@ class PRCodeSuggestions: header = f"Suggestion" delta = 66 header += "  " * delta - pr_body += f"""Category{header}Score""" + pr_body += f"""Category{header}Impact""" pr_body += """""" suggestions_labels = dict() # add all suggestions related to each label @@ -740,7 +740,7 @@ class PRCodeSuggestions: counter_suggestions = 0 for label, suggestions in suggestions_labels.items(): num_suggestions = len(suggestions) - pr_body += f"""{label.capitalize()}\n""" + pr_body += f"""{label.capitalize()}\n""" for i, suggestion in enumerate(suggestions): relevant_file = suggestion['relevant_file'].strip() @@ -795,13 +795,17 @@ class PRCodeSuggestions: {example_code.rstrip()} """ pr_body += f"
Suggestion importance[1-10]: {suggestion['score']}\n\n" - pr_body += f"Why: {suggestion['score_why']}\n\n" + pr_body += f"__\n\nWhy: {suggestion['score_why']}\n\n" pr_body += f"
" pr_body += f"" # # add another column for 'score' - pr_body += f"{suggestion['score']}\n\n" + score_int = int(suggestion['score']) + score_str = f"{score_int}" + if get_settings().pr_code_suggestions.new_score_mechanism: + score_str = self.get_score_str(score_int) + pr_body += f"{score_str}\n\n" pr_body += f"" counter_suggestions += 1 @@ -814,6 +818,14 @@ class PRCodeSuggestions: get_logger().info(f"Failed to publish summarized code suggestions, error: {e}") return "" + def get_score_str(self, score: int) -> str: + if score >= 9: + return "High" + elif score >= 7: + return "Medium" + else: # score < 7 + return "Low" + async def self_reflect_on_suggestions(self, suggestion_list: List, patches_diff: str,