fixup! [runtime][java] Support use Long-Term Memory in action.

address comments.

Author: wenjin272

e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/VectorStoreLongTermMemoryTest.java

@@ -50,6 +50,8 @@
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.TestInfo;
 import org.mockito.Mockito;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -71,6 +73,8 @@
  */
 @Disabled("Should setup Elasticsearch server.")
 public class VectorStoreLongTermMemoryTest {
+    private static final Logger LOG = LoggerFactory.getLogger(VectorStoreLongTermMemoryTest.class);
+
     private static final String NAME = "chat-history";
     private final VectorStoreLongTermMemory ltm;
     private MemorySet memorySet;
@@ -318,7 +322,7 @@ public void testUsingLtmInAction() throws Exception {
             agentsEnv.execute();
 
             // check async compaction
-            System.out.print(es.get(null, "ltm_test_job-2-test-ltm", Collections.emptyMap()));
+            LOG.debug(es.get(null, "ltm_test_job-2-test-ltm", Collections.emptyMap()).toString());
             Assertions.assertEquals(1, es.size("ltm_test_job-2-test-ltm"));
         } finally {
             es.deleteCollection("ltm_test_job-1-test-ltm");

runtime/src/main/java/org/apache/flink/agents/runtime/context/RunnerContextImpl.java

@@ -30,6 +30,7 @@
 import org.apache.flink.agents.plan.AgentPlan;
 import org.apache.flink.agents.plan.utils.JsonUtils;
 import org.apache.flink.agents.runtime.memory.CachedMemoryStore;
+import org.apache.flink.agents.runtime.memory.KeyedMemory;
 import org.apache.flink.agents.runtime.memory.MemoryObjectImpl;
 import org.apache.flink.agents.runtime.memory.VectorStoreLongTermMemory;
 import org.apache.flink.agents.runtime.metrics.FlinkAgentsMetricGroupImpl;
@@ -107,7 +108,7 @@ public void switchActionContext(String actionName, MemoryContext memoryContext,
         this.actionName = actionName;
         this.memoryContext = memoryContext;
         if (ltm != null) {
-            ltm.switchContext(key);
+            ((KeyedMemory) ltm).switchContext(key);
         }
     }