feat(import): enrich baidu and bytedance release metadata
This commit is contained in:
66
scripts/import_bytedance_data_test.go
Normal file
66
scripts/import_bytedance_data_test.go
Normal file
@@ -0,0 +1,66 @@
|
||||
//go:build llm_script
|
||||
|
||||
package main
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestEnrichBytedanceModelMetadataUsesSpecificFamilyRules(t *testing.T) {
|
||||
cases := []struct {
|
||||
modelID string
|
||||
wantReleaseDate string
|
||||
wantSourceURL string
|
||||
}{
|
||||
{
|
||||
modelID: "bytedance-doubao-seed-1.6-thinking",
|
||||
wantReleaseDate: "2025-06-11",
|
||||
wantSourceURL: "https://developer.volcengine.com/articles/7517188354606104612",
|
||||
},
|
||||
{
|
||||
modelID: "bytedance-doubao-1.5-thinking-pro",
|
||||
wantReleaseDate: "2025-04-17",
|
||||
wantSourceURL: "https://developer.volcengine.com/articles/7496718897794039827",
|
||||
},
|
||||
{
|
||||
modelID: "bytedance-seedance-1.0-lite",
|
||||
wantReleaseDate: "2025-05-13",
|
||||
wantSourceURL: "https://developer.volcengine.com/articles/7504284064976502823",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range cases {
|
||||
enriched := enrichBytedanceModelMetadata(ModelPricing{
|
||||
ModelID: tc.modelID,
|
||||
SourceURL: "https://www.volcengine.com/docs/82379/1099320",
|
||||
})
|
||||
|
||||
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 TestEnrichBytedanceModelMetadataFallsBackToPricingSource(t *testing.T) {
|
||||
enriched := enrichBytedanceModelMetadata(ModelPricing{
|
||||
ModelID: "bytedance-deepseek-r1",
|
||||
SourceURL: "https://www.volcengine.com/docs/82379/1099320",
|
||||
})
|
||||
|
||||
if enriched.ReleaseDate != "" {
|
||||
t.Fatalf("unexpected release date: %q", enriched.ReleaseDate)
|
||||
}
|
||||
if enriched.ModelSourceURL != "https://www.volcengine.com/docs/82379/1099320" {
|
||||
t.Fatalf("model source url = %q, want pricing source fallback", enriched.ModelSourceURL)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBytedanceReleaseDateValueReturnsNilForUnknownDate(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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user