Update retriever.py

retriever.py

@@ -4,7 +4,14 @@ from sentence_transformers import SentenceTransformer, util
 model = SentenceTransformer("all-MiniLM-L6-v2")
 
 def get_relevant_passages(query, df, top_k=20):
-    corpus = df["description"].astype(str).tolist()
+    # Concatenate all fields into a single string per row
+    corpus = df.apply(lambda row: f"{row['description']} "
+                                   f"Test types: {', '.join(row['test_type'])}. "
+                                   f"Adaptive support: {row['adaptive_support']}. "
+                                   f"Remote support: {row['remote_support']}. "
+                                   f"Duration: {row['duration']} minutes.",
+                      axis=1).tolist()
+
     corpus_embeddings = model.encode(corpus, convert_to_tensor=True)
     query_embedding = model.encode(query, convert_to_tensor=True)