Update tools/WikipediaTool.py

tools/WikipediaTool.py

@@ -1,18 +1,16 @@
 from smolagents import Tool
 import wikipedia
-
-class WikipediaTool(Tool):
-    name = "wikipedia_reader"
-    description = "Fetches the text content of a Wikipedia page given a topic."
+class LocalWikipediaTool(Tool):
+    name = "wikipedia_search"
+    description = "Search Wikipedia and return a summary."
     inputs = {
-        "topic": {"type": "string", "description": "The Wikipedia page topic to fetch."}
+        "query": {"type": "string", "description": "Search term"}
     }
     output_type = "string"
 
-    def forward(self, topic: str) -> str:
+    def forward(self, query: str) -> str:
         try:
-            return wikipedia.summary(topic, sentences=5)  # or full page with wikipedia.page(topic).content
-        except wikipedia.exceptions.DisambiguationError as e:
-            return f"Disambiguation error. Options: {e.options}"
-        except Exception as e:
-            return f"Error fetching Wikipedia page: {e}"
+            page = wikipedia.page(query)
+            return page.content  # full article text
+        except Exception:
+            return ""