added duckduckgo

app.py

@@ -24,7 +24,7 @@ from langgraph.graph.message import add_messages
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from langgraph.prebuilt import ToolNode, tools_condition
 
-from tools import fetch_website, get_wiki_full, youtube_transcript, python_repl_tool
+from tools import fetch_website, get_wiki_full, youtube_transcript, python_repl_tool, duckduckgo_search_results
 
 # Initialize the Hugging Face model
 hf_model_name = "openai/gpt-oss-120b" # "Qwen/Qwen2.5-72B-Instruct"
@@ -45,7 +45,8 @@ tools_list = [
     fetch_website,
     get_wiki_full,
     youtube_transcript,
-    python_repl_tool
+    python_repl_tool,
+    duckduckgo_search_results
 ]
 
 llm_with_tools = chat_model.bind_tools(
@@ -62,6 +63,13 @@ def assistant(state: AgentState):
     # System message
     textual_description_of_tool = dedent(
         """
+        duckduckgo_search_results(query: str) -> list[dict]:
+            Perform a web search using DuckDuckGo and return the results.
+            Args:
+                query: The search query string.
+            Returns:
+                A list of search results, where each result is a dictionary that includes the snippet, title, and link.
+
         fetch_website(url: str) -> str:
             Fetch the content of a website.
             Args:
@@ -143,12 +151,11 @@ class BasicAgent:
                 content=question # + '/nothink'
             )
         ]
-        response = await agent_graph.ainvoke({"messages": messages}, config={"recursion_limit": 20})
+        response = await agent_graph.ainvoke({"messages": messages}, config={"recursion_limit": 10})
         response_text = response['messages'][-1].content
         # return response_text.split('</think>')[-1]
         return extract_answer(response_text)
 
-
 async def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,

requirements.txt

@@ -3,6 +3,8 @@ requests
 wikipedia
 youtube-transcript-api
 pytube
+duckduckgo-search
+ddgs
 
 langchain
 langchain_experimental

tools.py

@@ -4,7 +4,7 @@ from bs4 import BeautifulSoup
 
 from langchain.tools import tool
 from langchain_community.utilities import WikipediaAPIWrapper
-from langchain_community.tools import WikipediaQueryRun
+from langchain_community.tools import WikipediaQueryRun, DuckDuckGoSearchRun, DuckDuckGoSearchResults
 from langchain_community.document_loaders import YoutubeLoader, WebBaseLoader
 from langchain_experimental.utilities import PythonREPL
 
@@ -12,6 +12,20 @@ from langchain_experimental.utilities import PythonREPL
 # Initialize Python REPL
 python_repl = PythonREPL()
 
+@tool
+def duckduckgo_search_results(query: str) -> list[dict]:
+    """Perform a DuckDuckGo search for the given query and return the results.
+    Args:
+        query: The search query string.
+    Returns:
+        A list of search results, where each result is a dictionary that includes the snippet, title, and link.
+    """
+    try:
+        search = DuckDuckGoSearchResults(output_format="list")
+        return search.invoke(query)
+    except Exception as e:
+        return f"Error performing search: {str(e)}"
+
 @tool
 def fetch_website(url:str) -> str:
     """Fetch the content of a website.