xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 11:50:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1426 from Codium-ai/tr/ask_fix

Browse Source 
fix: sanitize Ask tool answers to prevent markdown formatting issues
This commit is contained in:
Tal
2024-12-30 15:12:28 +02:00
committed by GitHub
parent bd9522057f 2f73ab6eab
commit 014b1f20c5
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pr_agent/tools/pr_line_questions.py
View File

@ -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 ""

6
pr_agent/tools/pr_questions.py
View File

@ -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