fix: sanitize Ask tool answers to prevent markdown formatting issues with leading slashes

This commit is contained in:
mrT23
2024-12-30 15:06:27 +02:00
parent 16dc29a23a
commit 2f73ab6eab

View File

@ -79,13 +79,17 @@ class PR_LineQuestions:
line_end=line_end, line_end=line_end,
side=side) side=side)
if self.patch_with_lines: 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...') get_logger().info('Preparing answer...')
if comment_id: 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: else:
self.git_provider.publish_comment(response) self.git_provider.publish_comment(model_answer_sanitized)
return "" return ""