ai-gateway/router/relay-router.go
2023-04-26 21:50:09 +08:00

21 lines
492 B
Go

package router
import (
"github.com/gin-gonic/gin"
"one-api/controller"
"one-api/middleware"
)
func SetRelayRouter(router *gin.Engine) {
relayV1Router := router.Group("/v1")
relayV1Router.Use(middleware.TokenAuth(), middleware.Distribute())
{
relayV1Router.Any("/*path", controller.Relay)
}
relayDashboardRouter := router.Group("/dashboard")
relayDashboardRouter.Use(middleware.TokenAuth(), middleware.Distribute())
{
relayDashboardRouter.Any("/*path", controller.Relay)
}
}