fixing the circuit breaker issue for remote model (#3652) (#3654)

Signed-off-by: Dhrubo Saha <[email protected]>
(cherry picked from commit 70391fc)

Co-authored-by: Dhrubo Saha <[email protected]>

Author: opensearch-trigger-bot[bot]

plugin/src/main/java/org/opensearch/ml/task/MLTaskRunner.java

@@ -87,7 +87,6 @@ protected void handleAsyncMLTaskComplete(MLTask mlTask) {
     public void run(FunctionName functionName, Request request, TransportService transportService, ActionListener<Response> listener) {
         if (!request.isDispatchTask()) {
             log.debug("Run ML request {} locally", request.getRequestID());
-            checkOpenCircuitBreaker(mlCircuitBreakerService, mlStats);
             checkCBAndExecute(functionName, request, listener);
             return;
         }

plugin/src/test/java/org/opensearch/ml/task/TaskRunnerTests.java

@@ -139,8 +139,20 @@ public void testRun_CircuitBreakerOpen() {
         TransportService transportService = mock(TransportService.class);
         ActionListener listener = mock(ActionListener.class);
         MLTaskRequest request = new MLTaskRequest(false);
-        expectThrows(CircuitBreakingException.class, () -> mlTaskRunner.run(FunctionName.REMOTE, request, transportService, listener));
+        expectThrows(CircuitBreakingException.class, () -> mlTaskRunner.run(FunctionName.BATCH_RCF, request, transportService, listener));
         Long value = (Long) mlStats.getStat(MLNodeLevelStat.ML_CIRCUIT_BREAKER_TRIGGER_COUNT).getValue();
         assertEquals(1L, value.longValue());
     }
+
+    public void testRun_NoCircuitbreakerforRemote() {
+        when(mlCircuitBreakerService.checkOpenCB()).thenReturn(thresholdCircuitBreaker);
+        when(thresholdCircuitBreaker.getName()).thenReturn("Memory Circuit Breaker");
+        when(thresholdCircuitBreaker.getThreshold()).thenReturn(87);
+        TransportService transportService = mock(TransportService.class);
+        ActionListener listener = mock(ActionListener.class);
+        MLTaskRequest request = new MLTaskRequest(false);
+        mlTaskRunner.run(FunctionName.REMOTE, request, transportService, listener);
+        Long value = (Long) mlStats.getStat(MLNodeLevelStat.ML_CIRCUIT_BREAKER_TRIGGER_COUNT).getValue();
+        assertEquals(0L, value.longValue());
+    }
 }