101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
//go:build llm_script
|
|
|
|
package main
|
|
|
|
import "testing"
|
|
|
|
func TestEnrichBaiduModelMetadataUsesSpecificFamilyRules(t *testing.T) {
|
|
cases := []struct {
|
|
modelID string
|
|
wantReleaseDate string
|
|
wantSourceURL string
|
|
}{
|
|
{
|
|
modelID: "baidu-ernie-5.0",
|
|
wantReleaseDate: "2026-01-22",
|
|
wantSourceURL: "https://cloud.baidu.com/news/news_eacd0f0b-0ca3-4963-aec8-5e6b9ebef9ba",
|
|
},
|
|
{
|
|
modelID: "baidu-ernie-x1.1",
|
|
wantReleaseDate: "2025-09-09",
|
|
wantSourceURL: "https://cloud.baidu.com/news/news_be713ff4-8477-4852-88f1-9cc56c406d6a",
|
|
},
|
|
{
|
|
modelID: "baidu-ernie-4.5-turbo-128k",
|
|
wantReleaseDate: "2025-04-25",
|
|
wantSourceURL: "https://cloud.baidu.com/article/3887765",
|
|
},
|
|
{
|
|
modelID: "baidu-ernie-x1-turbo-32k",
|
|
wantReleaseDate: "2025-04-25",
|
|
wantSourceURL: "https://cloud.baidu.com/article/3887765",
|
|
},
|
|
{
|
|
modelID: "baidu-ernie-4.5-8k",
|
|
wantReleaseDate: "2025-03-16",
|
|
wantSourceURL: "https://cloud.baidu.com/article/3835921",
|
|
},
|
|
{
|
|
modelID: "baidu-ernie-x1-8k",
|
|
wantReleaseDate: "2025-03-16",
|
|
wantSourceURL: "https://cloud.baidu.com/article/3835921",
|
|
},
|
|
{
|
|
modelID: "baidu-ernie-character",
|
|
wantReleaseDate: "2024-03-22",
|
|
wantSourceURL: "https://cloud.baidu.com/news/news_667c065f-0bd7-475d-98c2-901763d0ee77",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
enriched := enrichBaiduModelMetadata(ModelPricing{
|
|
ModelID: tc.modelID,
|
|
SourceURL: "https://cloud.baidu.com/doc/qianfan/s/wmh4sv6ya",
|
|
})
|
|
|
|
if enriched.ReleaseDate != tc.wantReleaseDate {
|
|
t.Fatalf("%s release date = %q, want %q", tc.modelID, enriched.ReleaseDate, tc.wantReleaseDate)
|
|
}
|
|
if enriched.ModelSourceURL != tc.wantSourceURL {
|
|
t.Fatalf("%s source url = %q, want %q", tc.modelID, enriched.ModelSourceURL, tc.wantSourceURL)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEnrichBaiduModelMetadataFallsBackToPricingSource(t *testing.T) {
|
|
enriched := enrichBaiduModelMetadata(ModelPricing{
|
|
ModelID: "baidu-deepseek-r1",
|
|
SourceURL: "https://cloud.baidu.com/doc/qianfan/s/wmh4sv6ya",
|
|
})
|
|
|
|
if enriched.ReleaseDate != "" {
|
|
t.Fatalf("unexpected release date: %q", enriched.ReleaseDate)
|
|
}
|
|
if enriched.ModelSourceURL != "https://cloud.baidu.com/doc/qianfan/s/wmh4sv6ya" {
|
|
t.Fatalf("model source url = %q, want pricing source fallback", enriched.ModelSourceURL)
|
|
}
|
|
}
|
|
|
|
func TestEnrichBaiduModelMetadataSupportsSourceOnlyRules(t *testing.T) {
|
|
enriched := enrichBaiduModelMetadata(ModelPricing{
|
|
ModelID: "baidu-ernie-5.1",
|
|
SourceURL: "https://cloud.baidu.com/doc/qianfan/s/wmh4sv6ya",
|
|
})
|
|
|
|
if enriched.ReleaseDate != "" {
|
|
t.Fatalf("unexpected release date: %q", enriched.ReleaseDate)
|
|
}
|
|
if enriched.ModelSourceURL != "https://cloud.baidu.com/product/wenxinworkshop.html" {
|
|
t.Fatalf("model source url = %q, want product source", enriched.ModelSourceURL)
|
|
}
|
|
}
|
|
|
|
func TestBaiduReleaseDateValueReturnsNilForUnknownDate(t *testing.T) {
|
|
if value := releaseDateValue(""); value != nil {
|
|
t.Fatalf("blank release date should stay nil, got %v", value)
|
|
}
|
|
if value := releaseDateValue("invalid"); value != nil {
|
|
t.Fatalf("invalid release date should stay nil, got %v", value)
|
|
}
|
|
}
|