retriever.py

from sentence_transformers.util import cos_sim
from embedder import get_model
import numpy as np

def retrieve_chunks(index, texts, question, top_k=15):
    model = get_model()
    q_embedding = model.encode([question], convert_to_numpy=True, normalize_embeddings=True)[0]

    scores, indices = index.search(np.array([q_embedding]), top_k)
    selected = [texts[i] for i in indices[0]]
    return selected