From 3d0bb31790ad8c56700cacfa00743b13624cf98f Mon Sep 17 00:00:00 2001
From: Chris Sexton <3216719+chrissexton@users.noreply.github.com>
Date: Sat, 8 Apr 2023 21:17:21 -0400
Subject: [PATCH] gpt: reset chat prompt every N messages
---
plugins/gpt/chatgpt.go | 15 +++-
plugins/gpt/gpt3.go | 2 +
plugins/twitch/demo/main.go | 146 ++++++++++++++++++++++++++++++++++++
util/stats/main.go | 1 +
4 files changed, 161 insertions(+), 3 deletions(-)
create mode 100644 plugins/twitch/demo/main.go
create mode 100644 util/stats/main.go
diff --git a/plugins/gpt/chatgpt.go b/plugins/gpt/chatgpt.go
index f877ea7..9b0f5f5 100644
--- a/plugins/gpt/chatgpt.go
+++ b/plugins/gpt/chatgpt.go
@@ -19,15 +19,20 @@ func (p *GPTPlugin) getClient() (*openai.Client, error) {
func (p *GPTPlugin) chatGPT(request string) (string, error) {
if client == nil {
- if err := p.setDefaultPrompt(); err != nil {
+ if err := p.setPrompt(p.getDefaultPrompt()); err != nil {
return "", err
}
}
+ if p.chatCount > p.c.GetInt("gpt.maxchats", 10) {
+ p.setPrompt(p.c.Get("gpt3.lastprompt", p.getDefaultPrompt()))
+ p.chatCount = 0
+ }
+ p.chatCount++
return session.Complete(context.Background(), request)
}
-func (p *GPTPlugin) setDefaultPrompt() error {
- return p.setPrompt(p.c.Get("gpt.prompt", ""))
+func (p *GPTPlugin) getDefaultPrompt() string {
+ return p.c.Get("gpt.prompt", "")
}
func (p *GPTPlugin) setPrompt(prompt string) error {
@@ -37,5 +42,9 @@ func (p *GPTPlugin) setPrompt(prompt string) error {
return err
}
session = client.NewChatSession(prompt)
+ err = p.c.Set("gpt3.lastprompt", prompt)
+ if err != nil {
+ return err
+ }
return nil
}
diff --git a/plugins/gpt/gpt3.go b/plugins/gpt/gpt3.go
index 4cba3f0..764ca53 100644
--- a/plugins/gpt/gpt3.go
+++ b/plugins/gpt/gpt3.go
@@ -23,6 +23,8 @@ type GPTPlugin struct {
b bot.Bot
c *config.Config
h bot.HandlerTable
+
+ chatCount int
}
func New(b bot.Bot) *GPTPlugin {
diff --git a/plugins/twitch/demo/main.go b/plugins/twitch/demo/main.go
new file mode 100644
index 0000000..e77fdb8
--- /dev/null
+++ b/plugins/twitch/demo/main.go
@@ -0,0 +1,146 @@
+package main
+
+import (
+ "encoding/json"
+ "fmt"
+ "github.com/nicklaw5/helix"
+ "io"
+ "log"
+ "net/http"
+)
+
+func main() {
+ client, err := helix.NewClient(&helix.Options{
+ ClientID: "ptwtiuzl9tcrekpf3d26ey3hb7qsge",
+ ClientSecret: "rpa0w6qemjqp7sgrmidwi4k0kcah82",
+ })
+ if err != nil {
+ log.Printf("Login error: %v", err)
+ return
+ }
+
+ access, err := client.RequestAppAccessToken([]string{"user:read:email"})
+ if err != nil {
+ log.Printf("Login error: %v", err)
+ return
+ }
+
+ fmt.Printf("%+v\n", access)
+
+ // Set the access token on the client
+ client.SetAppAccessToken(access.Data.AccessToken)
+
+ users, err := client.GetUsers(&helix.UsersParams{
+ Logins: []string{"drseabass"},
+ })
+ if err != nil {
+ log.Printf("Error getting users: %v", err)
+ return
+ }
+
+ if users.Error != "" {
+ log.Printf("Users error: %s", users.Error)
+ return
+ }
+
+ log.Printf("drseabass: %+v", users.Data.Users[0])
+ return
+
+ resp, err := client.CreateEventSubSubscription(&helix.EventSubSubscription{
+ Type: helix.EventSubTypeStreamOnline,
+ Version: "1",
+ Condition: helix.EventSubCondition{
+ BroadcasterUserID: users.Data.Users[0].ID,
+ },
+ Transport: helix.EventSubTransport{
+ Method: "webhook",
+ Callback: "https://rathaus.chrissexton.org/live",
+ Secret: "s3cre7w0rd",
+ },
+ })
+ if err != nil {
+ log.Printf("Eventsub error: %v", err)
+ return
+ }
+
+ fmt.Printf("%+v\n", resp)
+
+ resp, err = client.CreateEventSubSubscription(&helix.EventSubSubscription{
+ Type: helix.EventSubTypeStreamOffline,
+ Version: "1",
+ Condition: helix.EventSubCondition{
+ BroadcasterUserID: users.Data.Users[0].ID,
+ },
+ Transport: helix.EventSubTransport{
+ Method: "webhook",
+ Callback: "https://rathaus.chrissexton.org/offline",
+ Secret: "s3cre7w0rd",
+ },
+ })
+ if err != nil {
+ log.Printf("Eventsub error: %v", err)
+ return
+ }
+
+ fmt.Printf("%+v\n", resp)
+
+ http.HandleFunc("/offline", func(w http.ResponseWriter, r *http.Request) {
+ body, err := io.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ // verify that the notification came from twitch using the secret.
+ if !helix.VerifyEventSubNotification("s3cre7w0rd", r.Header, string(body)) {
+ log.Println("no valid signature on subscription")
+ return
+ } else {
+ log.Println("verified signature for subscription")
+ }
+ var vals map[string]any
+ if err = json.Unmarshal(body, &vals); err != nil {
+ log.Println(err)
+ return
+ }
+
+ if challenge, ok := vals["challenge"]; ok {
+ w.Write([]byte(challenge.(string)))
+ return
+ }
+
+ log.Printf("got offline webhook: %v\n", vals)
+ w.WriteHeader(200)
+ w.Write([]byte("ok"))
+ })
+ http.HandleFunc("/live", func(w http.ResponseWriter, r *http.Request) {
+ body, err := io.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ // verify that the notification came from twitch using the secret.
+ if !helix.VerifyEventSubNotification("s3cre7w0rd", r.Header, string(body)) {
+ log.Println("no valid signature on subscription")
+ return
+ } else {
+ log.Println("verified signature for subscription")
+ }
+ var vals map[string]any
+ if err = json.Unmarshal(body, &vals); err != nil {
+ log.Println(err)
+ return
+ }
+
+ if challenge, ok := vals["challenge"]; ok {
+ w.Write([]byte(challenge.(string)))
+ return
+ }
+
+ log.Printf("got live webhook: %v\n", vals)
+ w.WriteHeader(200)
+ w.Write([]byte("ok"))
+ })
+ http.ListenAndServe("0.0.0.0:1337", nil)
+}
diff --git a/util/stats/main.go b/util/stats/main.go
new file mode 100644
index 0000000..43b4fd5
--- /dev/null
+++ b/util/stats/main.go
@@ -0,0 +1 @@
+package stats