Update retriever.py

retriever.py

@@ -1,5 +1,7 @@
 import datasets
 from langchain.docstore.document import Document
+from langchain_community.retrievers import BM25Retriever
+from langchain.tools import Tool
 
 # Load the dataset
 games_dataset = datasets.load_dataset("nateraw/list-of-bestselling-pc-games", split="train")
@@ -21,9 +23,6 @@ docs = [
     for game in games_dataset
 ]
 
-from langchain_community.retrievers import BM25Retriever
-from langchain.tools import Tool
-
 bm25_retriever = BM25Retriever.from_documents(docs)
 
 def extract_text(query: str) -> str:
@@ -34,7 +33,7 @@ def extract_text(query: str) -> str:
     else:
         return "No matching game information found."
 
-guest_info_tool = Tool(
+game_info_tool = Tool(
     name="game_info_retriever",
     func=extract_text,
     description="Retrieves detailed information about games based on their name or relation."