105 lines
3.3 KiB
Go
105 lines
3.3 KiB
Go
//go:build llm_script
|
|
|
|
package main
|
|
|
|
import "testing"
|
|
|
|
func TestEnrichZhipuModelMetadataUsesSpecificFamilyRules(t *testing.T) {
|
|
cases := []struct {
|
|
modelID string
|
|
wantReleaseDate string
|
|
wantSourceURL string
|
|
wantConfidence string
|
|
wantSourceKind string
|
|
}{
|
|
{
|
|
modelID: "glm-5.1-32k",
|
|
wantReleaseDate: "2026-04-07",
|
|
wantSourceURL: "https://www.zhipuai.cn/zh/research",
|
|
wantConfidence: "official_primary",
|
|
wantSourceKind: "official_announcement",
|
|
},
|
|
{
|
|
modelID: "glm-5-turbo-32k",
|
|
wantReleaseDate: "2026-03-15",
|
|
wantSourceURL: "https://www.zhipuai.cn/en/research/155",
|
|
wantConfidence: "official_primary",
|
|
wantSourceKind: "official_announcement",
|
|
},
|
|
{
|
|
modelID: "glm-5-32k",
|
|
wantReleaseDate: "2026-02-11",
|
|
wantSourceURL: "https://www.zhipuai.cn/zh/research/154",
|
|
wantConfidence: "official_primary",
|
|
wantSourceKind: "official_announcement",
|
|
},
|
|
{
|
|
modelID: "glm-4.7-flash",
|
|
wantReleaseDate: "2026-01-19",
|
|
wantSourceURL: "https://www.zhipuai.cn/zh/news/148",
|
|
wantConfidence: "official_primary",
|
|
wantSourceKind: "official_announcement",
|
|
},
|
|
{
|
|
modelID: "glm-4.6v-flashx",
|
|
wantReleaseDate: "2025-12-07",
|
|
wantSourceURL: "https://www.zhipuai.cn/zh/research/144",
|
|
wantConfidence: "official_primary",
|
|
wantSourceKind: "official_announcement",
|
|
},
|
|
{
|
|
modelID: "glm-tts-clone",
|
|
wantReleaseDate: "2025-12-10",
|
|
wantSourceURL: "https://www.zhipuai.cn/zh/research",
|
|
wantConfidence: "official_primary",
|
|
wantSourceKind: "official_announcement",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
enriched := enrichZhipuModelMetadata(ModelPricing{
|
|
ModelID: tc.modelID,
|
|
SourceURL: "https://open.bigmodel.cn/pricing",
|
|
})
|
|
|
|
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)
|
|
}
|
|
if enriched.DateConfidence != tc.wantConfidence {
|
|
t.Fatalf("%s date confidence = %q, want %q", tc.modelID, enriched.DateConfidence, tc.wantConfidence)
|
|
}
|
|
if enriched.DateSourceKind != tc.wantSourceKind {
|
|
t.Fatalf("%s date source kind = %q, want %q", tc.modelID, enriched.DateSourceKind, tc.wantSourceKind)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEnrichZhipuModelMetadataFallsBackToPricingSource(t *testing.T) {
|
|
enriched := enrichZhipuModelMetadata(ModelPricing{
|
|
ModelID: "glm-unknown-preview",
|
|
SourceURL: "https://open.bigmodel.cn/pricing",
|
|
})
|
|
|
|
if enriched.ReleaseDate != "" {
|
|
t.Fatalf("unexpected release date: %q", enriched.ReleaseDate)
|
|
}
|
|
if enriched.ModelSourceURL != "https://open.bigmodel.cn/pricing" {
|
|
t.Fatalf("model source url = %q, want pricing source fallback", enriched.ModelSourceURL)
|
|
}
|
|
if enriched.DateConfidence != "unknown" || enriched.DateSourceKind != "unknown" {
|
|
t.Fatalf("unexpected fallback date metadata: confidence=%q kind=%q", enriched.DateConfidence, enriched.DateSourceKind)
|
|
}
|
|
}
|
|
|
|
func TestZhipuReleaseDateValueReturnsNilForUnknownDate(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)
|
|
}
|
|
}
|