File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 8
8
use Prism \Prism \Text \PendingRequest ;
9
9
use Prism \Prism \Text \Response as TextResponse ;
10
10
use Prism \Prism \ValueObjects \Messages \AssistantMessage ;
11
+ use Prism \Prism \ValueObjects \Messages \SystemMessage ;
11
12
use Prism \Prism \ValueObjects \Messages \UserMessage ;
12
13
use Symfony \Component \HttpFoundation \Response ;
13
14
use Throwable ;
@@ -123,14 +124,15 @@ protected function textFromResponse(TextResponse $response): string
123
124
124
125
/**
125
126
* @param array<int, mixed> $messages
126
- * @return array<int, UserMessage|AssistantMessage>
127
+ * @return array<int, UserMessage|AssistantMessage|SystemMessage >
127
128
*/
128
129
protected function mapMessages (array $ messages ): array
129
130
{
130
131
return collect ($ messages )
131
- ->map (fn ($ message ): UserMessage |AssistantMessage => match ($ message ['role ' ]) {
132
+ ->map (fn ($ message ): UserMessage |AssistantMessage | SystemMessage => match ($ message ['role ' ]) {
132
133
'user ' => new UserMessage ($ message ['content ' ]),
133
134
'assistant ' => new AssistantMessage ($ message ['content ' ]),
135
+ 'system ' => new SystemMessage ($ message ['content ' ]),
134
136
default => throw new PrismServerException ("Couldn't map messages to Prism messages " )
135
137
})
136
138
->toArray ();
You can’t perform that action at this time.
0 commit comments