Add wikipedia_summary tool

app.py

@@ -33,6 +33,30 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+import wikipedia
+
+@tool
+def wikipedia_summary(query: str) -> str:
+    """
+    A tool that fetches a Wikipedia summary for a given topic.
+
+    Args:
+        query: The topic to search for in Wikipedia.
+
+    Returns:
+        A short summary of the topic from Wikipedia.
+    """
+
+    try:
+        summary = wikipedia.summary(query, sentences=2)
+        return f"Here is a short summary of {query} from Wikipedia: {summary}"
+    except wikipedia.exceptions.DisambiguationError as e:
+        return f"The topic '{query}' is ambiguous. Try a more specific term. Suggestions: {e.options[:5]}"
+    except wikipedia.exceptions.PageError:
+        return f"Sorry, no page was found for '{query}' on Wikipedia."
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
 
 final_answer = FinalAnswerTool()