Spaces:
Running
Running
Daniel Foley
commited on
Commit
·
b609620
1
Parent(s):
fea960e
passed original query to reranker
Browse files
RAG.py
CHANGED
@@ -153,7 +153,7 @@ def RAG(llm: Any, query: str, index_name: str, embeddings: Any, top: int = 10, k
|
|
153 |
return "No documents found for your query.", []
|
154 |
|
155 |
# Rerank documents
|
156 |
-
reranked = rerank(documents=retrieved, query=
|
157 |
if not reranked:
|
158 |
return "Unable to process the retrieved documents.", []
|
159 |
|
|
|
153 |
return "No documents found for your query.", []
|
154 |
|
155 |
# Rerank documents
|
156 |
+
reranked = rerank(documents=retrieved, query=query)
|
157 |
if not reranked:
|
158 |
return "Unable to process the retrieved documents.", []
|
159 |
|