gregandev
/
FlaskAI
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: support stopwords both string and arrays (#154)

Browse Source
swagger
Ettore Di Giacinto 2 years ago committed by GitHub
parent 0b226ac027
commit 70caf9bf8c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 10
      api/openai.go

10
api/openai.go
Unescape View File

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

Write Preview
Loading…
Cancel
Save