Update helpers.py

helpers.py

@@ -226,8 +226,9 @@ def prompt_order(queries):
 def augment_prompt(query: str, k: int = 10):
     queries = []
     queries.append(query)
-
-    vectorstore = get_vectorstore()
+    
+    if vectorstore is None:
+        vectorstore = get_vectorstore()
 
     retriever = vectorstore.as_retriever(search_kwargs={"k": k})
     results = retriever.invoke(query)