tool

agent.py

@@ -21,7 +21,6 @@ from langchain_community.retrievers import BM25Retriever
 
 from langgraph.prebuilt import ToolNode, tools_condition
 
-# load system prompt
 from prompts import system_prompt
 
 
@@ -29,63 +28,6 @@ from prompts import system_prompt
 load_dotenv()
 
 
-# maths tool
-@tool
-def add(a:int, b:int) -> int:
-    """add two numbers.
-    args:
-        a: first int
-        b: second int
-    """
-    return a + b
-
-
-@tool
-def subtract(a:int, b:int) -> int:
-    """subtract two numbers.
-    args:
-        a: first int
-        b: second int
-    """
-    return a - b
-
-
-@tool
-def multiply(a:int, b:int) -> int:
-    """multiply two numbers.
-    args:
-        a: first int
-        b: second int
-    """
-    return a * b
-
-
-@tool
-def divide(a:int, b:int) -> float:
-    """divide two numbers.
-    args:
-        a: first int
-        b: second int
-    """
-    try:
-        # Attempt the division
-        result = a / b
-        return result
-    except ZeroDivisionError:
-        # Handle the case where b is zero
-        raise ValueError("Cannot divide by zero.")
-
-
-@tool
-def modulus(a:int, b:int) -> int:
-    """modulus remainder of two numbers.
-    args:
-        a: first int
-        b: second int
-    """
-    return a % b
-
-
 # wikipedia search tool
 @tool
 def search_wiki(query: str) -> str:
@@ -143,11 +85,6 @@ sys_msg = SystemMessage(content=system_prompt)
 
 
 tools = [
-    add,
-    subtract,
-    multiply,
-    divide,
-    modulus,
     search_web,
     search_wiki,
     search_arxiv