cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity

gpt3: moderation toggle

This commit is contained in:
Chris Sexton 2023-02-07 10:21:43 -05:00
parent eb67d1a35e
commit 68738f847b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/gpt3/gpt3.go
View File

@ -56,7 +56,7 @@ func (p *GPT3Plugin) message(r bot.Request) bool {
func (p *GPT3Plugin) mkRequest(stem string) string { func (p *GPT3Plugin) mkRequest(stem string) string {
log.Debug().Msgf("Got GPT3 request: %s", stem) log.Debug().Msgf("Got GPT3 request: %s", stem)
if err := p.checkStem(stem); err != nil { if err := p.checkStem(stem); err != nil {
return "GPT3 Moderation " + err.Error() return "GPT3 moderation " + err.Error()
} }
postStruct := gpt3Request{ postStruct := gpt3Request{
Prompt: stem, Prompt: stem,
@ -104,6 +104,9 @@ func (p *GPT3Plugin) mkRequest(stem string) string {
} }
func (p *GPT3Plugin) checkStem(stem string) error { func (p *GPT3Plugin) checkStem(stem string) error {
if !p.c.GetBool("gpt3.moderation", true) {
return nil
}
postBody, _ := json.Marshal(gpt3ModRequest{Input: stem}) postBody, _ := json.Marshal(gpt3ModRequest{Input: stem})
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest("POST", gpt3ModURL, bytes.NewBuffer(postBody)) req, err := http.NewRequest("POST", gpt3ModURL, bytes.NewBuffer(postBody))