feat: update group

service/aiproxy/controller/group.go

@@ -274,3 +274,28 @@ func CreateGroup(c *gin.Context) {
 	}
 	middleware.SuccessResponse(c, nil)
 }
+
+func UpdateGroup(c *gin.Context) {
+	group := c.Param("group")
+	if group == "" {
+		middleware.ErrorResponse(c, http.StatusOK, "invalid parameter")
+		return
+	}
+	req := CreateGroupRequest{}
+	err := json.NewDecoder(c.Request.Body).Decode(&req)
+	if err != nil {
+		middleware.ErrorResponse(c, http.StatusOK, "invalid parameter")
+		return
+	}
+	err = model.UpdateGroup(group, &model.Group{
+		RPMRatio: req.RPMRatio,
+		RPM:      req.RPM,
+		TPMRatio: req.TPMRatio,
+		TPM:      req.TPM,
+	})
+	if err != nil {
+		middleware.ErrorResponse(c, http.StatusOK, err.Error())
+		return
+	}
+	middleware.SuccessResponse(c, nil)
+}

service/aiproxy/model/group.go

@@ -131,6 +131,25 @@ func DeleteGroupsByIDs(ids []string) (err error) {
 	})
 }
 
+func UpdateGroup(id string, group *Group) (err error) {
+	if id == "" {
+		return errors.New("group id is empty")
+	}
+	defer func() {
+		if err == nil {
+			if err := CacheDeleteGroup(id); err != nil {
+				log.Error("cache delete group failed: " + err.Error())
+			}
+		}
+	}()
+	result := DB.
+		Clauses(clause.Returning{}).
+		Where("id = ?", id).
+		Select("rpm_ratio", "rpm", "tpm_ratio", "tpm").
+		Updates(group)
+	return HandleUpdateResult(result, ErrGroupNotFound)
+}
+
 func UpdateGroupUsedAmountAndRequestCount(id string, amount float64, count int) (err error) {
 	group := &Group{ID: id}
 	defer func() {

service/aiproxy/router/api.go

@@ -49,6 +49,7 @@ func SetAPIRouter(router *gin.Engine) {
 		groupRoute := apiRouter.Group("/group")
 		{
 			groupRoute.POST("/:group", controller.CreateGroup)
+			groupRoute.PUT("/:group", controller.UpdateGroup)
 			groupRoute.GET("/:group", controller.GetGroup)
 			groupRoute.DELETE("/:group", controller.DeleteGroup)
 			groupRoute.POST("/:group/status", controller.UpdateGroupStatus)