Skip to content

Commit

Permalink
returning ranked docs only
Browse files Browse the repository at this point in the history
  • Loading branch information
Dephoh committed Dec 11, 2024
1 parent 8547e22 commit 706b16d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion RAG.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ def rerank(documents: List[Document], query: str, max_workers: int = 3) -> List[
reranker = BM25Retriever.from_documents(full_docs, k=min(10, len(full_docs)))
reranked_docs = reranker.invoke(query)
logging.info(f"Finished reranking: {time.time()-start}")
return full_docs
return reranked_docs


def parse_xml_and_query(query:str,xml_string:str) -> str:
Expand Down

0 comments on commit 706b16d

Please sign in to comment.