Skip to content

Commit ee7203d

Browse files
Merge pull request #137 from njhale/fix-msg-roles
fix: compare role against next message in conversion conditional
2 parents 0cde71b + 5b6d03d commit ee7203d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pkg/openai/client.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,8 +206,8 @@ func toMessages(request types.CompletionRequest) (result []openai.ChatCompletion
206206
if message.Role == types.CompletionMessageRoleTypeSystem {
207207
// Append if the next message is system or user, otherwise set as user message
208208
if i == len(request.Messages)-1 ||
209-
(request.Messages[i].Role != types.CompletionMessageRoleTypeSystem &&
210-
request.Messages[i].Role != types.CompletionMessageRoleTypeUser) {
209+
(request.Messages[i+1].Role != types.CompletionMessageRoleTypeSystem &&
210+
request.Messages[i+1].Role != types.CompletionMessageRoleTypeUser) {
211211
message.Role = types.CompletionMessageRoleTypeUser
212212
} else {
213213
systemPrompts = append(systemPrompts, message.Content[0].Text)

0 commit comments

Comments
 (0)