Update src/vector_db.py

src/vector_db.py

@@ -6,6 +6,11 @@ import pyarrow as pa
 import pandas as pd
 import numpy as np
 import tqdm
+import logging
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
 
 class VectorDB:
 
@@ -90,8 +95,10 @@ class VectorDB:
     def retrieve_prefiltered_hits(self, query, k):
 
         query_vec = self.retriever.encode(query)
+        logger.info('encoded')
         documents = self.table.search(query_vec, vector_column_name=self.vector_column).limit(k).to_list()
         names = [doc[self.name_column] for doc in documents]
         descriptions = [doc[self.description_column] for doc in documents]
+        logger.info('done topK lookup')
 
         return names, descriptions