muffin/goMuffin
1
1
Fork 0
Code Issues 1 Pull requests Projects Releases 3 Packages Wiki Activity Actions

fix: memory

Browse source
This commit is contained in:
Siwoo Jeon 2025-06-04 23:02:48 +09:00
parent 3755e58ec7
commit 4228adf7f3
Signed by: migan
GPG key ID: 036E9A8C5E8E48DA
3 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
chatbot/chatbot.go
View file

@ -3,6 +3,7 @@ package chatbot
import ( import (
"context" "context"
"fmt" "fmt"
"log"
"math/rand" "math/rand"
"sync" "sync"
@ -173,6 +174,8 @@ func getAIResponse(c *Chatbot, user *discordgo.User, question string) (string, e
return "", err return "", err
} }
log.Printf("%s TOKEN: %d", user.ID, result.UsageMetadata.PromptTokenCount)
return resultText, nil return resultText, nil
} }

5
chatbot/memory.go
View file

@ -16,7 +16,6 @@ func SaveMemory(data *databases.InsertMemory) error {
func GetMemory(userId string) ([]*genai.Content, error) { func GetMemory(userId string) ([]*genai.Content, error) {
var data []databases.Memory var data []databases.Memory
MAX_LENGTH := 50
memory := []*genai.Content{} memory := []*genai.Content{}
cur, err := databases.Database.Memory.Find(context.TODO(), bson.D{{Key: "user_id", Value: userId}}) 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) cur.All(context.TODO(), &data)
if len(data) > MAX_LENGTH {
data = data[MAX_LENGTH:]
}
for _, data := range data { for _, data := range data {
memory = append(memory, memory = append(memory,
genai.NewContentFromText(data.Content, genai.RoleUser), genai.NewContentFromText(data.Content, genai.RoleUser),

2
configs/version.go
View file

@ -7,7 +7,7 @@ import (
"git.wh64.net/muffin/goMuffin/utils" "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] var updatedString string = utils.RegexpDecimals.FindAllStringSubmatch(MUFFIN_VERSION, -1)[3][0]