16 lines
343 B
Go
16 lines
343 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.POST("/chat/completions", controller.Relay)
|
||
|
|
}
|
||
|
|
}
|