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