Update app.py

app.py

@@ -144,6 +144,32 @@ def get_today_matches() -> str:
     
     return result
 
+@tool
+def wiki_search(query: str) -> str:
+    """
+    Fetches a short summary from Wikipedia for a given topic.
+
+    Args:
+        query: Topic to search on Wikipedia.
+
+    Returns:
+        A short summary of the topic.
+    """
+
+    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
+
+    try:
+        response = requests.get(url, timeout=10)
+
+        if response.status_code != 200:
+            return "No Wikipedia page found for this topic."
+
+        data = response.json()
+
+        return data.get("extract", "No summary available.")
+
+    except Exception as e:
+        return f"Error fetching Wikipedia data: {str(e)}"
 
 final_answer = FinalAnswerTool()
 
@@ -164,7 +190,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_image, get_current_time_in_timezone, get_live_cricket_score, get_today_matches], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_image, get_current_time_in_timezone, get_live_cricket_score, get_today_matches, wiki_search], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,