fix monitoring manual mark attribution
This commit is contained in:
@@ -535,7 +535,17 @@ func TestBuildMonitoringRawPayloadQwenIncludesSearchResultsInCitationInputs(t *t
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildMonitoringCitationFactCanDisableContentResolution(t *testing.T) {
|
||||
func TestMonitoringPlatformResolvesContentCitationsForAllKnownModels(t *testing.T) {
|
||||
for _, platformID := range []string{"doubao", "qwen", "kimi", "deepseek", "wenxin", "yuanbao"} {
|
||||
t.Run(platformID, func(t *testing.T) {
|
||||
if !monitoringPlatformResolvesContentCitations(platformID) {
|
||||
t.Fatalf("monitoringPlatformResolvesContentCitations(%q) = false, want true", platformID)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildMonitoringCitationFactResolvesContentForQwen(t *testing.T) {
|
||||
aliasArticleID := int64(42)
|
||||
aliasPublishRecordID := int64(84)
|
||||
normalizedURL := "https://example.com/article"
|
||||
@@ -561,15 +571,15 @@ func TestBuildMonitoringCitationFactCanDisableContentResolution(t *testing.T) {
|
||||
t.Fatalf("resolved PublishRecordID = %v, want %d", resolvedFact.PublishRecordID, aliasPublishRecordID)
|
||||
}
|
||||
|
||||
sourceOnlyFact, ok := buildMonitoringCitationFact(input, aliasMap, false)
|
||||
sourceOnlyFact, ok := buildMonitoringCitationFact(input, aliasMap, monitoringPlatformResolvesContentCitations("qwen"))
|
||||
if !ok {
|
||||
t.Fatal("buildMonitoringCitationFact() without content resolution returned !ok")
|
||||
t.Fatal("buildMonitoringCitationFact() for qwen returned !ok")
|
||||
}
|
||||
if sourceOnlyFact.ArticleID != nil {
|
||||
t.Fatalf("source-only ArticleID = %v, want nil", sourceOnlyFact.ArticleID)
|
||||
if sourceOnlyFact.ArticleID == nil || *sourceOnlyFact.ArticleID != aliasArticleID {
|
||||
t.Fatalf("qwen ArticleID = %v, want %d", sourceOnlyFact.ArticleID, aliasArticleID)
|
||||
}
|
||||
if sourceOnlyFact.PublishRecordID != nil {
|
||||
t.Fatalf("source-only PublishRecordID = %v, want nil", sourceOnlyFact.PublishRecordID)
|
||||
if sourceOnlyFact.PublishRecordID == nil || *sourceOnlyFact.PublishRecordID != aliasPublishRecordID {
|
||||
t.Fatalf("qwen PublishRecordID = %v, want %d", sourceOnlyFact.PublishRecordID, aliasPublishRecordID)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user