Update agent.py

agent.py

@@ -13,6 +13,8 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 
+from duckduckgo_search import ddg
+
 load_dotenv()
 
 @tool
@@ -105,19 +107,28 @@ def wiki_search(query: str) -> str:
 
 @tool
 def web_search(query: str) -> str:
-    """Search Tavily for a query and return maximum 3 results.
+    """Search Tavily/DuckDuckGo for a query and return maximum 3 results.
     
     Args:
         query: The search query."""
 
+    """ 
     search_docs = TavilySearchResults(max_results=3).invoke(query=query)
     formatted_search_docs = "\n\n---\n\n".join(
         [
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
             for doc in search_docs
         ])
-    return {"web_results": formatted_search_docs}
+    """
+    search_docs = ddg(query, max_results=3)
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc["href"]}" page=""/>\n{doc["body"]}\n</Document>'
+            for doc in search_docs
+        ])
 
+    
+    return {"web_results": formatted_search_docs}
 
 # load the system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f: