ai-gateway/router/relay-router.go
2023-04-25 10:57:37 +08:00

16 lines
331 B
Go

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