diff --git a/dao/db.go b/dao/db.go
index 6050f57..742dfa2 100644
--- a/dao/db.go
+++ b/dao/db.go
@@ -6,6 +6,7 @@ import (
 	"gorm.io/driver/mysql"
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
+	"log"
 )
 
 var DB *gorm.DB
@@ -59,6 +60,11 @@ func Init() error {
 		fmt.Println("model table:", err)
 		return err
 	}
+	err = db.AutoMigrate(&FunctionModel{})
+	if err != nil {
+		log.Println("function model table:", err)
+		return err
+	}
 
 	err = db.AutoMigrate(&Session{})
 	if err != nil {
diff --git a/main.go b/main.go
index 3263608..10896c5 100644
--- a/main.go
+++ b/main.go
@@ -31,13 +31,14 @@ 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
-	handler.SetUpSessionGroup(r) // Session
-	handler.SetUpModelGroup(r)   // Model
-	handler.SetUpIMGroup(r)      // IM
+	r.Use(JWTAuthMiddleware())     // 使用 JWT 认证中间件
+	handler.SetUpUserGroup(r)      // User
+	handler.SetUpToolGroup(r)      // Tool
+	handler.SetUpFileGroup(r)      // File
+	handler.SetUpSessionGroup(r)   // Session
+	handler.SetUpModelGroup(r)     // Model
+	handler.SetUpFuncModelGroup(r) // FuncModel
+	handler.SetUpIMGroup(r)        // IM
 	defer dao.Close()
 	defer worker.CloseRedis()
 	//定时任务