MedQAAssistant.retrieve_chunks_for_query:v0
1
2
3
4
5
6
7
8
9
10
import weave
from medrag_multi_modal.retrieval.text_retrieval.bm25s_retrieval import BM25sRetriever
@weave.op()
def retrieve_chunks_for_query(self, query: str) -> list[dict]:
retriever_kwargs = {"top_k": self.top_k_chunks_for_query}
if not isinstance(self.retriever, BM25sRetriever):
retriever_kwargs["metric"] = self.retrieval_similarity_metric
return self.retriever.predict(query, **retriever_kwargs)