39 lines
825 B
Go
39 lines
825 B
Go
package chatbot
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.wh64.net/muffin/goMuffin/databases"
|
|
"go.mongodb.org/mongo-driver/v2/bson"
|
|
"google.golang.org/genai"
|
|
)
|
|
|
|
func SaveMemory(data *databases.Memory) error {
|
|
_, err := databases.Database.Memory.InsertOne(context.TODO(), *data)
|
|
return err
|
|
}
|
|
|
|
func GetMemory(chatId bson.ObjectID) ([]*genai.Content, error) {
|
|
var data []databases.Memory
|
|
|
|
memory := []*genai.Content{}
|
|
|
|
cur, err := databases.Database.Memory.Find(context.TODO(), databases.User{ChatId: chatId})
|
|
if err != nil {
|
|
return memory, err
|
|
}
|
|
|
|
err = cur.All(context.TODO(), &data)
|
|
if err != nil {
|
|
return memory, err
|
|
}
|
|
|
|
for _, data := range data {
|
|
memory = append(memory,
|
|
genai.NewContentFromText(data.Content, genai.RoleUser),
|
|
genai.NewContentFromText(data.Answer, genai.RoleModel),
|
|
)
|
|
}
|
|
|
|
return memory, nil
|
|
}
|