diff --git a/main.go b/main.go
index 3bb60b3..4a4dd33 100644
--- a/main.go
+++ b/main.go
@@ -31,10 +31,12 @@ func main() {
 		panic("failed to connect redis:" + err.Error())
 	}
 	r.Use(handler.CrosHandler())
-	r.Use(JWTAuthMiddleware()) // 使用 JWT 认证中间件
-	handler.SetUpUserGroup(r)  // User
-	handler.SetUpToolGroup(r)  // Tool
-	handler.SetUpFileGroup(r)  // File
+	r.Use(JWTAuthMiddleware())   // 使用 JWT 认证中间件
+	handler.SetUpUserGroup(r)    // User
+	handler.SetUpToolGroup(r)    // Tool
+	handler.SetUpFileGroup(r)    // File
+	handler.SetUpSessionGroup(r) // Session
+	handler.SetUpModelGroup(r)   // Model
 	defer dao.Close()
 	defer worker.CloseRedis()
 	//定时任务