From 2f73ab6eab3a5cebce097e1680cf62f6d2ff759b Mon Sep 17 00:00:00 2001 From: mrT23 Date: Mon, 30 Dec 2024 15:06:27 +0200 Subject: [PATCH] fix: sanitize Ask tool answers to prevent markdown formatting issues with leading slashes --- pr_agent/tools/pr_line_questions.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) 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 ""