Skip to content

datarobot_genai.drtools.calculator

calculator

calculator 

calculator(expression: str) -> str

Calculate a math expression, e.g. '15 * 7'.

Source code in datarobot_genai/drtools/calculator.py 
19
20
21
22
23
24
25
26
27
def calculator(expression: Annotated[str, "A math expression to calculate, e.g. '15 * 7'."]) -> str:
    """Calculate a math expression, e.g. '15 * 7'."""
    if not expression or not expression.strip():
        return "Error: expression cannot be empty"

    allowed = set("0123456789+-*/().% ")
    if not all(c in allowed for c in expression):
        return "Error: invalid characters in expression"
    return str(eval(expression))  # noqa: S307