diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionService.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionService.java index f1fb0fd4..183bdd13 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionService.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionService.java @@ -1,5 +1,7 @@ package com.iqser.red.service.redaction.v1.server.redaction.service; +import static com.iqser.red.service.redaction.v1.server.redaction.model.Dictionary.RECOMMENDATION_PREFIX; + import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; @@ -263,7 +265,7 @@ public class EntityRedactionService { .containsKey(sectionNumber)) { nerEntities.getResult().get(sectionNumber).forEach(res -> { if (cellstarts == null || cellstarts.isEmpty()) { - nerValuesPerType.computeIfAbsent(res.getType(), (a) -> new HashSet<>()) + nerValuesPerType.computeIfAbsent(RECOMMENDATION_PREFIX + res.getType(), (a) -> new HashSet<>()) .add(new String(Base64.decodeBase64(res.getValue().getBytes()))); } else { boolean intersectsCellStart = false; @@ -273,7 +275,7 @@ public class EntityRedactionService { } } if (!intersectsCellStart) { - nerValuesPerType.computeIfAbsent(res.getType(), (a) -> new HashSet<>()) + nerValuesPerType.computeIfAbsent(RECOMMENDATION_PREFIX + res.getType(), (a) -> new HashSet<>()) .add(new String(Base64.decodeBase64(res.getValue().getBytes()))); } }