Update retriever.py

retriever.py

@@ -3,7 +3,6 @@ from langchain_community.retrievers import BM25Retriever
 from langchain.docstore.document import Document
 import datasets
 
-
 class GuestInfoRetrieverTool(Tool):
     name = "guest_info_retriever"
     description = "Retrieves detailed information about gala guests based on their name or relation."
@@ -18,7 +17,6 @@ class GuestInfoRetrieverTool(Tool):
     def __init__(self, docs):
         self.is_initialized = False
         self.retriever = BM25Retriever.from_documents(docs)
-       
 
     def forward(self, query: str):
         results = self.retriever.get_relevant_documents(query)
@@ -27,7 +25,6 @@ class GuestInfoRetrieverTool(Tool):
         else:
             return "No matching guest information found."
 
-
 def load_guest_dataset():
     # Load the dataset
     guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
@@ -47,4 +44,4 @@ def load_guest_dataset():
     ]
 
     # Return the tool
-    return GuestInfoRetrieverTool(docs)
+    return GuestInfoRetrieverTool(docs)