fix failed ut (#4459)

Signed-off-by: Yaliang Wu <[email protected]>

Author: ylwu-amzn

common/src/test/java/org/opensearch/ml/common/agent/MLAgentTest.java

@@ -690,11 +690,11 @@ public void writeTo_ReadFrom_ContextManagementName() throws IOException {
         );
 
         BytesStreamOutput output = new BytesStreamOutput();
-        output.setVersion(CommonValue.VERSION_3_3_0);
+        output.setVersion(CommonValue.VERSION_3_4_0);
         agent.writeTo(output);
 
         StreamInput streamInput = output.bytes().streamInput();
-        streamInput.setVersion(CommonValue.VERSION_3_3_0);
+        streamInput.setVersion(CommonValue.VERSION_3_4_0);
         MLAgent deserializedAgent = new MLAgent(streamInput);
 
         assertEquals("template_name", deserializedAgent.getContextManagementTemplateName());
@@ -731,11 +731,11 @@ public void writeTo_ReadFrom_ContextManagementInline() throws IOException {
         );
 
         BytesStreamOutput output = new BytesStreamOutput();
-        output.setVersion(CommonValue.VERSION_3_3_0);
+        output.setVersion(CommonValue.VERSION_3_4_0);
         agent.writeTo(output);
 
         StreamInput streamInput = output.bytes().streamInput();
-        streamInput.setVersion(CommonValue.VERSION_3_3_0);
+        streamInput.setVersion(CommonValue.VERSION_3_4_0);
         MLAgent deserializedAgent = new MLAgent(streamInput);
 
         assertNull(deserializedAgent.getContextManagementTemplateName());

plugin/src/main/java/org/opensearch/ml/action/memorycontainer/TransportCreateMemoryContainerAction.java

@@ -123,7 +123,7 @@ public TransportCreateMemoryContainerAction(
         this.remoteMemoryStoreHelper = remoteMemoryStoreHelper;
         trustedConnectorEndpointsRegex = ML_COMMONS_TRUSTED_CONNECTOR_ENDPOINTS_REGEX.get(settings);
         this.memoryContainerPipelineHelper = memoryContainerPipelineHelper;
-        clusterService
+        this.clusterService
             .getClusterSettings()
             .addSettingsUpdateConsumer(ML_COMMONS_TRUSTED_CONNECTOR_ENDPOINTS_REGEX, it -> trustedConnectorEndpointsRegex = it);
     }