cdupland commited on
Commit
ae06536
·
1 Parent(s): 22c6fdf

CHange k parameter ford common namespace retiever

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -131,7 +131,7 @@ def get_retreive_answer(enterprise_id,prompt,index,common_id):
131
 
132
  retriever_commun = vector_store_commun.as_retriever(
133
  search_type="similarity_score_threshold",
134
- search_kwargs={"k": 3, "score_threshold": 0.1},
135
  )
136
 
137
  commun_context = retriever_commun.invoke(prompt)
 
131
 
132
  retriever_commun = vector_store_commun.as_retriever(
133
  search_type="similarity_score_threshold",
134
+ search_kwargs={"k": 5, "score_threshold": 0.1},
135
  )
136
 
137
  commun_context = retriever_commun.invoke(prompt)