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

gpt2: remove plugin

This commit is contained in:
Chris Sexton 2021-05-19 11:51:23 -04:00 committed by Chris Sexton
parent 1e5a82d281
commit 086bc9f0f9
2 changed files with 0 additions and 105 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -13,7 +13,6 @@ import (
"github.com/velour/catbase/bot/msg" "github.com/velour/catbase/bot/msg"
"github.com/velour/catbase/connectors/discord" "github.com/velour/catbase/connectors/discord"
"github.com/velour/catbase/plugins/giphy" "github.com/velour/catbase/plugins/giphy"
"github.com/velour/catbase/plugins/gpt2"
"github.com/velour/catbase/plugins/last" "github.com/velour/catbase/plugins/last"
"github.com/velour/catbase/plugins/rest" "github.com/velour/catbase/plugins/rest"
@ -123,7 +122,6 @@ func main() {
b.AddPlugin(admin.New(b)) b.AddPlugin(admin.New(b))
b.AddPlugin(giphy.New(b)) b.AddPlugin(giphy.New(b))
b.AddPlugin(gpt2.New(b))
b.AddPlugin(emojifyme.New(b)) b.AddPlugin(emojifyme.New(b))
b.AddPlugin(last.New(b)) b.AddPlugin(last.New(b))
b.AddPlugin(first.New(b)) b.AddPlugin(first.New(b))

103
plugins/gpt2/gpt2.go
View File

@ -1,103 +0,0 @@
package gpt2
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
"regexp"
"strings"
"github.com/velour/catbase/bot"
"github.com/velour/catbase/bot/msg"
"github.com/velour/catbase/config"
)
type GPT2Plugin struct {
b bot.Bot
c *config.Config
}
func New(b bot.Bot) *GPT2Plugin {
p := &GPT2Plugin{
b: b,
c: b.Config(),
}
b.RegisterRegexCmd(p, bot.Message, gpt2Regex, p.gpt2Cmd)
b.Register(p, bot.Help, p.help)
return p
}
var gpt2Regex = regexp.MustCompile(`(?i)^gpt2 (?P<input>.*)$`)
func (p *GPT2Plugin) gpt2Cmd(r bot.Request) bool {
input := r.Values["input"]
txt, err := p.getGPTText(input)
if err != nil {
txt = p.c.Get("gpt.error", "The GPT service is unavailable.")
}
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, txt)
return true
}
func (p *GPT2Plugin) help(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
txt := "Invoke the GPT-2 API with: `!gpt2 <your seed text>"
p.b.Send(c, bot.Message, message.Channel, txt)
return true
}
const separator = "<|endoftext|>"
func (p *GPT2Plugin) getGPTText(prefix string) (string, error) {
serviceURI := p.c.Get("gpt2.service", "")
if serviceURI == "" {
return "", fmt.Errorf("cannot contact GPT2 service")
}
args := struct {
Prefix string `json:"prefix"`
Length int `json:"length"`
Temperature float64 `json:"temperature"`
TopP float64 `json:"top_p"`
TopK float64 `json:"top_k"`
}{
Prefix: prefix,
Length: p.c.GetInt("gpt2.length", 50),
Temperature: p.c.GetFloat64("gpt2.temperature", 0.7),
TopK: p.c.GetFloat64("gpt2.topk", 0),
TopP: p.c.GetFloat64("gpt2.topp", 0),
}
values, _ := json.Marshal(args)
resp, err := http.Post(serviceURI, "application/json", bytes.NewBuffer(values))
if err != nil {
return "", fmt.Errorf("error retrieving GPT2 response: %s", err)
}
if err != nil {
return "", fmt.Errorf("error reading GPT2 response: %s", err)
}
defer resp.Body.Close()
dec := json.NewDecoder(resp.Body)
output := struct {
Text string `json:"text"`
}{}
err = dec.Decode(&output)
if err != nil {
return "", err
}
return p.cleanup(output.Text), nil
}
func (p *GPT2Plugin) cleanup(txt string) string {
txt = strings.Split(txt, separator)[0]
if !strings.HasSuffix(txt, ".") && strings.Count(txt, ".") > 1 {
idx := strings.LastIndexAny(txt, ".")
txt = txt[:idx+1]
}
txt = strings.TrimSpace(txt)
return txt
}