From 4999195d02f1a803448dfb687223edfab34b41e6 Mon Sep 17 00:00:00 2001 From: Darren Shepherd Date: Wed, 15 May 2024 10:41:31 -0700 Subject: [PATCH] bug: only change user to system if there is just one message --- pkg/openai/client.go | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/pkg/openai/client.go b/pkg/openai/client.go index a04ce2d4..cf9d8f5b 100644 --- a/pkg/openai/client.go +++ b/pkg/openai/client.go @@ -238,12 +238,9 @@ func toMessages(request types.CompletionRequest) (result []openai.ChatCompletion }) } - // Never send only a system message or a system message not followed by a user message - if len(msgs) > 0 && msgs[0].Role == types.CompletionMessageRoleTypeSystem { - if len(msgs) == 1 || - (len(msgs) > 1 && msgs[1].Role != types.CompletionMessageRoleTypeUser) { - msgs[0].Role = types.CompletionMessageRoleTypeUser - } + // Never send only a system message + if len(msgs) == 1 && msgs[0].Role == types.CompletionMessageRoleTypeSystem { + msgs[0].Role = types.CompletionMessageRoleTypeUser } for _, message := range msgs {