from typing import List
from core.faiss_vector import FaissIndex

class SemanticRefiner:
    def __init__(self):
        self.index = FaissIndex()

    def build_from_results(self, results: List[dict]):
        texts = [r.get("snippet","") for r in results if r.get("snippet")]
        if texts:
            self.index.add(texts)

    def refine(self, query: str, k: int = 10):
        return self.index.search(query, k)