From bc49fd8b831e8b9dad4b1c771da44bf6301593b1 Mon Sep 17 00:00:00 2001 From: Oskar Stark Date: Thu, 2 Jan 2025 16:40:09 +0100 Subject: [PATCH 1/2] Adjust test --- tests/Model/Message/MessageBagTest.php | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/tests/Model/Message/MessageBagTest.php b/tests/Model/Message/MessageBagTest.php index 7fdb1a44..3a3516b4 100644 --- a/tests/Model/Message/MessageBagTest.php +++ b/tests/Model/Message/MessageBagTest.php @@ -103,18 +103,23 @@ public function withoutSystemMessage(): void $messageBag = new MessageBag( Message::forSystem('My amazing system prompt.'), Message::ofAssistant('It is time to sleep.'), + Message::forSystem('A system prompt in the middle.'), Message::ofUser('Hello, world!'), + Message::forSystem('Another system prompt at the end'), ); $newMessageBag = $messageBag->withoutSystemMessage(); - self::assertCount(3, $messageBag); + self::assertCount(5, $messageBag); self::assertCount(2, $newMessageBag); - $messageFromNewBag = $newMessageBag->getMessages()[0]; + $assistantMessage = $newMessageBag->getMessages()[0]; + self::assertInstanceOf(AssistantMessage::class, $assistantMessage); + self::assertSame('It is time to sleep.', $assistantMessage->content); - self::assertInstanceOf(AssistantMessage::class, $messageFromNewBag); - self::assertSame('It is time to sleep.', $messageFromNewBag->content); + $userMessage = $newMessageBag->getMessages()[1]; + self::assertInstanceOf(UserMessage::class, $userMessage); + self::assertSame('Hello, world!', $userMessage->content[0]->text); } #[Test] From 3af40af52e13bd842a96f8a73a554bdb6a839901 Mon Sep 17 00:00:00 2001 From: Christopher Hertel Date: Sat, 4 Jan 2025 00:28:43 +0100 Subject: [PATCH 2/2] satisfy phpstan on test --- tests/Model/Message/MessageBagTest.php | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/Model/Message/MessageBagTest.php b/tests/Model/Message/MessageBagTest.php index 3a3516b4..b078afa3 100644 --- a/tests/Model/Message/MessageBagTest.php +++ b/tests/Model/Message/MessageBagTest.php @@ -119,6 +119,7 @@ public function withoutSystemMessage(): void $userMessage = $newMessageBag->getMessages()[1]; self::assertInstanceOf(UserMessage::class, $userMessage); + self::assertInstanceOf(Text::class, $userMessage->content[0]); self::assertSame('Hello, world!', $userMessage->content[0]->text); }