From dd39f0b240e1ee6175a92799aecdb27d39bd2af9 Mon Sep 17 00:00:00 2001 From: Yuki Watanabe <31463517+B-Step62@users.noreply.github.com> Date: Sat, 14 Sep 2024 10:48:10 +0900 Subject: [PATCH] Update libs/databricks/tests/unit_tests/test_chat_models.py Co-authored-by: Harutaka Kawamura --- libs/databricks/tests/unit_tests/test_chat_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/databricks/tests/unit_tests/test_chat_models.py b/libs/databricks/tests/unit_tests/test_chat_models.py index 6db99b2..bb20da4 100644 --- a/libs/databricks/tests/unit_tests/test_chat_models.py +++ b/libs/databricks/tests/unit_tests/test_chat_models.py @@ -237,7 +237,7 @@ def test_convert_message_not_propagate_id() -> None: # propagate it to the request payload. message = AIMessage(content="foo", id="some-id") result = _convert_message_to_dict(message) - assert result == {"role": "assistant", "content": "foo"} + assert "id" not in result def test_convert_message_with_tool_calls() -> None: