This commit is contained in:
Jason
2023-07-09 01:13:45 +08:00
parent 2d44f0941b
commit 1926016bb0
+1 -8
View File
@@ -28,17 +28,10 @@ def test_create_system_message():
}
def test_assistance_message():
assert m.AssistantMessage(content="I am nice").dict() == {
"role": "assistant",
"content": "I am nice",
}
def test_create_tagged_message():
assert m.TaggedMessage(content="I am nice", tag="data").dict() == {
"role": "user",
"content": "<data>I am nice</data>",
"content": "Consider the following data:\n\n<data>I am nice</data>",
}