time logging and grpc for neural search

qdrant · Oct 21, 2023 · cc1ca8a · cc1ca8a
1 parent 3b6a330
commit cc1ca8a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/qdrant_demo/neural_searcher.py b/qdrant_demo/neural_searcher.py
@@ -1,3 +1,4 @@
+import time
 from typing import List
 
 from qdrant_client import QdrantClient
@@ -10,14 +11,16 @@ class NeuralSearcher:
 
     def __init__(self, collection_name: str):
         self.collection_name = collection_name
-        self.qdrant_client = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY)
+        self.qdrant_client = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY, prefer_grpc=True)
         self.qdrant_client.set_model(EMBEDDINGS_MODEL)
 
     def search(self, text: str, filter_: dict = None) -> List[dict]:
+        start_time = time.time()
         hits = self.qdrant_client.query(
             collection_name=self.collection_name,
             query_text=text,
             query_filter=Filter(**filter_) if filter_ else None,
             limit=5
         )
+        print(f"Search took {time.time() - start_time} seconds")
         return [hit.metadata for hit in hits]