gavinzli commited on
Commit
1494f1e
·
1 Parent(s): 776586f

Update DocRetriever to change search type to 'similarity' and adjust score threshold handling

Browse files
Files changed (1) hide show
  1. app/retriever/__init__.py +2 -2
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='similarity_score_threshold',
34
  search_kwargs={
35
  "k": k,
36
  "filter": _filter,
37
- "score_threshold": .7
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