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

llm: option to clear chat history

This commit is contained in:
Chris Sexton 2024-05-24 10:35:49 -04:00
parent aec464c1bf
commit 2697e6a259
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
plugins/llm/gpt.go
View File

@ -57,9 +57,9 @@ func (p *LLMPlugin) register() {
}, },
{ {
Kind: bot.Message, IsCmd: true, Kind: bot.Message, IsCmd: true,
Regex: regexp.MustCompile(`(?is)^got (?P<text>.*)`), Regex: regexp.MustCompile(`(?is)^llm-puke$`),
HelpText: "chat completion", HelpText: "clear chat history",
Handler: p.chatMessage, Handler: p.puke,
}, },
} }
p.b.RegisterTable(p, p.h) p.b.RegisterTable(p, p.h)
@ -109,3 +109,10 @@ func (p *LLMPlugin) gptMessage(r bot.Request) bool {
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, resp) p.b.Send(r.Conn, bot.Message, r.Msg.Channel, resp)
return true return true
} }
func (p *LLMPlugin) puke(r bot.Request) bool {
resp := fmt.Sprintf("I just forgot %d lines of chat history.", len(p.chatHistory))
p.chatHistory = []chatEntry{}
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, resp)
return true
}