Spaces:
Sleeping
Sleeping
Update helpers.py
Browse files- 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
|
|
|
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)
|