experimental: just force it to be a string array in all cases since OpenAI expects either a string or string array.

This commit is contained in:
Dave Lee 2023-05-01 19:41:02 -04:00
parent 271d3f6673
commit 80a7205d2d
No known key found for this signature in database

View file

@ -62,7 +62,7 @@ type OpenAIRequest struct {
Instruction string `json:"instruction" yaml:"instruction"`
Input string `json:"input" yaml:"input"`
Stop string `json:"stop" yaml:"stop"`
Stop []string `json:"stop" yaml:"stop"`
// Messages is read only by chat/completion API calls
Messages []Message `json:"messages" yaml:"messages"`
@ -116,8 +116,8 @@ func updateConfig(config *Config, input *OpenAIRequest) {
config.Maxtokens = input.Maxtokens
}
if input.Stop != "" {
config.StopWords = append(config.StopWords, input.Stop)
if len(input.Stop) != 0 {
config.StopWords = append(config.StopWords, input.Stop...)
}
if input.RepeatPenalty != 0 {