feat: Add overwrite to load index

presets/ragengine/main.py

@@ -310,10 +310,11 @@ async def persist_index(
 )
 async def load_index(
         index_name: str,
-        path: str = Query(DEFAULT_VECTOR_DB_PERSIST_DIR, description="Path to load the index from")
+        path: str = Query(DEFAULT_VECTOR_DB_PERSIST_DIR, description="Path to load the index from"),
+        overwrite: bool = Query(False, description="Overwrite the existing index if it already exists")
 ):
     try:
-        await rag_ops.load(index_name, path)
+        await rag_ops.load(index_name, path, overwrite)
         return {"message": f"Successfully loaded index {index_name} from {path}."}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Loading failed: {str(e)}")

presets/ragengine/vector_store/base.py

@@ -290,21 +290,21 @@
             logger.error(f"Failed to persist index {index_name}. Error: {str(e)}")
             raise HTTPException(status_code=500, detail=f"Persistence failed: {str(e)}")
 
-    async def load(self, index_name: str, path: str):
+    async def load(self, index_name: str, path: str, overwrite: bool):
         """Common logic for loading an index."""
         # Check path existence before acquiring any lock
         if not os.path.exists(path):
             raise HTTPException(status_code=404, detail=f"Path does not exist: {path}")
         if self.use_rwlock:
             async with self.rwlock.writer_lock:
-                await self._load_internal(index_name, path)
+                await self._load_internal(index_name, path, overwrite)
         else:
-            await self._load_internal(index_name, path)
+            await self._load_internal(index_name, path, overwrite)
 
-    async def _load_internal(self, index_name: str, path: str):
+    async def _load_internal(self, index_name: str, path: str, overwrite: bool):
         """Common logic for loading an index."""
         try:
-            if index_name in self.index_map:
+            if index_name in self.index_map and not overwrite:
                 raise HTTPException(
                     status_code=409,
                     detail=f"Index '{index_name}' already exists. Use a different name or delete the existing index first."

presets/ragengine/vector_store_manager/manager.py

@@ -46,9 +46,9 @@ async def persist(self, index_name: str, path: str) -> None:
         """Persist existing index."""
         return await self.vector_store.persist(index_name, path)
 
-    async def load(self, index_name: str, path: str) -> None:
+    async def load(self, index_name: str, path: str, overwrite: bool) -> None:
         """Load existing index."""
-        return await self.vector_store.load(index_name, path)
+        return await self.vector_store.load(index_name, path, overwrite)
 
     async def shutdown(self):
         """Shutdown the manager."""