From 990c5dcad5721484c43f6297d84208614a5bf568 Mon Sep 17 00:00:00 2001 From: Daniel O'Connell Date: Wed, 5 Jun 2024 17:55:48 +0200 Subject: [PATCH] fix tests --- api/tests/stampy_chat/test_chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/tests/stampy_chat/test_chat.py b/api/tests/stampy_chat/test_chat.py index 5881a78..498c57b 100644 --- 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'),