quoc-khanh commited on
Commit
749d94e
·
verified ·
1 Parent(s): b09977a

Update helpers.py

Browse files
Files changed (1) hide show
  1. helpers.py +1 -1
helpers.py CHANGED
@@ -227,7 +227,7 @@ def augment_prompt(query: str, k: int = 10):
227
  queries = []
228
  queries.append(query)
229
 
230
- if vectorstore is None:
231
  vectorstore = get_vectorstore()
232
 
233
  retriever = vectorstore.as_retriever(search_kwargs={"k": k})
 
227
  queries = []
228
  queries.append(query)
229
 
230
+ if not vectorstore:
231
  vectorstore = get_vectorstore()
232
 
233
  retriever = vectorstore.as_retriever(search_kwargs={"k": k})