Update app.py

app.py

@@ -21,31 +21,17 @@ load_dotenv()
 
 
 
-@tool
-def search_wikipedia(query: str) -> str:
-    """
-    Fetches a summary of a Wikipedia page for a given query.
-    Args:
-        query: The search term to look up on Wikipedia.
-    Returns:
-        str: A summary of the Wikipedia page if successful, or an error message if the request fails.
-    Raises:
-        requests.exceptions.RequestException: If there is an issue with the HTTP request.
-    """
-    url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
+import wikipedia
 
-    try:
-        response = requests.get(url)
-        response.raise_for_status()
-
-        data = response.json()
-        title = data["title"]
-        extract = data["extract"]
-
-        return f"Summary for {title}: {extract}"
-
-    except requests.exceptions.RequestException as e:
-        return f"Error fetching Wikipedia data: {str(e)}"
+@tool
+class WikipediaSearchTool(Tool):
+    def use(self, query: str) -> str:
+        try:
+            return wikipedia.summary(query, sentences=2)
+        except wikipedia.exceptions.DisambiguationError as e:
+            return f"Disambiguation Error: {e.options}"
+        except wikipedia.exceptions.PageError:
+            return "No page found."