Add Agent Memory Search API and UI functionality

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

Author: Copilot

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

@@ -38,6 +38,7 @@
 import io.sentrius.sso.core.services.UserService;
 import io.sentrius.sso.core.services.agents.AgentClientService;
 import io.sentrius.sso.core.services.agents.AgentContextService;
+import io.sentrius.sso.core.services.agents.AgentMemorySearchService;
 import io.sentrius.sso.core.services.agents.AgentService;
 import io.sentrius.sso.core.services.auditing.AuditService;
 import io.sentrius.sso.core.services.security.CryptoService;
@@ -85,6 +86,7 @@ public class AgentApiController extends BaseController {
     final ZeroTrustRequestService ztatRequestService;
     final AgentContextService agentContextService;
     final AgentClientService agentClientService;
+    final AgentMemorySearchService agentMemorySearchService;
     final AppConfig appConfig;
 
     public AgentApiController(
@@ -96,7 +98,8 @@ public AgentApiController(
         ATPLPolicyService atplPolicyService,
         ZeroTrustAccessTokenService ztatService, ZeroTrustRequestService ztrService, AgentService agentService,
         ProvenanceKafkaProducer provenanceKafkaProducer, ZeroTrustRequestService ztatRequestService,
-        AgentContextService agentContextService, AgentClientService agentClientService, AppConfig appConfig
+        AgentContextService agentContextService, AgentClientService agentClientService, 
+        AgentMemorySearchService agentMemorySearchService, AppConfig appConfig
     ) {
         super(userService, systemOptions, errorOutputService);
         this.auditService = auditService;
@@ -111,6 +114,7 @@ public AgentApiController(
         this.ztatRequestService = ztatRequestService;
         this.agentContextService = agentContextService;
         this.agentClientService = agentClientService;
+        this.agentMemorySearchService = agentMemorySearchService;
         this.appConfig = appConfig;
     }
 
@@ -826,4 +830,50 @@ public ResponseEntity<AgentContextDTO> createContext(
         return ResponseEntity.ok(dto);
     }
 
+    @GetMapping("/memory/search")
+    @LimitAccess(applicationAccess = {ApplicationAccessEnum.CAN_MANAGE_APPLICATION})
+    public ResponseEntity<?> searchAgentMemory(
+        @RequestParam(required = false) String content,
+        @RequestParam(required = false) String agent,
+        @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime startDate,
+        @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime endDate,
+        @RequestParam(defaultValue = "0") int page,
+        @RequestParam(defaultValue = "20") int size
+    ) {
+        try {
+            log.info("Searching agent memory with parameters - content: '{}', agent: '{}', startDate: {}, endDate: {}, page: {}, size: {}", 
+                     content, agent, startDate, endDate, page, size);
+
+            // Determine search type and execute appropriate search
+            if (content != null && !content.trim().isEmpty()) {
+                if (startDate != null && endDate != null) {
+                    // Search by content and date range
+                    var results = agentMemorySearchService.searchByContentAndDateRange(
+                        content.trim(), startDate, endDate, page, size);
+                    return ResponseEntity.ok(results);
+                } else if (agent != null && !agent.trim().isEmpty()) {
+                    // Search by specific agent and content
+                    var results = agentMemorySearchService.searchByAgentAndContent(
+                        agent.trim(), content.trim(), page, size);
+                    return ResponseEntity.ok(results);
+                } else {
+                    // Search by content only
+                    var results = agentMemorySearchService.searchByContent(content.trim(), page, size);
+                    return ResponseEntity.ok(results);
+                }
+            } else if (agent != null && !agent.trim().isEmpty()) {
+                // Search by agent only
+                var results = agentMemorySearchService.searchByAgent(agent.trim(), page, size);
+                return ResponseEntity.ok(results);
+            } else {
+                // Return all memories with pagination
+                var results = agentMemorySearchService.getAllMemories(page, size);
+                return ResponseEntity.ok(results);
+            }
+        } catch (Exception e) {
+            log.error("Error searching agent memory", e);
+            return ResponseEntity.status(500).body("Error searching agent memory: " + e.getMessage());
+        }
+    }
+
 }

api/src/main/java/io/sentrius/sso/controllers/view/AgentController.java

@@ -65,5 +65,10 @@ public String listConnections(Model m, @RequestParam("agentId") String agentId)
         return "sso/agents/agent_comms";
     }
 
+    @GetMapping("/memory/search")
+    @LimitAccess(applicationAccess = {ApplicationAccessEnum.CAN_MANAGE_APPLICATION})
+    public String searchAgentMemory(Model m) {
+        return "sso/agents/memory_search";
+    }
 
 }

api/src/main/resources/templates/fragments/sidebar.html

@@ -55,6 +55,16 @@
                     <i class="fa-solid fa-user"></i> <span class="ms-1 d-none d-sm-inline">Manage Agent/Users</span>
                 </a>
             </li>
+            <li th:if="${#sets.contains(operatingUser.authorizationType.accessSet, 'CAN_MANAGE_APPLICATION')}">
+                <a href="/sso/v1/agent/list" class="nav-link px-0 align-middle">
+                    <i class="fas fa-robot"></i> <span class="ms-1 d-none d-sm-inline">Agents</span>
+                </a>
+            </li>
+            <li th:if="${#sets.contains(operatingUser.authorizationType.accessSet, 'CAN_MANAGE_APPLICATION')}">
+                <a href="/sso/v1/agent/memory/search" class="nav-link px-0 align-middle">
+                    <i class="fas fa-search"></i> <span class="ms-1 d-none d-sm-inline">Agent Memory</span>
+                </a>
+            </li>
             <li th:if="${#sets.contains(operatingUser.authorizationType.accessSet, 'CAN_MANAGE_APPLICATION')}">
                 <a href="/sso/v1/atpl/" class="nav-link px-0 align-middle">
                     <i class="fas fa-shield-alt"></i> <span class="ms-1 d-none d-sm-inline">Trust Policies</span>