Spaces:
Running
Running
updates suggestion searcher
Browse files- suggestion_searcher.py +2 -2
suggestion_searcher.py
CHANGED
|
@@ -17,7 +17,6 @@ class SuggestionSearcher:
|
|
| 17 |
|
| 18 |
dense_query = self.dense_model.encode(text).tolist()
|
| 19 |
sparse_query = next(self.sparse_model.query_embed(text))
|
| 20 |
-
# late_query = next(self.late_interaction_model.query_embed(text))
|
| 21 |
|
| 22 |
prefetch = [
|
| 23 |
models.Prefetch(
|
|
@@ -34,11 +33,12 @@ class SuggestionSearcher:
|
|
| 34 |
|
| 35 |
search_result = self.qdrant_client.query_points(
|
| 36 |
collection_name= self.collection_name,
|
|
|
|
|
|
|
| 37 |
prefetch=prefetch,
|
| 38 |
query=models.FusionQuery(
|
| 39 |
fusion=models.Fusion.RRF,
|
| 40 |
),
|
| 41 |
-
# using=LATE_INTERACTION_MODEL,
|
| 42 |
with_payload=True,
|
| 43 |
limit = 5
|
| 44 |
).points
|
|
|
|
| 17 |
|
| 18 |
dense_query = self.dense_model.encode(text).tolist()
|
| 19 |
sparse_query = next(self.sparse_model.query_embed(text))
|
|
|
|
| 20 |
|
| 21 |
prefetch = [
|
| 22 |
models.Prefetch(
|
|
|
|
| 33 |
|
| 34 |
search_result = self.qdrant_client.query_points(
|
| 35 |
collection_name= self.collection_name,
|
| 36 |
+
search_params=models.SearchParams(
|
| 37 |
+
quantization=models.QuantizationSearchParams(rescore=False),hnsw_ef=64, exact=False),
|
| 38 |
prefetch=prefetch,
|
| 39 |
query=models.FusionQuery(
|
| 40 |
fusion=models.Fusion.RRF,
|
| 41 |
),
|
|
|
|
| 42 |
with_payload=True,
|
| 43 |
limit = 5
|
| 44 |
).points
|