diff --git a/dao/user.go b/dao/user.go
index 6629b43..ab8a160 100644
--- a/dao/user.go
+++ b/dao/user.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"go.mongodb.org/mongo-driver/bson"
 	"gorm.io/gorm"
+	"log"
 )
 
 type User struct {
@@ -300,6 +301,7 @@ func CreateUserUIConfigInfo(config proto.UserUIConfigInfo) (string, error) {
 }
 
 func GetUserUIConfigInfo(userID int) (proto.UserUIConfigInfo, error) {
+	log.Println("get user ui config info database:", proto.Config.MONGO_DATABASE, " collection:", UserUIConfigCollection)
 	collection := mongoClient.Database(proto.Config.MONGO_DATABASE).Collection(UserUIConfigCollection)
 	var config proto.UserUIConfigInfo
 	err := collection.FindOne(context.TODO(), bson.M{"user_id": userID}).Decode(&config)
diff --git a/main.go b/main.go
index 7e0be58..4e81eb9 100644
--- a/main.go
+++ b/main.go
@@ -47,6 +47,7 @@ func main() {
 	handler.SetUpIMGroup(r)            // IM
 	defer dao.Close()
 	defer worker.CloseRedis()
+	defer dao.CloseMongoDB()
 	//定时任务
 	c := cron.New(cron.WithSeconds())
 	// 添加每 10 秒执行一次的任务
diff --git a/proto/user.go b/proto/user.go
index a6cbead..7d6d4f8 100644
--- a/proto/user.go
+++ b/proto/user.go
@@ -1,6 +1,7 @@
 package proto
 
 import (
+	"go.mongodb.org/mongo-driver/bson/primitive"
 	"time"
 )
 
@@ -125,15 +126,16 @@ type DeleteThirdPartyLoginReq struct {
 
 // 用户对前端交互的配置信息结构,mongodb
 type UserUIConfigInfo struct {
-	UserID        int                      `json:"user_id" form:"user_id"`                 // 用户ID
-	Theme         string                   `json:"theme" form:"theme"`                     // 主题
-	Language      string                   `json:"language" form:"language"`               // 语言
-	FontSize      int                      `json:"font_size" form:"font_size"`             // 字体大小
-	GenAIFunction UserUIFunctionConfigInfo `json:"gen_ai_function" form:"gen_ai_function"` // 生成AI功能配置
-	KBaseFunction UserUIFunctionConfigInfo `json:"k_base_function" form:"k_base_function"` // 知识库功能配置
+	ID            primitive.ObjectID       `bson:"_id,omitempty"`                                                 // 自动生成ID
+	UserID        int                      `json:"user_id" form:"user_id" bson:"user_id"`                         // 用户ID
+	Theme         string                   `json:"theme" form:"theme" bson:"theme"`                               // 主题
+	Language      string                   `json:"language" form:"language" bson:"language"`                      // 语言
+	FontSize      int                      `json:"font_size" form:"font_size" bson:"font_size"`                   // 字体大小
+	GenAIFunction UserUIFunctionConfigInfo `json:"gen_ai_function" form:"gen_ai_function" bson:"gen_ai_function"` // 生成AI功能配置
+	KBaseFunction UserUIFunctionConfigInfo `json:"k_base_function" form:"k_base_function" bson:"k_base_function"` // 知识库功能配置
 }
 
 type UserUIFunctionConfigInfo struct {
-	ModelID   int `json:"model_id" form:"model_id"`     // 模型ID,选择的模型
-	SessionID int `json:"session_id" form:"session_id"` // 会话ID,上一次的会话
+	ModelID   int `json:"model_id" form:"model_id" bson:"model_id"`       // 模型ID,选择的模型
+	SessionID int `json:"session_id" form:"session_id" bson:"session_id"` // 会话ID,上一次的会话
 }