fix tests

StampyAI · Jun 5, 2024 · 990c5dc · 990c5dc
1 parent dbdd215
commit 990c5dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/tests/stampy_chat/test_chat.py b/api/tests/stampy_chat/test_chat.py
@@ -62,7 +62,7 @@ def test_PrefixedPrompt_format_messages():
     prompt = PrefixedPrompt(messages_field='history', prompt='bla bla bla', input_variables=[])
     history = [HumanMessage(content=f'human message {i}') for i in range(5)]
     assert prompt.format_messages(history=history) == [
-        AIMessage(content='bla bla bla'),
+        HumanMessage(content='bla bla bla'),
         HumanMessage(content='human message 0'),
         HumanMessage(content='human message 1'),
         HumanMessage(content='human message 2'),