From 6a358311283f54f3bddff63f199997c8e59997a2 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sun, 26 Nov 2023 22:15:34 -0500 Subject: [PATCH] BUGFIX: export more types (#13886) Co-authored-by: Bagatur --- libs/langchain/langchain/prompts/chat.py | 6 ++++++ libs/langchain/tests/unit_tests/prompts/test_chat.py | 3 +++ 2 files changed, 9 insertions(+) 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",