restoring reteaver

Signed-off-by: apple muncy <[email protected]>

app.py

@@ -44,10 +44,11 @@ chat = ChatHuggingFace(llm=llm, verbose=True)
 
 # Define available tools
 tools = [
-#        load_guest_dataset, 
+#         
         search_tool, 
         weather_info_tool, 
-        hub_stats_tool
+        hub_stats_tool,
+        load_guest_dataset
         ]
 
 # Bind tools to the chat model

retriever.py

@@ -4,6 +4,7 @@ from langchain.docstore.document import Document
 import datasets
 
 class GuestInfoRetrieverTool(Tool):
+    """retrievs guest data"""
     name = "guest_info_retriever"
 
     description = "Retrieves detailed information about gala guests based on their name or relation."
@@ -28,6 +29,7 @@ class GuestInfoRetrieverTool(Tool):
 
 
 def load_guest_dataset():
+    """Get data set  """
     # Load the dataset
     guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
     # Convert dataset entries into Document objects