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

Update helpers.py

Browse files
Files changed (1) hide show
  1. helpers.py +6 -1
helpers.py CHANGED
@@ -227,8 +227,13 @@ def augment_prompt(query: str, k: int = 10):
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})
234
  results = retriever.invoke(query)
 
227
  queries = []
228
  queries.append(query)
229
 
230
+ if "vectorstore" not in globals():
231
+ print("Không tìm thấy vectorstore. Đang tạo mới...")
232
  vectorstore = get_vectorstore()
233
+ print("✅ Vectorstore đã được tạo thành công!")
234
+ else:
235
+ print("✅ Vectorstore đã tồn tại, không cần tạo lại.")
236
+
237
 
238
  retriever = vectorstore.as_retriever(search_kwargs={"k": k})
239
  results = retriever.invoke(query)