| from smolagents import DuckDuckGoSearchTool, Tool | |
| import wikipediaapi | |
| 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] | |
| my_tool_list = [WikipediaSearchTool(),DuckDuckGoSearchTool()] |