| # 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 | |
| from fastapi import Request | |
| def search(request: Request, query: str) -> list[dict]: | |
| embedding = get_embedding(request, query) | |
| ids = retrieve_ids(embedding) | |
| contexts = fetch_contexts_by_ids(ids) | |
| reranked = rerank(request, query, contexts) | |
| return reranked | |