Update tools.py

tools.py

@@ -1,7 +1,30 @@
 from langchain_community.retrievers import BM25Retriever
 from langchain.tools import Tool
 from langchain_community.tools import DuckDuckGoSearchRun
+langchain_community.document_loaders.wikipedia.WikipediaLoader
 
 
 search_tool = DuckDuckGoSearchRun()
 
+def wiki_loader(query: str, lang: str='en', load_max_docs: int=2):
+    """
+    Fetches content from Wikipedia based on a given query.
+
+    Parameters:
+    - query (str): The search query for Wikipedia.
+    - lang (str): The language of the Wikipedia to search in. Default is 'en'.
+    - load_max_docs (int): The maximum number of documents to load. Default is 2.
+
+    Returns:
+    - list: A list of documents containing the fetched Wikipedia content.
+    """
+    try:
+        # Initialize the WikipediaLoader with the given query, language, and max documents
+        loader = WikipediaLoader(query=query, lang=lang, load_max_docs=load_max_docs)
+        # Load the documents
+        documents = loader.load()
+        print(documents)
+        return documents
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return []