Просмотр исходного кода

fix: 如果模型管理有自定义配置则不合并默认配置

Seefs 5 дней назад
Родитель
Сommit
4950db666f
1 измененных файлов с 11 добавлено и 6 удалено
  1. 11 6
      model/pricing.go

+ 11 - 6
model/pricing.go

@@ -196,20 +196,25 @@ func updatePricing() {
 		modelSupportEndpointsStr[ability.Model] = endpoints
 	}
 
-	// 再补充模型自定义端点
+	// 再补充模型自定义端点:若配置有效则替换默认端点,不做合并
 	for modelName, meta := range metaMap {
 		if strings.TrimSpace(meta.Endpoints) == "" {
 			continue
 		}
 		var raw map[string]interface{}
 		if err := json.Unmarshal([]byte(meta.Endpoints), &raw); err == nil {
-			endpoints := modelSupportEndpointsStr[modelName]
-			for k := range raw {
-				if !common.StringsContains(endpoints, k) {
-					endpoints = append(endpoints, k)
+			endpoints := make([]string, 0, len(raw))
+			for k, v := range raw {
+				switch v.(type) {
+				case string, map[string]interface{}:
+					if !common.StringsContains(endpoints, k) {
+						endpoints = append(endpoints, k)
+					}
 				}
 			}
-			modelSupportEndpointsStr[modelName] = endpoints
+			if len(endpoints) > 0 {
+				modelSupportEndpointsStr[modelName] = endpoints
+			}
 		}
 	}