Upload retrieval.py

retrieval.py

@@ -9,14 +9,14 @@ retrieved_docs = None
 
 # Retrieval Function
 def retrieve_documents(query, top_k=5):
-    faiss_index_path = f"rag7_index.faiss"
+    faiss_index_path = f"data_local/rag7_index.faiss"
     index = faiss.read_index(faiss_index_path)
 
     query_embedding = np.array(embedding_model.embed_documents([query]), dtype=np.float32)
 
     _, nearest_indices = index.search(query_embedding, top_k)
 
-    with open(f"rag7_docs.json", "r") as f:
+    with open(f"data_local/rag7_docs.json", "r") as f:
         documents = json.load(f)  # Contains all documents for this dataset
 
     retrieved_docs = [Document(page_content=documents[i]) for i in nearest_indices[0]]