Update app.py

app.py

@@ -8,7 +8,7 @@ import base64
 from langchain_core.messages import HumanMessage
 from langchain_openai import ChatOpenAI
 from langchain_community.tools import DuckDuckGoSearchResults
-
+from langchain_google_community import GoogleSearchAPIWrapper
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -40,7 +40,8 @@ tools = [
     multiply,
     add,
     subtract,
-    DuckDuckGoSearchResults()
+    GoogleSearchAPIWrapper(k=10).run,
+    DuckDuckGoSearchResults(),
 ]
 ###### state and behavior
 from typing import TypedDict, Annotated, Optional
@@ -65,6 +66,12 @@ from langchain_core.messages import HumanMessage, SystemMessage
 def assistant(state: AgentState, llm_with_tools):
     # System message
     textual_description_of_tool = """
+
+GoogleSearchAPIWrapper(k=10).run():
+    Search for Top k = 10 information in the internet through GoogleSearch engine
+    Returns:
+        Snipnet of information from search engine.
+        
 DuckDuckGoSearchResults():
     Search for information in the internet through DuckDuckGo engine
 
@@ -115,6 +122,13 @@ class BasicAgent:
         api_key = os.environ["OPENAI_API_KEY"]
         if not api_key:
             raise ValueError("OPENAI_API_KEY environment variable not set or loaded.")
+            
+        if not os.environ["GOOGLE_API_KEY"]:
+            raise ValueError("GOOGLE_API_KEY environment variable not set or loaded.")
+
+        if not os.environ["GOOGLE_CSE_ID"]:
+            raise ValueError("GOOGLE_CSE_ID environment variable not set or loaded.")
+            
         # self.visionLLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # multi-modal LLM
         self.LLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # manager LLM
         self.LLM_with_tools = self.LLM.bind_tools(tools, parallel_tool_calls=False)