fix: rate limit streaming prediction

ntua-unit-of-control-and-informatics · Dec 19, 2024 · db9ae59 · db9ae59
1 parent 53e01a5
commit db9ae59
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main/kotlin/org/jaqpot/api/service/model/ModelService.kt b/src/main/kotlin/org/jaqpot/api/service/model/ModelService.kt
@@ -290,6 +290,11 @@ class ModelService(
         throw ResponseStatusException(HttpStatus.BAD_REQUEST, "Unknown dataset type", null)
     }
 
+    @PreAuthorize("@predictModelAuthorizationLogic.decide(#root, #modelId)")
+    @WithRateLimitProtectionByUser(
+        limit = 30,
+        intervalInSeconds = 60 * 60
+    ) // 30 requests per hour, up to 100 predictions per request
     fun streamPredictWithModel(
         modelId: Long,
         datasetId: Long,