File size: 468 Bytes
e6bc25d 2aa7bf4 4fdc679 2aa7bf4 835d057 4fdc679 835d057 4fdc679 835d057 4fdc679 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# 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
|