quoc-khanh commited on
Commit
b49f336
·
verified ·
1 Parent(s): e6eef93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -201,7 +201,8 @@ def augment_prompt(query: str, queries_list=None, k: int = 10):
201
  # see_metadata_fin = metadata
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
  else:
206
  retriever = vectorstore.as_retriever(search_kwargs={"k": k, "fetch_k": 20, "filter": metadata})
207
 
 
201
  # see_metadata_fin = metadata
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