Add load_guest_dataset() to retriever.py

retriever.py

@@ -4,22 +4,27 @@ from langchain.docstore.document 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
-]
+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)
+
 
 class GuestInfoRetrieverTool(Tool):
     name = "guest_info_retriever"
@@ -42,6 +47,3 @@ class GuestInfoRetrieverTool(Tool):
             return "\n\n".join([doc.page_content for doc in results[:3]])
         else:
             return "No matching guest information found."
-
-# Initialize the tool
-guest_info_tool = GuestInfoRetrieverTool(docs)