Update tools/tools.py

tools/tools.py

@@ -13,11 +13,46 @@ from dotenv import load_dotenv
 from smolagents import tool, LiteLLMModel
 import google.generativeai as genai
 from pytesseract import image_to_string
+from smolagents import WikipediaSearchTool, tool
 
 load_dotenv()
 
 MODEL_ID = "groq/meta-llama/llama-4-scout-17b-16e-instruct"
 
+_wiki_tool = WikipediaSearchTool(
+    user_agent="GAIA-Agent (alandiogo200@gmail.com)",  # troque pelo seu email real
+    content_type="summary",
+)
+
+@tool
+def wikipedia_search(query: str, max_chars: int = 2000) -> str:
+    """
+    Searches Wikipedia and returns a summary, truncated as a safety net.
+    Args:
+        query (str): The search term. Use the exact Wikipedia page title (e.g. "Mercedes_Sosa").
+        max_chars (int): Maximum characters to return (default 2000).
+    Returns:
+        str: Wikipedia summary content or an error message.
+    """
+    result = _wiki_tool.forward(query)
+    if result and len(result) > max_chars:
+        return result[:max_chars] + "\n\n[...content truncated, use a more specific query if you need more detail...]"
+    return result
+
+@tool
+def wikipedia_search(query: str, max_chars: int = 2000) -> str:
+    """
+    Searches Wikipedia and returns a summary, truncated as a safety net.
+    Args:
+        query (str): The search term. Use the exact Wikipedia page title (e.g. "Mercedes_Sosa").
+        max_chars (int): Maximum characters to return (default 2000).
+    Returns:
+        str: Wikipedia summary content or an error message.
+    """
+    result = _wiki_tool.forward(query)
+    if result and len(result) > max_chars:
+        return result[:max_chars] + "\n\n[...content truncated, use a more specific query if you need more detail...]"
+    return result
 #  Vision Tool 
 @tool
 def vision_tool(prompt: str, image_list: List[Image.Image]) -> str: