Update tools/WikipediaTool.py

tools/WikipediaTool.py

@@ -10,7 +10,19 @@ class LocalWikipediaTool(Tool):
 
     def forward(self, query: str) -> str:
         try:
-            page = wikipedia.page(query)
-            return page.content  # full article text
-        except Exception:
-            return ""
+            # search for related pages
+            results = wikipedia.search(query)
+
+            if not results:
+                return "No Wikipedia 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}"
+
+        except Exception as e:
+            return f"Wikipedia error: {e}"