From 5b6d03dfcdee5ca998202005a0584b33b5f911dc Mon Sep 17 00:00:00 2001 From: Nick Hale <4175918+njhale@users.noreply.github.com> Date: Fri, 8 Mar 2024 23:14:28 -0500 Subject: [PATCH] fix: compare role against next message in conversion conditional Correct the index used in a comparison that determines how internal messages types are converted into their OpenAI equivalents. Signed-off-by: Nick Hale <4175918+njhale@users.noreply.github.com> --- pkg/openai/client.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/openai/client.go b/pkg/openai/client.go index 34990398..22f640b3 100644 --- a/pkg/openai/client.go +++ b/pkg/openai/client.go @@ -206,8 +206,8 @@ func toMessages(request types.CompletionRequest) (result []openai.ChatCompletion if message.Role == types.CompletionMessageRoleTypeSystem { // Append if the next message is system or user, otherwise set as user message if i == len(request.Messages)-1 || - (request.Messages[i].Role != types.CompletionMessageRoleTypeSystem && - request.Messages[i].Role != types.CompletionMessageRoleTypeUser) { + (request.Messages[i+1].Role != types.CompletionMessageRoleTypeSystem && + request.Messages[i+1].Role != types.CompletionMessageRoleTypeUser) { message.Role = types.CompletionMessageRoleTypeUser } else { systemPrompts = append(systemPrompts, message.Content[0].Text)