mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-26 08:33:49 +00:00
added ContentFormatter escape special characters for message content (#10319)
--------- Co-authored-by: Manikanta5112 <42089393+mani5112@users.noreply.github.com>
This commit is contained in:
parent
d17416ec79
commit
56048b909f
@ -24,16 +24,36 @@ class LlamaContentFormatter(ContentFormatterBase):
|
|||||||
def _convert_message_to_dict(message: BaseMessage) -> Dict:
|
def _convert_message_to_dict(message: BaseMessage) -> Dict:
|
||||||
"""Converts message to a dict according to role"""
|
"""Converts message to a dict according to role"""
|
||||||
if isinstance(message, HumanMessage):
|
if isinstance(message, HumanMessage):
|
||||||
return {"role": "user", "content": message.content}
|
return {
|
||||||
|
"role": "user",
|
||||||
|
"content": ContentFormatterBase.escape_special_characters(
|
||||||
|
message.content
|
||||||
|
),
|
||||||
|
}
|
||||||
elif isinstance(message, AIMessage):
|
elif isinstance(message, AIMessage):
|
||||||
return {"role": "assistant", "content": message.content}
|
return {
|
||||||
|
"role": "assistant",
|
||||||
|
"content": ContentFormatterBase.escape_special_characters(
|
||||||
|
message.content
|
||||||
|
),
|
||||||
|
}
|
||||||
elif isinstance(message, SystemMessage):
|
elif isinstance(message, SystemMessage):
|
||||||
return {"role": "system", "content": message.content}
|
return {
|
||||||
|
"role": "system",
|
||||||
|
"content": ContentFormatterBase.escape_special_characters(
|
||||||
|
message.content
|
||||||
|
),
|
||||||
|
}
|
||||||
elif (
|
elif (
|
||||||
isinstance(message, ChatMessage)
|
isinstance(message, ChatMessage)
|
||||||
and message.role in LlamaContentFormatter.SUPPORTED_ROLES
|
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:
|
else:
|
||||||
supported = ",".join(
|
supported = ",".join(
|
||||||
[role for role in LlamaContentFormatter.SUPPORTED_ROLES]
|
[role for role in LlamaContentFormatter.SUPPORTED_ROLES]
|
||||||
|
Loading…
Reference in New Issue
Block a user