Codex
bb7968b612
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
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)
|
|
}
|
|
}
|