xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2026-01-01 12:35:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ Model list return by group

Browse Source
This commit is contained in:
Martial BE
2023-12-28 15:48:20 +08:00
parent 4d43dce64b
commit 72983ac734
6 changed files with 95 additions and 548 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
router/api-router.go
View File

@@ -67,7 +67,7 @@ func SetApiRouter(router *gin.Engine) {
{
channelRoute.GET("/", controller.GetAllChannels)
channelRoute.GET("/search", controller.SearchChannels)
channelRoute.GET("/models", controller.ListModels)
channelRoute.GET("/models", controller.ListModelsForAdmin)
channelRoute.GET("/:id", controller.GetChannel)
channelRoute.GET("/test", controller.TestAllChannels)
channelRoute.GET("/test/:id", controller.TestChannel)

2
router/relay-router.go
View File

@@ -11,7 +11,7 @@ func SetRelayRouter(router *gin.Engine) {
router.Use(middleware.CORS())
// https://platform.openai.com/docs/api-reference/introduction
modelsRouter := router.Group("/v1/models")
modelsRouter.Use(middleware.TokenAuth())
modelsRouter.Use(middleware.TokenAuth(), middleware.Distribute())
{
modelsRouter.GET("", controller.ListModels)
modelsRouter.GET("/:model", controller.RetrieveModel)