Update retriever_m3.py

retriever_m3.py

@@ -36,6 +36,9 @@ df = pd.read_parquet(parquet_path,engine="pyarrow")
 # inference client is defined at app.py, here just the function
 
 def find_similar_foundations_api(query: str, client, top_k: int = 5):
+    if "Title" not in df.columns or "Purpose" not in df.columns:
+        raise ValueError("Dataset must contain 'Title' and 'Purpose' columns.")
+    
     # Compute similarity against all missions
     results = client.sentence_similarity(
         {
@@ -53,8 +56,8 @@ def find_similar_foundations_api(query: str, client, top_k: int = 5):
     )
 
     return [
-        {"score": float(score), "Title": foundation, "Purpose": mission}
-        for score, foundation, mission in scored[:top_k]
+        {"score": float(score), "Title": title, "Purpose": purpose}
+        for score, title, purpose in scored[:top_k]
     ]