Remove test

Author: phrocker

ai-agent/src/main/java/io/sentrius/agent/services/EndpointRegistry.java

@@ -11,11 +11,9 @@
 import io.sentrius.sso.core.dto.agents.AgentMemoryDTO;
 import io.sentrius.sso.core.dto.agents.MemoryQueryDTO;
 import io.sentrius.sso.core.dto.capabilities.EndpointDescriptor;
-import io.sentrius.sso.core.dto.ztat.TokenDTO;
-import io.sentrius.sso.core.embeddings.EmbeddingService;
+import io.sentrius.sso.core.embeddings.EmbeddingServiceIfc;
 import io.sentrius.sso.core.exceptions.ZtatException;
 import io.sentrius.sso.core.services.agents.AgentClientService;
-import jakarta.annotation.PostConstruct;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
@@ -29,7 +27,7 @@ public class EndpointRegistry {
     Map<String, EndpointDescriptor> descriptorMap = new HashMap<>();
 
     private final AgentClientService agentClientService;
-    private final EmbeddingService embeddingService;
+    private final EmbeddingServiceIfc embeddingService;
 
     public void loadEndpoints(AgentExecution dto) throws ZtatException, JsonProcessingException {
         List<EndpointDescriptor> endpoints = agentClientService.getAvailableEndpoints(dto); // however you get them

ai-agent/src/main/java/io/sentrius/agent/services/EndpointSearcher.java

@@ -5,11 +5,9 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
-import java.util.stream.Collectors;
-import com.fasterxml.jackson.core.JsonProcessingException;
 import io.sentrius.sso.core.dto.capabilities.EndpointDescriptor;
 import io.sentrius.sso.core.dto.ztat.TokenDTO;
-import io.sentrius.sso.core.embeddings.EmbeddingService;
+import io.sentrius.sso.core.embeddings.EmbeddingServiceIfc;
 import io.sentrius.sso.core.exceptions.ZtatException;
 import io.sentrius.sso.core.services.endpoints.CosineSimilarity;
 import lombok.extern.slf4j.Slf4j;
@@ -19,11 +17,12 @@
 @Service
 public class EndpointSearcher {
 
-    private final EmbeddingService embeddingService;
+    private final EmbeddingServiceIfc embeddingService;
     private final EndpointRegistry endpointRegistry;
 
-    public EndpointSearcher(EmbeddingService embeddingService,
-                            EndpointRegistry endpointRegistry
+    public EndpointSearcher(
+        EmbeddingServiceIfc embeddingService,
+        EndpointRegistry endpointRegistry
     ) {
 
         this.embeddingService = embeddingService;

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

@@ -5,7 +5,7 @@
 import io.sentrius.sso.core.controllers.BaseController;
 import io.sentrius.sso.core.dto.agents.AgentMemoryDTO;
 import io.sentrius.sso.core.dto.agents.MemoryQueryDTO;
-import io.sentrius.sso.core.embeddings.EmbeddingService;
+import io.sentrius.sso.core.embeddings.EmbeddingServiceIfc;
 import io.sentrius.sso.core.model.agents.AgentMemory;
 import io.sentrius.sso.core.model.security.enums.ApplicationAccessEnum;
 import io.sentrius.sso.core.services.ErrorOutputService;
@@ -24,14 +24,11 @@
 import org.springframework.format.annotation.DateTimeFormat;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
-import org.springframework.security.core.Authentication;
 import org.springframework.web.bind.annotation.*;
 
 import jakarta.validation.Valid;
-import java.security.Principal;
 import java.time.LocalDateTime;
 import java.util.ArrayList;
-import java.util.Collection;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -46,11 +43,11 @@ public class AgentMemoryController extends BaseController {
     private final PersistentAgentMemoryStore memoryStore;
     private final VectorAgentMemoryStore vectorMemoryStore;
     private final UserAttributeService userAttributeService;
-    private final EmbeddingService embeddingService;
+    private final EmbeddingServiceIfc embeddingService;
 
     public AgentMemoryController(PersistentAgentMemoryStore memoryStore, VectorAgentMemoryStore vectorMemoryStore, UserService userService, SystemOptions systemOptions, ErrorOutputService errorOutputService,
                                  UserAttributeService userAttributeService,
-                                 EmbeddingService embeddingService
+                                 EmbeddingServiceIfc embeddingService
     ) {
         super(userService, systemOptions, errorOutputService);
         this.memoryStore = memoryStore;

…so/core/embeddings/EmbeddingService.java …core/embeddings/EmbeddingServiceIfc.javacore/src/main/java/io/sentrius/sso/core/embeddings/EmbeddingService.java renamed to core/src/main/java/io/sentrius/sso/core/embeddings/EmbeddingServiceIfc.java core/src/main/java/io/sentrius/sso/core/embeddings/EmbeddingService.java renamed to core/src/main/java/io/sentrius/sso/core/embeddings/EmbeddingServiceIfc.java

@@ -5,7 +5,7 @@
 import io.sentrius.sso.core.dto.ztat.TokenDTO;
 import io.sentrius.sso.core.exceptions.ZtatException;
 
-public interface EmbeddingService {
+public interface EmbeddingServiceIfc {
     float[] embed(TokenDTO dto, String text) throws ZtatException, JsonProcessingException;
     List<float[]> embed(TokenDTO dto, List<String> text) throws ZtatException, JsonProcessingException;
 }

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

@@ -25,16 +25,13 @@
 @Service
 public class EmbeddingService {
 
-    private final RestTemplate restTemplate;
     private final String integrationProxyUrl;
     private final ZeroTrustClientService zeroTrustClientService;
 
     public EmbeddingService(
-        RestTemplate restTemplate,
         @Value("${sentrius.integration.proxyUrl:http://localhost:8081}") String integrationProxyUrl,
         ZeroTrustClientService zeroTrustClientService
     ) {
-        this.restTemplate = restTemplate;
         this.integrationProxyUrl = integrationProxyUrl;
         this.zeroTrustClientService = zeroTrustClientService;
     }
@@ -149,49 +146,7 @@ public float[] embed(String text) {
         }
     }
 
-    /**
-     * Generate embeddings for multiple texts in batch via integration proxy
-     */
-    public Map<String, float[]> embedBatch(List<String> texts) {
-        if (texts == null || texts.isEmpty()) {
-            return new HashMap<>();
-        }
 
-        try {
-            String url = integrationProxyUrl + "/api/v1/embeddings/generate/batch";
-            HttpHeaders headers = createAuthHeaders();
-            if (headers == null) {
-                log.warn("No authentication context available for batch embedding generation");
-                return new HashMap<>();
-            }
-            
-            Map<String, Object> requestBody = new HashMap<>();
-            requestBody.put("texts", texts);
-            
-            HttpEntity<Map<String, Object>> entity = new HttpEntity<>(requestBody, headers);
-            
-            ResponseEntity<Map> response = restTemplate.exchange(url, HttpMethod.POST, entity, Map.class);
-            
-            if (response.getStatusCode().is2xxSuccessful() && response.getBody() != null) {
-                Map<String, Object> responseBody = response.getBody();
-                
-                if (responseBody.containsKey("embeddings")) {
-                    @SuppressWarnings("unchecked")
-                    Map<String, float[]> embeddings = (Map<String, float[]>) responseBody.get("embeddings");
-                    
-                    log.debug("Generated batch embeddings for {} texts", embeddings.size());
-                    return embeddings;
-                }
-            }
-            
-            log.warn("Failed to generate batch embeddings - unexpected response format");
-            return new HashMap<>();
-            
-        } catch (Exception e) {
-            log.error("Error generating batch embeddings for {} texts", texts.size(), e);
-            return new HashMap<>();
-        }
-    }
 
     /**
      * Calculate cosine similarity between two embeddings

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

@@ -4,7 +4,7 @@
 import java.util.Map;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import io.sentrius.sso.core.dto.ztat.TokenDTO;
-import io.sentrius.sso.core.embeddings.EmbeddingService;
+import io.sentrius.sso.core.embeddings.EmbeddingServiceIfc;
 import io.sentrius.sso.core.exceptions.ZtatException;
 import io.sentrius.sso.core.utils.JsonUtil;
 import org.springframework.beans.factory.annotation.Value;
@@ -13,7 +13,7 @@
 
 @Service
 @ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.SERVLET)
-public class LLMService implements EmbeddingService {
+public class LLMService implements EmbeddingServiceIfc {
 
     final ZeroTrustClientService zeroTrustClientService;