From 39b0792cb66974eaab97aba0634fdf27b58bf37c Mon Sep 17 00:00:00 2001 From: "andrea.terzani" Date: Mon, 14 Apr 2025 16:59:14 +0200 Subject: [PATCH] Refactor DeletionService for improved code clarity and maintainability --- .../java/com/olympus/apollo/services/DeletionService.java | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/olympus/apollo/services/DeletionService.java b/src/main/java/com/olympus/apollo/services/DeletionService.java index ad80796..aaef865 100644 --- a/src/main/java/com/olympus/apollo/services/DeletionService.java +++ b/src/main/java/com/olympus/apollo/services/DeletionService.java @@ -78,13 +78,16 @@ public class DeletionService { public void deleteRecordsOnlyFromVectorStore(DeletionRequest deletionRequest) { try { + KSDocument ksDocument = ksDocumentRepository.findById(deletionRequest.getKsDocumentId()).get(); + ksDocument.setIngestionStatus("DELETING"); + ksDocumentRepository.save(ksDocument); String rag_filter = "KsDocumentId=='"+deletionRequest.getKsDocumentId()+"'"; + logger.info("Starting deletion"); vectorStore.delete(rag_filter); //elimino dal vectorStore ma mantengo il record - KSDocument ksDocument = ksDocumentRepository.findById(deletionRequest.getKsDocumentId()).get(); ksDocument.setIngestionStatus("LOADED"); Date now = new Date(); ksDocument.setIngestionDate(now);