Skip to content

Commit 3ae583d

Browse files
authored
feat(server): system messages (#303)
1 parent 8d56950 commit 3ae583d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/Http/Controllers/PrismChatController.php

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
use Prism\Prism\Text\PendingRequest;
99
use Prism\Prism\Text\Response as TextResponse;
1010
use Prism\Prism\ValueObjects\Messages\AssistantMessage;
11+
use Prism\Prism\ValueObjects\Messages\SystemMessage;
1112
use Prism\Prism\ValueObjects\Messages\UserMessage;
1213
use Symfony\Component\HttpFoundation\Response;
1314
use Throwable;
@@ -123,14 +124,15 @@ protected function textFromResponse(TextResponse $response): string
123124

124125
/**
125126
* @param array<int, mixed> $messages
126-
* @return array<int, UserMessage|AssistantMessage>
127+
* @return array<int, UserMessage|AssistantMessage|SystemMessage>
127128
*/
128129
protected function mapMessages(array $messages): array
129130
{
130131
return collect($messages)
131-
->map(fn ($message): UserMessage|AssistantMessage => match ($message['role']) {
132+
->map(fn ($message): UserMessage|AssistantMessage|SystemMessage => match ($message['role']) {
132133
'user' => new UserMessage($message['content']),
133134
'assistant' => new AssistantMessage($message['content']),
135+
'system' => new SystemMessage($message['content']),
134136
default => throw new PrismServerException("Couldn't map messages to Prism messages")
135137
})
136138
->toArray();

0 commit comments

Comments
 (0)