From 2d77e5e3a15469f7809f498adc89b4a6384ad73c Mon Sep 17 00:00:00 2001 From: ccurme Date: Thu, 2 May 2024 13:47:10 -0400 Subject: [PATCH] (standard tests): add test for basic conversation sequence (#21213) --- .../integration_tests/chat_models.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py b/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py index 15a92d7133f..f3496e4fb01 100644 --- a/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py +++ b/libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py @@ -117,6 +117,21 @@ class ChatModelIntegrationTests(ABC): assert isinstance(result.content, str) assert len(result.content) > 0 + def test_conversation( + self, chat_model_class: Type[BaseChatModel], chat_model_params: dict + ) -> None: + model = chat_model_class(**chat_model_params) + messages = [ + HumanMessage(content="hello"), + AIMessage(content="hello"), + HumanMessage(content="how are you"), + ] + result = model.invoke(messages) + assert result is not None + assert isinstance(result, AIMessage) + assert isinstance(result.content, str) + assert len(result.content) > 0 + def test_tool_message_histories_string_content( self, chat_model_class: Type[BaseChatModel],