Spaces:
Runtime error
Runtime error
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:
|