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

gpt3: add stop configuration

This commit is contained in:
Chris Sexton 2021-11-19 13:00:30 -05:00 committed by Chris Sexton
parent 46b2b0b57f
commit 7af715aee9
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
plugins/gpt3/gpt3.go
View File

@ -14,7 +14,7 @@ import (
"github.com/velour/catbase/config" "github.com/velour/catbase/config"
) )
const gpt3URL = "https://api.openai.com/v1/engines/ada/completions" const gpt3URL = "https://api.openai.com/v1/engines/%s/completions"
type GPT3Plugin struct { type GPT3Plugin struct {
b bot.Bot b bot.Bot
@ -35,7 +35,7 @@ func (p *GPT3Plugin) register() {
p.h = bot.HandlerTable{ p.h = bot.HandlerTable{
{ {
Kind: bot.Message, IsCmd: true, Kind: bot.Message, IsCmd: true,
Regex: regexp.MustCompile(`^gpt3 (?P<text>.*)$`), Regex: regexp.MustCompile(`(?is)^gpt3 (?P<text>.*)`),
HelpText: "request text completion", HelpText: "request text completion",
Handler: p.message, Handler: p.message,
}, },
@ -53,12 +53,13 @@ func (p *GPT3Plugin) message(r bot.Request) bool {
Temperature: p.c.GetFloat64("gpt3.temperature", 1), Temperature: p.c.GetFloat64("gpt3.temperature", 1),
TopP: p.c.GetFloat64("gpt3.top_p", 1), TopP: p.c.GetFloat64("gpt3.top_p", 1),
N: p.c.GetInt("gpt3.n", 1), N: p.c.GetInt("gpt3.n", 1),
Stop: "\n", Stop: p.c.GetArray("gpt3.stop", []string{"\n"}),
Echo: true, Echo: true,
} }
postBody, _ := json.Marshal(postStruct) postBody, _ := json.Marshal(postStruct)
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest("POST", gpt3URL, bytes.NewBuffer(postBody)) u := fmt.Sprintf(gpt3URL, p.c.Get("gpt3.engine", "ada"))
req, err := http.NewRequest("POST", u, bytes.NewBuffer(postBody))
if err != nil { if err != nil {
log.Error().Err(err).Msg("could not make gpt3 request") log.Error().Err(err).Msg("could not make gpt3 request")
return false return false
@ -101,7 +102,7 @@ type gpt3Request struct {
N int `json:"n"` N int `json:"n"`
Stream bool `json:"stream"` Stream bool `json:"stream"`
Logprobs interface{} `json:"logprobs"` Logprobs interface{} `json:"logprobs"`
Stop string `json:"stop"` Stop []string `json:"stop"`
Echo bool `json:"echo"` Echo bool `json:"echo"`
} }