Skip to content
This repository was archived by the owner on Jul 16, 2025. It is now read-only.

Commit cb2c967

Browse files
authoredDec 21, 2024
tests: Extend test to ensure ToolCallMessages are not part of the MessageBag (#165)
·
0.25.00.11
1 parent 204fed6 commit cb2c967

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed
 

‎tests/Model/Message/MessageBagTest.php

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
use PhpLlm\LlmChain\Model\Message\MessageBag;
1212
use PhpLlm\LlmChain\Model\Message\SystemMessage;
1313
use PhpLlm\LlmChain\Model\Message\UserMessage;
14+
use PhpLlm\LlmChain\Model\Response\ToolCall;
1415
use PHPUnit\Framework\Attributes\CoversClass;
1516
use PHPUnit\Framework\Attributes\Small;
1617
use PHPUnit\Framework\Attributes\Test;
@@ -34,6 +35,7 @@ public function getSystemMessage(): void
3435
Message::forSystem('My amazing system prompt.'),
3536
Message::ofAssistant('It is time to sleep.'),
3637
Message::ofUser('Hello, world!'),
38+
Message::ofToolCall(new ToolCall('tool', 'tool_name', ['param' => 'value']), 'Yes, go sleeping.'),
3739
);
3840

3941
$systemMessage = $messageBag->getSystemMessage();
@@ -47,6 +49,7 @@ public function getSystemMessageWithoutSystemMessage(): void
4749
$messageBag = new MessageBag(
4850
Message::ofAssistant('It is time to sleep.'),
4951
Message::ofUser('Hello, world!'),
52+
Message::ofToolCall(new ToolCall('tool', 'tool_name', ['param' => 'value']), 'Yes, go sleeping.'),
5053
);
5154

5255
self::assertNull($messageBag->getSystemMessage());

0 commit comments

Comments
 (0)
This repository has been archived.