wikipedia_searcher.py

from langchain_community.tools import WikipediaQueryRun
from langchain_community.utilities import WikipediaAPIWrapper

class WikipediaSearcher:
    def __init__(self):
        # WikipediaAPIWrapper handles the direct Wikipedia API calls
        self.wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())

    def search(self, query: str) -> str:
        # Returns a concise summary or article snippet for the query
        try:
            result = self.wikipedia.run(query)
            return result
        except Exception as e:
            return f"Error retrieving Wikipedia data: {e}"