From 4dd50fe151eb1ee96d4d8dc354b48ebccae26430 Mon Sep 17 00:00:00 2001
From: powerli2002 <845160705@qq.com>
Date: Wed, 22 May 2024 17:24:27 +0800
Subject: [PATCH] fix: Fixed an issue where clearing faiss database was invalid
 when using flask4modelcache_demo

---
 modelcache/manager/vector_data/faiss.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/modelcache/manager/vector_data/faiss.py b/modelcache/manager/vector_data/faiss.py
index f035ded..0f8445c 100644
--- a/modelcache/manager/vector_data/faiss.py
+++ b/modelcache/manager/vector_data/faiss.py
@@ -34,7 +34,10 @@ def search(self, data: np.ndarray, top_k: int = -1, model=None):
         return list(zip(dist[0], ids))
 
     def rebuild_col(self, ids=None):
-        return True
+        try:
+            self._index.reset()
+        except Exception as e:
+            return f"An error occurred during index rebuild: {e}"
 
     def rebuild(self, ids=None):
         return True