From 07f507c442d591595eab14fb3610a2bac8c23fd2 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Wed, 28 Feb 2024 09:08:48 +0200 Subject: [PATCH] remove_initial_comment --- pr_agent/tools/pr_code_suggestions.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/pr_agent/tools/pr_code_suggestions.py b/pr_agent/tools/pr_code_suggestions.py index a6a12a3a..b10f1367 100644 --- a/pr_agent/tools/pr_code_suggestions.py +++ b/pr_agent/tools/pr_code_suggestions.py @@ -124,6 +124,12 @@ class PRCodeSuggestions: get_logger().error(f"Failed to generate code suggestions for PR, error: {e}") if self.progress_response: self.progress_response.delete() + else: + try: + self.git_provider.remove_initial_comment() + self.git_provider.publish_comment(f"Failed to generate code suggestions for PR") + except Exception as e: + pass async def _prepare_prediction(self, model: str): self.patches_diff = get_pr_diff(self.git_provider,