Update tooling.py

tooling.py

@@ -42,6 +42,24 @@ class ModelMathTool(Tool):
 
         return result
 
+class WikipediaTool(Tool):
+    name = "wikip_tool"
+    description = "Searches Wikipedia and provides summary about the queried topic."
 
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "Topic of wikipedia search",
+        }
+    }
+
+    output_type = "string"
+
+
+    def __init__(self):
+        import wikipedia
+
+    def forward(self, query: str) -> str:
+        return wikipedia.summary(query, sentences=3)