Update agent.py

agent.py

@@ -6,6 +6,9 @@ from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
 
+from langchain_community.tools import DuckDuckGoSearchResults
+
+
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.document_loaders import ArxivLoader
@@ -43,6 +46,25 @@ def divide(a: int,b: int) -> float:
 def web_search(query: str) -> str:
   """ Search for a query on web and return best 2 result."""
 
+  search = DuckDuckGoSearchResults(num_results=2)
+  results = search.invoke(query)
+
+  '''formatted_results = "\n\n-----\n\n".join(
+      [
+          #f'<Result: source = "{result.metadata["source"]}", page = "{result.metadata.get("page","")}">\n{result.page_content}\n </Result>'
+          f'<Result: source = "{result.get("url", "")}", page = "{result.get("title","")}">\n{result.get("content","")}\n </Result>'
+          for result in results
+      ]
+  )'''
+  return {"web_results" : results}
+
+
+
+    
+'''@tool
+def web_search(query: str) -> str:
+  """ Search for a query on web and return best 2 result."""
+
   search_results = TavilySearchResults(max_results = 2).invoke(input=query)
 
   formatted_search_results = "\n\n-----\n\n".join(
@@ -52,7 +74,7 @@ def web_search(query: str) -> str:
           for result in search_results
       ]
   )
-  return {"web_results" : formatted_search_results}
+  return {"web_results" : formatted_search_results}'''
 
 @tool
 def wikipedia_search(query: str) -> str: