77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
|
//go:build llm_script
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
const default360PricingURL = "https://ai.360.com/open/models"
|
||
|
|
|
||
|
|
var platform360CardPattern = regexp.MustCompile(`(?s)([A-Za-z0-9._/-]+)\n([^\n]+)\n.*?(?:输入价格|Input Price)\s*:\s*¥([\d.]+)\s*/\s*1M tokens.*?(?:输出价格|Output Price)\s*:\s*¥([\d.]+)\s*/\s*1M tokens.*?(?:上下文|Context)\s*:\s*([\d,]+)`)
|
||
|
|
|
||
|
|
func parse360PricingCatalog(raw string) ([]officialPricingRecord, error) {
|
||
|
|
matches := platform360CardPattern.FindAllStringSubmatch(raw, -1)
|
||
|
|
if len(matches) == 0 {
|
||
|
|
return nil, fmt.Errorf("unexpected 360 pricing content")
|
||
|
|
}
|
||
|
|
|
||
|
|
records := make([]officialPricingRecord, 0, len(matches))
|
||
|
|
for _, match := range matches {
|
||
|
|
modelName := strings.TrimSpace(match[1])
|
||
|
|
providerName := normalize360Provider(match[2], modelName)
|
||
|
|
providerNameCn, providerCountry, providerWebsite := providerMetadata(providerName)
|
||
|
|
record := officialPricingRecord{
|
||
|
|
ModelID: normalizeExternalID("360", modelName),
|
||
|
|
ModelName: modelName,
|
||
|
|
ProviderName: providerName,
|
||
|
|
ProviderNameCn: providerNameCn,
|
||
|
|
ProviderCountry: providerCountry,
|
||
|
|
ProviderWebsite: providerWebsite,
|
||
|
|
OperatorName: "360 Open Platform",
|
||
|
|
OperatorNameCn: "360 智脑开放平台",
|
||
|
|
OperatorCountry: "CN",
|
||
|
|
OperatorWebsite: "https://ai.360.com",
|
||
|
|
OperatorType: "relay",
|
||
|
|
Region: "CN",
|
||
|
|
Currency: "CNY",
|
||
|
|
InputPrice: mustParseSubscriptionPrice(match[3]),
|
||
|
|
OutputPrice: mustParseSubscriptionPrice(match[4]),
|
||
|
|
ContextLength: parseContextLengthCommon(match[5]),
|
||
|
|
SourceURL: default360PricingURL,
|
||
|
|
ModelSourceURL: default360PricingURL,
|
||
|
|
DateConfidence: "unknown",
|
||
|
|
DateSourceKind: "official_product_page",
|
||
|
|
Modality: detectModality(modelName),
|
||
|
|
}
|
||
|
|
record.IsFree = record.InputPrice == 0 && record.OutputPrice == 0
|
||
|
|
records = append(records, record)
|
||
|
|
}
|
||
|
|
return records, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func normalize360Provider(raw string, modelName string) string {
|
||
|
|
switch strings.ToLower(strings.TrimSpace(raw)) {
|
||
|
|
case "deepseek", "深度求索":
|
||
|
|
return "DeepSeek"
|
||
|
|
case "moonshot ai", "月之暗面":
|
||
|
|
return "Moonshot AI"
|
||
|
|
case "qwen", "阿里巴巴", "通义千问":
|
||
|
|
return "Qwen"
|
||
|
|
case "zhipu", "智谱":
|
||
|
|
return "Zhipu AI"
|
||
|
|
case "字节跳动":
|
||
|
|
return "ByteDance"
|
||
|
|
case "360智脑":
|
||
|
|
return "360"
|
||
|
|
default:
|
||
|
|
providerByPath := providerFromModelPath(modelName)
|
||
|
|
if providerByPath != "unknown" {
|
||
|
|
return providerByPath
|
||
|
|
}
|
||
|
|
return strings.TrimSpace(raw)
|
||
|
|
}
|
||
|
|
}
|