feat: create tools for agents

tools/__init__.py

@@ -0,0 +1,13 @@
+from tools.math_tools import add, div, mod, mult, sub
+from tools.search_tools import arxiv_search, internet_search, wiki_search
+
+__all__ = [
+    "add",
+    "div",
+    "mod",
+    "mult",
+    "sub",
+    "arxiv_search",
+    "internet_search",
+    "wiki_search",
+]

tools/math_tools.py

@@ -0,0 +1,77 @@
+from langchain_core.tools import tool
+
+
+@tool
+def add(a: float, b: float) -> float:
+    """Adds two numbers and returns the result rounded to 2 decimal places.
+
+    Args:
+        a (float): First number to be added
+        b (float): Second number to be added
+
+    Returns:
+        float: The sum of a and b, rounded to 2 decimal places
+    """
+    return round((a + b), 2)
+
+
+@tool
+def sub(a: float, b: float) -> float:
+    """Subtracts the second number from the first and returns the result rounded to 2 decimal places.
+
+    Args:
+        a (float): Number to subtract from
+        b (float): Number to subtract
+
+    Returns:
+        float: The difference between a and b, rounded to 2 decimal places
+    """
+    return round((a - b), 2)
+
+
+@tool
+def mult(a: float, b: float) -> float:
+    """Multiplies two numbers and returns the result rounded to 2 decimal places.
+
+    Args:
+        a (float): First number to multiply
+        b (float): Second number to multiply
+
+    Returns:
+        float: The product of a and b, rounded to 2 decimal places
+    """
+    return round((a * b), 2)
+
+
+@tool
+def div(a: float, b: float) -> float:
+    """Divides the first number by the second and returns the result rounded to 2 decimal places.
+
+    Args:
+        a (float): Number to be divided (dividend)
+        b (float): Number to divide by (divisor)
+
+    Raises:
+        ValueError: If the divisor (b) is zero
+
+    Returns:
+        float: The quotient of a divided by b, rounded to 2 decimal places
+    """
+    if b == 0:
+        raise ValueError("Cannot divide by zero!")
+
+    return round((a / b), 2)
+
+
+@tool
+def mod(a: float, b: float) -> float:
+    """Calculates the remainder of dividing the first number by the second and returns the result rounded to 2 decimal places.
+
+    Args:
+        a (float): Number to be divided (dividend)
+        b (float): Number to divide by (divisor)
+
+    Returns:
+        float: The remainder of a divided by b, rounded to 2 decimal places
+    """
+    return round((a % b), 2)

tools/search_tools.py

@@ -0,0 +1,99 @@
+from typing import Dict, List
+
+from langchain_community.document_loaders import ArxivLoader, WikipediaLoader
+from langchain_core.tools import tool
+from langchain_tavily import TavilySearch
+
+
+@tool
+def internet_search(query: str) -> Dict[str, List[Dict[str, str]]]:
+    """Perform a web search using Tavily Search API.
+
+    This tool searches the web for relevant information based on the provided query.
+    It returns up to 2 most relevant results with their sources, titles, and content.
+
+    Args:
+        query (str): The search query to look up on the web.
+
+    Returns:
+        Dict[str, List[Dict[str, str]]]: A dictionary containing a list of search results.
+            Each result is a dictionary with keys:
+            - Source: URL of the webpage
+            - Title: Title of the webpage
+            - Content: Main content/text from the webpage
+    """
+    response = TavilySearch(max_results=1).invoke(query)
+
+    formatted_answer = [
+        {
+            "Source": result["url"],
+            "Title": result["title"],
+            "Content": result["content"],
+        }
+        for result in response["results"]
+    ]
+
+    return {"web_results": formatted_answer}
+
+
+@tool
+def wiki_search(query: str) -> Dict[str, List[Dict[str, str]]]:
+    """Search Wikipedia articles using the provided query.
+
+    This tool searches Wikipedia for articles matching the query and returns
+    up to 2 most relevant results with their sources, titles, and content.
+
+    Args:
+        query (str): The search query to look up on Wikipedia.
+
+    Returns:
+        Dict[str, List[Dict[str, str]]]: A dictionary containing a list of Wikipedia results.
+            Each result is a dictionary with keys:
+            - Source: URL of the Wikipedia article
+            - Title: Title of the Wikipedia article
+            - Content: Main content/text from the article
+    """
+    docs = WikipediaLoader(query=query, load_max_docs=2).load()
+
+    formatted_answer = [
+        {
+            "Source": doc.metadata["source"],
+            "Title": doc.metadata["title"],
+            "Content": doc.page_content,
+        }
+        for doc in docs
+    ]
+
+    return {"wiki_results": formatted_answer}
+
+
+@tool
+def arxiv_search(query: str) -> Dict[str, List[Dict[str, str]]]:
+    """Search academic papers on arXiv using the provided query.
+
+    This tool searches arXiv for academic papers matching the query and returns
+    up to 2 most relevant results with their sources, titles, and content.
+
+    Args:
+        query (str): The search query to look up on arXiv.
+
+    Returns:
+        Dict[str, List[Dict[str, str]]]: A dictionary containing a list of arXiv results.
+            Each result is a dictionary with keys:
+            - Source: URL of the arXiv paper
+            - Title: Title of the academic paper
+            - Content: Main content/abstract of the paper
+    """
+    docs = ArxivLoader(query=query, load_max_docs=2).load()
+
+    formatted_answer = [
+        {
+            "Published": doc.metadata["Published"],
+            "Authors": doc.metadata["Authors"],
+            "Title": doc.metadata["Title"],
+            "Content": doc.page_content,
+        }
+        for doc in docs
+    ]
+
+    return {"arxiv_results": formatted_answer}