fix lint issue

aigateway/router/aigateway.go

@@ -3,39 +3,48 @@ package router
 import (
 	"fmt"
 
+	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"opencsg.com/csghub-server/aigateway/handler"
 	"opencsg.com/csghub-server/api/middleware"
+	"opencsg.com/csghub-server/builder/instrumentation"
 	"opencsg.com/csghub-server/common/config"
-	"opencsg.com/csghub-server/common/i18n"
 )
 
 func NewRouter(config *config.Config) (*gin.Engine, error) {
 	r := gin.New()
-	r.Use(gin.Recovery())
-	r.Use(middleware.Log())
+	middleware.SetInfraMiddleware(r, config, instrumentation.Aigateway)
+	r.Use(cors.New(cors.Config{
+		AllowCredentials: true,
+		AllowHeaders:     []string{"*"},
+		AllowMethods:     []string{"*"},
+		AllowAllOrigins:  true,
+	}))
 	//to access model,fintune with any kind of tokens in auth header
-	i18n.InitLocalizersFromEmbedFile()
-	r.Use(middleware.ModifyAcceptLanguageMiddleware(), middleware.LocalizedErrorMiddleware())
 	r.Use(middleware.Authenticator(config))
-	mustLogin := middleware.MustLogin()
+	middlewareCollection := middleware.MiddlewareCollection{}
+	middlewareCollection.Auth.NeedLogin = middleware.MustLogin()
+	middlewareCollection.Auth.NeedPhoneVerified = middleware.NeedPhoneVerified(config)
 
 	v1Group := r.Group("/v1")
 
 	openAIhandler, err := handler.NewOpenAIHandlerFromConfig(config)
 	if err != nil {
 		return nil, fmt.Errorf("error creating openai handler :%w", err)
 	}
-	v1Group.GET("/models", mustLogin, openAIhandler.ListModels)
-	v1Group.GET("/models/:model", mustLogin, openAIhandler.GetModel)
-	v1Group.POST("/chat/completions", mustLogin, openAIhandler.Chat)
-	v1Group.POST("/embeddings", mustLogin, openAIhandler.Embedding)
+	v1Group.GET("/models", middlewareCollection.Auth.NeedLogin, openAIhandler.ListModels)
+	v1Group.GET("/models/:model", middlewareCollection.Auth.NeedLogin, openAIhandler.GetModel)
+	v1Group.POST("/chat/completions", middlewareCollection.Auth.NeedLogin, openAIhandler.Chat)
+	v1Group.POST("/embeddings", middlewareCollection.Auth.NeedLogin, openAIhandler.Embedding)
 
 	mcpProxy, err := handler.NewMCPProxyHandler(config)
 	if err != nil {
 		return nil, fmt.Errorf("error creating mcp proxy handler :%w", err)
 	}
 	CreateMCPRoute(v1Group, mcpProxy)
+	if err := extendRoutes(v1Group, middlewareCollection, config); err != nil {
+		return nil, fmt.Errorf("error creating extended routes :%w", err)
+	}
 	return r, nil
 }
 

aigateway/router/api_ce.go

@@ -0,0 +1,13 @@
+//go:build !ee && !saas
+
+package router
+
+import (
+	"github.com/gin-gonic/gin"
+	"opencsg.com/csghub-server/api/middleware"
+	"opencsg.com/csghub-server/common/config"
+)
+
+func extendRoutes(_ *gin.RouterGroup, _ middleware.MiddlewareCollection, _ *config.Config) error {
+	return nil
+}