[email protected] commited on
Commit
6fe1dfe
·
1 Parent(s): 87fc54f

fix: Increase similarity search limit in retriever from 3 to 10

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -74,7 +74,7 @@ class Rag:
74
  self.retriever = self.document_vector_store.as_retriever(
75
  search_type="similarity_score_threshold",
76
  search_kwargs={
77
- "k": 3,
78
  "score_threshold": 0.5,
79
  },
80
  )
 
74
  self.retriever = self.document_vector_store.as_retriever(
75
  search_type="similarity_score_threshold",
76
  search_kwargs={
77
+ "k": 10,
78
  "score_threshold": 0.5,
79
  },
80
  )