Update tools.py

Added loading the dataset

tools.py

@@ -29,8 +29,17 @@ class GuestInfoRetrieverTool(Tool):
         else:
             return "No matching guest information found."
 
+from datasets import load_dataset
+from langchain_core.documents import Document
+
+# Load the dataset from Hugging Face
+dataset = load_dataset("agents-course/unit3-invitees")["train"]
+
+# Create Document objects from the "info" column
+docs = [Document(page_content=row["info"]) for row in dataset]
+
 # Initialize the tool
-# guest_info_tool = GuestInfoRetrieverTool(docs)
+guest_info_tool = GuestInfoRetrieverTool(docs)