feat: delete summary index too

superagent-ai · Mar 27, 2024 · 4ba3418 · 4ba3418
1 parent 306ee63
commit 4ba3418
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/api/delete.py b/api/delete.py
@@ -3,6 +3,7 @@
 from models.delete import RequestPayload, ResponsePayload
 from vectordbs import get_vector_service
 from vectordbs.base import BaseVectorDatabase
+from utils.summarise import SUMMARY_SUFFIX
 
 router = APIRouter()
 
@@ -16,8 +17,15 @@ async def delete(payload: RequestPayload):
         encoder=encoder,
         dimensions=payload.encoder.dimensions,
     )
+    summary_vector_service: BaseVectorDatabase = get_vector_service(
+        index_name=f"{payload.index_name}{SUMMARY_SUFFIX}",
+        credentials=payload.vector_database,
+        encoder=encoder,
+        dimensions=payload.encoder.dimensions,
+    )
 
     for file in payload.files:
         data = await vector_service.delete(file_url=file.url)
+        await summary_vector_service.delete(file_url=file.url)
 
     return ResponsePayload(success=True, data=data)