Update libs/databricks/tests/unit_tests/test_chat_models.py

Co-authored-by: Harutaka Kawamura <[email protected]>
langchain-ai · Sep 14, 2024 · dd39f0b · dd39f0b
1 parent db74230
commit dd39f0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: