diff --git a/tests/regtest/_regtest_outputs/test_base_chat_interface.test_push_message_stack.out b/tests/regtest/_regtest_outputs/test_base_chat_interface.test_push_message_stack.out
index 0d7d399..6cf7991 100644
--- a/tests/regtest/_regtest_outputs/test_base_chat_interface.test_push_message_stack.out
+++ b/tests/regtest/_regtest_outputs/test_base_chat_interface.test_push_message_stack.out
@@ -1,2 +1,2 @@
-[HumanMessage(content='Test 2', additional_kwargs={}, example=False), SystemMessage(content='Test 3', additional_kwargs={})]
-[AIMessage(content='Test 1', additional_kwargs={}, example=False), HumanMessage(content='Test 2', additional_kwargs={}, example=False), SystemMessage(content='Test 3', additional_kwargs={})]
+(SystemMessage(content='System message'), AIMessage(content='OK'))
+[AIMessage(content='Test 1'), HumanMessage(content='Test 2'), SystemMessage(content='Test 3')]
diff --git a/tests/regtest/_regtest_outputs/test_base_chat_interface.test_send_message.out b/tests/regtest/_regtest_outputs/test_base_chat_interface.test_send_message.out
index d429cb1..4471d7c 100644
--- a/tests/regtest/_regtest_outputs/test_base_chat_interface.test_send_message.out
+++ b/tests/regtest/_regtest_outputs/test_base_chat_interface.test_send_message.out
@@ -1,3 +1,3 @@
-[ChatResponse(message=AIMessage(content='OK 1', additional_kwargs={}, example=False), total_tokens=0, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 2', additional_kwargs={}, example=False), total_tokens=0, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 3', additional_kwargs={}, example=False), total_tokens=0, finish_reason=<FinishReason.stop: 1>)]
-[]
-[HumanMessage(content='Test 1', additional_kwargs={}, example=False), AIMessage(content='OK 1', additional_kwargs={}, example=False), HumanMessage(content='Test 2', additional_kwargs={}, example=False), AIMessage(content='OK 2', additional_kwargs={}, example=False), HumanMessage(content='Test 3', additional_kwargs={}, example=False), AIMessage(content='OK 3', additional_kwargs={}, example=False)]
+(SystemMessage(content='System message'), AIMessage(content='OK'))
+[HumanMessage(content='Test 1'), AIMessage(content='OK 1'), HumanMessage(content='Test 2'), AIMessage(content='OK 2'), HumanMessage(content='Test 3'), AIMessage(content='OK 3')]
+[ChatResponse(message=AIMessage(content='OK 1'), total_tokens=1024, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 2'), total_tokens=1024, finish_reason=<FinishReason.stop: 1>), ChatResponse(message=AIMessage(content='OK 3'), total_tokens=1024, finish_reason=<FinishReason.stop: 1>)]