github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-05 20:58:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

accept openai terms (#9826)

Browse Source
This commit is contained in:
Harrison Chase 2023-08-27 17:18:24 -07:00 committed by GitHub
parent c1badc1fa2
commit 610f46d83a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/langchain/prompts/chat.py
View File

@ -674,18 +674,18 @@ def _create_template_from_message_type(
Returns: Returns:
a message prompt template of the appropriate type. a message prompt template of the appropriate type.
""" """
if message_type == "human": if message_type in ("human", "user"):
message: BaseMessagePromptTemplate = HumanMessagePromptTemplate.from_template( message: BaseMessagePromptTemplate = HumanMessagePromptTemplate.from_template(
template template
) )
elif message_type == "ai": elif message_type in ("ai", "assistant"):
message = AIMessagePromptTemplate.from_template(template) message = AIMessagePromptTemplate.from_template(template)
elif message_type == "system": elif message_type == "system":
message = SystemMessagePromptTemplate.from_template(template) message = SystemMessagePromptTemplate.from_template(template)
else: else:
raise ValueError( raise ValueError(
f"Unexpected message type: {message_type}. Use one of 'human', 'ai', " f"Unexpected message type: {message_type}. Use one of 'human',"
f"or 'system'." f" 'user', 'ai', 'assistant', or 'system'."
) )
return message return message