cneicy
ceddfd3d6b
Sync production source including routing, billing, Astra detection and public uptime. Update custom build instructions and license packaging.
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package controller
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/QuantumNous/new-api/common"
|
|
"github.com/QuantumNous/new-api/model"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func GetModelPricingConfig(c *gin.Context) {
|
|
snapshot, err := model.GetModelPricingSnapshot(c.QueryArray("model"))
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, snapshot)
|
|
}
|
|
|
|
func PreviewModelPricingConversion(c *gin.Context) {
|
|
var request struct {
|
|
ModelName string `json:"model_name"`
|
|
Pricing model.PricingValues `json:"pricing"`
|
|
}
|
|
if err := common.DecodeJson(c.Request.Body, &request); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
preview, err := model.PreviewModelPricingConversion(request.ModelName, request.Pricing)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, preview)
|
|
}
|
|
|
|
func PreviewModelPricing(c *gin.Context) {
|
|
var request struct {
|
|
ModelName string `json:"model_name"`
|
|
Pricing model.PricingValues `json:"pricing"`
|
|
}
|
|
if err := common.DecodeJson(c.Request.Body, &request); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
preview, err := model.PreviewModelPricing(request.ModelName, request.Pricing)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
common.ApiSuccess(c, model.ModelPricingDescription{
|
|
Effective: preview,
|
|
CacheWriteMode: model.ResolveCacheWriteMode(request.ModelName, request.Pricing),
|
|
BillingDetails: model.ResolveLegacyBillingDetails(request.ModelName, preview, request.Pricing),
|
|
})
|
|
}
|
|
|
|
func UpdateModelPricingConfig(c *gin.Context) {
|
|
var request struct {
|
|
Changes []model.ModelPricingChange `json:"changes"`
|
|
}
|
|
if err := common.DecodeJson(c.Request.Body, &request); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
if err := model.UpdateModelPricing(request.Changes); err != nil {
|
|
status := http.StatusBadRequest
|
|
if errors.Is(err, model.ErrModelPricingConflict) {
|
|
status = http.StatusConflict
|
|
}
|
|
c.JSON(status, gin.H{"success": false, "message": err.Error()})
|
|
return
|
|
}
|
|
names := make([]string, 0, len(request.Changes))
|
|
for _, change := range request.Changes {
|
|
names = append(names, change.ModelName)
|
|
}
|
|
recordManageAudit(c, "model.pricing.update", map[string]any{"models": names})
|
|
common.ApiSuccess(c, gin.H{"updated_models": names})
|
|
}
|