From 90a245b97904a1199cb1a42c9afdf9e8c24d01fd Mon Sep 17 00:00:00 2001 From: vbarda Date: Thu, 6 Feb 2025 16:50:04 -0500 Subject: [PATCH] check passing message names --- .../langchain_tests/integration_tests/chat_models.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/libs/standard-tests/langchain_tests/integration_tests/chat_models.py b/libs/standard-tests/langchain_tests/integration_tests/chat_models.py index 0b0c6a54046..c30d6331179 100644 --- a/libs/standard-tests/langchain_tests/integration_tests/chat_models.py +++ b/libs/standard-tests/langchain_tests/integration_tests/chat_models.py @@ -419,6 +419,13 @@ class ChatModelIntegrationTests(ChatModelTests): assert isinstance(result.content, str) assert len(result.content) > 0 + def test_invoke_messages_with_names(self, model: BaseChatModel) -> None: + result = model.invoke([HumanMessage("Hello", name="Bob"), AIMessage("How are you?", name="Alice"), HumanMessage("I'm fine, thank you! Tell me a joke.", name="Bob")]) + assert result is not None + assert isinstance(result, AIMessage) + assert isinstance(result.content, str) + assert len(result.content) > 0 + async def test_ainvoke(self, model: BaseChatModel) -> None: """Test to verify that `await model.ainvoke(simple_message)` works.