xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-17 05:33:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 fix: Fix model name not URL encoded in pricing

Browse Source
This commit is contained in:
Martial BE
2024-04-01 09:47:58 +08:00
parent 91a113d09b
commit 6300ce4835
3 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
controller/pricing.go
View File

@@ -3,6 +3,7 @@ package controller
import (
"errors"
"net/http"
"net/url"
"one-api/common"
"one-api/model"
"one-api/relay/util"
@@ -80,10 +81,12 @@ func AddPrice(c *gin.Context) {
func UpdatePrice(c *gin.Context) {
modelName := c.Param("model")
if modelName == "" {
if modelName == "" || len(modelName) < 2 {
common.APIRespondWithError(c, http.StatusOK, errors.New("model name is required"))
return
}
modelName = modelName[1:]
modelName, _ = url.PathUnescape(modelName)
var price model.Price
if err := c.ShouldBindJSON(&price); err != nil {
@@ -104,10 +107,12 @@ func UpdatePrice(c *gin.Context) {
func DeletePrice(c *gin.Context) {
modelName := c.Param("model")
if modelName == "" {
if modelName == "" || len(modelName) < 2 {
common.APIRespondWithError(c, http.StatusOK, errors.New("model name is required"))
return
}
modelName = modelName[1:]
modelName, _ = url.PathUnescape(modelName)
if err := util.PricingInstance.DeletePrice(modelName); err != nil {
common.APIRespondWithError(c, http.StatusOK, err)