fix: feature controller (#629)

epam · Dec 27, 2024 · 8c414cc · 8c414cc
1 parent 8594dc7
commit 8c414cc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/server/src/main/java/com/epam/aidial/core/server/controller/DeploymentFeatureController.java b/server/src/main/java/com/epam/aidial/core/server/controller/DeploymentFeatureController.java
@@ -34,7 +34,8 @@ public DeploymentFeatureController(Proxy proxy, ProxyContext context) {
     }
 
     public Future<?> handle(String deploymentId, Function<Deployment, String> endpointGetter, boolean requireEndpoint) {
-        DeploymentController.selectDeployment(context, deploymentId).map(dep -> {
+        // make sure request.body() called before request.resume()
+        return DeploymentController.selectDeployment(context, deploymentId).map(dep -> {
             String endpoint = endpointGetter.apply(dep);
             context.setDeployment(dep);
             context.getRequest().body()
@@ -45,8 +46,6 @@ public Future<?> handle(String deploymentId, Function<Deployment, String> endpoi
             handleRequestError(deploymentId, error);
             return null;
         });
-
-        return Future.succeededFuture();
     }
 
     @SneakyThrows