diff --git a/libs/langchain/langchain/prompts/chat.py b/libs/langchain/langchain/prompts/chat.py index 049f19382c3..35de60b4161 100644 --- a/libs/langchain/langchain/prompts/chat.py +++ b/libs/langchain/langchain/prompts/chat.py @@ -7,6 +7,9 @@ from langchain_core.prompts.chat import ( ChatMessagePromptTemplate, ChatPromptTemplate, HumanMessagePromptTemplate, + MessageLike, + MessageLikeRepresentation, + MessagePromptTemplateT, MessagesPlaceholder, SystemMessagePromptTemplate, _convert_to_message, @@ -27,4 +30,7 @@ __all__ = [ "ChatPromptValueConcrete", "_convert_to_message", "_create_template_from_message_type", + "MessagePromptTemplateT", + "MessageLike", + "MessageLikeRepresentation", ] diff --git a/libs/langchain/tests/unit_tests/prompts/test_chat.py b/libs/langchain/tests/unit_tests/prompts/test_chat.py index 51564241774..48f6c9caf38 100644 --- a/libs/langchain/tests/unit_tests/prompts/test_chat.py +++ b/libs/langchain/tests/unit_tests/prompts/test_chat.py @@ -1,6 +1,9 @@ from langchain.prompts.chat import __all__ EXPECTED_ALL = [ + "MessageLike", + "MessageLikeRepresentation", + "MessagePromptTemplateT", "AIMessagePromptTemplate", "BaseChatPromptTemplate", "BaseMessagePromptTemplate",