Daniel Foley commited on
Commit
b609620
·
1 Parent(s): fea960e

passed original query to reranker

Browse files
Files changed (1) hide show
  1. RAG.py +1 -1
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=new_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