aiproxy/core/controller/monitor.go

package controller

import (
	"net/http"
	"strconv"
	"strings"

	"github.com/gin-gonic/gin"
	"github.com/labring/aiproxy/core/middleware"
	"github.com/labring/aiproxy/core/monitor"
)

// GetAllChannelModelErrorRates godoc
//
//	@Summary		Get all channel model error rates
//	@Description	Returns a list of all channel model error rates
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Success		200	{object}	middleware.APIResponse{data=map[int64]map[string]float64}
//	@Router			/api/monitor/ [get]
func GetAllChannelModelErrorRates(c *gin.Context) {
	rates, err := monitor.GetAllChannelModelErrorRates(c.Request.Context())
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, rates)
}

// GetChannelModelErrorRates godoc
//
//	@Summary		Get channel model error rates
//	@Description	Returns a list of channel model error rates
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Param			id	path		int	true	"Channel ID"
//	@Success		200	{object}	middleware.APIResponse{data=[]map[string]float64}
//	@Router			/api/monitor/{id} [get]
func GetChannelModelErrorRates(c *gin.Context) {
	channelID, err := strconv.ParseInt(c.Param("id"), 10, 64)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusBadRequest, "Invalid channel ID")
		return
	}

	rates, err := monitor.GetChannelModelErrorRates(c.Request.Context(), channelID)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, rates)
}

// ClearAllModelErrors godoc
//
//	@Summary		Clear all model errors
//	@Description	Clears all model errors
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Success		200	{object}	middleware.APIResponse
//	@Router			/api/monitor/ [delete]
func ClearAllModelErrors(c *gin.Context) {
	err := monitor.ClearAllModelErrors(c.Request.Context())
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, nil)
}

// ClearChannelAllModelErrors godoc
//
//	@Summary		Clear channel all model errors
//	@Description	Clears all model errors for a specific channel
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Param			id	path		int	true	"Channel ID"
//	@Success		200	{object}	middleware.APIResponse
//	@Router			/api/monitor/{id} [delete]
func ClearChannelAllModelErrors(c *gin.Context) {
	channelID, err := strconv.ParseInt(c.Param("id"), 10, 64)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusBadRequest, "Invalid channel ID")
		return
	}

	err = monitor.ClearChannelAllModelErrors(c.Request.Context(), int(channelID))
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, nil)
}

// ClearChannelModelErrors godoc
//
//	@Summary		Clear channel model errors
//	@Description	Clears model errors for a specific channel and model
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Param			id		path		int		true	"Channel ID"
//	@Param			model	path		string	true	"Model name"
//	@Success		200		{object}	middleware.APIResponse
//	@Router			/api/monitor/{id}/{model} [delete]
func ClearChannelModelErrors(c *gin.Context) {
	model := strings.TrimPrefix(c.Param("model"), "/")

	channelID, err := strconv.ParseInt(c.Param("id"), 10, 64)
	if err != nil {
		middleware.ErrorResponse(c, http.StatusBadRequest, "Invalid channel ID")
		return
	}

	err = monitor.ClearChannelModelErrors(c.Request.Context(), model, int(channelID))
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, nil)
}

// GetModelsErrorRate godoc
//
//	@Summary		Get models error rate
//	@Description	Returns a list of models error rate
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Success		200	{object}	middleware.APIResponse{data=map[string]float64}
//	@Router			/api/monitor/models [get]
func GetModelsErrorRate(c *gin.Context) {
	rates, err := monitor.GetModelsErrorRate(c.Request.Context())
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, rates)
}

// GetAllBannedModelChannels godoc
//
//	@Summary		Get all banned model channels
//	@Description	Returns a list of all banned model channels
//	@Tags			monitor
//	@Produce		json
//	@Security		ApiKeyAuth
//	@Success		200	{object}	middleware.APIResponse{data=map[string][]int64}
//	@Router			/api/monitor/banned_channels [get]
func GetAllBannedModelChannels(c *gin.Context) {
	channels, err := monitor.GetAllBannedModelChannels(c.Request.Context())
	if err != nil {
		middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
		return
	}

	middleware.SuccessResponse(c, channels)
}