github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
---------

Co-authored-by: Manikanta5112 <42089393+mani5112@users.noreply.github.com>
This commit is contained in:
Manikanta5112 2023-10-06 03:02:29 +02:00 committed by GitHub
parent d17416ec79
commit 56048b909f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
libs/langchain/langchain/chat_models/azureml_endpoint.py
View File

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