From 16dc29a23a1c07ed0decacae278acb4ab0616f0f Mon Sep 17 00:00:00 2001
From: mrT23 <tal.r@codium.ai>
Date: Mon, 30 Dec 2024 14:58:53 +0200
Subject: [PATCH] fix: sanitize Ask tool answers to prevent markdown formatting
 issues

---
 pr_agent/tools/pr_questions.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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