Add userId to agent memory markings and persist conversation history (#49)

* Initial plan

* Fix agent memory markings to include userId for privacy scoping

Co-authored-by: phrocker <[email protected]>

* Apply same userId marking fix to ChatWSHandler websocket path

Co-authored-by: phrocker <[email protected]>

* Add persistent memory storage for user messages and agent responses in ChatWSHandler

Co-authored-by: phrocker <[email protected]>

* Update

---------

Co-authored-by: copilot-swe-agent[bot] <[email protected]>
Co-authored-by: phrocker <[email protected]>
Co-authored-by: Marc Parisi <[email protected]>

Author: Copilot

api/src/main/java/io/sentrius/sso/controllers/api/agents/AgentMemoryController.java

@@ -63,7 +63,7 @@ public AgentMemoryController(PersistentAgentMemoryStore memoryStore, VectorAgent
     public ResponseEntity<List<AgentMemoryDTO>> storeMemories(
         @RequestParam(name = "agentId") String agentId,
         @RequestBody List<AgentMemoryDTO> memoryDTOs,
-        @RequestParam(defaultValue = "false") boolean generateEmbedding,
+        @RequestParam(defaultValue = "true") boolean generateEmbedding,
         HttpServletRequest request, HttpServletResponse response) {
 
 

core/src/main/java/io/sentrius/sso/core/dto/agents/AgentExecutionContextDTO.java

@@ -86,6 +86,10 @@ public void addToPersistentMemory(String key, JsonNode value, String classificat
             // Add to data list with persistent flag
             longTermMemories.put(key, memoryMeta);
 
+            log.info("Added persistent memory meta for {}", key);
+
+        } else {
+            log.warn("AgentContext is null, cannot add to persistent memory");
         }
     }
 
@@ -121,10 +125,13 @@ public boolean hasPersistentMemory() {
     }
 
     public Map<String, JsonNode> flushPersistentMemory() {
-        var persistentItems = getPersistentMemoryItems();
+        // Deep copy each entry to avoid returning references to mutable JsonNodes
+        Map<String, JsonNode> copy = new HashMap<>();
+        for (Map.Entry<String, JsonNode> entry : getPersistentMemoryItems().entrySet()) {
+            copy.put(entry.getKey(), entry.getValue().deepCopy());
+        }
         longTermMemories.clear();
-        return persistentItems;
-
+        return copy;
 
     }
 

core/src/main/java/io/sentrius/sso/core/services/agents/AgentClientService.java

@@ -27,6 +27,7 @@
 import io.sentrius.sso.core.utils.JsonUtil;
 import io.sentrius.sso.provenance.ProvenanceEvent;
 import lombok.extern.slf4j.Slf4j;
+import org.aspectj.weaver.loadtime.Agent;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnWebApplication;
 import org.springframework.stereotype.Service;
@@ -320,7 +321,8 @@ public void storeMemory(AgentExecution execution, String agentId, AgentMemoryDTO
         String url = "/api/v1/agents/memory/store";
         try {
             log.info("Storing memory for agentId {}: {}", agentId, memory.getMemoryKey());
-            zeroTrustClientService.callPostOnApi(execution, url,memory,
+            List<AgentMemoryDTO> dto = List.of(memory);
+            zeroTrustClientService.callPostOnApi(execution, url,dto,
                 Maps.immutableEntry("agentId",List.of(agentId)) );
         } catch (ZtatException e) {
             log.error("Failed to store memory for key {}", e.getMessage());