Update app.py

app.py

@@ -61,15 +61,19 @@ def search_wikipedia(search_term: str, results: int = 5) -> str:
 @tool
 def view_wikipedia(article_name: str) -> str:
     """
-    Retrieve full text for a given Wikipedia article.
-    """
-    import wikipedia
-    wikipedia_page = wikipedia.page(article_name)
-    article_title = wikipedia_page.title
-    article_text = wikipedia_page.content
-    return f"### Title: {article_title} ### Text: {article_text}"
+    View the content of a specific Wikipedia article.
 
+    Args:
+        article_name (str): The exact title of the Wikipedia article to view.
 
+    Returns:
+        str: The first 2000 characters of the article's content.
+    """
+    try:
+        page = wikipedia.page(article_name)
+        return page.content[:2000]
+    except Exception as e:
+        return f"Error fetching article '{article_name}': {e}"
 # --- Initialize tools ---
 youtube_transcript_extractor = load_tool("Mightypeacock/tool-YoutubeTranscript", trust_remote_code=True)
 visit_webpage = VisitWebpageTool()