Spaces:
Running
Running
[email protected]
commited on
Commit
·
6fe1dfe
1
Parent(s):
87fc54f
fix: Increase similarity search limit in retriever from 3 to 10
Browse files
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":
|
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 |
)
|