diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-08 00:23:52 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-08 00:23:52 +0200 |
commit | 8211705debf9d1335223c606275f46c43c78d8a2 (patch) | |
tree | f09f902c7d31b2035813c42cbb4a47e720fa363b /rag/db/vector.py | |
parent | 95f47c4900a96d91daaef93bf87094ed3d4da43c (diff) |
Updates
Diffstat (limited to 'rag/db/vector.py')
-rw-r--r-- | rag/db/vector.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/rag/db/vector.py b/rag/db/vector.py index 4aa62cc..bbbbf32 100644 --- a/rag/db/vector.py +++ b/rag/db/vector.py @@ -50,6 +50,9 @@ class VectorDB: def search(self, query: List[float], limit: int = 4) -> List[ScoredPoint]: log.debug("Searching for vectors...") hits = self.client.search( - collection_name=self.collection_name, query_vector=query, limit=limit + collection_name=self.collection_name, + query_vector=query, + limit=limit, + score_threshold=0.6, ) return hits |