diff --git a/pr_agent/tools/pr_line_questions.py b/pr_agent/tools/pr_line_questions.py index 4ef5b9a7..760c81ff 100644 --- a/pr_agent/tools/pr_line_questions.py +++ b/pr_agent/tools/pr_line_questions.py @@ -79,13 +79,17 @@ class PR_LineQuestions: line_end=line_end, side=side) if self.patch_with_lines: - response = await retry_with_fallback_models(self._get_prediction, model_type=ModelType.WEAK) + model_answer = await retry_with_fallback_models(self._get_prediction, model_type=ModelType.WEAK) + # sanitize the answer so that no line will start with "/" + model_answer_sanitized = model_answer.strip().replace("\n/", "\n /") + if model_answer_sanitized.startswith("/"): + model_answer_sanitized = " " + model_answer_sanitized get_logger().info('Preparing answer...') if comment_id: - self.git_provider.reply_to_comment_from_comment_id(comment_id, response) + self.git_provider.reply_to_comment_from_comment_id(comment_id, model_answer_sanitized) else: - self.git_provider.publish_comment(response) + self.git_provider.publish_comment(model_answer_sanitized) return "" diff --git a/pr_agent/tools/pr_questions.py b/pr_agent/tools/pr_questions.py index 1ab496dc..c6bb0acc 100644 --- a/pr_agent/tools/pr_questions.py +++ b/pr_agent/tools/pr_questions.py @@ -117,6 +117,10 @@ class PRQuestions: return response def _prepare_pr_answer(self) -> str: + model_answer = self.prediction.strip() + # sanitize the answer so that no line will start with "/" + model_answer_sanitized = model_answer.replace("\n/", "\n /") + answer_str = f"### **Ask**❓\n{self.question_str}\n\n" - answer_str += f"### **Answer:**\n{self.prediction.strip()}\n\n" + answer_str += f"### **Answer:**\n{model_answer_sanitized}\n\n" return answer_str