PII 마스킹 — 입력 전처리
@Component
public class PiiMaskingService {
// 마스킹 규칙
private static final Map<String, String> PATTERNS = Map.of(
"CARD", "(\\d{4}[- ]?){3}\\d{4}",
"SSN", "\\d{6}-\\d{7}",
"PHONE","01[016789]-?\\d{3,4}-?\\d{4}",
"EMAIL","[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\\.[a-zA-Z]{2,}"
);
public MaskedInput mask(String text) {
Map<String, String> tokens = new HashMap<>();
String masked = text;
for (var entry : PATTERNS.entrySet()) {
var m = Pattern.compile(entry.getValue())
.matcher(masked);
while (m.find()) {
// PII → 토큰으로 치환
String token = "[" + entry.getKey()
+ "_" + tokens.size() + "]";
tokens.put(token, m.group());
masked = masked.replace(m.group(), token);
}
}
return new MaskedInput(masked, tokens);
}
// LLM 응답에서 토큰을 원래 값으로 복원
public String unmask(String response,
Map<String,String> tokens) {
String result = response;
for (var e : tokens.entrySet())
result = result.replace(e.getKey(), e.getValue());
return result;
}
}
출력 PII 검사 — 응답 필터링
@Component
public class OutputGuardrail {
private final PiiMaskingService piiService;
public String filter(String response) {
// 응답에 PII가 포함됐는지 검사
MaskedInput check = piiService.mask(response);
if (!check.getTokens().isEmpty()) {
// PII 발견: 응답에서 마스킹
log.warn("PII detected in LLM output: {}",
check.getTokens().keySet());
return check.getMaskedText();
}
return response;
}
}
// 전체 파이프라인
public String safeChat(String userId, String rawInput) {
// 1. 입력 PII 마스킹
var masked = piiService.mask(rawInput);
// 2. 마스킹된 텍스트로 LLM 호출
String llmResponse = llm.call(masked.getMaskedText());
// 3. 응답에서 추가 PII 필터링
String filtered = outputGuardrail.filter(llmResponse);
// 4. 토큰 복원 (필요 시)
return piiService.unmask(filtered, masked.getTokens());
}