diff --git a/sdk/src/main/java/com/atlan/model/lineage/FluentLineage.java b/sdk/src/main/java/com/atlan/model/lineage/FluentLineage.java index 4e661dc1ef..7e2163b9ff 100644 --- a/sdk/src/main/java/com/atlan/model/lineage/FluentLineage.java +++ b/sdk/src/main/java/com/atlan/model/lineage/FluentLineage.java @@ -11,6 +11,7 @@ import com.atlan.model.enums.AtlanStatus; import com.atlan.model.fields.*; import com.atlan.model.relations.Reference; +import com.fasterxml.jackson.annotation.JsonProperty; import java.util.List; import java.util.stream.Collectors; import java.util.stream.Stream; @@ -81,6 +82,11 @@ public static FluentLineageBuilder builder(AtlanClient client, String startingGu @Builder.Default FilterList.Condition relationshipsCondition = FilterList.Condition.AND; + /** Whether to include immediate neighbors of the starting asset in the response. */ + @Builder.Default + @JsonProperty("immediateNeighbours") + Boolean immediateNeighbors = false; + /** * Assets to include in the results. Any assets not matching these filters will not be included in the results, * but will still be traversed in the lineage so that any assets beyond them are still considered for inclusion @@ -148,6 +154,9 @@ public static FluentLineageBuilder builder(AtlanClient client, String startingGu } request.relationshipTraversalFilters(relationships.build()); } + if (immediateNeighbors != null) { + request.immediateNeighbors(immediateNeighbors); + } if (includesInResults != null) { FilterList.FilterListBuilder entities = FilterList.builder().condition(includesCondition); for (LineageFilter asset : includesInResults) {