Update agent.py

agent.py

@@ -25,6 +25,9 @@ def gemini_generate(prompt: str) -> str:
 
 @tool
 def calculator(expr: str) -> str:
+     """
+    Safely evaluates a math expression and returns the result.
+    """
     try:
         return str(eval(expr, {"__builtins__": {}}))
     except Exception as e:
@@ -32,6 +35,9 @@ def calculator(expr: str) -> str:
 
 @tool
 def wiki_search(query: str) -> str:
+     """
+    Fetches up to 2 Wikipedia pages for the query and concatenates their text.
+    """
     docs = WikipediaLoader(query=query, load_max_docs=2).load()
     return "\n\n".join(d.page_content for d in docs)