Update app.py

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 from typing import Dict, Any, List, TypedDict, Optional
 from langgraph.graph import Graph, StateGraph
 from langgraph.prebuilt import ToolNode
-from tools import create_calculator_tool, create_search_tool
+from tools import create_calculator_tool, create_search_tool, simple_search
 print("trial")
 # (Keep Constants as is)
 # --- Constants ---
@@ -158,17 +158,16 @@ class BasicAgent:
     def _use_search(self, state: AgentState) -> AgentState:
         """Use the search tool."""
         try:
-            # Create search state with input from search_query
-            search_input = {
-                "query": state["search_query"],
-                "max_results": 3
-            }
-            result = self.search_tool.invoke({"input": search_input})
+            # Use the simplified search function
+            search_results = simple_search(
+                query=state["search_query"],
+                max_results=3
+            )
             
             state["history"].append({
                 'step': 'search',
                 'query': state["search_query"],
-                'results': [str(r) for r in result['output']['results']]
+                'results': search_results
             })
             state["needs_more_info"] = False
             state["current_step"] = 'final_answer'