Final_Assignment_Template / calculus_tools.py
kokluch's picture
Add calculus tools
0b5b1f5
raw
history blame contribute delete
979 Bytes
from langchain_core.tools import tool
@tool
def add(a: float, b: float) -> float:
"""
Add two float numbers.
Args:
a: float
b float
Returns:
a + b
"""
return a + b
@tool
def substract(a: float, b: float) -> float:
"""
Substract two float numbers.
Args:
a: float
b float
Returns:
a - b
"""
return a - b
@tool
def multiple(a: float, b: float) -> float:
"""
Multiple two float numbers.
Args:
a: float
b float
Returns:
a * b
"""
return a * b
@tool
def divide(a: float, b: float) -> float:
"""
Divide two float numbers.
Args:
a: float
b float
Returns:
a / b
"""
if b is not 0:
return a / b
else:
raise "Illegal operation. Unable to divide by zero."