Update retrieval.py

retrieval.py

@@ -36,6 +36,7 @@ def compute_anchor_scores_for_hadiths(
 def hybrid_search_fixed(query,
                         df,
                         bm25,
+                        preprocess_query,
                         model,
                         hadith_embeddings,
                         anchor_index,
@@ -60,7 +61,7 @@ def hybrid_search_fixed(query,
     eps = 1e-8
 
     # 1) BM25 candidates and scores
-    bm25_indices, bm25_scores = bm25_retrieve(query, bm25, top_k=top_bm25)
+    bm25_indices, bm25_scores = bm25_retrieve(query, bm25, preprocess_query,top_k=top_bm25)
     # make dictionary mapping hadith_idx -> bm25_score
     bm25_map = {int(idx): float(score) for idx, score in zip(bm25_indices, bm25_scores)}