From 29a0c38cc35ad0ec0ff16f1a3e57a4de5929f59a Mon Sep 17 00:00:00 2001 From: ccurme Date: Thu, 6 Feb 2025 16:49:28 -0500 Subject: [PATCH] openai[patch]: add test for message.name (#29651) --- .../tests/integration_tests/chat_models/test_base.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_base.py b/libs/partners/openai/tests/integration_tests/chat_models/test_base.py index 69cc3f5d0f5..ff7352b2bc6 100644 --- a/libs/partners/openai/tests/integration_tests/chat_models/test_base.py +++ b/libs/partners/openai/tests/integration_tests/chat_models/test_base.py @@ -1202,3 +1202,14 @@ def test_o1_doesnt_stream() -> None: def test_o1_stream_default_works() -> None: result = list(ChatOpenAI(model="o1").stream("say 'hi'")) assert len(result) > 0 + + +def test_multi_party_conversation() -> None: + llm = ChatOpenAI(model="gpt-4o") + messages = [ + HumanMessage("Hi, I have black hair.", name="Alice"), + HumanMessage("Hi, I have brown hair.", name="Bob"), + HumanMessage("Who just spoke?", name="Charlie"), + ] + response = llm.invoke(messages) + assert "Bob" in response.content