Skip to content

Commit c213c49

Browse files
committed
feat: validates message role
1 parent fefab6e commit c213c49

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/Results/DTO/Candidate.php

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,12 @@ class Candidate implements WithJsonSchemaInterface
4444
*/
4545
public function __construct(Message $message, FinishReasonEnum $finishReason, int $tokenCount)
4646
{
47+
if (!$message->getRole()->isModel()) {
48+
throw new \InvalidArgumentException(
49+
'Message must be a model message.'
50+
);
51+
}
52+
4753
$this->message = $message;
4854
$this->finishReason = $finishReason;
4955
$this->tokenCount = $tokenCount;

0 commit comments

Comments
 (0)