OPIK-1386 move guardrails logic to a new resource failing test green

Author: idoberko2

Diff for: apps/opik-backend/src/main/java/com/comet/opik/api/resources/v1/priv/GuardrailsResource.java

@@ -0,0 +1,63 @@
+package com.comet.opik.api.resources.v1.priv;
+
+import com.codahale.metrics.annotation.Timed;
+import com.comet.opik.api.GuardrailBatch;
+import com.comet.opik.domain.GuardrailsService;
+import com.comet.opik.infrastructure.auth.RequestContext;
+import com.comet.opik.infrastructure.ratelimit.RateLimited;
+import com.comet.opik.utils.AsyncUtils;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.media.Content;
+import io.swagger.v3.oas.annotations.media.Schema;
+import io.swagger.v3.oas.annotations.parameters.RequestBody;
+import io.swagger.v3.oas.annotations.responses.ApiResponse;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import jakarta.inject.Provider;
+import jakarta.validation.Valid;
+import jakarta.validation.constraints.NotNull;
+import jakarta.ws.rs.Consumes;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import lombok.NonNull;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+
+import static com.comet.opik.utils.AsyncUtils.setRequestContext;
+
+@Path("/v1/private/guardrails")
+@Produces(MediaType.APPLICATION_JSON)
+@Consumes(MediaType.APPLICATION_JSON)
+@Timed
+@Slf4j
+@RequiredArgsConstructor(onConstructor_ = @jakarta.inject.Inject)
+@Tag(name = "Guardrails", description = "Guardrails related resources")
+public class GuardrailsResource {
+    private final @NonNull GuardrailsService guardrailsService;
+    private final @NonNull Provider<RequestContext> requestContext;
+
+    @POST
+    @Operation(operationId = "addGuardrailsBatch", summary = "Batch guardrails for traces", description = "Batch guardrails for traces", responses = {
+            @ApiResponse(responseCode = "204", description = "No Content")})
+    @RateLimited
+    public Response addGuardrailsBatch(
+            @RequestBody(content = @Content(schema = @Schema(implementation = GuardrailBatch.class))) @NotNull @Valid GuardrailBatch batch) {
+
+        String workspaceId = requestContext.get().getWorkspaceId();
+
+        log.info("Add guardrails batch, size {} on  workspaceId '{}'", batch.guardrails().size(),
+                workspaceId);
+
+        guardrailsService.addTraceGuardrails(batch.guardrails())
+                .contextWrite(ctx -> setRequestContext(ctx, requestContext))
+                .retryWhen(AsyncUtils.handleConnectionError())
+                .block();
+
+        log.info("Done adding guardrails batch, size {} on  workspaceId '{}'", batch.guardrails().size(),
+                workspaceId);
+
+        return Response.noContent().build();
+    }
+}

Diff for: apps/opik-backend/src/main/java/com/comet/opik/api/resources/v1/priv/TracesResource.java

@@ -9,7 +9,6 @@
 import com.comet.opik.api.FeedbackScore;
 import com.comet.opik.api.FeedbackScoreBatch;
 import com.comet.opik.api.FeedbackScoreNames;
-import com.comet.opik.api.GuardrailBatch;
 import com.comet.opik.api.ProjectStats;
 import com.comet.opik.api.Trace;
 import com.comet.opik.api.Trace.TracePage;
@@ -26,7 +25,6 @@
 import com.comet.opik.domain.CommentDAO;
 import com.comet.opik.domain.CommentService;
 import com.comet.opik.domain.FeedbackScoreService;
-import com.comet.opik.domain.GuardrailsService;
 import com.comet.opik.domain.Streamer;
 import com.comet.opik.domain.TraceService;
 import com.comet.opik.domain.workspaces.WorkspaceMetadata;
@@ -92,7 +90,6 @@ public class TracesResource {
 
     private final @NonNull TraceService service;
     private final @NonNull FeedbackScoreService feedbackScoreService;
-    private final @NonNull GuardrailsService guardrailsService;
     private final @NonNull CommentService commentService;
     private final @NonNull FiltersFactory filtersFactory;
     private final @NonNull WorkspaceMetadataService workspaceMetadataService;
@@ -429,30 +426,6 @@ public Response scoreBatchOfTraces(
         return Response.noContent().build();
     }
 
-    @PUT
-    @Path("/guardrails")
-    @Operation(operationId = "addGuardrailsBatch", summary = "Batch guardrails for traces", description = "Batch guardrails for traces", responses = {
-            @ApiResponse(responseCode = "204", description = "No Content")})
-    @RateLimited
-    public Response addGuardrailsBatch(
-            @RequestBody(content = @Content(schema = @Schema(implementation = GuardrailBatch.class))) @NotNull @Valid GuardrailBatch batch) {
-
-        String workspaceId = requestContext.get().getWorkspaceId();
-
-        log.info("Add guardrails batch for traces, size {} on  workspaceId '{}'", batch.guardrails().size(),
-                workspaceId);
-
-        guardrailsService.addTraceGuardrails(batch.guardrails())
-                .contextWrite(ctx -> setRequestContext(ctx, requestContext))
-                .retryWhen(AsyncUtils.handleConnectionError())
-                .block();
-
-        log.info("Done adding guardrails batch for traces, size {} on  workspaceId '{}'", batch.guardrails().size(),
-                workspaceId);
-
-        return Response.noContent().build();
-    }
-
     @GET
     @Path("/stats")
     @Operation(operationId = "getTraceStats", summary = "Get trace stats", description = "Get trace stats", responses = {