fix: RouteController doesn't mark upstream as failed in case of unsuc…

…cessful http code #602 (#603)
epam · Dec 5, 2024 · c8a17e5 · c8a17e5
1 parent 99cfcab
commit c8a17e5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/src/main/java/com/epam/aidial/core/server/controller/RouteController.java b/server/src/main/java/com/epam/aidial/core/server/controller/RouteController.java
@@ -158,6 +158,8 @@ private void handleProxyResponse(HttpClientResponse proxyResponse) {
 
         if (responseStatusCode == 200) {
             context.getUpstreamRoute().succeed();
+        } else {
+            context.getUpstreamRoute().fail(proxyResponse);
         }
 
         BufferingReadStream proxyResponseStream = new BufferingReadStream(proxyResponse,