quoc-khanh commited on
Commit
1d3ae20
·
verified ·
1 Parent(s): ccc469e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -202,7 +202,7 @@ def augment_prompt(query: str, queries_list=None, k: int = 10):
202
  global last_vector_docs
203
  if metadata == {}:
204
  # retriever = vectorstore.as_retriever(search_kwargs={"k": 25, "fetch_k": 50, "filter": {'Tai lieu ve': 'Đề án'}})
205
- retriever = vectorstore.as_retriever(search_kwargs={"k": 30})
206
  else:
207
  retriever = vectorstore.as_retriever(search_kwargs={"k": k, "fetch_k": 20, "filter": metadata})
208
 
 
202
  global last_vector_docs
203
  if metadata == {}:
204
  # retriever = vectorstore.as_retriever(search_kwargs={"k": 25, "fetch_k": 50, "filter": {'Tai lieu ve': 'Đề án'}})
205
+ retriever = vectorstore.as_retriever(search_kwargs={"k": 20})
206
  else:
207
  retriever = vectorstore.as_retriever(search_kwargs={"k": k, "fetch_k": 20, "filter": metadata})
208