From d4fe58080158609b88f3d51438c2f98fa24b20d4 Mon Sep 17 00:00:00 2001
From: Xun Zhang <xunzh@amazon.com>
Date: Wed, 5 Mar 2025 13:18:25 -0800
Subject: [PATCH] update CB setting to 100 to bypass memory check

Signed-off-by: Xun Zhang <xunzh@amazon.com>
---
 .../opensearch/ml/rest/RestMLRemoteInferenceIT.java   | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java
index bbaa6b32ce..0b8e713f06 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java
@@ -222,6 +222,17 @@ public void testPredictWithAutoDeployAndTTL_RemoteModel() throws IOException, In
             System.out.println("OPENAI_KEY is null");
             return;
         }
+        Response updateCBSettingResponse = TestHelper
+            .makeRequest(
+                client(),
+                "PUT",
+                "_cluster/settings",
+                null,
+                "{\"persistent\":{\"plugins.ml_commons.jvm_heap_memory_threshold\":100}}",
+                ImmutableList.of(new BasicHeader(HttpHeaders.USER_AGENT, ""))
+            );
+        assertEquals(200, updateCBSettingResponse.getStatusLine().getStatusCode());
+
         Response response = createConnector(completionModelConnectorEntity);
         Map responseMap = parseResponseToMap(response);
         String connectorId = (String) responseMap.get("connector_id");