added search tools

agent.py

@@ -1,4 +1,5 @@
 from langchain.tools import Tool
+from langchain.utilities import WikipediaAPIWrapper, ArxivAPIWrapper, DuckDuckGoSearchRun
 import math
 
 ## ----- TOOLS DEFINITION ----- ##
@@ -123,15 +124,45 @@ def logarithm(a: float, base: float = math.e) -> float:
         raise ValueError("Logarithm arguments must be positive")
     return math.log(a, base)
 
+# ** Search Tools ** #
+
+# DuckDuckGo Web Search
+duckduckgo_search = DuckDuckGoSearchRun()
+web_search_tool = Tool.from_function(
+    func=duckduckgo_search.run,
+    name="Web Search",
+    description="Use this tool to search the internet for general-purpose queries."
+)
+
+# Wikipedia Search
+wikipedia_search = WikipediaAPIWrapper()
+wikipedia_tool = Tool.from_function(
+    func=wikipedia_search.run,
+    name="Wikipedia Search",
+    description="Use this tool to search Wikipedia for factual or encyclopedic information."
+)
+
+# ArXiv Search
+arxiv_search = ArxivAPIWrapper()
+arxiv_tool = Tool.from_function(
+    func=arxiv_search.run,
+    name="ArXiv Search",
+    description="Use this tool to search ArXiv for scientific papers. Input should be a research topic or query."
+)
+
 ## ----- TOOLS LIST ----- ##
 
 tools = [
-    add_numbers,
-    subtract_numbers,
-    multiply_numbers,
-    divide_numbers,
-    power,
-    modulus,
-    square_root,
-    logarithm
-]
+    # Math
+    Tool.from_function(func=add_numbers, name="Add Numbers", description="Add two numbers."),
+    Tool.from_function(func=subtract_numbers, name="Subtract Numbers", description="Subtract two numbers."),
+    Tool.from_function(func=multiply_numbers, name="Multiply Numbers", description="Multiply two numbers."),
+    Tool.from_function(func=divide_numbers, name="Divide Numbers", description="Divide two numbers."),
+    Tool.from_function(func=power, name="Power", description="Raise one number to the power of another."),
+    Tool.from_function(func=modulus, name="Modulus", description="Compute the modulus (remainder) of a division."),
+    Tool.from_function(func=square_root, name="Square Root", description="Compute the square root of a number."),
+    Tool.from_function(func=logarithm, name="Logarithm", description="Compute the logarithm of a number with a given base."),
+    # Search
+    web_search_tool,
+    wikipedia_tool,
+    arxiv_tool