Update tools.py

tools.py

@@ -96,18 +96,18 @@ def tool_register(cls: BaseModel):
 
 
 @tool_register
-class get_documents(ToolBase):
-    """Retrieves general information from Wikipedia based on the user's query.    """
+class retrieve_wiki_data(ToolBase):
+    """Retrieves relevant information from Wikipedia based on the user’s query."""
     
-    logging.info("@tool_register: get_documents()")  
+    logging.info("@tool_register: retrieve_wiki_data()")  
     
-    query: str = Field(description="Search query to retrieve relevant documents.")
+    query: str = Field(description="The user's input or question, used to search.")
     logging.info(f"query: {query}")
 
     @classmethod
     def invoke(cls, input: Dict) -> str:
 
-        logging.info(f"get_documents.invoke() input: {input}")
+        logging.info(f"retrieve_wiki_data.invoke() input: {input}")
         # Check if the input is a dictionary
 
         query = input.get("query", None)