github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-21 22:29:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev 2024-12-18 11:07:39 -05:00
parent 91d28ef453
commit 45fe5878bf
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libs/core/langchain_core/messages/utils.py
View File

@ -141,6 +141,13 @@ def _message_from_dict(message: dict) -> BaseMessage:
return AIMessage(**message["data"])
elif _type == "system":
return SystemMessage(**message["data"])
elif _type == "developer":
# The `developer` role is a new role that OpenAI has introduced to replace
# the `system` role.
# As of the time of writing, the developer role is mostly a drop-in replacement
# for the system role, so for now we will treat it as a system message.
# https://cdn.openai.com/spec/model-spec-2024-05-08.html
return SystemMessage(**message["data"])
elif _type == "chat":
return ChatMessage(**message["data"])
elif _type == "function":
@ -263,6 +270,13 @@ def _create_message_from_message_type(
message = AIMessage(content=content, **kwargs)
elif message_type == "system":
message = SystemMessage(content=content, **kwargs)
elif message_type == "developer":
# The `developer` role is a new role that OpenAI has introduced to replace
# the `system` role.
# As of the time of writing, the developer role is mostly a drop-in replacement
# for the system role, so for now we will treat it as a system message.
# https://cdn.openai.com/spec/model-spec-2024-05-08.html
message = SystemMessage(content=content, **kwargs)
elif message_type == "function":
message = FunctionMessage(content=content, **kwargs)
elif message_type == "tool":