diff --git a/opencat.go b/opencat.go index fecc219..c102cab 100644 --- a/opencat.go +++ b/opencat.go @@ -14,7 +14,6 @@ import ( "os" "github.com/duke-git/lancet/v2/fileutil" - "github.com/gin-contrib/cors" "github.com/gin-gonic/gin" "github.com/google/uuid" "gorm.io/gorm" @@ -143,6 +142,7 @@ func main() { } port := os.Getenv("PORT") r := gin.Default() + r.Use(team.CORS()) group := r.Group("/1") { group.Use(team.AuthMiddleware()) @@ -167,7 +167,7 @@ func main() { // 初始化用户 r.POST("/1/users/init", team.Handleinit) - r.Any("/v1/*proxypath", cors.Default(), router.HandleProxy) + r.Any("/v1/*proxypath", router.HandleProxy) // r.POST("/v1/chat/completions", router.HandleProy) // r.GET("/v1/models", router.HandleProy) diff --git a/pkg/team/middleware.go b/pkg/team/middleware.go index ab01e79..7cbd49b 100644 --- a/pkg/team/middleware.go +++ b/pkg/team/middleware.go @@ -6,6 +6,7 @@ import ( "opencatd-open/store" "strings" + "github.com/gin-contrib/cors" "github.com/gin-gonic/gin" ) @@ -57,3 +58,12 @@ func AuthMiddleware() gin.HandlerFunc { c.Next() } } + +func CORS() gin.HandlerFunc { + config := cors.DefaultConfig() + config.AllowAllOrigins = true + config.AllowCredentials = true + config.AllowMethods = []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"} + config.AllowHeaders = []string{"*"} + return cors.New(config) +}