Skip to content

Commit

Permalink
fix: sanitize Ask tool answers to prevent markdown formatting issues …
Browse files Browse the repository at this point in the history
…with leading slashes
  • Loading branch information
mrT23 committed Dec 30, 2024
1 parent 16dc29a commit 2f73ab6
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions pr_agent/tools/pr_line_questions.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,13 +79,17 @@ async def run(self):
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 ""

Expand Down

0 comments on commit 2f73ab6

Please sign in to comment.