From e33e1a5f79821ae992dc0fb19128fecdb70cdd2e Mon Sep 17 00:00:00 2001 From: deiflaender Date: Wed, 12 Jan 2022 13:03:28 +0100 Subject: [PATCH] RED-3172: Add AI Entities as recommendations --- .../v1/server/redaction/service/EntityRedactionService.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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()))); } }