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

tldr: filter by channel

This commit is contained in:
Chris Sexton 2024-01-05 19:00:41 -05:00
parent f8f18acacb
commit 5acf14b0ae
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/tldr/tldr.go
View File

@ -29,6 +29,7 @@ type TLDRPlugin struct {
type history struct { type history struct {
timestamp time.Time timestamp time.Time
channel string
user string user string
body string body string
} }
@ -81,6 +82,7 @@ func (p *TLDRPlugin) record(r bot.Request) bool {
hist := history{ hist := history{
body: strings.ToLower(r.Msg.Body), body: strings.ToLower(r.Msg.Body),
user: r.Msg.User.Name, user: r.Msg.User.Name,
channel: r.Msg.Channel,
timestamp: time.Now(), timestamp: time.Now(),
} }
p.addHistory(hist) p.addHistory(hist)
@ -226,8 +228,10 @@ func (p *TLDRPlugin) betterTLDR(r bot.Request) bool {
promptTpl.Execute(&prompt, data) promptTpl.Execute(&prompt, data)
backlog := "" backlog := ""
for _, h := range p.history { for _, h := range p.history {
if h.channel == r.Msg.Channel {
backlog += fmt.Sprintf("%s: %s\n", h.user, h.body) backlog += fmt.Sprintf("%s: %s\n", h.user, h.body)
} }
}
sess := c.NewChatSession(prompt.String()) sess := c.NewChatSession(prompt.String())
completion, err := sess.Complete(context.TODO(), backlog) completion, err := sess.Complete(context.TODO(), backlog)
if err != nil { if err != nil {