Upload app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import os
 import requests
 
-from langchain_community.tools import DuckDuckGoSearchResults
+from langchain_community.tools import DuckDuckGoSearchResults, \
+    WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
 from langfuse import Langfuse, get_client
 from langfuse.langchain import CallbackHandler
 import pandas as pd
@@ -16,6 +18,11 @@ tools = [
     multiply, add, subtract, divide, modulus,
 ]
 tools.append(DuckDuckGoSearchResults())
+tools.append(
+    WikipediaQueryRun(
+        api_wrapper=WikipediaAPIWrapper()
+    )
+)
 
 Langfuse(
     public_key=os.environ.get("LANGFUSE_PUBLIC_KEY"),