mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-08-31 02:41:34 +00:00
34 lines
1.0 KiB
Go
34 lines
1.0 KiB
Go
package router
|
|
|
|
import (
|
|
"github.com/QuantumNous/new-api/controller"
|
|
"github.com/QuantumNous/new-api/middleware"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func SetVideoRouter(router *gin.Engine) {
|
|
videoSharedRouter := router.Group("/v1")
|
|
videoSharedRouter.Use(middleware.RouteTag("relay"))
|
|
videoSharedRouter.Use(middleware.TokenAuth())
|
|
videoSharedRouter.Use(middleware.SystemPerformanceCheck())
|
|
videoSharedRouter.POST(
|
|
"/video/generations",
|
|
middleware.PinTaskPluginEndpoint(),
|
|
middleware.TaskPluginEndpointOnly(middleware.ModelRequestRateLimit()),
|
|
middleware.PrepareTaskPluginEndpoint(),
|
|
middleware.Distribute(),
|
|
func(c *gin.Context) {
|
|
controller.RelayTaskPluginEndpoint(c, controller.RelayTask)
|
|
},
|
|
)
|
|
|
|
videoV1Router := router.Group("/v1")
|
|
videoV1Router.Use(middleware.RouteTag("relay"))
|
|
videoV1Router.Use(middleware.TokenAuth(), middleware.Distribute())
|
|
{
|
|
videoV1Router.GET("/video/generations/:task_id", controller.RelayTaskFetch)
|
|
videoV1Router.POST("/videos/:video_id/remix", controller.RelayTask)
|
|
}
|
|
}
|