Skip to content

Commit 8535eca

Browse files
Zhangxunmtakolarkunnu
authored andcommitted
update CB setting to 100 to bypass memory check (opensearch-project#3627)
Signed-off-by: Xun Zhang <xunzh@amazon.com>
1 parent 4240810 commit 8535eca

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java

+11
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,17 @@ public void testPredictWithAutoDeployAndTTL_RemoteModel() throws IOException, In
222222
System.out.println("OPENAI_KEY is null");
223223
return;
224224
}
225+
Response updateCBSettingResponse = TestHelper
226+
.makeRequest(
227+
client(),
228+
"PUT",
229+
"_cluster/settings",
230+
null,
231+
"{\"persistent\":{\"plugins.ml_commons.jvm_heap_memory_threshold\":100}}",
232+
ImmutableList.of(new BasicHeader(HttpHeaders.USER_AGENT, ""))
233+
);
234+
assertEquals(200, updateCBSettingResponse.getStatusLine().getStatusCode());
235+
225236
Response response = createConnector(completionModelConnectorEntity);
226237
Map responseMap = parseResponseToMap(response);
227238
String connectorId = (String) responseMap.get("connector_id");

0 commit comments

Comments
 (0)