Spaces:
Sleeping
Sleeping
Daniel Foley
commited on
Commit
·
706b16d
1
Parent(s):
8547e22
returning ranked docs only
Browse files
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
|
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:
|