Create vector_store.py

vector_store.py

@@ -0,0 +1,14 @@
+# vector_store.py
+import faiss
+import numpy as np
+
+def create_faiss_index(vectors):
+    dim = vectors[0].shape[0]
+    index = faiss.IndexFlatL2(dim)
+    index.add(np.array(vectors))
+    return index
+
+def search_similar_cvs(query_vector, index, k=3):
+    query_vector = np.array([query_vector])
+    distances, indices = index.search(query_vector, k)
+    return indices[0].tolist()