Procházet zdrojové kódy

修复gpts无法设置倍率的问题

CaIon před 2 roky
rodič
revize
a29e3765f4
1 změnil soubory, kde provedl 4 přidání a 0 odebrání
  1. 4 0
      common/model-ratio.go

+ 4 - 0
common/model-ratio.go

@@ -15,6 +15,7 @@ import (
 // 1 === ¥0.014 / 1k tokens
 var ModelRatio = map[string]float64{
 	"midjourney":                50,
+	"gpt-4-gizmo-*":             15,
 	"gpt-4":                     15,
 	"gpt-4-0314":                15,
 	"gpt-4-0613":                15,
@@ -89,6 +90,9 @@ func UpdateModelRatioByJSONString(jsonStr string) error {
 }
 
 func GetModelRatio(name string) float64 {
+	if strings.HasPrefix(name, "gpt-4-gizmo") {
+		name = "gpt-4-gizmo-*"
+	}
 	ratio, ok := ModelRatio[name]
 	if !ok {
 		SysError("model ratio not found: " + name)