fb44c8bf03
包含Go API项目的所有源代码、配置文件、Docker配置、文档和前端资源 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
25 lines
679 B
Go
25 lines
679 B
Go
package router
|
|
|
|
import (
|
|
"one-api/controller"
|
|
"one-api/middleware"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func SetVideoRouter(router *gin.Engine) {
|
|
videoV1Router := router.Group("/v1")
|
|
videoV1Router.Use(middleware.TokenAuth(), middleware.Distribute())
|
|
{
|
|
videoV1Router.POST("/video/generations", controller.RelayTask)
|
|
videoV1Router.GET("/video/generations/:task_id", controller.RelayTask)
|
|
}
|
|
|
|
klingV1Router := router.Group("/kling/v1")
|
|
klingV1Router.Use(middleware.KlingRequestConvert(), middleware.TokenAuth(), middleware.Distribute())
|
|
{
|
|
klingV1Router.POST("/videos/text2video", controller.RelayTask)
|
|
klingV1Router.POST("/videos/image2video", controller.RelayTask)
|
|
}
|
|
}
|