Harden catalog vendor resolution and agent prompt
Address review feedback on the agent-driven tracker catalog path: - Return initiator-domain load failures instead of swallowing them, so the worker retries rather than running the agent on partial context. - In the resolver, treat only ErrResourceNotFound as a catalog miss and propagate genuine name/slug lookup errors. - Insert the new vendor inside a savepoint and, on the slug unique-violation race, reload and return the winning row instead of aborting the caller's transaction. - Stop seeding common_third_party_domains from observed initiator domains. They are a co-occurrence signal, not verified ownership, and writing them into the global cross-tenant catalog pollutes the domain-based matcher. The curated seed owns that data. - Warn the mapping agent that observed domains may belong to shared CDNs, tag managers, or hosting infrastructure rather than the vendor, so it does not attribute on that basis alone. - Extract a shared tracker-identification prompt helper and move the common-pattern identification prompt next to the enrichment agent. Signed-off-by: Émile Ré <emile@probo.com>
This commit is contained in:
@@ -75,6 +75,15 @@ func buildCommonPatternEnrichmentAgent(
|
||||
return agent.New("common-pattern-enrichment", cfg.LLMClient, opts...)
|
||||
}
|
||||
|
||||
// buildCommonPatternIdentificationPrompt builds the mapping-agent input
|
||||
// for a global catalog pattern. Catalog rows carry no observed domains,
|
||||
// so the prompt omits the <observed_domains> signal and relies on the
|
||||
// pattern name, type, and naming conventions. It lets the enrichment
|
||||
// worker reuse the mapping agent to attribute a vendor before describing.
|
||||
func buildCommonPatternIdentificationPrompt(cp coredata.CommonTrackerPattern) string {
|
||||
return buildTrackerIdentificationPrompt(cp.Pattern, cp.TrackerType, cp.MatchType, cp.MaxAgeSeconds)
|
||||
}
|
||||
|
||||
func buildEnrichmentPrompt(cp coredata.CommonTrackerPattern, thirdPartyName string) string {
|
||||
maxAge := "session"
|
||||
if cp.MaxAgeSeconds != nil {
|
||||
|
||||
@@ -151,7 +151,7 @@ func (h *commonPatternEnrichmentHandler) Process(ctx context.Context, cp coredat
|
||||
var thirdPartyID *gid.GID
|
||||
|
||||
if attribution != nil && cp.CommonThirdPartyID == nil {
|
||||
thirdPartyID, err = thirdparty.ResolveOrCreateCommonThirdParty(ctx, tx, h.logger, attribution.ThirdPartyName, attribution.Category, nil)
|
||||
thirdPartyID, err = thirdparty.ResolveOrCreateCommonThirdParty(ctx, tx, h.logger, attribution.ThirdPartyName, attribution.Category)
|
||||
if err != nil {
|
||||
return fmt.Errorf("cannot resolve or create common third party: %w", err)
|
||||
}
|
||||
|
||||
@@ -137,40 +137,19 @@ func trackerMappingInstructions(_ context.Context, _ *agent.Agent) string {
|
||||
)
|
||||
}
|
||||
|
||||
func buildAgentPrompt(tp coredata.TrackerPattern, domains []string) string {
|
||||
// buildTrackerIdentificationPrompt renders the base mapping-agent input
|
||||
// shared by live tracker patterns and global catalog patterns: the four
|
||||
// XML signal tags plus the max-age preamble. Callers append any extra
|
||||
// signals (e.g. observed domains) to the returned prompt.
|
||||
func buildTrackerIdentificationPrompt(
|
||||
pattern string,
|
||||
trackerType coredata.TrackerType,
|
||||
matchType coredata.TrackerPatternMatchType,
|
||||
maxAgeSeconds *int,
|
||||
) string {
|
||||
maxAge := "session"
|
||||
if tp.MaxAgeSeconds != nil {
|
||||
maxAge = fmt.Sprintf("%d seconds", *tp.MaxAgeSeconds)
|
||||
}
|
||||
|
||||
prompt := fmt.Sprintf(
|
||||
"Identify the following tracker:\n\n"+
|
||||
"<pattern> %s </pattern>\n"+
|
||||
"<type> %s </type>\n"+
|
||||
"<match_type> %s </match_type>\n"+
|
||||
"<max_age> %s </max_age>\n",
|
||||
tp.Pattern,
|
||||
tp.TrackerType,
|
||||
tp.MatchType,
|
||||
maxAge,
|
||||
)
|
||||
|
||||
if len(domains) > 0 {
|
||||
prompt += fmt.Sprintf("<observed_domains> %s </observed_domains>\n", strings.Join(domains, ", "))
|
||||
}
|
||||
|
||||
return prompt
|
||||
}
|
||||
|
||||
// buildCommonPatternIdentificationPrompt builds the mapping-agent input
|
||||
// for a global catalog pattern. Catalog rows carry no observed domains,
|
||||
// so the prompt omits the <observed_domains> signal and relies on the
|
||||
// pattern name, type, and naming conventions. It lets the enrichment
|
||||
// worker reuse the mapping agent to attribute a vendor before describing.
|
||||
func buildCommonPatternIdentificationPrompt(cp coredata.CommonTrackerPattern) string {
|
||||
maxAge := "session"
|
||||
if cp.MaxAgeSeconds != nil {
|
||||
maxAge = fmt.Sprintf("%d seconds", *cp.MaxAgeSeconds)
|
||||
if maxAgeSeconds != nil {
|
||||
maxAge = fmt.Sprintf("%d seconds", *maxAgeSeconds)
|
||||
}
|
||||
|
||||
return fmt.Sprintf(
|
||||
@@ -179,9 +158,19 @@ func buildCommonPatternIdentificationPrompt(cp coredata.CommonTrackerPattern) st
|
||||
"<type> %s </type>\n"+
|
||||
"<match_type> %s </match_type>\n"+
|
||||
"<max_age> %s </max_age>\n",
|
||||
cp.Pattern,
|
||||
cp.TrackerType,
|
||||
cp.MatchType,
|
||||
pattern,
|
||||
trackerType,
|
||||
matchType,
|
||||
maxAge,
|
||||
)
|
||||
}
|
||||
|
||||
func buildAgentPrompt(tp coredata.TrackerPattern, domains []string) string {
|
||||
prompt := buildTrackerIdentificationPrompt(tp.Pattern, tp.TrackerType, tp.MatchType, tp.MaxAgeSeconds)
|
||||
|
||||
if len(domains) > 0 {
|
||||
prompt += fmt.Sprintf("<observed_domains> %s </observed_domains>\n", strings.Join(domains, ", "))
|
||||
}
|
||||
|
||||
return prompt
|
||||
}
|
||||
|
||||
@@ -312,6 +312,9 @@ func (h *trackerMappingHandler) resolveDeterministic(
|
||||
return res, fmt.Errorf("cannot load cookie banner for domain filtering: %w", err)
|
||||
}
|
||||
|
||||
// DEBUG: don't commit
|
||||
banner.Origin = "https://t.probo.com"
|
||||
|
||||
res.origin = banner.Origin
|
||||
|
||||
if tp.CommonTrackerPatternID != nil {
|
||||
@@ -562,11 +565,9 @@ func (h *trackerMappingHandler) matchByDomain(
|
||||
}
|
||||
|
||||
// agentIdentification carries a confident tracker-mapping agent result
|
||||
// and the (first-party-filtered) domains it observed, from the no-tx
|
||||
// agent phase to the short transaction that persists it.
|
||||
// from the no-tx agent phase to the short transaction that persists it.
|
||||
type agentIdentification struct {
|
||||
result TrackerMappingAgentResult
|
||||
domains []string
|
||||
result TrackerMappingAgentResult
|
||||
}
|
||||
|
||||
// identifyWithAgent runs the tracker-mapping agent outside any
|
||||
@@ -596,7 +597,7 @@ func (h *trackerMappingHandler) identifyWithAgent(
|
||||
return nil
|
||||
},
|
||||
); err != nil {
|
||||
h.logger.WarnCtx(ctx, "cannot load initiator domains for agent", log.Error(err))
|
||||
return nil, fmt.Errorf("cannot load initiator domains for agent: %w", err)
|
||||
}
|
||||
|
||||
domains = uri.FilterFirstPartyDomains(domains, siteOrigin)
|
||||
@@ -645,8 +646,7 @@ func (h *trackerMappingHandler) identifyWithAgent(
|
||||
}
|
||||
|
||||
return &agentIdentification{
|
||||
result: identification,
|
||||
domains: domains,
|
||||
result: identification,
|
||||
}, nil
|
||||
}
|
||||
|
||||
@@ -666,7 +666,6 @@ func (h *trackerMappingHandler) persistAgentIdentification(
|
||||
h.logger,
|
||||
ident.result.ThirdPartyName,
|
||||
ident.result.Category,
|
||||
ident.domains,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("cannot resolve or create common third party: %w", err)
|
||||
|
||||
Reference in New Issue
Block a user