vector_store.py

# 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()