Update agent.py

agent.py

@@ -17,33 +17,38 @@ from langchain_huggingface import (
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain_openai import ChatOpenAI
-from langchain_community.utilities import SerpAPIWrapper
+from duckduckgo_search import DDGS
 
 load_dotenv()
 
 ### =============== SEARCH TOOLS =============== ###
 
 @tool
-def serpapi_search(query: str) -> str:
-    """Search the web using SerpAPI.
+def duckduckgo_search(query: str) -> str:
+    """Search the web using DuckDuckGo.
     Args:
         query: The search query."""
     try:
-        # Get API key from environment variable
-        api_key = os.getenv("SERPAPI_API_KEY")
-        if not api_key:
-            return {"search_results": "Error: SERPAPI_API_KEY not found in environment variables."}
+        # Initialize DuckDuckGo search
+        ddgs = DDGS()
         
-        # Initialize SerpAPIWrapper with the API key
-        search = SerpAPIWrapper(serpapi_api_key=api_key)
+        # Perform the search and get the first 5 results
+        results = list(ddgs.text(query, max_results=5))
         
-        # Perform the search
-        results = search.run(query)
-        
-        if not results or results.strip() == "":
+        if not results:
             return {"search_results": "No search results found."}
         
-        return {"search_results": results}
+        # Format the results
+        formatted_results = []
+        for i, result in enumerate(results, 1):
+            formatted_result = f"{i}. {result.get('title', 'No title')}\n"
+            formatted_result += f"   URL: {result.get('href', 'No URL')}\n"
+            formatted_result += f"   {result.get('body', 'No description')}\n"
+            formatted_results.append(formatted_result)
+        
+        search_output = "\n".join(formatted_results)
+        
+        return {"search_results": search_output}
     except Exception as e:
         return {"search_results": f"Error performing search: {str(e)}"}
 
@@ -80,7 +85,7 @@ print(system_prompt)
 sys_msg = SystemMessage(content=system_prompt)
 
 tools = [
-    serpapi_search,
+    duckduckgo_search,
     save_and_read_file,
 ]
 
@@ -137,7 +142,7 @@ def build_graph(provider: str = "openai"):
 # test
 if __name__ == "__main__":
     question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
-    graph = build_graph(provider="openai")
+    graph = build_graph(provider="huggingface")
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]: