add wiki tool

agent.py

@@ -5,6 +5,7 @@ from langgraph.graph import StateGraph, START, END
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFacePipeline
 from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
 from langchain_core.tools import tool
+from langchain_community.document_loaders import WikipediaLoader
 from ddgs import DDGS
 from dotenv import load_dotenv
 from groq import Groq
@@ -58,6 +59,28 @@ def web_search(keywords: str, max_results:int = 5) -> str:
             output += f"Results: {result['title']}\n{result['body']}\n{result['href']}\n\n"
         return(output)
 
+@tool
+def wiki_search(query: str) -> str:
+    """
+    Search Wikipedia for a query and return a maximum of 2 results
+
+    Use cases:
+    When the question requires the use of information from wikipedia
+
+    Args:
+    query: The search query
+    """
+
+    search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+            for doc in search_docs
+        ])
+    return {"wiki_results": formatted_search_docs}
+
+    
+
 # @tool
 # def get_image_file(task_id):
 #     """