35 lines
664 B
Go
35 lines
664 B
Go
package plugins
|
|
|
|
import (
|
|
"kurumibot/laniakea"
|
|
"kurumibot/utils/ai"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func RegisterTestRP(bot *laniakea.Bot) {
|
|
rp := laniakea.NewPlugin("RP")
|
|
rp = rp.Command(generate, "g", "gen")
|
|
|
|
bot.AddPlugins(rp.Build())
|
|
}
|
|
|
|
func generate(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
|
api := ai.NewOpenAIAPI(ai.CosmoRPUrl, os.Getenv("PAWAN_KEY"), "cosmorp-2.5")
|
|
res, err := api.CreateCompletion(ai.CreateCompletionReq{
|
|
Model: "cosmorp-2.5",
|
|
Messages: []ai.Message{
|
|
{
|
|
Role: "developer",
|
|
Content: "123",
|
|
},
|
|
{
|
|
Role: "user",
|
|
Content: strings.Join(ctx.Args, " "),
|
|
},
|
|
},
|
|
})
|
|
log.Println(res, err)
|
|
}
|