updated code

app.py

@@ -21,12 +21,12 @@ class BasicAgent:
         
         # Initialize Google Generative AI client
         # It's recommended to set GOOGLE_API_KEY as an environment variable
-        gemini_api_key = "AIzaSyAwgCoL6IjGotj0y27Rirb5GvtVLTDF8jc"
-        if not gemini_api_key:
+        self.gemini_api_key = "AIzaSyAwgCoL6IjGotj0y27Rirb5GvtVLTDF8jc"
+        if not self.gemini_api_key:
             print("Warning: GOOGLE_API_KEY not found. Gemini API calls might fail.")
             print("Please set the GOOGLE_API_KEY environment variable.")
         
-        genai.configure(api_key=gemini_api_key)
+        genai.configure(api_key=self.gemini_api_key)
         self.gemini_client = genai
         self.gemini_model = "gemini-2.5-flash" # Specify Gemini Flash model
 
@@ -55,7 +55,7 @@ class BasicAgent:
         """Call LLM with the given prompt - supports multiple providers"""
         try:
             # Option 1: Google Gemini Flash
-            if os.getenv("GOOGLE_API_KEY"):
+            if self.gemini_api_key:
                 model = self.gemini_client.GenerativeModel(self.gemini_model)
                 response = model.generate_content(
                     prompt,
@@ -164,7 +164,7 @@ class BasicAgent:
                     
                     elif tool_name == 'web_search':
                         query = match.group(1).strip('"\'')
-                        result = search_tool(query)
+                        result = search_tool.invoke(query)
                         enhanced_response = enhanced_response.replace(match.group(0), f"search result: {result}")
                         
                 except Exception as e: