Skip to content

calculator

calculator

Calculator tool — safe math evaluation via ast module.

Classes

CalculatorTool

Bases: BaseTool

Safe math calculator using AST-based evaluation.

Functions

safe_eval 

safe_eval(expression: str) -> float

Evaluate a math expression safely — Rust backend with Python fallback.

Source code in src/openjarvis/tools/calculator.py 
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
def safe_eval(expression: str) -> float:
    """Evaluate a math expression safely — Rust backend with Python fallback."""
    try:
        from openjarvis._rust_bridge import get_rust_module

        _rust = get_rust_module()
        return float(_rust.CalculatorTool().execute(expression))
    except ImportError:
        import ast as _ast

        # Support ^ as the power operator (common math/calculator notation).
        expression = expression.replace("^", "**")
        try:
            tree = _ast.parse(expression, mode="eval")
        except SyntaxError as exc:
            raise ValueError(f"Syntax error in expression: {exc}") from exc
        try:
            return float(_safe_eval_node(tree.body))
        except ZeroDivisionError:
            return math.inf