Update app.py

wiki_tool added

app.py

@@ -8,6 +8,9 @@ import os
 from tavily import TavilyClient
 from Gradio_UI import GradioUI
 
+import wikipedia
+import textwrap
+
 # Example tool template - can be modified for your needs
 
 @tool
@@ -20,6 +23,51 @@ def my_custom_tool(arg1: str, arg2: int) -> str:  # Return type specification is
     return "What magic will you build?"
 
 
+@tool
+def wiki_search(query: str, sentences: int = 3) -> str:
+    """Search Wikipedia and return a summary of the topic.
+    Args:
+        query: The topic to search for on Wikipedia
+        sentences: Number of sentences to return in the summary (default: 3)
+    Returns:
+        str: A summary of the Wikipedia article
+    """
+    try:
+        # Set language to English for consistency
+        wikipedia.set_lang("en")
+        
+        # Search for the page
+        search_results = wikipedia.search(query)
+        if not search_results:
+            return f"No Wikipedia results found for: {query}"
+            
+        try:
+            # Get the page
+            page = wikipedia.page(search_results[0], auto_suggest=False)
+            
+            # Get summary and full URL
+            summary = wikipedia.summary(search_results[0], sentences=sentences, auto_suggest=False)
+            url = page.url
+            
+            # Format the response
+            response = f"Wikipedia article: {page.title}\n"
+            response += f"URL: {url}\n\n"
+            response += f"Summary:\n{textwrap.fill(summary, width=80)}\n"
+            
+            return response
+            
+        except wikipedia.DisambiguationError as e:
+            # Handle disambiguation pages
+            options = e.options[:5]  # Show first 5 options
+            response = f"'{query}' may refer to multiple topics. Here are some options:\n\n"
+            for i, option in enumerate(options, 1):
+                response += f"{i}. {option}\n"
+            return response
+            
+    except Exception as e:
+        return f"Error searching Wikipedia: {str(e)}"
+
+        
 @tool
 def tavily_search(query: str) -> str:
     """A tool that performs web search using Tavily API and formats the results.
@@ -84,7 +132,7 @@ with open("prompts.yaml", 'r') as stream:
 # Initialize agent with tools
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, tavily_search, get_current_time_in_timezone, image_generation_tool], # Added tavily_search
+    tools=[final_answer, tavily_search,wiki_search, get_current_time_in_timezone, image_generation_tool],
     max_steps=6,
     verbosity_level=1,
     grammar=None,