Update search_utils.py

search_utils.py

@@ -182,14 +182,15 @@ class SemanticSearch:
         try:
             shard_start = time.time()
             distances, indices = index.search(query_embedding, top_k)
-            valid_mask = (indices[0] >= 0) & (indices[0] < index.ntotal)
-            valid_indices = indices[0][valid_mask].tolist()
-            valid_distances = distances[0][valid_mask].tolist()
+            # valid_mask = (indices[0] >= 0) & (indices[0] < index.ntotal)
+            valid_indices = indices[0]
+            valid_distances = distances[0]
             
             if len(valid_indices) != top_k:
                 self.logger.debug(f"Shard {shard_idx}: Found {len(valid_indices)} valid results out of {top_k}")
                 
             global_indices = [self._global_index(shard_idx, idx) for idx in valid_indices]
+            logger.info(f"Global Indice {global_indices}")
             self.logger.debug(f"Shard {shard_idx} search completed in {time.time() - shard_start:.3f}s")
             return valid_distances, global_indices
         except Exception as e: