graph test

app.py

@@ -8,7 +8,6 @@ import gradio as gr
 import pandas as pd
 import requests
 from langchain_community.retrievers import WikipediaRetriever
-
 from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
 from langchain_community.utilities import WikipediaAPIWrapper
 
@@ -21,13 +20,13 @@ from langchain_core.messages import (
     ToolCall,
     ToolMessage,
 )
+from langchain_google_community.search import GoogleSearchAPIWrapper
 from langchain_openai import ChatOpenAI
 from langgraph.graph import END, START, StateGraph
 
 
 hf_token = os.getenv("OPENAI_API_KEY")
 
-
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -178,8 +177,8 @@ class BasicAgent:
 
     def web_search(self, state: AnswerState) -> Dict[str, Any]:
         print("[web_search] Searching for: " + str(state["search_request"]))
-        search_tool = DuckDuckGoSearchRun()
-        results = search_tool.invoke(str(state["search_request"]))
+        search_tool = GoogleSearchAPIWrapper()
+        results = search_tool.run(str(state["search_request"]))
         print(f"Search results: {results}")
         state["messages"].append(
             ChatMessage(

requirements.txt

@@ -9,3 +9,4 @@ langchain-community
 ddgs
 duckduckgo-search
 wikipedia
+langchain-google-community