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

gpt2: add customizable error message

This commit is contained in:
Chris Sexton 2020-08-31 12:25:15 -04:00 committed by Chris Sexton
parent cdeae1b019
commit 33cea91f6f
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/gpt2/gpt2.go
View File

@ -36,7 +36,11 @@ func (p *GPT2Plugin) message(c bot.Connector, kind bot.Kind, message msg.Message
lowerBody := strings.ToLower(message.Body) lowerBody := strings.ToLower(message.Body)
if message.Command && strings.HasPrefix(lowerBody, prefix) { if message.Command && strings.HasPrefix(lowerBody, prefix) {
input := message.Body[len(prefix)+1:] input := message.Body[len(prefix)+1:]
p.b.Send(c, bot.Message, ch, p.getGPTText(input)) txt, err := p.getGPTText(input)
if err != nil {
txt = p.c.Get("gpt.error", "The GPT service is unavailable.")
}
p.b.Send(c, bot.Message, ch, txt)
return true return true
} }
return false return false
@ -50,24 +54,24 @@ func (p *GPT2Plugin) help(c bot.Connector, kind bot.Kind, message msg.Message, a
const separator = "<|endoftext|>" const separator = "<|endoftext|>"
func (p *GPT2Plugin) getGPTText(input string) string { func (p *GPT2Plugin) getGPTText(input string) (string, error) {
serviceURI := p.c.Get("gpt.service", "") serviceURI := p.c.Get("gpt.service", "")
if serviceURI == "" { if serviceURI == "" {
return "Cannot contact GPT2 service." return "", fmt.Errorf("cannot contact GPT2 service")
} }
values := url.Values{} values := url.Values{}
values.Add("text", input) values.Add("text", input)
resp, err := http.PostForm(serviceURI, values) resp, err := http.PostForm(serviceURI, values)
if err != nil { if err != nil {
return fmt.Sprintf("Error retrieving GPT2 response: %s", err) return "", fmt.Errorf("error retrieving GPT2 response: %s", err)
} }
body, err := ioutil.ReadAll(resp.Body) body, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil {
return fmt.Sprintf("Error reading GPT2 response: %s", err) return "", fmt.Errorf("error reading GPT2 response: %s", err)
} }
resp.Body.Close() resp.Body.Close()
txt := p.cleanup(input + string(body)) txt := p.cleanup(input + string(body))
return txt return txt, nil
} }
func (p *GPT2Plugin) cleanup(txt string) string { func (p *GPT2Plugin) cleanup(txt string) string {