fix: channel listing

controller/model.go

@@ -332,13 +332,14 @@ func init() {
 	}
 }
 
-func ListModels(c *gin.Context) {
+func ListModels(c *gin.Context, isChannel bool) {
 	userId := c.GetInt("id")
 	userGroup, _ := model.CacheGetUserGroup(userId)
 
 	// Only return the models that the channel and user have access to
 	var modalList []OpenAIModels
 
+	if !isChannel {
 		for _, modalData := range openAIModels {
 			channel, err := model.CacheGetRandomSatisfiedChannel(userGroup, modalData.Id)
 
@@ -350,6 +351,7 @@ func ListModels(c *gin.Context) {
 				modalList = append(modalList, modalData)
 			}
 		}
+	}
 
 	c.JSON(200, gin.H{
 		"object": "list",

router/api-router.go

@@ -65,7 +65,9 @@ func SetApiRouter(router *gin.Engine) {
 		{
 			channelRoute.GET("/", controller.GetAllChannels)
 			channelRoute.GET("/search", controller.SearchChannels)
-			channelRoute.GET("/models", controller.ListModels)
+			channelRoute.GET("/models", func(c *gin.Context) {
+				controller.ListModels(c, true)
+			})
 			channelRoute.GET("/:id", controller.GetChannel)
 			channelRoute.GET("/test", controller.TestAllChannels)
 			channelRoute.GET("/test/:id", controller.TestChannel)

router/relay-router.go

@@ -12,7 +12,9 @@ func SetRelayRouter(router *gin.Engine) {
 	modelsRouter := router.Group("/v1/models")
 	modelsRouter.Use(middleware.TokenAuth())
 	{
-		modelsRouter.GET("/", controller.ListModels)
+		modelsRouter.GET("/", func(c *gin.Context) {
+			controller.ListModels(c, false)
+		})
 		modelsRouter.GET("/:model", controller.RetrieveModel)
 	}
 	relayV1Router := router.Group("/v1")