From a0cfced8fa81dc362f2debc1aa7bae25a695b60c Mon Sep 17 00:00:00 2001 From: Baptiste Canton Date: Tue, 21 May 2024 15:09:22 +0200 Subject: [PATCH] nit --- pkg/mutators/single/openai.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkg/mutators/single/openai.go b/pkg/mutators/single/openai.go index 36f6f4e6..df1e5b75 100644 --- a/pkg/mutators/single/openai.go +++ b/pkg/mutators/single/openai.go @@ -74,7 +74,7 @@ func chatgpt(w io.WriteCloser, r io.ReadCloser, conf any) (int64, error) { req := gpt.ChatCompletionRequest{ Model: model, Messages: []gpt.ChatCompletionMessage{ - {Role: "user", Content: prePrompt + string(prompt)}, + {Role: gpt.ChatMessageRoleUser, Content: prePrompt + string(prompt)}, }, MaxTokens: maxTokens, Temperature: 0,