Update retriever.py

retriever.py

@@ -54,19 +54,24 @@ def load_guest_dataset():
         List of Document objects containing guest information
     """
     # Load the dataset from Hugging Face Hub
-    # This dataset contains information about fictional gala guests
     guest_dataset = datasets.load_dataset(
         "agents-course/unit3-invitees", 
         split="train"
     )
     
     # Convert to Document objects for the retriever
+    # Dataset has fields: name, relation, description, email
     docs = [
         Document(
-            page_content=guest["text"],
-            metadata={"source": f"guest_{i}"}
+            page_content="\n".join([
+                f"Name: {guest['name']}",
+                f"Relation: {guest['relation']}",
+                f"Description: {guest['description']}",
+                f"Email: {guest['email']}"
+            ]),
+            metadata={"name": guest["name"]}
         )
-        for i, guest in enumerate(guest_dataset)
+        for guest in guest_dataset
     ]
     
     return docs