fix

src/agent.py

@@ -37,15 +37,21 @@ class WikiTitleFinder(Tool):
 
 class WikiContentFetcher(Tool):
     name = "wiki_page"
-    description = "Fetch Wikipedia page content."
-    inputs = {"page_title": {"type": "string", "description": "Wikipedia page title."}}
+    description = "Fetch full content of a Wikipedia page as markdown."
+    inputs = {"page_title": {"type": "string", "description": "Exact Wikipedia page title."}}
     output_type = "string"
 
     def forward(self, page_title: str) -> str:
         try:
-            return to_markdown(wiki.page(page_title).html())
+            page = wiki.page(page_title, auto_suggest=False)
+            markdown_content = to_markdown(page.html())
+            return markdown_content[:15000]  # Prevent token explosion
         except wiki.exceptions.PageError:
-            return f"'{page_title}' not found."
+            return f"Page '{page_title}' not found."
+        except wiki.exceptions.DisambiguationError as e:
+            return f"Multiple pages found: {e.options[:5]}"
+        except Exception as e:
+            return f"Wikipedia fetch error: {str(e)[:200]}"
 
 class MathSolver(Tool):
     name = "math_solver"

tools/WikipediaTool.py

@@ -2,27 +2,20 @@ from smolagents import Tool
 import wikipedia
 class LocalWikipediaTool(Tool):
     name = "wikipedia_search"
-    description = "Use websearch tool first to get the facts, and if needed use this Fallback tool: Use Wikipedia after the web search tool."
-    inputs = {
-        "query": {"type": "string", "description": "Search term"}
-    }
+    description = "Search Wikipedia and return summary. Use only after web search."
+    inputs = {"query": {"type": "string", "description": "Search term"}}
     output_type = "string"
 
     def forward(self, query: str) -> str:
         try:
-            # search for related pages
-            results = wikipedia.search(query)
-
+            results = wikipedia.search(query, results=3)
             if not results:
-                return "No Wikipedia results found."
+                return "No results found."
 
-            # pick first result
             page_title = results[0]
-
-            # get a short summary (2–3 sentences)
-            summary = wikipedia.summary(page_title, sentences=3)
-
-            return f"Title: {page_title}\nSummary: {summary}"
+            summary = wikipedia.summary(page_title, sentences=5, auto_suggest=False)
+            
+            return f"Title: {page_title}\n\nSummary: {summary}"
 
         except Exception as e:
-            return f"Wikipedia error: {e}"
+            return f"Wikipedia error: {str(e)[:150]}"