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()]