File size: 767 Bytes
dadbcfb 27f4cb5 dadbcfb 09e1c05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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()] |