# rag/service/search.py
from models.embedder import get_embedding
from modules.retriever import retrieve_ids
from modules.corpus import fetch_contexts_by_ids
from models.reranker import rerank

def search(query: str) -> list[dict]:
    embedding = get_embedding(query)
    ids = retrieve_ids(embedding)
    contexts = fetch_contexts_by_ids(ids)
    reranked = rerank(query, contexts)
    return reranked