Update helpers.py

helpers.py

@@ -15,6 +15,8 @@ import shutil
 import requests
 from bs4 import BeautifulSoup
 
+from file_loader import get_vectorstore
+
 async def get_urls_splits(url='https://nct.neu.edu.vn/', char='https://nct.neu.edu.vn/'):
     reqs = requests.get(url)
     soup = BeautifulSoup(reqs.text, 'html.parser')
@@ -195,6 +197,8 @@ def augment_prompt(query: str, k: int = 10):
     queries = []
     queries.append(query)
 
+    vectorstore = get_vectorstore()
+
     retriever = vectorstore.as_retriever(search_kwargs={"k": k})
     results = retriever.invoke(query)