Spaces:
Running
Running
File size: 391 Bytes
b02630d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """Wikipedia search tool."""
from langchain_community.utilities import WikipediaAPIWrapper
class WikiTool:
"""Wrapper for Wikipedia-based QA."""
def __init__(self) -> None:
self.api = WikipediaAPIWrapper(top_k_results=3, lang="en")
def query(self, query: str) -> str:
"""Search Wikipedia and return a summarized answer."""
return self.api.run(query)
|