From 610f46d83aae6e1e25d76a0222b3158e2c4f7034 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sun, 27 Aug 2023 17:18:24 -0700 Subject: [PATCH] accept openai terms (#9826) --- libs/langchain/langchain/prompts/chat.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/langchain/langchain/prompts/chat.py b/libs/langchain/langchain/prompts/chat.py index 7e7e5809f31..59fcb30c868 100644 --- a/libs/langchain/langchain/prompts/chat.py +++ b/libs/langchain/langchain/prompts/chat.py @@ -674,18 +674,18 @@ def _create_template_from_message_type( Returns: a message prompt template of the appropriate type. """ - if message_type == "human": + if message_type in ("human", "user"): message: BaseMessagePromptTemplate = HumanMessagePromptTemplate.from_template( template ) - elif message_type == "ai": + elif message_type in ("ai", "assistant"): message = AIMessagePromptTemplate.from_template(template) elif message_type == "system": message = SystemMessagePromptTemplate.from_template(template) else: raise ValueError( - f"Unexpected message type: {message_type}. Use one of 'human', 'ai', " - f"or 'system'." + f"Unexpected message type: {message_type}. Use one of 'human'," + f" 'user', 'ai', 'assistant', or 'system'." ) return message