adds filter to query suggestions

app.py

@@ -17,14 +17,16 @@ ALLOWED_API_KEY = str(API_KEY)
 
 @app.get("/api/search")
 async def search(q: str):
-    xss = nh3.clean(q)
+    query = q.lower()
+    xss = nh3.clean(query)
     data = await doc_searcher.search(text=xss)
     return data
 
 @app.get("/api/suggestions")
-async def get_suggestions(q: str):
-    xss = nh3.clean(q)
-    data = await suggestion_searcher.search(text=xss)
+async def get_suggestions(q: str, type: int):
+    query = q.lower()
+    xss = nh3.clean(query)
+    data = await suggestion_searcher.search(text=xss,type=type)
     return data
 
 

suggestion_searcher.py

@@ -1,4 +1,5 @@
 from qdrant_client import QdrantClient
+from qdrant_client.models import Filter, FieldCondition, MatchValue
 from fastembed import SparseTextEmbedding, LateInteractionTextEmbedding
 from qdrant_client import QdrantClient, models
 from sentence_transformers import SentenceTransformer
@@ -13,7 +14,7 @@ class SuggestionSearcher:
         self.late_interaction_model = LateInteractionTextEmbedding(LATE_INTERACTION_MODEL)
         self.qdrant_client = QdrantClient(QDRANT_URL,api_key=QDRANT_API_KEY,timeout=30)
     
-    async def search(self, text: str):
+    async def search(self, text: str,type:int):
         
         dense_query = self.dense_model.encode(text).tolist()
         sparse_query = next(self.sparse_model.query_embed(text))
@@ -41,10 +42,19 @@ class SuggestionSearcher:
                 )
         ]
 
+        self.qdrant_client.scroll
         search_result = self.qdrant_client.query_points(
             collection_name= self.collection_name,
+            query_filter=Filter(
+                must=[
+                    FieldCondition(
+                        key="tip",
+                        match=MatchValue(value=type)
+                    )
+                ]
+            ),
             search_params=models.SearchParams(
-                hnsw_ef=128, 
+                hnsw_ef=64, 
                 exact=False, 
                 quantization=models.QuantizationSearchParams(
                         rescore=True,