Daniel Foley commited on
Commit
706b16d
·
1 Parent(s): 8547e22

returning ranked docs only

Browse files
Files changed (1) hide show
  1. RAG.py +1 -1
RAG.py CHANGED
@@ -118,7 +118,7 @@ def rerank(documents: List[Document], query: str, max_workers: int = 3) -> List[
118
  reranker = BM25Retriever.from_documents(full_docs, k=min(10, len(full_docs)))
119
  reranked_docs = reranker.invoke(query)
120
  logging.info(f"Finished reranking: {time.time()-start}")
121
- return full_docs
122
 
123
 
124
  def parse_xml_and_query(query:str,xml_string:str) -> str:
 
118
  reranker = BM25Retriever.from_documents(full_docs, k=min(10, len(full_docs)))
119
  reranked_docs = reranker.invoke(query)
120
  logging.info(f"Finished reranking: {time.time()-start}")
121
+ return reranked_docs
122
 
123
 
124
  def parse_xml_and_query(query:str,xml_string:str) -> str: