Skip to content

fix(insights): attribute AI-bot-applied tags and tier as Generated #30595

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Shreyansh100704 wants to merge 3 commits into from
+39 −1
Open

fix(insights): attribute AI-bot-applied tags and tier as Generated #30595

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
61025e2
fix(insights): attribute AI-bot-applied tags and tier as Generated
Shreyansh100704 Jul 28, 2026
3166216
Merge branch 'main' into fix/ai-generated-tag-tier-attribution
Shreyansh100704 Jul 28, 2026
eadf24e
Merge branch 'main' into fix/ai-generated-tag-tier-attribution
Shreyansh100704 Jul 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 11 additions & 1 deletion openmetadata-service/src/main/java/org/openmetadata/service/search/SearchIndexUtils.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -565,7 +565,7 @@ private static void processTagAndTierSources(
if (tagFQN == null || labelType == null) {
continue;
}
String tagSource = labelType.value();
String tagSource = resolveTagSource(tag, labelType);
Map<String, Integer> bucket =
tagFQN.startsWith("Tier.")
? tagAndTierSources.getTierSources()
Expand All @@ -574,6 +574,16 @@ private static void processTagAndTierSources(
}
}

// AI-bot-applied tags count as Generated regardless of labelType, mirroring
// getDescriptionSource. Applier is recorded per-tag in appliedBy (set server-side).
private static String resolveTagSource(TagLabel tag, TagLabel.LabelType labelType) {
String tagSource = labelType.value();
if (tag.getAppliedBy() != null && AI_BOTS.contains(tag.getAppliedBy())) {
tagSource = TagLabel.LabelType.GENERATED.value();
}
return tagSource;
}

private static void processEntityTagSources(
EntityInterface entity, TagAndTierSources tagAndTierSources) {
processTagAndTierSources(entity.getTags(), tagAndTierSources);
Expand Down
28 changes: 28 additions & 0 deletions openmetadata-service/src/test/java/org/openmetadata/service/search/SearchIndexUtilsTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -219,6 +219,34 @@ void testDescriptionAndTagSourceProcessing() {
assertEquals(2, descriptionSources.get(ChangeSource.SUGGESTED.value()));
}

@Test
void testAiBotAppliedTagsCountAsGenerated() {
TagLabel botTag =
new TagLabel()
.withTagFQN("PII.Sensitive")
.withLabelType(TagLabel.LabelType.MANUAL)
.withAppliedBy("collateaiapplicationbot");
TagLabel botTier =
new TagLabel()
.withTagFQN("Tier.Tier1")
.withLabelType(TagLabel.LabelType.MANUAL)
.withAppliedBy("aiautomationapplicationbot");
TagLabel humanTag =
new TagLabel()
.withTagFQN("PersonalData.Personal")
.withLabelType(TagLabel.LabelType.MANUAL)
.withAppliedBy("analyst");
Table table = new Table().withTags(List.of(botTag, botTier, humanTag));

SearchIndexUtils.TagAndTierSources tagAndTierSources =
SearchIndexUtils.processTagAndTierSources(table);

assertEquals(1, tagAndTierSources.getTagSources().get(TagLabel.LabelType.GENERATED.value()));
assertEquals(1, tagAndTierSources.getTierSources().get(TagLabel.LabelType.GENERATED.value()));
assertEquals(1, tagAndTierSources.getTagSources().get(TagLabel.LabelType.MANUAL.value()));
assertNull(tagAndTierSources.getTierSources().get(TagLabel.LabelType.MANUAL.value()));
}

@Test
void testSearchAggregationBuilderHelpers() {
SearchAggregationNode termNode = SearchAggregation.terms("team", "owner.name", 25);
Expand Down
Loading