| import faiss | |
| import numpy as np | |
| from sentence_transformers import SentenceTransformer | |
| def retrieve_relevant_chunks(query,embedder, k=5, index_path="faiss_index.idx", chunks_path="text_chunks.npy"): | |
| index = faiss.read_index(index_path) | |
| chunks = np.load(chunks_path, allow_pickle=True) | |
| query_embedding = embedder.encode([query], convert_to_numpy=True) | |
| distances, indices = index.search(query_embedding, k) | |
| retrieved_chunks = [chunks[i] for i in indices[0]] | |
| return retrieved_chunks | |