diff --git a/chatbot/chatbot.go b/chatbot/chatbot.go index 62d5767..d860ebb 100644 --- a/chatbot/chatbot.go +++ b/chatbot/chatbot.go @@ -3,6 +3,7 @@ package chatbot import ( "context" "fmt" + "log" "math/rand" "sync" @@ -173,6 +174,8 @@ func getAIResponse(c *Chatbot, user *discordgo.User, question string) (string, e return "", err } + log.Printf("%s TOKEN: %d", user.ID, result.UsageMetadata.PromptTokenCount) + return resultText, nil } diff --git a/chatbot/memory.go b/chatbot/memory.go index db6d706..b2df66b 100644 --- a/chatbot/memory.go +++ b/chatbot/memory.go @@ -16,7 +16,6 @@ func SaveMemory(data *databases.InsertMemory) error { func GetMemory(userId string) ([]*genai.Content, error) { var data []databases.Memory - MAX_LENGTH := 50 memory := []*genai.Content{} cur, err := databases.Database.Memory.Find(context.TODO(), bson.D{{Key: "user_id", Value: userId}}) @@ -26,10 +25,6 @@ func GetMemory(userId string) ([]*genai.Content, error) { cur.All(context.TODO(), &data) - if len(data) > MAX_LENGTH { - data = data[MAX_LENGTH:] - } - for _, data := range data { memory = append(memory, genai.NewContentFromText(data.Content, genai.RoleUser), diff --git a/configs/version.go b/configs/version.go index b33ddae..4e52270 100644 --- a/configs/version.go +++ b/configs/version.go @@ -7,7 +7,7 @@ import ( "git.wh64.net/muffin/goMuffin/utils" ) -const MUFFIN_VERSION = "0.0.0-madeleine_canary.250604a-muffin-ai" +const MUFFIN_VERSION = "0.0.0-madeleine_canary.250604b-muffin-ai" var updatedString string = utils.RegexpDecimals.FindAllStringSubmatch(MUFFIN_VERSION, -1)[3][0]