Update app.py

app.py

@@ -19,18 +19,25 @@ from groq import Groq
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # ---------- Tools ----------
+from langchain_core.tools import tool
+from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
+from langchain_community.tools.tavily_search import TavilySearchResults
+
 @tool
 def wiki_search(query: str) -> str:
+    """Search Wikipedia for a given query and return content from up to 2 relevant pages."""
     docs = WikipediaLoader(query=query, load_max_docs=2).load()
     return "\n\n".join([doc.page_content for doc in docs])
 
 @tool
 def web_search(query: str) -> str:
+    """Search the web using the Tavily API and return content from up to 3 search results."""
     docs = TavilySearchResults(max_results=3).invoke(query)
     return "\n\n".join([doc.page_content for doc in docs])
 
 @tool
 def arvix_search(query: str) -> str:
+    """Search academic papers on Arxiv for a given query and return up to 3 result summaries."""
     docs = ArxivLoader(query=query, load_max_docs=3).load()
     return "\n\n".join([doc.page_content[:1000] for doc in docs])