Adapting the local database for faiss integration.

Author: peng3307165

modelcache/manager/vector_data/faiss.py

@@ -23,7 +23,7 @@ def mul_add(self, datas: List[VectorData], model=None):
         ids = np.array(id_array)
         self._index.add_with_ids(np_data, ids)
 
-    def search(self, data: np.ndarray, top_k: int = -1):
+    def search(self, data: np.ndarray, top_k: int = -1, model=None):
         if self._index.ntotal == 0:
             return None
         if top_k == -1:
@@ -33,6 +33,9 @@ def search(self, data: np.ndarray, top_k: int = -1):
         ids = [int(i) for i in ids[0]]
         return list(zip(dist[0], ids))
 
+    def rebuild_col(self, ids=None):
+        return True
+
     def rebuild(self, ids=None):
         return True