Update retriever.py

retriever.py

@@ -6,6 +6,21 @@ from langchain_community.retrievers import BM25Retriever
 import datasets
 from langchain.docstore.document import Document
 
+# Load dataset and create documents
+guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
+
+docs = [
+    Document(
+        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 guest in guest_dataset
+]
 
 ###########################
 # Retriever - changed to LangGraph version
@@ -27,29 +42,34 @@ guest_info_tool = Tool(
 )
 ###########################
 
+# # (Optional) Remove or update this function if unused
+# def load_guest_dataset():
+#     # This function is no longer needed as docs are created above
+#     pass
+
+
+
+# # no change from smolagents to LangGraph
+# def load_guest_dataset():
+#     # Load the dataset
+#     guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
 
+#     # Convert dataset entries into Document objects
+#     docs = [
+#         Document(
+#             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 guest in guest_dataset
+#     ]
 
-# no change from smolagents to LangGraph
-def load_guest_dataset():
-    # Load the dataset
-    guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
-
-    # Convert dataset entries into Document objects
-    docs = [
-        Document(
-            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 guest in guest_dataset
-    ]
-
-    # Return the tool
-    return GuestInfoRetrieverTool(docs)
+#     # Return the tool
+#     return GuestInfoRetrieverTool(docs)