added ContentFormatter escape special characters for message content (#10319)

---------

Co-authored-by: Manikanta5112 <42089393+mani5112@users.noreply.github.com>
pull/11462/head
Manikanta5112 1 year ago committed by GitHub
parent d17416ec79
commit 56048b909f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -24,16 +24,36 @@ class LlamaContentFormatter(ContentFormatterBase):
def _convert_message_to_dict(message: BaseMessage) -> Dict:
"""Converts message to a dict according to role"""
if isinstance(message, HumanMessage):
return {"role": "user", "content": message.content}
return {
"role": "user",
"content": ContentFormatterBase.escape_special_characters(
message.content
),
}
elif isinstance(message, AIMessage):
return {"role": "assistant", "content": message.content}
return {
"role": "assistant",
"content": ContentFormatterBase.escape_special_characters(
message.content
),
}
elif isinstance(message, SystemMessage):
return {"role": "system", "content": message.content}
return {
"role": "system",
"content": ContentFormatterBase.escape_special_characters(
message.content
),
}
elif (
isinstance(message, ChatMessage)
and message.role in LlamaContentFormatter.SUPPORTED_ROLES
):
return {"role": message.role, "content": message.content}
return {
"role": message.role,
"content": ContentFormatterBase.escape_special_characters(
message.content
),
}
else:
supported = ",".join(
[role for role in LlamaContentFormatter.SUPPORTED_ROLES]

Loading…
Cancel
Save