Upload app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 import os
 import requests
 
-from langchain_community.tools import DuckDuckGoSearchResults, \
-    WikipediaQueryRun
+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
 import pandas as pd
@@ -17,7 +17,11 @@ from src.tools import (
 tools = [
     multiply, add, subtract, divide, modulus,
 ]
-tools.append(DuckDuckGoSearchResults())
+tools.append(
+    TavilySearch(
+        max_results=5,
+    )
+)
 tools.append(
     WikipediaQueryRun(
         api_wrapper=WikipediaAPIWrapper()