Upload app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import os
 import requests
 
-from langchain_community.tools import WikipediaQueryRun
-from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_tavily import TavilySearch
 from langfuse import Langfuse, get_client
 from langfuse.langchain import CallbackHandler
@@ -11,22 +9,17 @@ import pandas as pd
 
 from src.agent import Agent
 from src.tools import (
-    multiply, add, subtract, divide, modulus
+    multiply, add, subtract, divide, modulus, wiki_search
 )
 
 tools = [
-    multiply, add, subtract, divide, modulus,
+    multiply, add, subtract, divide, modulus, wiki_search
 ]
 tools.append(
     TavilySearch(
         max_results=5,
     )
 )
-tools.append(
-    WikipediaQueryRun(
-        api_wrapper=WikipediaAPIWrapper()
-    )
-)
 
 Langfuse(
     public_key=os.environ.get("LANGFUSE_PUBLIC_KEY"),