Update my_tools.py

my_tools.py

@@ -1,6 +1,23 @@
-from smolagents import DuckDuckGoSearchTool, PythonInterpreterTool
+from smolagents import DuckDuckGoSearchTool, Tool
+import wikipediaapi
 
-my_tool_list = [
-    DuckDuckGoSearchTool(),        # 用来查网页
-    PythonInterpreterTool(),       # 可以跑代码、算数、分析 CSV
-]
+class WikipediaSearchTool(Tool):
+    name = "wikipedia_search"
+    description = "查找英文维基百科的页面简介，输入应为一个词或短词组"
+    inputs = {
+        "query": {"type": "string", "description": "维基百科搜索关键词，例如人名/专名"}
+    }
+    output_type = "string"
+    def __init__(self, lang="en"):
+        super().__init__()
+        self.wiki = wikipediaapi.Wikipedia(language=lang, user_agent="celum")
+    def forward(self, query: str):
+        page = self.wiki.page(query)
+        if not page.exists():
+            return "No Wikipedia page found."
+        return page.summary[:1000]
+
+wiki_tool = WikipediaSearchTool()
+ddg_tool = DuckDuckGoSearchTool()  # smolagents 自带
+
+my_tool_list = [wiki_tool, ddg_tool]