Update retriever.py

retriever.py

@@ -4,6 +4,23 @@ from langchain_core.tools import Tool
 from langchain_core.documents import Document
 import datasets
 
+# 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
+]
+
 bm25_retriever = BM25Retriever.from_documents(docs)
 
 def extract_text(query: str) -> str:
@@ -21,22 +38,7 @@ guest_info_tool = Tool(
 )
 
 
-# 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
-]
 
 
 '''