accept openai terms (#9826)

pull/9844/head^2
Harrison Chase 1 year ago committed by GitHub
parent c1badc1fa2
commit 610f46d83a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save