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