Skip to content

Commit af628f3

Browse files
fix: do not block request when model detects no content violation
1 parent b1ff448 commit af628f3

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/main/java/io/gravitee/policy/ai/prompt/guard/rails/AiPromptGuardRailsPolicy.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,9 @@ private CompletableSource checkContent(HttpPlainExecutionContext ctx) {
8181
.flatMapCompletable(classifierResults -> {
8282
var detectedContentTypes = detectClassifierResultContentTypes(classifierResults, sensitivityThreshold);
8383
if (
84-
configuration.parseContentChecks().isEmpty() ||
85-
configuration.parseContentChecks().stream().anyMatch(detectedContentTypes::contains)
84+
!detectedContentTypes.isEmpty() &&
85+
(configuration.parseContentChecks().isEmpty() ||
86+
configuration.parseContentChecks().stream().anyMatch(detectedContentTypes::contains))
8687
) {
8788
logMetrics(detectedContentTypes, ctx, configuration.requestPolicy().getAction());
8889
if (RequestPolicy.BLOCK_REQUEST.equals(configuration.requestPolicy())) {

0 commit comments

Comments
 (0)