From 4228adf7f336f51f8d09914079c00364c1fb7bbf Mon Sep 17 00:00:00 2001
From: Siwoo Jeon <me@migan.co.kr>
Date: Wed, 4 Jun 2025 23:02:48 +0900
Subject: [PATCH] fix: memory

---
 chatbot/chatbot.go | 3 +++
 chatbot/memory.go  | 5 -----
 configs/version.go | 2 +-
 3 files changed, 4 insertions(+), 6 deletions(-)

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]