Update app.py

app.py

@@ -1175,7 +1175,7 @@ def cache_response(query, response, ttl=3600):
 
 
 tavily_tool = TavilySearchResults(
-    max_results=7,
+    max_results=4,
     search_depth="basic",
     include_answer=True,
     include_raw_content=False,
@@ -1187,38 +1187,10 @@ tavily_tool = TavilySearchResults(
     # args_schema=...,       # overwrite default args_schema: BaseModel
 )
 
-class CachedTavilyTool:
-    def __init__(self, tavily_instance):
-        self.tavily_instance = tavily_instance
-
-    def search(self, query):
-        """Perform Tavily search with caching."""
-        # Check for cached response
-        cached_response = get_cached_response(query)
-        if cached_response:
-            print("Cache hit!")
-            return cached_response
-
-        # Perform Tavily search
-        print("Cache miss! Searching Tavily...")
-        try:
-            results = self.tavily_instance.search(query)
-            if not results:
-                return "No relevant information found."
-
-            # Cache the results
-            cache_response(query, results)
-            return results
-        except Exception as e:
-            print(f"Error during Tavily search: {e}")
-            return "An error occurred while processing the Tavily search."
-
-# Create a cached instance of TavilyTool
-cached_tavily_tool = CachedTavilyTool(tavily_tool)
 # Compile all tool functions into a list
 tools = [
     knowledge_base_tool,  # Tool for querying the knowledge base and retrieving responses
-    cached_tavily_tool,
+    tavily_tool,
     # google_search_tool,  # Tool for performing a Google search and retrieving search result snippets
 ]