Update DocRetriever to change search type to 'similarity' and adjust score threshold handling
Browse files
app/retriever/__init__.py
CHANGED
@@ -30,11 +30,11 @@ class DocRetriever(BaseRetriever):
|
|
30 |
_filter.update({"user_id": req.user_id})
|
31 |
print(_filter)
|
32 |
self.retriever = vectorstore.as_retriever(
|
33 |
-
search_type='
|
34 |
search_kwargs={
|
35 |
"k": k,
|
36 |
"filter": _filter,
|
37 |
-
"score_threshold": .
|
38 |
}
|
39 |
)
|
40 |
|
|
|
30 |
_filter.update({"user_id": req.user_id})
|
31 |
print(_filter)
|
32 |
self.retriever = vectorstore.as_retriever(
|
33 |
+
search_type='similarity',
|
34 |
search_kwargs={
|
35 |
"k": k,
|
36 |
"filter": _filter,
|
37 |
+
# "score_threshold": .3
|
38 |
}
|
39 |
)
|
40 |
|